Fix std_lsh FIRRTL primitive

calyxir · Jan 31, 2024 · 5c8fa19 · 5c8fa19
1 parent a2df85a
commit 5c8fa19
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tools/firrtl/generate-firrtl-with-primitives.py b/tools/firrtl/generate-firrtl-with-primitives.py
@@ -1,4 +1,5 @@
 import json
+import math
 import os
 import sys
 
@@ -13,6 +14,9 @@ def generate_replacement_map(inst):
         replacement_map["DIFF"] = replacement_map["OUT_WIDTH"] - replacement_map["IN_WIDTH"]
     elif inst["name"] == "std_slice":
         replacement_map["DIFF"] = replacement_map["IN_WIDTH"] - replacement_map["OUT_WIDTH"]
+    elif inst["name"] == "std_lsh":
+        width = replacement_map["WIDTH"]
+        replacement_map["BITS"] = math.ceil(math.log(width, 2)) + 1
 
     return replacement_map
 
@@ -62,4 +66,4 @@ def main():
     generate(sys.argv[1], sys.argv[2])
 
 if __name__ == '__main__':
-        main()
+        main()
diff --git a/tools/firrtl/templates/std_lsh.fir b/tools/firrtl/templates/std_lsh.fir
@@ -3,4 +3,4 @@
         input right : UInt<WIDTH>
         output out : UInt<WIDTH>
 
-        out <= dshl(left, right)
+        out <= dshl(left, bits(right, BITS, 0))