cleanup sgl-kernel kernels (#3175)

This commit is contained in:
Yineng Zhang
2025-01-27 19:11:01 +08:00
committed by GitHub
parent f8ca66fb49
commit 827aa8730b
8 changed files with 147 additions and 26 deletions

View File

@@ -88,7 +88,7 @@ sources = [
"src/sgl-kernel/csrc/int8_gemm_kernel.cu",
"src/sgl-kernel/csrc/fp8_gemm_kernel.cu",
"src/sgl-kernel/csrc/lightning_attention_decode_kernel.cu",
"src/sgl-kernel/csrc/rotary_embedding.cu",
"src/sgl-kernel/csrc/fused_add_rms_norm_kernel.cu",
"3rdparty/flashinfer/csrc/activation.cu",
"3rdparty/flashinfer/csrc/bmm_fp8.cu",
"3rdparty/flashinfer/csrc/norm.cu",