Skip to content

Commit f611c9f

Browse files
committed
[MXFP4] Add test for bf16&mxfp4 gemm
1 parent 616e8e9 commit f611c9f

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

examples/dequantize_gemm/test_example_dequantize_gemm.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import example_dequant_gemv_fp16xint4
44
import example_dequant_gemm_fp4_hopper
5+
import example_dequant_gemm_bf16_mxfp4_hopper
56

67

78
@tilelang.testing.requires_cuda
@@ -15,5 +16,11 @@ def test_example_dequant_gemm_fp4_hopper():
1516
example_dequant_gemm_fp4_hopper.main()
1617

1718

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+
1825
if __name__ == "__main__":
1926
tilelang.testing.main()

0 commit comments

Comments
 (0)