Fix a bug in WASM's GEMM

microsoft · Mar 22, 2024 · 1e158ea · 1e158ea
1 parent 6238e9c
commit 1e158ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/mlas/lib/wasm_simd/SgemmKernelWasmSimd.cpp b/onnxruntime/core/mlas/lib/wasm_simd/SgemmKernelWasmSimd.cpp
@@ -171,11 +171,11 @@ Return Value:
         if (k > 0) {
 
             Row0AElements0 = a[0];
-            Row0AElements1 = a[1];
+            Row0AElements1 = a[0];
 
             if (ProcessTwoRows) {
                 Row1AElements0 = a[lda];
-                Row1AElements1 = a[lda + 1];
+                Row1AElements1 = a[lda];
             }
 
             BElements0 = MlasLoadFloat32x4(B + 0);