Skip to content

Commit

Permalink
Update include/tuvx/linalg/linalg.inl
Browse files Browse the repository at this point in the history
Co-authored-by: Kyle Shores <[email protected]>
  • Loading branch information
AdityaDendukuri and K20shores authored Jun 5, 2024
1 parent 90a82fe commit e4f7aa9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion include/tuvx/linalg/linalg.inl
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ inline std::vector<T> tridiag_solve(trid_mat<T> A, std::vector<T> b) {
std::size_t N = b.size();
std::vector<T> x(N);
// forward pass
for (int i = 1; i < N; i++) {
for (std::size_t i = 1; i < N; i++) {
temp = A.ldiag[i - 1] / A.mdiag[i - 1];
A.mdiag[i] -= temp * A.udiag[i - 1];
b[i] -= temp * b[i - 1];
Expand Down

0 comments on commit e4f7aa9

Please sign in to comment.