Skip to content

Commit a195014

Browse files
authored
Merge pull request #128 from mipals/patch-2
Update jacobian.jl
2 parents dcbca35 + 3834a16 commit a195014

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/jacobian.jl

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
using ReverseDiff: JacobianTape, JacobianConfig, jacobian, jacobian!, compile
2+
using LinearAlgebra: mul!
23

34
#########
45
# setup #
56
#########
67

78
# some objective functions to work with
89
f(a, b) = (a + b) * (a * b)'
9-
g!(out, a, b) = A_mul_Bc!(out, a + b, a * b)
10+
g!(out, a, b) = mul!(out, a + b, a * b)
1011

1112
# pre-record JacobianTapes for `f` and `g` using inputs of shape 10x10 with Float64 elements
1213
const f_tape = JacobianTape(f, (rand(10, 10), rand(10, 10)))

0 commit comments

Comments
 (0)