minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[AMD] Align alternative sgl-kernel wheel (#17092)

Browse Source
This commit is contained in:
Alan Kao
2026-01-14 21:26:45 -08:00
committed by GitHub
parent 98096b5e02
commit 43fe3a4ddf
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
3rdparty/amd/sgl-kernel/CMakeLists_rocm.txt
Vendored
+3 -2
View File
@@ -102,14 +102,15 @@ ${PROJ_ROOT}/csrc/allreduce/deterministic_all_reduce.hip
${PROJ_ROOT}/csrc/allreduce/quick_all_reduce.hip ${PROJ_ROOT}/csrc/allreduce/quick_all_reduce.hip
${PROJ_ROOT}/csrc/common_extension_rocm.cc ${PROJ_ROOT}/csrc/common_extension_rocm.cc
${PROJ_ROOT}/csrc/elementwise/activation.hip ${PROJ_ROOT}/csrc/elementwise/activation.hip
${PROJ_ROOT}/csrc/elementwise/pos_enc.hip
${PROJ_ROOT}/csrc/elementwise/topk.hip ${PROJ_ROOT}/csrc/elementwise/topk.hip
${PROJ_ROOT}/csrc/grammar/apply_token_bitmask_inplace_hip.hip ${PROJ_ROOT}/csrc/grammar/apply_token_bitmask_inplace_hip.hip
${PROJ_ROOT}/csrc/kvcacheio/transfer.hip
${PROJ_ROOT}/csrc/moe/moe_align_kernel.hip ${PROJ_ROOT}/csrc/moe/moe_align_kernel.hip
${PROJ_ROOT}/csrc/moe/moe_topk_softmax_kernels.hip ${PROJ_ROOT}/csrc/moe/moe_topk_softmax_kernels.hip
${PROJ_ROOT}/csrc/moe/moe_topk_sigmoid_kernels.hip ${PROJ_ROOT}/csrc/moe/moe_topk_sigmoid_kernels.hip
${PROJ_ROOT}/csrc/sgl_diffusion/elementwise/timestep_embedding.hip
${PROJ_ROOT}/csrc/speculative/eagle_utils.hip ${PROJ_ROOT}/csrc/speculative/eagle_utils.hip
${PROJ_ROOT}/csrc/kvcacheio/transfer.hip
${PROJ_ROOT}/csrc/elementwise/pos_enc.hip
) )
set_source_files_properties( set_source_files_properties(
${SOURCES} ${SOURCES}
3rdparty/amd/sgl-kernel/rocm_hipify.py
Vendored
+3 -2
View File
@@ -17,14 +17,15 @@ sources = [
"csrc/allreduce/quick_all_reduce.cu", "csrc/allreduce/quick_all_reduce.cu",
"csrc/common_extension_rocm.cc", "csrc/common_extension_rocm.cc",
"csrc/elementwise/activation.cu", "csrc/elementwise/activation.cu",
"csrc/elementwise/pos_enc.cu",
"csrc/elementwise/topk.cu", "csrc/elementwise/topk.cu",
"csrc/grammar/apply_token_bitmask_inplace_cuda.cu", "csrc/grammar/apply_token_bitmask_inplace_cuda.cu",
"csrc/kvcacheio/transfer.cu",
"csrc/moe/moe_align_kernel.cu", "csrc/moe/moe_align_kernel.cu",
"csrc/moe/moe_topk_softmax_kernels.cu", "csrc/moe/moe_topk_softmax_kernels.cu",
"csrc/moe/moe_topk_sigmoid_kernels.cu", "csrc/moe/moe_topk_sigmoid_kernels.cu",
"csrc/sgl_diffusion/elementwise/timestep_embedding.cu",
"csrc/speculative/eagle_utils.cu", "csrc/speculative/eagle_utils.cu",
"csrc/kvcacheio/transfer.cu",
"csrc/elementwise/pos_enc.cu",
] ]
libraries = ["hiprtc", "amdhip64", "c10", "torch", "torch_python"] libraries = ["hiprtc", "amdhip64", "c10", "torch", "torch_python"]