@@ -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)...
8080end
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)...
136136end
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)...
195195end
196196
0 commit comments