support rename of CUDA internals (#122)

* support rename of internals

Author: vchuravy

Project.toml

@@ -14,7 +14,7 @@ StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 
 [compat]
 Adapt = "0.4, 1.0, 2.0"
-CUDA = "~1.0, ~1.1, =1.2.0"
+CUDA = "~1.0, ~1.1, ~1.2"
 Cassette = "0.3.2"
 LLVM = "1.5"
 MacroTools = "0.5"

src/backends/cuda.jl

@@ -295,11 +295,17 @@ end
 @inline Cassette.overdub(::CUDACtx, ::typeof(SpecialFunctions.erf), x::Union{Float32, Float64}) = CUDA.erf(x)
 @inline Cassette.overdub(::CUDACtx, ::typeof(SpecialFunctions.erfc), x::Union{Float32, Float64}) = CUDA.erfc(x)
 
+@static if Base.isbindingresolved(CUDA, :emit_shmem) && Base.isdefined(CUDA, :emit_shmem)
+    const emit_shmem = CUDA.emit_shmem
+else
+    const emit_shmem = CUDA._shmem
+end
+
 ###
 # GPU implementation of shared memory
 ###
 @inline function Cassette.overdub(ctx::CUDACtx, ::typeof(SharedMemory), ::Type{T}, ::Val{Dims}, ::Val{Id}) where {T, Dims, Id}
-    ptr = CUDA._shmem(Val(Id), T, Val(prod(Dims)))
+    ptr = emit_shmem(Val(Id), T, Val(prod(Dims)))
     CUDA.CuDeviceArray(Dims, ptr)
 end