Update cuda.jl

EnzymeAD · Jan 8, 2025 · 95570f5 · 95570f5
1 parent cf5fd60
commit 95570f5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/integration/cuda.jl b/test/integration/cuda.jl
@@ -1,4 +1,3 @@
-@static if !Sys.isapple()
 
 using Reactant
 using Test
@@ -17,6 +16,7 @@ function square!(x, y)
     return nothing
 end
 
+@static if !Sys.isapple()
 @testset "Square Kernel" begin
     oA = collect(1:1:64)
     A = Reactant.to_rarray(oA)
@@ -29,6 +29,7 @@ end
         @code_hlo optimize = :before_kernel square!(A, B)
     end
 end
+end
 
 function sin_kernel!(x, y)
     i = threadIdx().x
@@ -42,6 +43,7 @@ function sin!(x, y)
     return nothing
 end
 
+@static if !Sys.isapple()
 @testset "Sin Kernel" begin
     oA = collect(Float64, 1:1:64)
     A = Reactant.to_rarray(oA)
@@ -54,6 +56,7 @@ end
         @code_hlo optimize = :before_kernel sin!(A, B)
     end
 end
+end
 
 function smul_kernel!(x, y)
     i = threadIdx().x
@@ -68,6 +71,8 @@ function smul!(x)
     return nothing
 end
 
+@static if !Sys.isapple()
+
 # Broken pending jll update
 @static if false
 @testset "Constant Op Kernel" begin