Skip to content

Commit c6ab014

Browse files
committed
lint fix
1 parent ff3e04d commit c6ab014

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/deepseek_v32/test_tilelang_example_deepseek_v32.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def test_example_sparse_mla_fwd_pipelined():
3434
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
3535
def test_example_sparse_mla_bwd():
3636
test_sparse_mla_bwd(
37-
B=1, S=4096, SKV=4096, H=64, HKV=1, DQKV=576, DV=512, topk=2048, dtype=torch.bfloat16)
37+
B=1, S=4096, SKV=4096, H=64, HKV=1, DQKV=576, DV=512, topk=2048)
3838

3939

4040
if __name__ == "__main__":

0 commit comments

Comments
 (0)