Skip to content

Commit ee47adb

Browse files
committed
Fix soft_max data race
1 parent 6464269 commit ee47adb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ggml/src/ggml-webgpu/wgsl-shaders/soft_max.tmpl.wgsl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -300,6 +300,7 @@ fn main(@builtin(workgroup_id) wid: vec3<u32>,
300300
workgroupBarrier();
301301
}
302302
let row_max = scratch[0];
303+
workgroupBarrier();
303304

304305
var sum = 0.0f;
305306
col = lid.x;

0 commit comments

Comments
 (0)