fix: issue running Gemm on Windows (#166)

webonnx · May 23, 2023 · c772653 · c772653
1 parent 62533bd
commit c772653
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/wonnx/templates/matrix/gemm.wgsl b/wonnx/templates/matrix/gemm.wgsl
@@ -69,10 +69,7 @@ fn main(@builtin(global_invocation_id) global_id: vec3<u32>) {
 		);
 
 		product = mat_right * mat_left;
-
-		for(var index_mat: u32 = 0u; index_mat < {{ kernel_size }}u; index_mat = index_mat + 1u) {
-			tmpsum[index_mat] = tmpsum[index_mat] + product[index_mat];
-		}
+		tmpsum = tmpsum + product;
 	}
 
 	{% if i_lens | length == 3 %}