Skip to content

Commit 031d4ac

Browse files
committed
format
1 parent 5f6669e commit 031d4ac

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

testing/python/debug/test_device_assert.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ def test_device_assert_no_trigger():
1111
prev = os.environ.get("TILELANG_DEBUG")
1212
os.environ["TILELANG_DEBUG"] = "1"
1313
try:
14+
1415
@T.prim_func
1516
def program():
1617
with T.Kernel(threads=128):
@@ -26,6 +27,7 @@ def program():
2627
else:
2728
os.environ["TILELANG_DEBUG"] = prev
2829

30+
2931
if __name__ == "__main__":
3032

3133
tilelang.testing.main()

tilelang/language/print.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,8 +135,10 @@ def print_local_buffer_with_condition(condition: tir.PrimExpr,
135135

136136

137137
from tilelang.utils.target import check_cuda_availability
138+
138139
_IS_CUDA_AVAILABLE = check_cuda_availability()
139140

141+
140142
@macro
141143
def device_assert(condition: tir.PrimExpr, msg: str = ""):
142144
"""

0 commit comments

Comments
 (0)