From 084e2271d492b746669fbdb8eb1d4ba46be6ef24 Mon Sep 17 00:00:00 2001 From: Chris Duncan Date: Mon, 3 Feb 2025 15:36:40 -0800 Subject: [PATCH] Deprecate x and y temp vars for first two rounds. --- src/shaders/compute.wgsl | 84 +++++++++++++++------------------------- 1 file changed, 32 insertions(+), 52 deletions(-) diff --git a/src/shaders/compute.wgsl b/src/shaders/compute.wgsl index c09a1f7..dd229b4 100644 --- a/src/shaders/compute.wgsl +++ b/src/shaders/compute.wgsl @@ -156,57 +156,49 @@ fn main(id: vec3) { ****************************************************************************/ /** - * r=0, i=0, a=v[0], b=v[4], c=v[8], d=v[12] - * r=0, i=1, a=v[1], b=v[5], c=v[9], d=v[13] + * r=0, i=0, a=v[0], b=v[4], c=v[8], d=v[12], m[sigma]=0, m[sigma+1]=1 + * r=0, i=1, a=v[1], b=v[5], c=v[9], d=v[13], m[sigma]=2, m[sigma+1]=3 */ - x = vec4(m0, m2); - y = vec4(m1, m3); - v01 += v45 + vec4(0u, u32(v01.x + v45.x < v01.x), 0u, u32(v01.z + v45.z < v01.z)); - v01 += x + vec4(0u, u32(v01.x + x.x < v01.x), 0u, u32(v01.z + x.z < v01.z)); + v01 += vec4(m0, m2) + vec4(0u, u32(v01.x + m0.x < v01.x), 0u, u32(v01.z + m2.x < v01.z)); vCD = (vCD ^ v01).yxwz; v89 += vCD + vec4(0u, u32(v89.x + vCD.x < v89.x), 0u, u32(v89.z + vCD.z < v89.z)); v45 = ((v45 ^ v89) >> ROTATE_24) | ((v45 ^ v89).yxwz << ROTATE_8); v01 += v45 + vec4(0u, u32(v01.x + v45.x < v01.x), 0u, u32(v01.z + v45.z < v01.z)); - v01 += y + vec4(0u, u32(v01.x + y.x < v01.x), 0u, u32(v01.z + y.z < v01.z)); + v01 += vec4(m1, m3) + vec4(0u, u32(v01.x + m1.x < v01.x), 0u, u32(v01.z + m3.x < v01.z)); vCD = ((vCD ^ v01) >> ROTATE_16) | ((vCD ^ v01).yxwz << ROTATE_16); v89 += vCD + vec4(0u, u32(v89.x + vCD.x < v89.x), 0u, u32(v89.z + vCD.z < v89.z)); v45 = ((v45 ^ v89) << ROTATE_1) | ((v45 ^ v89).yxwz >> ROTATE_31); /** - * r=0, i=2, a=v[2], b=v[6], c=v[10], d=v[14] - * r=0, i=3, a=v[3], b=v[7], c=v[11], d=v[15] + * r=0, i=2, a=v[2], b=v[6], c=v[10], d=v[14], m[sigma]=4, m[sigma+1]=5 + * r=0, i=3, a=v[3], b=v[7], c=v[11], d=v[15], m[sigma]=6, m[sigma+1]=7 */ - x = vec4(m4, VEC2_0); - // y = VEC4_0; - v23 += v67 + vec4(0u, u32(v23.x + v67.x < v23.x), 0u, u32(v23.z + v67.z < v23.z)); - v23 += x + vec4(0u, u32(v23.x + x.x < v23.x), 0u, u32(v23.z + x.z < v23.z)); + v23 += vec4(m4, 0u, 0u) + vec4(0u, u32(v23.x + m4.x < v23.x), 0u, 0u); vEF = (vEF ^ v23).yxwz; vAB += vEF + vec4(0u, u32(vAB.x + vEF.x < vAB.x), 0u, u32(vAB.z + vEF.z < vAB.z)); v67 = ((v67 ^ vAB) >> ROTATE_24) | ((v67 ^ vAB).yxwz << ROTATE_8); v23 += v67 + vec4(0u, u32(v23.x + v67.x < v23.x), 0u, u32(v23.z + v67.z < v23.z)); - // v23 += y + vec4(0u, u32(v23.x + y.x < v23.x), 0u, u32(v23.z + y.z < v23.z)); + // NOP vEF = ((vEF ^ v23) >> ROTATE_16) | ((vEF ^ v23).yxwz << ROTATE_16); vAB += vEF + vec4(0u, u32(vAB.x + vEF.x < vAB.x), 0u, u32(vAB.z + vEF.z < vAB.z)); v67 = ((v67 ^ vAB) << ROTATE_1) | ((v67 ^ vAB).yxwz >> ROTATE_31); /** - * r=0, i=4, a=v[0], b=v[5], c=v[10], d=v[15] - * r=0, i=5, a=v[1], b=v[6], c=v[11], d=v[12] + * r=0, i=4, a=v[0], b=v[5], c=v[10], d=v[15], m[sigma]=8, m[sigma+1]=9 + * r=0, i=5, a=v[1], b=v[6], c=v[11], d=v[12], m[sigma]=10, m[sigma+1]=11 */ b = vec4(v45.zw, v67.xy); d = vec4(vEF.zw, vCD.xy); - // x = VEC4_0; - // y = VEC4_0; v01 += b + vec4(0u, u32(v01.x + b.x < v01.x), 0u, u32(v01.z + b.z < v01.z)); - // v01 += x + vec4(0u, u32(v01.x + x.x < v01.x), 0u, u32(v01.z + x.z < v01.z)); + // NOP d = (d ^ v01).yxwz; vAB += d + vec4(0u, u32(vAB.x + d.x < vAB.x), 0u, u32(vAB.z + d.z < vAB.z)); b = ((b ^ vAB) >> ROTATE_24) | ((b ^ vAB).yxwz << ROTATE_8); v01 += b + vec4(0u, u32(v01.x + b.x < v01.x), 0u, u32(v01.z + b.z < v01.z)); - // v01 += y + vec4(0u, u32(v01.x + y.x < v01.x), 0u, u32(v01.z + y.z < v01.z)); + // NOP d = ((d ^ v01) >> ROTATE_16) | ((d ^ v01).yxwz << ROTATE_16); vAB += d + vec4(0u, u32(vAB.x + d.x < vAB.x), 0u, u32(vAB.z + d.z < vAB.z)); b = ((b ^ vAB) << ROTATE_1) | ((b ^ vAB).yxwz >> ROTATE_31); @@ -217,21 +209,19 @@ fn main(id: vec3) { vCD = vec4(d.zw, vCD.zw); /** - * r=0, i=6, a=v[2], b=v[7], c=v[8], d=v[13] - * r=0, i=7, a=v[3], b=v[4], c=v[9], d=v[14] + * r=0, i=6, a=v[2], b=v[7], c=v[8], d=v[13], m[sigma]=12, m[sigma+1]=13 + * r=0, i=7, a=v[3], b=v[4], c=v[9], d=v[14], m[sigma]=14, m[sigma+1]=15 */ b = vec4(v67.zw, v45.xy); d = vec4(vCD.zw, vEF.xy); - // x = VEC4_0; - // y = VEC4_0; v23 += b + vec4(0u, u32(v23.x + b.x < v23.x), 0u, u32(v23.z + b.z < v23.z)); - // v23 += x + vec4(0u, u32(v23.x + x.x < v23.x), 0u, u32(v23.z + x.z < v23.z)); + // NOP d = (d ^ v23).yxwz; v89 += d + vec4(0u, u32(v89.x + d.x < v89.x), 0u, u32(v89.z + d.z < v89.z)); b = ((b ^ v89) >> ROTATE_24) | ((b ^ v89).yxwz << ROTATE_8); v23 += b + vec4(0u, u32(v23.x + b.x < v23.x), 0u, u32(v23.z + b.z < v23.z)); - // v23 += y + vec4(0u, u32(v23.x + y.x < v23.x), 0u, u32(v23.z + y.z < v23.z)); + // NOP d = ((d ^ v23) >> ROTATE_16) | ((d ^ v23).yxwz << ROTATE_16); v89 += d + vec4(0u, u32(v89.x + d.x < v89.x), 0u, u32(v89.z + d.z < v89.z)); b = ((b ^ v89) << ROTATE_1) | ((b ^ v89).yxwz >> ROTATE_31); @@ -249,57 +239,49 @@ fn main(id: vec3) { ****************************************************************************/ /** - * r=1, i=0, a=v[0], b=v[4], c=v[8], d=v[12] - * r=1, i=1, a=v[1], b=v[5], c=v[9], d=v[13] + * r=1, i=0, a=v[0], b=v[4], c=v[8], d=v[12], m[sigma]=?, m[sigma+1]=4 + * r=1, i=1, a=v[1], b=v[5], c=v[9], d=v[13], m[sigma]=?, m[sigma+1]=? */ - x = vec4(VEC2_0, m4); - // y = VEC4_0; - v01 += v45 + vec4(0u, u32(v01.x + v45.x < v01.x), 0u, u32(v01.z + v45.z < v01.z)); - v01 += x + vec4(0u, u32(v01.x + x.x < v01.x), 0u, u32(v01.z + x.z < v01.z)); + v01 += vec4(0u, 0u, m4) + vec4(0u, 0u, 0u, u32(v01.z + m4.x < v01.z)); vCD = (vCD ^ v01).yxwz; v89 += vCD + vec4(0u, u32(v89.x + vCD.x < v89.x), 0u, u32(v89.z + vCD.z < v89.z)); v45 = ((v45 ^ v89) >> ROTATE_24) | ((v45 ^ v89).yxwz << ROTATE_8); v01 += v45 + vec4(0u, u32(v01.x + v45.x < v01.x), 0u, u32(v01.z + v45.z < v01.z)); - // v01 += y + vec4(0u, u32(v01.x + y.x < v01.x), 0u, u32(v01.z + y.z < v01.z)); + // NOP vCD = ((vCD ^ v01) >> ROTATE_16) | ((vCD ^ v01).yxwz << ROTATE_16); v89 += vCD + vec4(0u, u32(v89.x + vCD.x < v89.x), 0u, u32(v89.z + vCD.z < v89.z)); v45 = ((v45 ^ v89) << ROTATE_1) | ((v45 ^ v89).yxwz >> ROTATE_31); /** - * r=1, i=2, a=v[2], b=v[6], c=v[10], d=v[14] - * r=1, i=3, a=v[3], b=v[7], c=v[11], d=v[15] + * r=1, i=2, a=v[2], b=v[6], c=v[10], d=v[14], m[sigma]=?, m[sigma+1]=? + * r=1, i=3, a=v[3], b=v[7], c=v[11], d=v[15], m[sigma]=?, m[sigma+1]=? */ - // x = VEC4_0; - // y = VEC4_0; - v23 += v67 + vec4(0u, u32(v23.x + v67.x < v23.x), 0u, u32(v23.z + v67.z < v23.z)); - // v23 += x + vec4(0u, u32(v23.x + x.x < v23.x), 0u, u32(v23.z + x.z < v23.z)); + // NOP vEF = (vEF ^ v23).yxwz; vAB += vEF + vec4(0u, u32(vAB.x + vEF.x < vAB.x), 0u, u32(vAB.z + vEF.z < vAB.z)); v67 = ((v67 ^ vAB) >> ROTATE_24) | ((v67 ^ vAB).yxwz << ROTATE_8); v23 += v67 + vec4(0u, u32(v23.x + v67.x < v23.x), 0u, u32(v23.z + v67.z < v23.z)); - // v23 += y + vec4(0u, u32(v23.x + y.x < v23.x), 0u, u32(v23.z + y.z < v23.z)); + // NOP vEF = ((vEF ^ v23) >> ROTATE_16) | ((vEF ^ v23).yxwz << ROTATE_16); vAB += vEF + vec4(0u, u32(vAB.x + vEF.x < vAB.x), 0u, u32(vAB.z + vEF.z < vAB.z)); v67 = ((v67 ^ vAB) << ROTATE_1) | ((v67 ^ vAB).yxwz >> ROTATE_31); /** - * r=1, i=4, a=v[0], b=v[5], c=v[10], d=v[15] - * r=1, i=5, a=v[1], b=v[6], c=v[11], d=v[12] + * r=1, i=4, a=v[0], b=v[5], c=v[10], d=v[15], m[sigma]=1, m[sigma+1]=0 + * r=1, i=5, a=v[1], b=v[6], c=v[11], d=v[12], m[sigma]=?, m[sigma+1]=2 */ b = vec4(v45.zw, v67.xy); d = vec4(vEF.zw, vCD.xy); - x = vec4(m1, m0); - y = vec4(VEC2_0, m2); v01 += b + vec4(0u, u32(v01.x + b.x < v01.x), 0u, u32(v01.z + b.z < v01.z)); - v01 += x + vec4(0u, u32(v01.x + x.x < v01.x), 0u, u32(v01.z + x.z < v01.z)); + v01 += vec4(m1, m0) + vec4(0u, u32(v01.x + m1.x < v01.x), 0u, u32(v01.z + m0.x < v01.z)); d = (d ^ v01).yxwz; vAB += d + vec4(0u, u32(vAB.x + d.x < vAB.x), 0u, u32(vAB.z + d.z < vAB.z)); b = ((b ^ vAB) >> ROTATE_24) | ((b ^ vAB).yxwz << ROTATE_8); v01 += b + vec4(0u, u32(v01.x + b.x < v01.x), 0u, u32(v01.z + b.z < v01.z)); - v01 += y + vec4(0u, u32(v01.x + y.x < v01.x), 0u, u32(v01.z + y.z < v01.z)); + v01 += vec4(0u, 0u, m2) + vec4(0u, 0u, 0u, u32(v01.z + m2.x < v01.z)); d = ((d ^ v01) >> ROTATE_16) | ((d ^ v01).yxwz << ROTATE_16); vAB += d + vec4(0u, u32(vAB.x + d.x < vAB.x), 0u, u32(vAB.z + d.z < vAB.z)); b = ((b ^ vAB) << ROTATE_1) | ((b ^ vAB).yxwz >> ROTATE_31); @@ -310,21 +292,19 @@ fn main(id: vec3) { vCD = vec4(d.zw, vCD.zw); /** - * r=1, i=6, a=v[2], b=v[7], c=v[8], d=v[13] - * r=1, i=7, a=v[3], b=v[4], c=v[9], d=v[14] + * r=1, i=6, a=v[2], b=v[7], c=v[8], d=v[13], m[sigma]=?, m[sigma+1]=? + * r=1, i=7, a=v[3], b=v[4], c=v[9], d=v[14], m[sigma]=?, m[sigma+1]=3 */ b = vec4(v67.zw, v45.xy); d = vec4(vCD.zw, vEF.xy); - // x = VEC4_0; - y = vec4(VEC2_0, m3); v23 += b + vec4(0u, u32(v23.x + b.x < v23.x), 0u, u32(v23.z + b.z < v23.z)); - // v23 += x + vec4(0u, u32(v23.x + x.x < v23.x), 0u, u32(v23.z + x.z < v23.z)); + // NOP d = (d ^ v23).yxwz; v89 += d + vec4(0u, u32(v89.x + d.x < v89.x), 0u, u32(v89.z + d.z < v89.z)); b = ((b ^ v89) >> ROTATE_24) | ((b ^ v89).yxwz << ROTATE_8); v23 += b + vec4(0u, u32(v23.x + b.x < v23.x), 0u, u32(v23.z + b.z < v23.z)); - v23 += y + vec4(0u, u32(v23.x + y.x < v23.x), 0u, u32(v23.z + y.z < v23.z)); + v23 += vec4(0u, 0u, m3) + vec4(0u, 0u, 0u, u32(v23.z + m3.x < v23.z)); d = ((d ^ v23) >> ROTATE_16) | ((d ^ v23).yxwz << ROTATE_16); v89 += d + vec4(0u, u32(v89.x + d.x < v89.x), 0u, u32(v89.z + d.z < v89.z)); b = ((b ^ v89) << ROTATE_1) | ((b ^ v89).yxwz >> ROTATE_31); -- 2.34.1