Support CuteDSL mm_fp4 backend (#18801)

2026-03-19 14:20:01 -07:00
parent d8ece7fb22
commit b42b9f6e1a
5 changed files with 109 additions and 219 deletions
@@ -81,5 +81,10 @@ class TestFP4GemmFlashinferTrtllm(FP4GemmBase, unittest.TestCase):
    backend = "flashinfer_trtllm"


+@unittest.skipIf(get_device_sm() < 100, "Test requires CUDA SM 100 or higher")
+class TestFP4GemmFlashinferCutedsl(FP4GemmBase, unittest.TestCase):
+    backend = "flashinfer_cutedsl"
+
+
 if __name__ == "__main__":
    unittest.main()