From: Neo Zhang Jianyu <redacted>
Date: Wed, 10 Dec 2025 08:59:57 +0000 (+0800)
Subject: fix softmax for iGPU (llama/17838)
X-Git-Tag: upstream/0.9.4.395~33
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=3fd1552d3ecd6c244f0a18893857722ffec4c296;p=pkg%2Fggml%2Fsources%2Fggml

fix softmax for iGPU (llama/17838)
---

diff --git a/src/ggml-sycl/softmax.cpp b/src/ggml-sycl/softmax.cpp
index 83b7c71b..b41124ac 100644
--- a/src/ggml-sycl/softmax.cpp
+++ b/src/ggml-sycl/softmax.cpp
@@ -277,7 +277,7 @@ static void soft_max_f32_sycl(const float *x, const T *mask,
     const int id       = get_current_device_id();
     const size_t smpbo = ggml_sycl_info().devices[id].smpbo;
 
-    if (nbytes_shared <= smpbo) {
+    if (nbytes_shared <= smpbo && ncols_x <= max_block_size) {
         launch_soft_max_kernels<32, 64, 128, 256, 512, 1024, 2048, 4096>(
             x, mask, sinks, dst, params, stream, block_dims, block_nums,
             nbytes_shared);