diff --git a/Correction/heat_gpu.cpp b/Correction/heat_gpu.cpp index fff45c00ae280fc01ace2a7d0d145a1fce3188cf..7b66094c8f88e40c49bea63cfae8ba69159231ef 100644 --- a/Correction/heat_gpu.cpp +++ b/Correction/heat_gpu.cpp @@ -21,15 +21,14 @@ void stepSlow(Matrix<T> ¤t, Matrix<T> &next) { nxt = next.dataSpan(), cols](int i) { auto idxs_cols = std::views::iota(1, cols - 1); - std::for_each( - idxs_cols.begin(), idxs_cols.end(), - [cur, nxt, cols, row = i * cols](int j) { - auto idx = row + j; // we need a 1d index for the spans - nxt[idx] = 0.25 * (cur[idx - 1] + - cur[idx + 1] + - cur[idx - cols] + - cur[idx + cols]); - }); + std::for_each(idxs_cols.begin(), idxs_cols.end(), + [cur, nxt, cols, row = i * cols](int j) { + auto idx = row + j; // we need a 1d index for the spans + nxt[idx] = 0.25 * (cur[idx - 1] + + cur[idx + 1] + + cur[idx - cols] + + cur[idx + cols]); + }); }); }