diff --git a/ext/ReactantCUDAExt.jl b/ext/ReactantCUDAExt.jl index 391be0bd9..9c9dd3b5f 100644 --- a/ext/ReactantCUDAExt.jl +++ b/ext/ReactantCUDAExt.jl @@ -952,12 +952,14 @@ function __init__() return nothing end -@static if !Sys.isapple() && Sys.ARCH != :aarch64 +# In Julia v1.11.3 precompiling this module caches bad code: +# . +@static if !Sys.isapple() Reactant.PrecompileTools.@setup_workload begin Reactant.initialize_dialect() client = Reactant.XLA.CPUClient(; checkcount=false) Reactant.PrecompileTools.@compile_workload begin - @static if Reactant.precompilation_supported() + @static if Reactant.precompilation_supported() && VERSION != v"1.11.3" function square_kernel!(x) i = CUDA.threadIdx().x x[i] *= x[i] diff --git a/test/integration/cuda.jl b/test/integration/cuda.jl index 1a95de9b6..9641609fe 100644 --- a/test/integration/cuda.jl +++ b/test/integration/cuda.jl @@ -17,9 +17,6 @@ function square!(x, y) return nothing end -# https://github.com/EnzymeAD/Reactant.jl/issues/614 -const skip_non_cuda_tests = true - @static if !Sys.isapple() @testset "Square Kernel" begin oA = collect(1:1:64)