fix: type error in LRN WGSL

webonnx · Mar 25, 2023 · 3ca9197 · 3ca9197
1 parent ccca5f7
commit 3ca9197
Showing 1 changed file with 1 addition and 1 deletion.
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 }}));
 }