Minor

gridap · Feb 20, 2025 · 1419e1f · 1419e1f
1 parent 0384e59
commit 1419e1f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Autodiff.jl b/src/Autodiff.jl
@@ -132,7 +132,7 @@ function distributed_autodiff_array_gradient(a,i_to_x,j_to_i)
   end
   j_to_ydual = a(i_to_xdual)
   j_to_result = map(i_to_cfg,j_to_i,j_to_ydual) do i_to_cfg,j_to_i,j_to_ydual
-    j_to_cfg = lazy_map(Reindex(i_to_cfg),j_to_i)
+    j_to_cfg = Arrays.autodiff_array_reindex(i_to_cfg,j_to_i)
     lazy_map(AutoDiffMap(),j_to_cfg,j_to_ydual)
   end
   return j_to_result
@@ -148,7 +148,7 @@ function distributed_autodiff_array_jacobian(a,i_to_x,j_to_i)
   end
   j_to_ydual = a(i_to_xdual)
   j_to_result = map(i_to_cfg,j_to_i,j_to_ydual) do i_to_cfg,j_to_i,j_to_ydual
-    j_to_cfg = lazy_map(Reindex(i_to_cfg),j_to_i)
+    j_to_cfg = Arrays.autodiff_array_reindex(i_to_cfg,j_to_i)
     lazy_map(AutoDiffMap(),j_to_cfg,j_to_ydual)
   end
   return j_to_result
@@ -202,11 +202,11 @@ function distributed_autodiff_array_gradient(a, i_to_x, j_to_i::AbstractArray{<:
 
   j_to_result = map(i_to_cfg,j_to_i,j_to_ydual_plus,j_to_ydual_minus) do i_to_cfg,j_to_i,j_to_ydual_plus,j_to_ydual_minus
     # Work for plus side
-    j_to_cfg_plus = lazy_map(Reindex(i_to_cfg),j_to_i.plus)
+    j_to_cfg_plus = Arrays.autodiff_array_reindex(i_to_cfg,j_to_i.plus)
     j_to_result_plus = lazy_map(AutoDiffMap(),j_to_cfg_plus,j_to_ydual_plus)
 
     # Work for minus side
-    j_to_cfg_minus = lazy_map(Reindex(i_to_cfg),j_to_i.minus)
+    j_to_cfg_minus = Arrays.autodiff_array_reindex(i_to_cfg,j_to_i.minus)
     j_to_result_minus = lazy_map(AutoDiffMap(),j_to_cfg_minus,j_to_ydual_minus)
 
     # Assemble on SkeletonTriangulation expects an array of interior of facets
@@ -234,11 +234,11 @@ function distributed_autodiff_array_jacobian(a, i_to_x, j_to_i::AbstractArray{<:
 
   j_to_result = map(i_to_cfg,j_to_i,j_to_ydual_plus,j_to_ydual_minus) do i_to_cfg,j_to_i,j_to_ydual_plus,j_to_ydual_minus
     # Work for plus side
-    j_to_cfg_plus = lazy_map(Reindex(i_to_cfg),j_to_i.plus)
+    j_to_cfg_plus = Arrays.autodiff_array_reindex(i_to_cfg,j_to_i.plus)
     j_to_result_plus = lazy_map(AutoDiffMap(),j_to_cfg_plus,j_to_ydual_plus)
 
     # Work for minus side
-    j_to_cfg_minus = lazy_map(Reindex(i_to_cfg),j_to_i.minus)
+    j_to_cfg_minus = Arrays.autodiff_array_reindex(i_to_cfg,j_to_i.minus)
     j_to_result_minus = lazy_map(AutoDiffMap(),j_to_cfg_minus,j_to_ydual_minus)
 
     # Merge the columns into a 2x2 block matrix