Skip to content

Commit 1fa290c

Browse files
committed
Small fixes
1 parent 5e68a00 commit 1fa290c

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/tensors/indexmanipulations.jl

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -542,10 +542,9 @@ function _add_transform_nonthreaded!(tdst, tsrc, p, transformer::GenericTreeTran
542542
for subtransformer in transformer.data
543543
# Special case without intermediate buffers whenever there is only a single block
544544
if length(subtransformer[1]) == 1
545-
_add_transform_single!(tdst, tsrc, p, α, β, subtransformer, backend...)
545+
_add_transform_single!(tdst, tsrc, p, subtransformer, α, β, backend...)
546546
else
547-
_add_transform_multi!(tdst, tsrc, p, α, β, subtransformer,
548-
buffers, backend...)
547+
_add_transform_multi!(tdst, tsrc, p, subtransformer, buffers, α, β, backend...)
549548
end
550549
end
551550
return nothing
@@ -554,7 +553,6 @@ end
554553
function _add_transform_threaded!(tdst, tsrc, p, transformer::GenericTreeTransformer,
555554
α, β, backend...;
556555
ntasks::Int=get_num_transformer_threads())
557-
buffersz = buffersize(transformer)
558556
nblocks = length(transformer.data)
559557

560558
counter = Threads.Atomic{Int}(1)

0 commit comments

Comments
 (0)