Skip to content

Commit 603225a

Browse files
authored
[ReactantCUDAExt] Skip precompile load on Julia v1.11.3 (#675)
1 parent 1e6037f commit 603225a

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

ext/ReactantCUDAExt.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -952,12 +952,14 @@ function __init__()
952952
return nothing
953953
end
954954

955-
@static if !Sys.isapple() && Sys.ARCH != :aarch64
955+
# In Julia v1.11.3 precompiling this module caches bad code:
956+
# <https://github.com/EnzymeAD/Reactant.jl/issues/614>.
957+
@static if !Sys.isapple()
956958
Reactant.PrecompileTools.@setup_workload begin
957959
Reactant.initialize_dialect()
958960
client = Reactant.XLA.CPUClient(; checkcount=false)
959961
Reactant.PrecompileTools.@compile_workload begin
960-
@static if Reactant.precompilation_supported()
962+
@static if Reactant.precompilation_supported() && VERSION != v"1.11.3"
961963
function square_kernel!(x)
962964
i = CUDA.threadIdx().x
963965
x[i] *= x[i]

test/integration/cuda.jl

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,6 @@ function square!(x, y)
1717
return nothing
1818
end
1919

20-
# https://github.com/EnzymeAD/Reactant.jl/issues/614
21-
const skip_non_cuda_tests = true
22-
2320
@static if !Sys.isapple()
2421
@testset "Square Kernel" begin
2522
oA = collect(1:1:64)

0 commit comments

Comments
 (0)