Skip to content

Commit cff4b0a

Browse files
committed
Keep full compatibility between CUDA v5 and v6
1 parent 4a142dd commit cff4b0a

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

ext/ReactantCUDAExt.jl

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,13 @@ using Reactant.Ops: @opcall
1212

1313
using Enzyme
1414
using Adapt: Adapt, adapt
15-
using CUDA: CUDA, CuDim, DenseCuArray, unsafe_cached_load, CUDACore
15+
using CUDA: CUDA, CuDim, DenseCuArray, unsafe_cached_load
16+
# Compatibility for CUDA v5 and v6
17+
if isdefined(CUDA, :CUDACore)
18+
using CUDA: CUDACore
19+
else
20+
const CUDACore = CUDA
21+
end
1622

1723
using GPUCompiler: GPUCompiler
1824
using GPUArraysCore: @allowscalar

0 commit comments

Comments
 (0)