Fixup buildkite handling for Mooncake

Author: kshyatt

test/mooncake.jl

@@ -13,14 +13,13 @@ is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
 m = 19
 for T in BLASFloats, n in (17, m, 23)
     TestSuite.seed_rng!(123)
-    if is_buildkite
-        if CUDA.functional()
-            TestSuite.test_mooncake(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
-        end
-        #=if AMDGPU.functional()
-            TestSuite.test_mooncake(ROCMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
-        end=# # not yet supported
-    else
+    if CUDA.functional()
+        TestSuite.test_mooncake(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end
+    #=if AMDGPU.functional()
+        TestSuite.test_mooncake(ROCMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
+    end=# # not yet supported
+    if !is_buildkite
         TestSuite.test_mooncake(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
     end
 end

test/runtests.jl

@@ -28,9 +28,6 @@ if !is_buildkite
     @safetestset "Image and Null Space" begin
         include("orthnull.jl")
     end
-    @safetestset "Mooncake" begin
-        include("mooncake.jl")
-    end
     @safetestset "ChainRules" begin
         include("chainrules.jl")
     end
@@ -71,6 +68,9 @@ end
 @safetestset "Projections" begin
     include("projections.jl")
 end
+@safetestset "Mooncake" begin
+    include("mooncake.jl")
+end
 
 using CUDA
 if CUDA.functional()