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 b9ff025 commit 47d8620Copy full SHA for 47d8620
src/Nonlinear/ReverseAD/forward_over_reverse.jl
@@ -55,7 +55,7 @@ function _eval_hessian_inner(
55
num_products = size(ex.seed_matrix, 2) # number of hessian-vector products
56
num_chunks = div(num_products, CHUNK)
57
@assert size(ex.seed_matrix, 1) == length(ex.rinfo.local_indices)
58
- for offset in 1:CHUNK:CHUNK*num_chunks
+ for offset in 1:CHUNK:(CHUNK*num_chunks)
59
_eval_hessian_chunk(d, ex, offset, CHUNK, Val(CHUNK))
60
end
61
# leftover chunk
0 commit comments