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

[1/2] Speed up prefill mla attention (#10156)

Browse Source
This commit is contained in:
fzyzcjy
2025-09-09 00:00:33 +08:00
committed by GitHub
parent 2c2b19b18b
commit 0096798ed6
6 changed files with 130 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
sgl-kernel/CMakeLists.txt
View File

@@ -259,6 +259,7 @@ set(SOURCES
"csrc/elementwise/activation.cu"
"csrc/elementwise/cast.cu"
"csrc/elementwise/copy.cu"
"csrc/elementwise/concat_mla.cu"
"csrc/elementwise/fused_add_rms_norm_kernel.cu"
"csrc/elementwise/rope.cu"
"csrc/common_extension.cc"