fix type for math dialect (#626)

Author: kaihsin

src/kirin/dialects/math/__init__.py

@@ -37,7 +37,7 @@ def atanh(x: float) -> float: ...
 
 
 @lowering.wraps(stmts.ceil)
-def ceil(x: float) -> float: ...
+def ceil(x: float) -> int: ...
 
 
 @lowering.wraps(stmts.copysign)
@@ -77,7 +77,7 @@ def fabs(x: float) -> float: ...
 
 
 @lowering.wraps(stmts.floor)
-def floor(x: float) -> float: ...
+def floor(x: float) -> int: ...
 
 
 @lowering.wraps(stmts.fmod)
@@ -149,7 +149,7 @@ def tanh(x: float) -> float: ...
 
 
 @lowering.wraps(stmts.trunc)
-def trunc(x: float) -> float: ...
+def trunc(x: float) -> int: ...
 
 
 @lowering.wraps(stmts.ulp)

src/kirin/dialects/math/_gen.py

@@ -114,6 +114,8 @@ class MathMethodTable(MethodTable):
     for name, obj, sig in builtin_math_functions():
         if "is" in name:
             ret_type = "bool"
+        elif name in {"trunc", "ceil", "floor"}:
+            ret_type = "int"
         else:
             ret_type = "float"
         f.write(textwrap.dedent(f"""

src/kirin/dialects/math/stmts.py

@@ -72,7 +72,7 @@ class ceil(ir.Statement):
     name = "ceil"
     traits = frozenset({ir.Pure(), lowering.FromPythonCall()})
     x: ir.SSAValue = info.argument(types.Float)
-    result: ir.ResultValue = info.result(types.Float)
+    result: ir.ResultValue = info.result(types.Int)
 
 
 @statement(dialect=dialect)
@@ -173,7 +173,7 @@ class floor(ir.Statement):
     name = "floor"
     traits = frozenset({ir.Pure(), lowering.FromPythonCall()})
     x: ir.SSAValue = info.argument(types.Float)
-    result: ir.ResultValue = info.result(types.Float)
+    result: ir.ResultValue = info.result(types.Int)
 
 
 @statement(dialect=dialect)
@@ -356,7 +356,7 @@ class trunc(ir.Statement):
     name = "trunc"
     traits = frozenset({ir.Pure(), lowering.FromPythonCall()})
     x: ir.SSAValue = info.argument(types.Float)
-    result: ir.ResultValue = info.result(types.Float)
+    result: ir.ResultValue = info.result(types.Int)
 
 
 @statement(dialect=dialect)