[refactor] Enable the new ast builder by default

python/taichi/lang/__init__.py

@@ -396,7 +396,7 @@ def __init__(self):
         self.gdb_trigger = False
         self.excepthook = False
         self.experimental_real_function = False
-        self.experimental_ast_refactor = False
+        self.experimental_ast_refactor = True
 
 
 def prepare_sandbox():

python/taichi/lang/ir_builder.py

@@ -904,6 +904,11 @@ def build_Continue(ctx, node):
     def build_Pass(ctx, node):
         return node
 
+    @staticmethod
+    def build_Raise(ctx, node):
+        node.exc = build_stmt(ctx, node.exc)
+        raise node.exc.ptr
+
 
 build_stmt = IRBuilder()
 

python/taichi/lang/linalg_impl.py

@@ -228,12 +228,13 @@ def svd(A, dt):
     Returns:
         Decomposed nxn matrices `U`, 'S' and `V`.
     """
-    if ti.static(A.n == 2):
+    if ti.static(A.n == 2):  # pylint: disable=R1705
         ret = svd2d(A, dt)
         return ret
-    if ti.static(A.n == 3):
+    elif ti.static(A.n == 3):
         return svd3d(A, dt)
-    raise Exception("SVD only supports 2D and 3D matrices.")
+    else:
+        raise Exception("SVD only supports 2D and 3D matrices.")
 
 
 @func
@@ -251,9 +252,11 @@ def polar_decompose(A, dt):
     Returns:
         Decomposed nxn matrices `U` and `P`.
     """
-    if ti.static(A.n == 2):
+    if ti.static(A.n == 2):  # pylint: disable=R1705
         ret = polar_decompose2d(A, dt)
         return ret
-    if ti.static(A.n == 3):
+    elif ti.static(A.n == 3):
         return polar_decompose3d(A, dt)
-    raise Exception("Polar decomposition only supports 2D and 3D matrices.")
+    else:
+        raise Exception(
+            "Polar decomposition only supports 2D and 3D matrices.")

tests/python/test_ast_refactor.py

@@ -967,3 +967,20 @@ def foo() -> ti.i32:
             return bar(1, 2, 3)
 
         foo()
+
+
+@ti.test(experimental_ast_refactor=True)
+def test_raise():
+    dim = 1
+    m = ti.Matrix.field(dim, dim, ti.f32)
+    ti.root.place(m)
+
+    with pytest.raises(Exception) as e:
+
+        @ti.kernel
+        def foo():
+            ti.polar_decompose(m, ti.f32)
+
+        foo()
+    assert e.value.args[
+        0] == "Polar decomposition only supports 2D and 3D matrices."