Tests Fix (#26)

* quick fix, checks if gpu available * quick fix, added check if gpu avaiable * fix tuple --------- Co-authored-by: tomsmierz <[email protected]> Co-authored-by: Łukasz Pawela <[email protected]>
euro-hpc-pl · Oct 25, 2024 · 9a32b6a · 9a32b6a
1 parent 970f4f3
commit 9a32b6a
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 18 deletions.
diff --git a/test/canonise.jl b/test/canonise.jl
@@ -6,7 +6,8 @@ d = [1, 2, 2, 2, 4, 2, 2]
 id = Dict(j => d[i] for (i, j) in enumerate(sites))
 
 @testset "Random QMps ($T)" for T in (Float32, Float64)
-    for toCUDA ∈ (true, false)
+    checks = CUDA.functional() ? (true, false) : (false, ) 
+    for toCUDA ∈ checks
         ψ = rand(QMps{T}, id, D)
         ϕ = rand(QMps{T}, id, D)
         @test is_consistent(ψ)
@@ -35,7 +36,7 @@ id = Dict(j => d[i] for (i, j) in enumerate(sites))
     end
 end
 
-@testset "Measure spectrum($T)" for T in (Float32, Float64)
+@testset "Measure spectrum ($T)" for T in (Float32, Float64)
     svd_mps = TensorMap{T}(
         1 => T[
             -0.694389933025747 -0.7195989305943268;;;

diff --git a/test/projectors.jl b/test/projectors.jl
@@ -31,23 +31,29 @@
     end
 
     @testset "Different devices" begin
+        checks = CUDA.functional() ? (true, false) : (false) 
         for T ∈ [Int16, Int32, Int64]
-            lp = PoolOfProjectors{T}()
-            p1 = Vector{T}([1, 1, 2, 2, 3, 3])
-            p2 = Vector{T}([1, 2, 1, 3])
-            k = add_projector!(lp, p1)
-            k = add_projector!(lp, p2)
-
-            @test typeof(get_projector!(lp, 2, :CPU)) <: Array{T,1}
-            @test typeof(get_projector!(lp, 1, :GPU)) <: CuArray{T,1}
-            @test length(lp, :GPU) == 1
-            @test length(lp, :CPU) == 2
-
-            @test typeof(get_projector!(lp, 1, :GPU)) <: CuArray{T,1}
-            @test length(lp, :GPU) == 1
-
-            @test typeof(get_projector!(lp, 2, :GPU)) <: CuArray{T,1}
-            @test length(lp, :GPU) == 2
+            for toCUDA ∈ checks
+                lp = PoolOfProjectors{T}()
+                p1 = Vector{T}([1, 1, 2, 2, 3, 3])
+                p2 = Vector{T}([1, 2, 1, 3])
+                k = add_projector!(lp, p1)
+                k = add_projector!(lp, p2)
+
+                if toCUDA
+                    @test typeof(get_projector!(lp, 1, :GPU)) <: CuArray{T,1}
+                    @test length(lp, :GPU) == 1
+
+                    @test typeof(get_projector!(lp, 1, :GPU)) <: CuArray{T,1}
+                    @test length(lp, :GPU) == 1
+
+                    @test typeof(get_projector!(lp, 2, :GPU)) <: CuArray{T,1}
+                    @test length(lp, :GPU) == 2
+                end
+
+                @test typeof(get_projector!(lp, 2, :CPU)) <: Array{T,1}
+                @test length(lp, :CPU) == 2   
+            end    
         end
     end
 end