Merge branch 'master' of github.com:taichi-dev/taichi into warnings_s…

…uppression_pr3
taichi-dev · May 7, 2022 · a9b3601 · a9b3601
2 parents 462c129 + 7324903
commit a9b3601
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 4 deletions.
diff --git a/python/taichi/lang/simt/warp.py b/python/taichi/lang/simt/warp.py
@@ -98,8 +98,9 @@ def match_all():
 
 
 def active_mask():
-    # TODO
-    pass
+    return expr.Expr(
+        _ti_core.insert_internal_func_call("cuda_active_mask",
+                                           expr.make_expr_group(), False))
 
 
 def sync(mask):

diff --git a/taichi/backends/metal/device.cpp b/taichi/backends/metal/device.cpp
@@ -246,6 +246,9 @@ class StreamImpl : public Stream {
     auto *cb = static_cast<CommandListImpl *>(cmdlist)->command_buffer();
     commit_command_buffer(cb);
 
+    // FIXME: Implement semaphore mechanism for Metal backend
+    //        and return the actual semaphore corresponding to the submitted
+    //        cmds.
     return nullptr;
   }
   StreamSemaphore submit_synced(

diff --git a/tests/python/test_simt.py b/tests/python/test_simt.py
@@ -251,8 +251,18 @@ def test_match_all():
 
 @test_utils.test(arch=ti.cuda)
 def test_active_mask():
-    # TODO
-    pass
+    a = ti.field(dtype=ti.u32, shape=32)
+
+    @ti.kernel
+    def foo():
+        ti.loop_config(block_dim=16)
+        for i in range(32):
+            a[i] = ti.simt.warp.active_mask()
+
+    foo()
+
+    for i in range(32):
+        assert a[i] == 65535
 
 
 @test_utils.test(arch=ti.cuda)