mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-03-15 08:38:24 +01:00
fix softmax r2r result wrong issue (llama/7811)
This commit is contained in:
parent
e08c62149b
commit
dc01aadb18
@ -9108,6 +9108,7 @@ static void soft_max_f32(const float * x, const float * mask, float * dst, const
|
||||
// find the sum of exps in the block
|
||||
tmp = warp_reduce_sum(tmp, item_ct1);
|
||||
if (block_size > WARP_SIZE) {
|
||||
item_ct1.barrier(sycl::access::fence_space::local_space);
|
||||
if (warp_id == 0) {
|
||||
buf[lane_id] = 0.f;
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user