From c772653fe97c0e9b523cdf530a71bff66ec51176 Mon Sep 17 00:00:00 2001 From: Tommy van der Vorst Date: Tue, 23 May 2023 07:42:13 +0200 Subject: [PATCH] fix: issue running Gemm on Windows (#166) --- wonnx/templates/matrix/gemm.wgsl | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/wonnx/templates/matrix/gemm.wgsl b/wonnx/templates/matrix/gemm.wgsl index 61891953..bef83a1a 100644 --- 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) { ); 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 %}