Skip to content

Commit 8b8f8c4

Browse files
sharanryChrisRackauckas
authored andcommitted
Avoid inv
1 parent f7b7c14 commit 8b8f8c4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

ext/LinearSolveEnzymeExt.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,9 @@ function EnzymeCore.EnzymeRules.forward(func::Const{typeof(LinearSolve.solve!)},
3636
end
3737

3838
dres = deepcopy(res)
39-
invA = inv(A)
4039
db = linsolve.dval.b
4140
dA = linsolve.dval.A
42-
dres.u .= invA * (db - dA * res.u)
41+
dres.u .= A \ (db - dA * res.u)
4342

4443
if RT <: DuplicatedNoNeed
4544
return dres

0 commit comments

Comments
 (0)