EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Add bf16 output option for dsv3_router_gemm kernel (#7999)

Browse Source
This commit is contained in:
Baizhou Zhang
2025-07-19 18:49:37 -07:00
committed by GitHub
parent 4540a4666a
commit 282eb59ff3
7 changed files with 465 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sgl-kernel/CMakeLists.txt
View File

@@ -222,7 +222,9 @@ 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/dsv3_router_gemm_bf16_out.cu"
"csrc/gemm/dsv3_router_gemm_entry.cu"
"csrc/gemm/dsv3_router_gemm_float_out.cu"
"csrc/gemm/fp8_blockwise_gemm_kernel.cu"
"csrc/gemm/fp8_gemm_kernel.cu"
"csrc/gemm/int8_gemm_kernel.cu"