Skip to content

Commit a707073

Browse files
committed
fix: update promote_op to use TensorOperations functions for consistency
1 parent e509e2b commit a707073

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ext/TensorOperationsChainRulesCoreExt.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ trivtuple(N) = ntuple(identity, N)
1818
@non_differentiable TensorOperations.tensorcontract_structure(args...)
1919
@non_differentiable TensorOperations.tensorcontract_type(args...)
2020
@non_differentiable TensorOperations.tensoralloc_contract(args...)
21-
@non_differentiable Base.promote_op(args...)
21+
@non_differentiable TensorOperations.promote_contract(args...)
22+
@non_differentiable TensorOperations.promote_add(args...)
2223

2324
# Cannot free intermediate tensors when using AD
2425
# Thus we change the forward passes: `istemp=false` and `tensorfree!` is a no-op

0 commit comments

Comments
 (0)