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 616e8e9 commit f611c9fCopy full SHA for f611c9f
examples/dequantize_gemm/test_example_dequantize_gemm.py
@@ -2,6 +2,7 @@
2
3
import example_dequant_gemv_fp16xint4
4
import example_dequant_gemm_fp4_hopper
5
+import example_dequant_gemm_bf16_mxfp4_hopper
6
7
8
@tilelang.testing.requires_cuda
@@ -15,5 +16,11 @@ def test_example_dequant_gemm_fp4_hopper():
15
16
example_dequant_gemm_fp4_hopper.main()
17
18
19
+@tilelang.testing.requires_cuda
20
+@tilelang.testing.requires_cuda_compute_version_ge(9, 0)
21
+def test_example_dequant_gemm_bf16_mxfp4_hopper():
22
+ example_dequant_gemm_bf16_mxfp4_hopper.main()
23
+
24
25
if __name__ == "__main__":
26
tilelang.testing.main()
0 commit comments