[Perf] Add Flashinfer DeepGEMM SM90 for SwapAB Optimization (#15514)

Co-authored-by: Brayden Zhong <b8zhong@users.noreply.github.com>
This commit is contained in:
b8zhong
2026-02-01 08:56:23 +08:00
committed by GitHub
co-authored by Brayden Zhong
parent 9951a1ae07
commit 398d13a189
3 changed files with 80 additions and 2 deletions
+5
View File
@@ -69,5 +69,10 @@ class TestFP8BlockwiseGemmFlashinferTrtllm(FP8BlockwiseGemmBase, unittest.TestCa
backend = "flashinfer_trtllm"
@unittest.skipIf(get_device_sm() != 90, "Test requires CUDA SM 90")
class TestFP8BlockwiseGemmFlashinferDeepGemm(FP8BlockwiseGemmBase, unittest.TestCase):
backend = "flashinfer_deepgemm"
if __name__ == "__main__":
unittest.main()