Просмотр исходного кода

[SYCL] fix softmax r2r result wrong issue (#7811)

pengxin99 1 год назад
Родитель
Сommit
d5c938cd77
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      ggml-sycl.cpp

+ 1 - 0
ggml-sycl.cpp

@@ -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;
         }