We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7db3a5f commit 9b59807Copy full SHA for 9b59807
1 file changed
lib/OptimizationBase/ext/OptimizationEnzymeExt.jl
@@ -195,8 +195,13 @@ function OptimizationBase.instantiate_function(
195
vdθ = Tuple((Array(r) for r in eachrow(I(length(θ)) * one(eltype(θ)))))
196
vdbθ = Tuple(zeros(eltype(θ), length(θ)) for i in eachindex(θ))
197
θ_arr = θ isa Array ? θ : Array(θ)
198
- G_arr = G isa Array ? G : Array(G)
199
- Enzyme.make_zero!(G_arr)
+ if G isa Array
+ G_arr = G
200
+ Enzyme.make_zero!(G)
201
+ else
202
+ G_arr = Array(G)
203
+ Enzyme.make_zero!(G_arr)
204
+ end
205
206
Enzyme.autodiff(
207
fmode,
@@ -208,8 +213,8 @@ function OptimizationBase.instantiate_function(
208
213
Const(p)
209
214
)
210
215
211
- if G !== G_arr
212
- G .= G_arr
216
+ if !(G isa Array)
217
+ copyto!(G, G_arr)
218
end
219
220
for i in eachindex(θ)
0 commit comments