Skip to content

Commit 8286110

Browse files
author
Katharine Hyatt
committed
Fix signatures
1 parent 496cbf2 commit 8286110

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

ext/TensorOperationsEnzymeExt/TensorOperationsEnzymeExt.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ function EnzymeRules.reverse(
7575
ba = map(ba_ -> getfield(ba_, :val), ba_dba)
7676
α = α_dα.val
7777
β = β_dβ.val
78-
dC, dA, dB, dα, dβ = TensorOperations.tensorcontract_pullback!(dC, dA, dB, Cval, Aval, Bval, α, β, pA_dpA.val, pB_dpB.val, pAB_dpAB.val, conjA_dconjA.val, conjB_dconjB.val, ba...)
78+
dC, dA, dB, dα, dβ = TensorOperations.tensorcontract_pullback!(dC, dA, dB, Cval, Aval, pA_dpA.val, conjA_dconjA.val, Bval, pB_dpB.val, conjB_dconjB.val, pAB_dpAB.val, α, β, ba...)
7979
return nothing, nothing, nothing, nothing, nothing, nothing, nothing, nothing, dα, dβ, map(ba_ -> nothing, ba)...
8080
end
8181

@@ -131,7 +131,7 @@ function EnzymeRules.reverse(
131131
ba = map(ba_ -> getfield(ba_, :val), ba_dba)
132132
dC = C_dC.dval
133133
dA = A_dA.dval
134-
dC, dA, dα, dβ = TensorOperations.tensoradd_pullback!(dC, dA, Cval, Aval, α, β, pA, conjA, ba...)
134+
dC, dA, dα, dβ = TensorOperations.tensoradd_pullback!(dC, dA, Cval, Aval, pA, conjA, α, β, ba...)
135135
return nothing, nothing, nothing, nothing, dα, dβ, map(ba_ -> nothing, ba)...
136136
end
137137

@@ -190,7 +190,7 @@ function EnzymeRules.reverse(
190190
ba = map(ba_ -> getfield(ba_, :val), ba_dba)
191191
dC = C_dC.dval
192192
dA = A_dA.dval
193-
dC, dA, dα, dβ = TensorOperations.tensortrace_pullback!(dC, dA, Cval, Aval, α, β, p, q, conjA, ba...)
193+
dC, dA, dα, dβ = TensorOperations.tensortrace_pullback!(dC, dA, Cval, Aval, p, q, conjA, α, β, ba...)
194194
return nothing, nothing, nothing, nothing, nothing, dα, dβ, map(ba_ -> nothing, ba)...
195195
end
196196

0 commit comments

Comments
 (0)