Run Enzyme tests only on CUDA CI machine

FluxML · Aug 20, 2024 · 3dfe290 · 3dfe290
1 parent d1ff714
commit 3dfe290
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -140,10 +140,14 @@ Random.seed!(0)
     @info "Skipping Distributed tests, set FLUX_TEST_DISTRIBUTED_MPI or FLUX_TEST_DISTRIBUTED_NCCL=true to run them."
   end
 
-  @testset "Enzyme" begin
-    Pkg.add(["CUDA", "cuDNN"])
-    import Enzyme
-    include("ext_enzyme/enzyme.jl")
+  if get(ENV, "FLUX_TEST_CUDA", "false") == "true"
+      @testset "Enzyme" begin
+        Pkg.add(["CUDA", "cuDNN"])
+        import Enzyme
+        include("ext_enzyme/enzyme.jl")
+      end
+  else
+    @info "Skipping Enzyme tests, set FLUX_TEST_CUDA=true to run them."
   end
 
 end