1

Author: Claude

ggml/src/ggml-vulkan/vulkan-shaders/ssm_scan.comp

@@ -111,12 +111,12 @@ void main() {
             int warp_id = tid / WARP_SIZE;
             int warp_offset = warp_id * WARP_SIZE;
 
-            warp_sdata[warp_offset + lane] = y;
+            warp_sdata[tid] = y;
             barrier();
 
             [[unroll]] for (int offset = WARP_SIZE / 2; offset > 0; offset >>= 1) {
                 if (lane < offset) {
-                    warp_sdata[warp_offset + lane] += warp_sdata[warp_offset + lane + offset];
+                    warp_sdata[tid] += warp_sdata[tid + offset];
                 }
                 barrier();
             }