[Calyx-FIRRTL] Debugged FIRRTL implementations of primitives (#1907)

* Debug std_reg FIRRTL implementation

* Fix std_const template typo

* Fix std_lsh FIRRTL primitive

* Fixed std_reg bug that was causing  to fail

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()

tools/firrtl/templates/std_const.fir

@@ -1,3 +1,3 @@
-    module std_const_WIDTH_VAL :
+    module std_const_WIDTH_VALUE :
         output out : UInt<WIDTH>
         out <= UInt(VALUE)

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))

tools/firrtl/templates/std_reg.fir

@@ -7,13 +7,17 @@
         output done : UInt<1>
 
         reg internal_reg : UInt<WIDTH>, clk
-        out <= UInt(0)
-        when eq(write_en, UInt(1)):
-            out <= in
-            done <= UInt(1)
+        reg done_reg : UInt<1>, clk
+
+        when eq(reset, UInt(1)):
+            internal_reg <= UInt(0)
+            done_reg <= UInt(0)
         else:
-            when eq(reset, UInt(1)):
-                done <= UInt(0)
-                out <= UInt(0)
+            when eq(write_en, UInt(1)):
+                internal_reg <= in
+                done_reg <= UInt(1)
             else:
-                done <= UInt(0)
+                done_reg <= UInt(0)
+
+        out <= internal_reg
+        done <= done_reg