add commit group

apache · May 19, 2022 · 8aa591e · 8aa591e
1 parent baaf4b8
commit 8aa591e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/python/unittest/test_tir_ptx_cp_async.py b/tests/python/unittest/test_tir_ptx_cp_async.py
@@ -40,6 +40,7 @@ def ptx_cp_async(
                 T.ptx_cp_async(A_shared.data, tx * 128 + 8 * i, A.data, tx * 128 + 8 * i, 16, dtype="float16")
                 )
 
+        T.evaluate(T.ptx_commit_group(dtype="float16"))
         T.evaluate(T.ptx_wait_group(0, dtype="float16"))
 
         for i in range(128):