fix caches

euro-hpc-pl · Mar 6, 2024 · 937c494 · 937c494
1 parent 99ca601
commit 937c494
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -27,6 +27,7 @@ Graphs = "1.9"
 LabelledGraphs = "0.4.4"
 Memoization = "0.2"
 MetaGraphs = "0.7"
+ProgressMeter = "1.10"
 SpinGlassNetworks = "1"
 SpinGlassTensors = "1"
 Statistics = "1.7.0"

diff --git a/src/SpinGlassEngine.jl b/src/SpinGlassEngine.jl
@@ -7,7 +7,6 @@ using SpinGlassNetworks
 using CUDA
 using TensorOperations
 using TensorCast
-using NNlib, NNlibCUDA
 using MetaGraphs
 using Memoization
 using LinearAlgebra, MKL

diff --git a/src/contractor.jl b/src/contractor.jl
@@ -535,7 +535,8 @@ Note: The memoization ensures that the dressed MPS is only constructed once for
 ) where T <: AbstractStrategy
 
     ψ = mps(ctr, i+1, indβ)
-    delete!(Memoization.caches[mps], ((ctr, i+1, indβ), ()))
+    caches = Memoization.find_caches(mps)
+    delete!(caches[mps], ((ctr, i+1, indβ), ()))
     if ctr.onGPU
         ψ = move_to_CUDA!(ψ)
     end