forgot cuda.functional for tests

Tractables · Oct 19, 2021 · 17e63c4 · 17e63c4
1 parent 8168c18
commit 17e63c4
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/test/structurelearner/rat_spn_tests.jl b/test/structurelearner/rat_spn_tests.jl
@@ -2,9 +2,7 @@ using Test
 using LogicCircuits
 using ProbabilisticCircuits
 using DataFrames
-
-
-
+using CUDA
 
 @testset "Random Region Graph tests" begin
 
@@ -65,8 +63,11 @@ end
     @test typeof(circuit) <: ProbCircuit
 
     estimate_parameters_em(circuit, data; pseudocount, use_gpu=false, update_per_batch = false)
-    estimate_parameters_em(circuit, data; pseudocount, use_gpu=true, update_per_batch = false)
     estimate_parameters_em(circuit, data; pseudocount, use_gpu=false, update_per_batch = true)
-    estimate_parameters_em(circuit, data; pseudocount, use_gpu=true, update_per_batch = true)
+
+    if CUDA.functional()
+        estimate_parameters_em(circuit, data; pseudocount, use_gpu=true, update_per_batch = false)
+        estimate_parameters_em(circuit, data; pseudocount, use_gpu=true, update_per_batch = true)
+    end
 
 end