Skip to content

Commit a501896

Browse files
committed
Did someone transpose the SOLVE_TRI result matrix? Perhaps...
1 parent 8794e13 commit a501896

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cpu/ops.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9698,7 +9698,7 @@ static void ggml_compute_forward_solve_tri_f32(const struct ggml_compute_params
96989698
for (int64_t t = 0; t < i00; ++t) {
96999699
sum += A_batch[i00 * n + t] * X_batch[t * k + i01];
97009700
}
9701-
9701+
97029702
const float diag = A_batch[i00 * n + i00];
97039703
GGML_ASSERT(diag != 0.0f && "Zero diagonal in triangular matrix");
97049704
X_batch[i00 * k + i01] = (B_batch[i00 * k + i01] - sum) / diag;

0 commit comments

Comments
 (0)