Remove multiple integer indexing in DataLayouts

CliMA · Aug 9, 2024 · 63a5628 · 63a5628
1 parent f8b7ad4
commit 63a5628
Show file tree

Hide file tree

Showing 21 changed files with 400 additions and 415 deletions.
diff --git a/ext/cuda/matrix_fields_single_field_solve.jl b/ext/cuda/matrix_fields_single_field_solve.jl
@@ -71,7 +71,8 @@ function _single_field_solve!(
     b_data = Fields.field_values(b)
     Nv = DataLayouts.nlevels(x_data)
     @inbounds for v in 1:Nv
-        x_data[v] = inv(A.λ) ⊠ b_data[v]
+        x_data[CartesianIndex(1, 1, 1, v, 1)] =
+            inv(A.λ) ⊠ b_data[CartesianIndex(1, 1, 1, v, 1)]
     end
 end