Merge pull request #10 from tensor4all/9-type-instability-in-ttcache

Fix type instability in TTCache
tensor4all · May 24, 2024 · 6137615 · 6137615
2 parents 4be3b5e + 09823cf
commit 6137615
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cachedtensortrain.jl b/src/cachedtensortrain.jl
@@ -7,7 +7,7 @@ abstract type BatchEvaluator{V} <: AbstractTensorTrain{V} end
 Cached evalulation of TT
 """
 struct TTCache{ValueType} <: BatchEvaluator{ValueType}
-    sitetensors::Vector{Array{ValueType}}
+    sitetensors::Vector{Array{ValueType,3}}
     cacheleft::Vector{Dict{MultiIndex,Vector{ValueType}}}
     cacheright::Vector{Dict{MultiIndex,Vector{ValueType}}}