From 90d909f97661a2a48b16c2a80fc4abd8d0fd6802 Mon Sep 17 00:00:00 2001 From: Sheehan Olver Date: Wed, 26 Mar 2025 11:48:56 +0000 Subject: [PATCH] broadcast -> map in viewblock to avoid allocations --- src/blockedarray.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/blockedarray.jl b/src/blockedarray.jl index 11962a5d..fe4838e8 100644 --- a/src/blockedarray.jl +++ b/src/blockedarray.jl @@ -244,12 +244,12 @@ end @inline view(block_arr::BlockedArray{<:Any, 0}) = view(block_arr.blocks) @inline function viewblock(block_arr::BlockedArray, block) - range = getindex.(axes(block_arr), Block.(block.n)) + range = map(getindex, axes(block_arr), map(Block, block.n)) return view(block_arr.blocks, range...) end @propagate_inbounds function _blockedindex_getindex(block_arr, blockindex) - I = getindex.(axes(block_arr), getindex.(Block.(blockindex.I), blockindex.α)) + I = map(getindex, axes(block_arr), map(getindex, map(Block,blockindex.I), blockindex.α)) block_arr.blocks[I...] end