Skip to content

Commit 2e9e075

Browse files
committed
also update tensoralloc
1 parent ce23b52 commit 2e9e075

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/tensors/tensoroperations.jl

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,13 @@ function TO.tensorstructure(t::AbstractTensorMap, iA::Int, conjA::Bool)
66
end
77

88
function TO.tensoralloc(
9-
::Type{TT}, structure::TensorMapSpace{S, N₁, N₂},
10-
istemp::Val, allocator = TO.DefaultAllocator()
11-
) where {T, S, N₁, N₂, TT <: AbstractTensorMap{T, S, N₁, N₂}}
9+
::Type{TT}, structure::TensorMapSpace, istemp::Val, allocator = TO.DefaultAllocator()
10+
) where {TT <: AbstractTensorMap}
1211
A = storagetype(TT)
1312
dim = fusionblockstructure(structure).totaldim
1413
data = TO.tensoralloc(A, dim, istemp, allocator)
15-
# return TT(data, structure)
16-
return TensorMap{T}(data, structure)
14+
TT′ = tensormaptype(spacetype(structure), numout(structure), numin(structure), typeof(data))
15+
return TT′(data, structure)
1716
end
1817

1918
function TO.tensorfree!(t::TensorMap, allocator = TO.DefaultAllocator())

0 commit comments

Comments
 (0)