Fixes typo in GemmBatchNoAtomicFunctorThreadK

ndgrigorian · oleksandr-pavlyk · commit 2f60f374bc12 · 2024-03-26T16:59:34.000-05:00
Closes #1570
diff --git a/dpctl/tensor/libtensor/include/kernels/linalg_functions/gemm.hpp b/dpctl/tensor/libtensor/include/kernels/linalg_functions/gemm.hpp
@@ -2190,7 +2190,7 @@ class GemmBatchNoAtomicFunctorThreadK
                 for (size_t vec_id = 1; vec_id < m_groups; ++vec_id) {
                     if (j + vec_id < m) {
                         res[total_offset + res_indexer(i * m + j + vec_id)] =
-                            local_sum[1];
+                            local_sum[vec_id];
                     }
                 }
             }

Original file line number	Diff line number	Diff line change
`@@ -2190,7 +2190,7 @@ class GemmBatchNoAtomicFunctorThreadK`
`2190`	`2190`	`for (size_t vec_id = 1; vec_id < m_groups; ++vec_id) {`
`2191`	`2191`	`if (j + vec_id < m) {`
`2192`	`2192`	`res[total_offset + res_indexer(i * m + j + vec_id)] =`
`2193`		`- local_sum[1];`
	`2193`	`+ local_sum[vec_id];`
`2194`	`2194`	`}`
`2195`	`2195`	`}`
`2196`	`2196`	`}`