[rtlir] Add visit_Constant to behavioral visitor

pymtl · Dec 7, 2023 · aa19bfb · aa19bfb
1 parent c8e2ab5
commit aa19bfb
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pymtl3/passes/rtlir/behavioral/BehavioralRTLIRGenL1Pass.py b/pymtl3/passes/rtlir/behavioral/BehavioralRTLIRGenL1Pass.py
@@ -435,6 +435,14 @@ def visit_Num( s, node ):
     ret.ast = node
     return ret
 
+  def visit_Constant( s, node ):
+    val = node.value
+    if isinstance( val, int ):
+      ret = bir.Number( val )
+      ret.ast = node
+      return ret
+    raise PyMTLSyntaxError( s.blk, node, 'invalid constant: not an integer!' )
+
   def visit_If( s, node ): raise NotImplementedError()
 
   def visit_For( s, node ): raise NotImplementedError()
@@ -646,3 +654,6 @@ def visit_Name( s, node ):
 
   def visit_Num( s, node ):
     return node.n
+
+  def visit_Constant( s, node ):
+    return node.value