ggml webgpu: actually add softmax, fix rms_norm offset (#16400)

* implement soft_max

* Fix soft_max data race

* Temporary fix, wait on each submit
This commit is contained in:
Reese Levine
2025-10-04 20:59:31 -07:00
committed by GitHub
parent 86df2c9ae4
commit 35266573b9
3 changed files with 10 additions and 1 deletions
@@ -300,6 +300,7 @@ fn main(@builtin(workgroup_id) wid: vec3<u32>,
workgroupBarrier();
}
let row_max = scratch[0];
workgroupBarrier();
var sum = 0.0f;
col = lid.x;