Skip to content

Commit ca216e7

Browse files
committed
lint fix
1 parent fb9739c commit ca216e7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

examples/deepseek_v32/test_tilelang_example_deepseek_v32.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,22 @@
66
from sparse_mla_fwd import test_sparse_mla_fwd
77
from sparse_mla_fwd_pipelined import test_sparse_mla_fwd_pipelined
88

9+
910
def test_example_topk_selector():
1011
test_topk_selector()
1112

13+
1214
def test_example_fp8_lighting_indexer():
1315
test_fp8_lighting_indexer()
1416

17+
1518
@tilelang.testing.requires_cuda
1619
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
1720
def test_example_sparse_mla_fwd():
1821
# small shapes for testing
1922
test_sparse_mla_fwd(S=1024, SKV=2048, H=128, HKV=1, DQK=576, DV=512, topk=256)
2023

24+
2125
@tilelang.testing.requires_cuda
2226
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
2327
def test_example_sparse_mla_fwd_pipelined():

0 commit comments

Comments
 (0)