File tree 2 files changed +4
-5
lines changed
2 files changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -952,12 +952,14 @@ function __init__()
952
952
return nothing
953
953
end
954
954
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 ()
956
958
Reactant. PrecompileTools. @setup_workload begin
957
959
Reactant. initialize_dialect ()
958
960
client = Reactant. XLA. CPUClient (; checkcount= false )
959
961
Reactant. PrecompileTools. @compile_workload begin
960
- @static if Reactant. precompilation_supported ()
962
+ @static if Reactant. precompilation_supported () && VERSION != v " 1.11.3 "
961
963
function square_kernel! (x)
962
964
i = CUDA. threadIdx (). x
963
965
x[i] *= x[i]
Original file line number Diff line number Diff line change @@ -17,9 +17,6 @@ function square!(x, y)
17
17
return nothing
18
18
end
19
19
20
- # https://github.com/EnzymeAD/Reactant.jl/issues/614
21
- const skip_non_cuda_tests = true
22
-
23
20
@static if ! Sys. isapple ()
24
21
@testset " Square Kernel" begin
25
22
oA = collect (1 : 1 : 64 )
You can’t perform that action at this time.
0 commit comments