remove post-inference decorator for backwards compatibility

Author: neelay893

src/kirin/dialects/vmath/rewrites/desugar.py

@@ -6,7 +6,6 @@
 from kirin.dialects.ilist import IListType
 
 from ..stmts import add as vadd, div as vdiv, mul as vmul, sub as vsub
-from .._dialect import dialect
 
 
 class DesugarBinOp(RewriteRule):
@@ -53,7 +52,6 @@ def replace_binop(self, node: ir.Statement) -> RewriteResult:
                 return RewriteResult()
 
 
-@dialect.post_inference
 class WalkDesugarBinop(RewriteRule):
     """
     Walks DesugarBinop. Needed for correct behavior when

test/dialects/vmath/test_desugar.py

@@ -28,17 +28,17 @@ def add_scalar_lhs():
     return add_kernel(x=3.0, y=[3.0, 4, 5])
 
 
-# def test_add_scalar_lhs():
-#     # out = add_scalar_lhs()
-#     VMathDesugar(add_scalar_lhs.dialects).unsafe_run(add_scalar_lhs)
-#     add_scalar_lhs.print()
-#     res = add_scalar_lhs()
-#     assert isinstance(res, IList)
-#     assert res.type.vars[0].typ is float
-#     assert np.allclose(np.asarray(res), np.array([6, 7, 8]))
+def test_add_scalar_lhs():
+    # out = add_scalar_lhs()
+    VMathDesugar(add_scalar_lhs.dialects).unsafe_run(add_scalar_lhs)
+    add_scalar_lhs.print()
+    res = add_scalar_lhs()
+    assert isinstance(res, IList)
+    assert res.type.vars[0].typ is float
+    assert np.allclose(np.asarray(res), np.array([6, 7, 8]))
 
 
 def test_typed_kernel_add():
-    # VMathDesugar(add_scalar_rhs_typed.dialects).unsafe_run(add_scalar_rhs_typed)
+    VMathDesugar(add_scalar_rhs_typed.dialects).unsafe_run(add_scalar_rhs_typed)
     add_scalar_rhs_typed.print()
     print(add_scalar_rhs_typed(IList([0, 1, 2]), 3.1))