Skip to content

Commit 368001a

Browse files
committed
simplify buffersize
1 parent 9e1d2e8 commit 368001a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/tensors/treetransformers.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ end
106106

107107
function buffersize(transformer::GenericTreeTransformer)
108108
return maximum(transformer.data; init=0) do (basistransform, structures_dst, _)
109-
return prod(structures_dst[1][1]) * max(size(basistransform)...)
109+
return prod(structures_dst[1][1]) * size(basistransform, 1)
110110
end
111111
end
112112

0 commit comments

Comments
 (0)