[tests] Always skip CUDA tests on non-CUDA machines

EnzymeAD · Jan 25, 2025 · 998d755 · 998d755
1 parent 2118ee2
commit 998d755
Showing 1 changed file with 23 additions and 12 deletions.
diff --git a/test/integration/cuda.jl b/test/integration/cuda.jl
@@ -16,11 +16,8 @@ function square!(x, y)
     return nothing
 end
 
-# GPUCompiler throws "Not implemented" errors on aarch64 before
-# <https://github.com/JuliaLang/julia/pull/57077> for some tests.
-const skip_tests =
-    Base.BinaryPlatforms.arch(Base.BinaryPlatforms.HostPlatform()) == "aarch64" &&
-    VERSION <= v"1.11.3"
+# https://github.com/EnzymeAD/Reactant.jl/issues/614
+const skip_non_cuda_tests = true
 
 @static if !Sys.isapple()
     @testset "Square Kernel" begin
@@ -32,7 +29,9 @@ const skip_tests =
             @test all(Array(A) .≈ (oA .* oA .* 100))
             @test all(Array(B) .≈ (oA .* 100))
         else
-            @static if !skip_tests
+            @static if skip_non_cuda_tests
+                @test false broken = true
+            else
                 @code_hlo optimize = :before_kernel square!(A, B)
             end
         end
@@ -61,7 +60,9 @@ end
             @test all(Array(A) .≈ oA .* sin.(oA .* 100))
             @test all(Array(B) .≈ (oA .* 100))
         else
-            @static if !skip_tests
+            @static if skip_non_cuda_tests
+                @test false broken = true
+            else
                 @code_hlo optimize = :before_kernel sin!(A, B)
             end
         end
@@ -89,7 +90,9 @@ end
             @jit smul!(A)
             @test all(Array(A) .≈ oA .* 15)
         else
-            @static if !skip_tests
+            @static if skip_non_cuda_tests
+                @test false broken = true
+            else
                 @code_hlo optimize = :before_kernel smul!(A)
             end
         end
@@ -116,7 +119,9 @@ tuplef2(a) = @cuda threads = 1 tuplef2!((5, a))
             @jit tuplef(A)
             @test all(Array(A) .≈ 3)
         else
-            @static if !skip_tests
+            @static if skip_non_cuda_tests
+                @test false broken = true
+            else
                 @code_hlo optimize = :before_kernel tuplef(A)
             end
         end
@@ -126,7 +131,9 @@ tuplef2(a) = @cuda threads = 1 tuplef2!((5, a))
             @jit tuplef2(A)
             @test all(Array(A) .≈ 5)
         else
-            @static if !skip_tests
+            @static if skip_non_cuda_tests
+                @test false broken = true
+            else
                 @code_hlo optimize = :before_kernel tuplef2(A)
             end
         end
@@ -136,7 +143,9 @@ tuplef2(a) = @cuda threads = 1 tuplef2!((5, a))
         @jit tuplef2(A)
         @test all(Array(A) .≈ 5)
     else
-        @static if !skip_tests
+        @static if skip_non_cuda_tests
+            @test false broken = true
+        else
             @code_hlo optimize = :before_kernel tuplef2(A)
         end
     end
@@ -164,7 +173,9 @@ end
             @jit aliased(a)
             @test all(Array(a) .== 9)
         else
-            @static if !skip_tests
+            @static if skip_non_cuda_tests
+                @test false broken = true
+            else
                 @code_hlo optimize = :before_kernel aliased(a)
             end
         end