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:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user