Add dsv3 router gemm kernel (#7627)

This commit is contained in:
Baizhou Zhang
2025-06-29 23:31:55 -07:00
committed by GitHub
parent 22352d47a9
commit 7248272ccc
8 changed files with 398 additions and 0 deletions

View File

@@ -222,6 +222,7 @@ set(SOURCES
"csrc/gemm/awq_kernel.cu"
"csrc/gemm/bmm_fp8.cu"
"csrc/gemm/dsv3_fused_a_gemm.cu"
"csrc/gemm/dsv3_router_gemm.cu"
"csrc/gemm/fp8_blockwise_gemm_kernel.cu"
"csrc/gemm/fp8_gemm_kernel.cu"
"csrc/gemm/int8_gemm_kernel.cu"