Skip to content

Commit e5cc6b9

Browse files
committed
more indexmanipulation overloads
1 parent 8dee6e8 commit e5cc6b9

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

src/tensors/indexmanipulations.jl

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,20 @@ function TK.add_transform!(
8484
tdst, only(tsrc), (p₁, p₂), fusiontreetransform, α, β, backend...
8585
)
8686
end
87+
function TK.add_transform!(
88+
tdst::BlockTensorMap,
89+
tsrc::TensorMap,
90+
(p₁, p₂)::Index2Tuple,
91+
fusiontreetransform,
92+
α::Number,
93+
β::Number,
94+
backend::AbstractBackend...,
95+
)
96+
# @assert length(tsrc) == 1 "source tensor must be a single tensor"
97+
return TK.add_transform!(
98+
only(tdst), tsrc, (p₁, p₂), fusiontreetransform, α, β, backend...
99+
)
100+
end
87101

88102
# we need to capture the other functions earlier to enjoy the fast transformers...
89103
for f! in (:add_permute!, :add_transpose!)

0 commit comments

Comments
 (0)