mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-03-16 00:55:24 +00:00
CUDA: fix shared memory access condition for mmv (llama/10740)
This commit is contained in:
parent
37df308a2a
commit
ce2b75d2fb
@ -57,7 +57,7 @@ static __global__ void mul_mat_vec(
|
||||
if (block_size > WARP_SIZE) {
|
||||
buf_iw[tid/WARP_SIZE] = sumf;
|
||||
__syncthreads();
|
||||
if (tid > WARP_SIZE) {
|
||||
if (tid >= WARP_SIZE) {
|
||||
return;
|
||||
}
|
||||
sumf = buf_iw[tid];
|
||||
|
Loading…
x
Reference in New Issue
Block a user