More generic safe similar

Author: ErikQQY

lib/NonlinearSolveBase/src/jacobian.jl

@@ -41,7 +41,7 @@ function construct_jacobian_cache(
                           (linsolve === nothing || needs_concrete_A(linsolve)))
     needs_jac = linsolve_needs_jac || concrete_jac(alg)
 
-    @bb fu_cache = similar(fu)
+    fu_cache = Utils.safe_similar(fu)
 
     if !has_analytic_jac && needs_jac
         if autodiff === nothing
@@ -80,9 +80,9 @@ function construct_jacobian_cache(
             end
         else
             if eltype(f.jac_prototype) <: Bool
-                similar(f.jac_prototype, promote_type(eltype(fu), eltype(u)))
+                Utils.safe_similar(f.jac_prototype, promote_type(eltype(fu), eltype(u)))
             else
-                similar(f.jac_prototype)
+                Utils.safe_similar(f.jac_prototype)
             end
         end
     end

lib/NonlinearSolveBase/src/utils.jl

@@ -101,13 +101,13 @@ restructure(y, x) = ArrayInterface.restructure(y, x)
 
 function safe_similar(x, args...; kwargs...)
     y = similar(x, args...; kwargs...)
-    return init_bigfloat_array!!(y)
+    return init_similar_array!!(y)
 end
 
-init_bigfloat_array!!(x) = x
+init_similar_array!!(x) = x
 
-function init_bigfloat_array!!(x::AbstractArray{<:BigFloat})
-    ArrayInterface.can_setindex(x) && fill!(x, BigFloat(0))
+function init_similar_array!!(x::AbstractArray{<:T}) where {T}
+    ArrayInterface.can_setindex(x) && fill!(x, T(0))
     return x
 end