]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
fix softmax r2r result wrong issue (llama/7811)
authorpengxin99 <redacted>
Fri, 7 Jun 2024 06:28:26 +0000 (14:28 +0800)
committerGeorgi Gerganov <redacted>
Sat, 15 Jun 2024 19:05:47 +0000 (22:05 +0300)
src/ggml-sycl.cpp

index 3ff76474d7e72e7373e564b7de830169bd92a77e..0a645b2e1db8023d8e8ad47fb50dc92919a40d86 100644 (file)
@@ -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;
         }