vulkan: fix l2_norm epsilon handling (#20350)

2026-03-12 06:39:41 +01:00
parent aa429cf507
commit 246ffc4b05
3 changed files with 3 additions and 3 deletions
@@ -36,7 +36,7 @@ void main() {
        barrier();
    }

-    const FLOAT_TYPE scale = inversesqrt(max(sum[0], FLOAT_TYPE(p.param1)));
+    const FLOAT_TYPE scale = 1.0f / max(sqrt(sum[0]), FLOAT_TYPE(p.param1));

    [[unroll]] for (uint i0 = tid; i0 < p.ne00; i0 += BLOCK_SIZE) {
        data_d[i3*p.nb13 + i2*p.nb12 + i1*p.nb11 + i0] = D_TYPE(scale * FLOAT_TYPE(data_a[i3*p.nb03 + i2*p.nb02 + i1*p.nb01 + i0]));