chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[sgl-kernel][3/N]Support Expert Specialization Grouped GEMM (#11674)

Browse Source
This commit is contained in:
Qi Yuhang
2025-10-15 13:39:31 -07:00
committed by GitHub
parent f226d3da2a
commit 6c01844f45
7 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sgl-kernel/include/sgl_kernel_ops.h
+2 -1
View File
@@ -835,4 +835,5 @@ void es_fp8_blockwise_scaled_grouped_mm(
const torch::Tensor& stride_b,
const torch::Tensor& stride_d,
const torch::Tensor& problem_sizes,
const torch::Tensor& expert_offsets);
const torch::Tensor& expert_offsets,
const torch::Tensor& workspace);