Skip to content

Commit 9e1d2e8

Browse files
committed
update transformer weight
1 parent c908015 commit 9e1d2e8

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/tensors/treetransformers.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,8 +96,10 @@ function GenericTreeTransformer(transform, p, Vdst, Vsrc)
9696
end
9797

9898
# Cost model for transforming a set of subblocks with fixed uncoupled sectors:
99-
# L x L x size(subblock) where L is the number of subblocks
100-
# this is L input blocks each going to L output blocks of given size
99+
# L x L x length(subblock) where L is the number of subblocks
100+
# this is L input blocks each going to L output blocks of given length
101+
# Note that it might be the case that the permutations are dominant, in which case the
102+
# actual cost model would scale like L x length(subblock)
101103
function _transformer_weight((matrix, structures_dst, structures_src))
102104
return length(matrix) * prod(structures_dst[1][1])
103105
end

0 commit comments

Comments
 (0)