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

fix moe_align_kernel shm init not sync bug (#3534)

Browse Source
This commit is contained in:
Xiaoyu Zhang
2025-02-13 16:47:00 +08:00
committed by GitHub
parent bf2a70872e
commit f076328bb7
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgl-kernel/src/sgl-kernel/csrc/moe_align_kernel.cu
View File

@@ -53,6 +53,8 @@ __global__ void moe_align_block_size_kernel(scalar_t* __restrict__ topk_ids, int
} }
} }
__syncthreads();
const size_t tokens_per_thread = CEILDIV(numel, blockDim.x); const size_t tokens_per_thread = CEILDIV(numel, blockDim.x);
const size_t start_idx = threadIdx.x * tokens_per_thread; const size_t start_idx = threadIdx.x * tokens_per_thread;