fix: type error in LRN WGSL

pixelspark · pixelspark · commit 3ca919798f00 · 2023-03-25T22:36:26.000+01:00
diff --git a/wonnx/templates/matrix/lrn.wgsl b/wonnx/templates/matrix/lrn.wgsl
@@ -19,5 +19,5 @@ fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
         square_sum += I * I;
     }
 
-    output_0.data[c] = input_0.data[ c ] / pow({{bias}} + ({{alpha}} / {{size}}.0) * square_sum,{{beta}});
+    output_0.data[c] = input_0.data[ c ] / pow({{ scalar_type }}({{ bias }}) + ({{ scalar_type }}({{ alpha }}) / {{ scalar_type }}({{ size }})) * square_sum, {{ scalar_type }}({{ beta }}));
 }

Original file line number	Diff line number	Diff line change
`@@ -19,5 +19,5 @@ fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {`
`19`	`19`	`square_sum += I * I;`
`20`	`20`	`}`
`21`	`21`
`22`		`- output_0.data[c] = input_0.data[ c ] / pow({{bias}} + ({{alpha}} / {{size}}.0) * square_sum,{{beta}});`
	`22`	`+ output_0.data[c] = input_0.data[ c ] / pow({{ scalar_type }}({{ bias }}) + ({{ scalar_type }}({{ alpha }}) / {{ scalar_type }}({{ size }})) * square_sum, {{ scalar_type }}({{ beta }}));`
`23`	`23`	`}`