From 985a095ea38eb47071b23f347c06813f0c4d420c Mon Sep 17 00:00:00 2001
From: Chris Duncan <chris@zoso.dev>
Date: Fri, 17 Jan 2025 10:47:09 -0800
Subject: [PATCH] Avoid main function call entirely if work was already found.

---
 src/shaders/compute.wgsl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/shaders/compute.wgsl b/src/shaders/compute.wgsl
index 79486d0..beec4a9 100644
--- a/src/shaders/compute.wgsl
+++ b/src/shaders/compute.wgsl
@@ -35,6 +35,7 @@ const ROTATE_31 = vec2(31u, 31u);
 */
 @compute @workgroup_size(8,8)
 fn search(@builtin(global_invocation_id) global_id: vec3<u32>) {
+	if (atomicLoad(&work.found) != 0u) { return; }
 	main(global_id);
 }
 
@@ -44,6 +45,7 @@ fn search(@builtin(global_invocation_id) global_id: vec3<u32>) {
 */
 @compute @workgroup_size(1)
 fn validate(@builtin(global_invocation_id) global_id: vec3<u32>) {
+	if (atomicLoad(&work.found) != 0u) { return; }
 	main(global_id);
 }
 
@@ -58,8 +60,6 @@ fn validate(@builtin(global_invocation_id) global_id: vec3<u32>) {
 * performance, but the variable name still contains the original index digit.
 */
 fn main(id: vec3<u32>) {
-	if (atomicLoad(&work.found) != 0u) { return; }
-
 	let threshold: u32 = ubo.threshold;
 
 	/**
-- 
2.34.1