Skip to content

Commit f548ec5

Browse files
committed
Update
1 parent 3074d60 commit f548ec5

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/Nonlinear/ReverseAD/mathoptinterface_api.jl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,12 @@ function MOI.initialize(d::NLPEvaluator, requested_features::Vector{Symbol})
153153
#
154154
for k in d.subexpression_order
155155
len = max_chunk * length(d.subexpressions[k].nodes)
156-
d.subexpressions[k].forward_storage_ϵ = zeros(len)
157-
d.subexpressions[k].partials_storage_ϵ = zeros(len)
158-
d.subexpressions[k].reverse_storage_ϵ = zeros(len)
156+
resize!(d.subexpressions[k].forward_storage_ϵ, len)
157+
fill!(d.subexpressions[k].forward_storage_ϵ, 0.0)
158+
resize!(d.subexpressions[k].partials_storage_ϵ, len)
159+
fill!(d.subexpressions[k].partials_storage_ϵ, 0.0)
160+
resize!(d.subexpressions[k].reverse_storage_ϵ, len)
161+
fill!(d.subexpressions[k].reverse_storage_ϵ, 0.0)
159162
end
160163
d.max_chunk = max_chunk
161164
if d.want_hess

0 commit comments

Comments
 (0)