From ac0af83d4814ef76594c39243d50818130c066f4 Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Thu, 26 Sep 2024 05:59:23 -0400
Subject: [PATCH] test: separate out the enzyme testing

---
 .github/workflows/Tests.yml | 1 +
 test/enzyme.jl              | 1 -
 test/runtests.jl            | 5 ++++-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/Tests.yml b/.github/workflows/Tests.yml
index b85feb422..33741cd5c 100644
--- a/.github/workflows/Tests.yml
+++ b/.github/workflows/Tests.yml
@@ -31,6 +31,7 @@ jobs:
           - "LinearSolveHYPRE"
           - "LinearSolvePardiso"
           - "LinearSolveBandedMatrices"
+          - "Enzyme"
     uses: "SciML/.github/.github/workflows/tests.yml@v1"
     with:
       group: "${{ matrix.group }}"
diff --git a/test/enzyme.jl b/test/enzyme.jl
index ac552a45a..323f3e607 100644
--- a/test/enzyme.jl
+++ b/test/enzyme.jl
@@ -1,7 +1,6 @@
 using Enzyme, ForwardDiff
 using LinearSolve, LinearAlgebra, Test
 using FiniteDiff
-using SafeTestsets
 
 n = 4
 A = rand(n, n);
diff --git a/test/runtests.jl b/test/runtests.jl
index 8d7b626fa..44be59058 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -15,13 +15,16 @@ if GROUP == "All" || GROUP == "Core"
     @time @safetestset "Non-Square Tests" include("nonsquare.jl")
     @time @safetestset "SparseVector b Tests" include("sparse_vector.jl")
     @time @safetestset "Default Alg Tests" include("default_algs.jl")
-    @time @safetestset "Enzyme Derivative Rules" include("enzyme.jl")
     @time @safetestset "Adjoint Sensitivity" include("adjoint.jl")
     @time @safetestset "Traits" include("traits.jl")
     @time @safetestset "BandedMatrices" include("banded.jl")
     @time @safetestset "Static Arrays" include("static_arrays.jl")
 end
 
+if GROUP == "All" || GROUP == "Enzyme"
+    @time @safetestset "Enzyme Derivative Rules" include("enzyme.jl")
+end
+
 if GROUP == "LinearSolveCUDA"
     Pkg.activate("gpu")
     Pkg.develop(PackageSpec(path = dirname(@__DIR__)))