reexport math constant to dialect (#448)

Author: kaihsin

src/kirin/dialects/math/__init__.py

@@ -1,10 +1,16 @@
 "math dialect, modeling functions in python's `math` stdlib"  # This file is generated by gen.py
 
+import math as pymath
+
 from kirin import lowering
 from kirin.dialects.math.dialect import dialect as dialect
 
 from . import stmts as stmts, interp as interp
 
+pi = pymath.pi
+e = pymath.e
+tau = pymath.tau
+
 
 @lowering.wraps(stmts.acos)
 def acos(x: float) -> float: ...

src/kirin/dialects/math/_gen.py

@@ -109,8 +109,12 @@ class MathMethodTable(MethodTable):
     f.write("# This file is generated by gen.py\n")
     f.write("from kirin.dialects.math.dialect import dialect as dialect\n")
     f.write("from . import stmts as stmts, interp as interp\n")
-
+    f.write("import math as pymath\n")
+    f.write("pi = pymath.pi\n")
+    f.write("e = pymath.e\n")
+    f.write("tau = pymath.tau\n")
     f.write("from kirin import lowering2\n")
+
     for name, obj, sig in builtin_math_functions():
         f.write(
             textwrap.dedent(

test/dialects/math/test_const.py

@@ -0,0 +1,9 @@
+import math as pymath
+
+from kirin.dialects import math
+
+
+def test_const():
+    assert math.pi == pymath.pi
+    assert math.e == pymath.e
+    assert math.tau == pymath.tau