From 230e9ee5c9569b70a31a654dbc0386f8d2450af0 Mon Sep 17 00:00:00 2001
From: Jeran <thejeran@gmail.com>
Date: Thu, 30 Apr 2026 16:13:30 +0200
Subject: [PATCH] Added IsNan to shaders w/out boilerplate

---
 src/components/computation/WGSLShaders.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/components/computation/WGSLShaders.ts b/src/components/computation/WGSLShaders.ts
index 42339bf6..aab2c557 100644
--- a/src/components/computation/WGSLShaders.ts
+++ b/src/components/computation/WGSLShaders.ts
@@ -492,6 +492,7 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(2) var<storage, read_write> outputData: array<${precision}>;
         @group(0) @binding(3) var<uniform> params: Params;
 
+        ${isNaNFunc}
         @compute @workgroup_size(16, 16, 1)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;
@@ -615,6 +616,7 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(2) var<storage, read_write> outputData: array<${precision}>;
         @group(0) @binding(3) var<uniform> params: Params;
 
+        ${isNaNFunc}
         @compute @workgroup_size(16, 16, 1)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;
@@ -697,6 +699,8 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(2) var<storage, read_write> outputData: array<${precision}>;
         @group(0) @binding(3) var<uniform> params: Params;
 
+        ${isNaNFunc}
+
         @compute @workgroup_size(16, 16, 1)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;
@@ -950,6 +954,7 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(2) var<storage, read_write> outputData: array<${precision}>;
         @group(0) @binding(3) var<uniform> params: Params;
 
+        ${isNaNFunc}
         @compute @workgroup_size(4, 4, 4)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;
@@ -1055,7 +1060,7 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(1) var<storage, read> secondData: array<${precision}>;
         @group(0) @binding(2) var<storage, read_write> outputData: array<${precision}>;
         @group(0) @binding(3) var<uniform> params: Params;
-
+        ${isNaNFunc}
         @compute @workgroup_size(4, 4, 4)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;
@@ -1169,7 +1174,7 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(1) var<storage, read> secondData: array<${precision}>;
         @group(0) @binding(2) var<storage, read_write> outputData: array<${precision}>;
         @group(0) @binding(3) var<uniform> params: Params;
-
+        ${isNaNFunc}
         @compute @workgroup_size(4, 4, 4)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;
@@ -1411,7 +1416,7 @@ export const createShaders = (precision: Precision) => {
         @group(0) @binding(0) var<storage, read> inputData: array<${precision}>;
         @group(0) @binding(1) var<storage, read_write> outputData: array<f32>;
         @group(0) @binding(2) var<uniform> params: Params;
-
+        ${isNaNFunc}
         @compute @workgroup_size(4, 4, 4)
         fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
             let zStride = params.zStride;