Update cgraph api 2 (#129)

* update compute graph APIs * fix * Update CGraph API v2
taichi-dev · Nov 22, 2023 · 2eaaa0f · 2eaaa0f
1 parent 54290d7
commit 2eaaa0f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/7_comet/assets/comet.py b/7_comet/assets/comet.py
@@ -48,8 +48,7 @@ def get_save_dir(name, arch):
 sym_arr = ti.graph.Arg(ti.graph.ArgKind.NDARRAY,
                        'arr',
                        ti.f32,
-                       ndim=2,
-                       element_shape=())
+                       ndim=2)
 @ti.kernel
 def img_to_ndarray(arr: ti.types.ndarray()):
     for I in grouped(img):