make all implicit methods GPU compatable

SciML · Apr 17, 2019 · 7308b33 · 7308b33
1 parent 87a97b6
commit 7308b33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nlsolve/utils.jl b/src/nlsolve/utils.jl
@@ -117,7 +117,7 @@ DiffEqBase.@def iipnlsolve begin
         W = WOperator(f, dt, true)
         J = nothing # is J = W.J better?
       else
-        J = fill(zero(uEltypeNoUnits),length(u),length(u)) # uEltype?
+        J = zero(u .* u')
         W = similar(J)
       end
     end