[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
taichi-dev · Jul 23, 2023 · 5987825 · 5987825
1 parent e1c7811
commit 5987825
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python/taichi/lang/simt/block.py b/python/taichi/lang/simt/block.py
@@ -16,18 +16,21 @@ def sync():
         return impl.call_internal("workgroupBarrier", with_runtime_context=False)
     raise ValueError(f"ti.block.shared_array is not supported for arch {arch}")
 
+
 def sync_all_nonzero(predicate):
     arch = impl.get_runtime().prog.config().arch
     if arch == _ti_core.cuda:
         return impl.call_internal("block_barrier_and_i32", predicate, with_runtime_context=False)
     raise ValueError(f"ti.block.sync_all_nonzero is not supported for arch {arch}")
 
+
 def sync_any_nonzero(predicate):
     arch = impl.get_runtime().prog.config().arch
     if arch == _ti_core.cuda:
         return impl.call_internal("block_barrier_or_i32", predicate, with_runtime_context=False)
     raise ValueError(f"ti.block.sync_any_nonzero is not supported for arch {arch}")
 
+
 def sync_count_nonzero(predicate):
     arch = impl.get_runtime().prog.config().arch
     if arch == _ti_core.cuda:

diff --git a/tests/python/test_simt.py b/tests/python/test_simt.py
@@ -34,6 +34,7 @@ def foo():
     for i in range(32):
         assert a[i] == 0
 
+
 @test_utils.test(arch=ti.cuda)
 def test_sync_all_nonzero():
     a = ti.field(dtype=ti.i32, shape=256)
@@ -62,7 +63,6 @@ def foo():
         assert a[i] == 0
 
 
-
 @test_utils.test(arch=ti.cuda)
 def test_any_nonzero():
     a = ti.field(dtype=ti.i32, shape=32)
@@ -90,6 +90,7 @@ def foo():
     for i in range(32):
         assert a[i] == 1
 
+
 @test_utils.test(arch=ti.cuda)
 def test_sync_any_nonzero():
     a = ti.field(dtype=ti.i32, shape=256)
@@ -117,6 +118,7 @@ def foo():
     for i in range(256):
         assert a[i] == 1
 
+
 @test_utils.test(arch=ti.cuda)
 def test_sync_count_nonzero():
     a = ti.field(dtype=ti.i32, shape=256)