Fixes typo in GemmBatchNoAtomicFunctorThreadK

ndgrigorian · ndgrigorian · commit da065b560dc8 · 2024-03-01T00:53:38.000-08: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
@@ -2100,7 +2100,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
`@@ -2100,7 +2100,7 @@ class GemmBatchNoAtomicFunctorThreadK`
`2100`	`2100`	`for (size_t vec_id = 1; vec_id < m_groups; ++vec_id) {`
`2101`	`2101`	`if (j + vec_id < m) {`
`2102`	`2102`	`res[total_offset + res_indexer(i * m + j + vec_id)] =`
`2103`		`- local_sum[1];`
	`2103`	`+ local_sum[vec_id];`
`2104`	`2104`	`}`
`2105`	`2105`	`}`
`2106`	`2106`	`}`