obscura/vllm
2
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2026-08-05 21:38:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

small adjustment to wvSplitKrc (#34410)

Browse Source 
Signed-off-by: Hashem Hashemi <[email protected]>
This commit is contained in:
Hashem Hashemi
2026-02-12 20:26:36 +00:00
committed by GitHub
parent 6d4e27ce29
commit fac4e96940
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
csrc/rocm/skinny_gemms.cu
+1 -2
View File
@@ -1568,8 +1568,7 @@ __global__ void __launch_bounds__(WvPrGrp* THRDS)
{
#endif
unsigned int kOff = k + (thrd * A_CHUNK);
unsigned int kOffcp =
k_str + kOff; // min__(K - A_CHUNK, k_str + kOff);
unsigned int kOffcp = min__(K - A_CHUNK, k_str + kOff);
for (unsigned int n = 0; n < N; n += CHUNKK * sprdN) {
__builtin_amdgcn_global_load_lds(
(int*)(&A[min__(