We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ff3e04d commit c6ab014Copy full SHA for c6ab014
examples/deepseek_v32/test_tilelang_example_deepseek_v32.py
@@ -34,7 +34,7 @@ def test_example_sparse_mla_fwd_pipelined():
34
@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
35
def test_example_sparse_mla_bwd():
36
test_sparse_mla_bwd(
37
- B=1, S=4096, SKV=4096, H=64, HKV=1, DQKV=576, DV=512, topk=2048, dtype=torch.bfloat16)
+ B=1, S=4096, SKV=4096, H=64, HKV=1, DQKV=576, DV=512, topk=2048)
38
39
40
if __name__ == "__main__":
0 commit comments