small bug fix with gpu and typos (#94)

euro-hpc-pl · Jan 13, 2025 · 7bef603 · 7bef603
1 parent 669dde8
commit 7bef603
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/examples/Project.toml b/examples/Project.toml
@@ -1,4 +1,5 @@
 [deps]
 Colors = "5ae59095-9a9b-59fe-a467-6f913c188581"
 Images = "916415d5-f1e6-5110-898d-aaa5f9f070e0"
+CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 SpinGlassPEPS = "2c514f87-1261-494e-8566-326879aaf4fe"
diff --git a/examples/inpaining.jl → examples/inpainting.jl b/examples/inpaining.jl → examples/inpainting.jl
@@ -1,4 +1,5 @@
 using SpinGlassPEPS
+using CUDA
 using Pkg
 using Logging
 
@@ -17,6 +18,7 @@ end
 disable_logging(LogLevel(1))
 
 instance = "$(@__DIR__)/instances/triplepoint4-plain-ring.h5"
+onGPU = CUDA.has_cuda_gpu()
 
 GEOMETRY = SquareSingleNode
 LAYOUT = GaugesEnergy
@@ -30,7 +32,7 @@ function bench_inpaining(::Type{T}, β::Real, max_states::Integer, bond_dim::Int
 	params = MpsParameters{T}(; bond_dim = bond_dim, method = :svd)
 	search_params = SearchParameters(; max_states = max_states)
 	net = PEPSNetwork{GEOMETRY{LAYOUT}, SPARSITY, T}(120, 120, potts_h, rotation(0))
-	ctr = MpsContractor{STRATEGY, GAUGE, T}(net, params; onGPU = true, beta = convert(Float64, β), graduate_truncation = true)
+	ctr = MpsContractor{STRATEGY, GAUGE, T}(net, params; onGPU = onGPU, beta = convert(Float64, β), graduate_truncation = true)
     droplets = SingleLayerDroplets(; max_energy = 100, min_size = 100 , metric = :hamming, mode=:RMF)
 	merge_strategy = merge_branches(ctr; merge_prob = :none, droplets_encoding = droplets)