change to ArrayInterface

SciML · Nov 22, 2024 · 4e19616 · 4e19616
1 parent c15f8a4
commit 4e19616
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl
@@ -4095,15 +4095,15 @@ function BVPFunction{iip, specialize, twopoint}(f, bc;
     end
 
     if jac_prototype !== nothing && colorvec === nothing &&
-       ArrayInterfaceCore.fast_matrix_colors(jac_prototype)
-        _colorvec = ArrayInterfaceCore.matrix_colors(jac_prototype)
+       ArrayInterface.fast_matrix_colors(jac_prototype)
+        _colorvec = ArrayInterface.matrix_colors(jac_prototype)
     else
         _colorvec = colorvec
     end
 
     if bcjac_prototype !== nothing && bccolorvec === nothing &&
-       ArrayInterfaceCore.fast_matrix_colors(bcjac_prototype)
-        _bccolorvec = ArrayInterfaceCore.matrix_colors(bcjac_prototype)
+       ArrayInterface.fast_matrix_colors(bcjac_prototype)
+        _bccolorvec = ArrayInterface.matrix_colors(bcjac_prototype)
     else
         _bccolorvec = bccolorvec
     end
@@ -4268,15 +4268,15 @@ function DynamicalBVPFunction{iip, specialize, twopoint}(f, bc;
     end
 
     if jac_prototype !== nothing && colorvec === nothing &&
-       ArrayInterfaceCore.fast_matrix_colors(jac_prototype)
-        _colorvec = ArrayInterfaceCore.matrix_colors(jac_prototype)
+       ArrayInterface.fast_matrix_colors(jac_prototype)
+        _colorvec = ArrayInterface.matrix_colors(jac_prototype)
     else
         _colorvec = colorvec
     end
 
     if bcjac_prototype !== nothing && bccolorvec === nothing &&
-       ArrayInterfaceCore.fast_matrix_colors(bcjac_prototype)
-        _bccolorvec = ArrayInterfaceCore.matrix_colors(bcjac_prototype)
+       ArrayInterface.fast_matrix_colors(bcjac_prototype)
+        _bccolorvec = ArrayInterface.matrix_colors(bcjac_prototype)
     else
         _bccolorvec = bccolorvec
     end