Skip to content

Commit 3ef076d

Browse files
committed
use @view!
1 parent 899bed8 commit 3ef076d

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/fusiontensor/base_interface.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
# This files defines Base functions for FusionTensor
22

3+
using BlockSparseArrays: @view!
4+
35
function Base.:*(x::Number, ft::FusionTensor)
46
return FusionTensor(
57
x * data_matrix(ft), codomain_axes(ft), domain_axes(ft), trees_block_mapping(ft)
@@ -158,6 +160,6 @@ function Base.view(ft::FusionTensor, f1f2::Tuple{<:SectorFusionTree,<:SectorFusi
158160
return view(ft, f1f2...)
159161
end
160162
function Base.view(ft::FusionTensor, f1::SectorFusionTree, f2::SectorFusionTree)
161-
charge_matrix = view(data_matrix(ft), trees_block_mapping(ft)[f1, f2])
163+
charge_matrix = @view! data_matrix(ft)[trees_block_mapping(ft)[f1, f2]]
162164
return reshape(charge_matrix, charge_block_size(ft, f1, f2))
163165
end

test/test_permutedims.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ end
146146
@test permutedims(ft, biperm) naive_permutedims(ft, biperm)
147147
@test permutedims(adjoint(ft), biperm) naive_permutedims(adjoint(ft), biperm)
148148

149-
ft = FusionTensor(sds22b, (g2, g2b), (g2, g2b))
149+
ft = FusionTensor(sds22b, (g2, g2b), (g2b, g2))
150150
@test permutedims(ft, biperm) naive_permutedims(ft, biperm)
151151
@test permutedims(adjoint(ft), biperm) naive_permutedims(adjoint(ft), biperm)
152152
end

0 commit comments

Comments
 (0)