Fix WOperator error when the Jacobian is lazy

SciML · Jul 16, 2019 · 2a541fd · 2a541fd
1 parent 9876d03
commit 2a541fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/derivative_utils.jl b/src/derivative_utils.jl
@@ -260,7 +260,7 @@ end
 function LinearAlgebra.mul!(Y::AbstractVecOrMat, W::WOperator, B::AbstractVecOrMat)
   if W._func_cache === nothing
     # Allocate cache only if needed
-    W._func_cache = Vector{eltype(W)}(undef, size(Y, 1))
+    W._func_cache = Vector{eltype(B)}(undef, size(Y, 1))
   end
   if W.transform
     # Compute mass_matrix * B