ggml : fix A indexing in simd_gemm scalar tail-column path (#25390)
`simd_gemm()` has an incorrect A-matrix index in the scalar tail-column path for full row blocks.
This commit is contained in:
@@ -78,7 +78,7 @@ static void simd_gemm(
|
|||||||
for (int64_t i = 0; i < GEMM_RM; i++) {
|
for (int64_t i = 0; i < GEMM_RM; i++) {
|
||||||
float a = C[i * N + jj];
|
float a = C[i * N + jj];
|
||||||
for (int64_t kk = 0; kk < K; kk++) {
|
for (int64_t kk = 0; kk < K; kk++) {
|
||||||
a += A[i + kk] * B[kk * N + jj];
|
a += A[i * K + kk] * B[kk * N + jj];
|
||||||
}
|
}
|
||||||
C[i * N + jj] = a;
|
C[i * N + jj] = a;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user