Import allowscalar (#716)

JuliaGPU · Jan 8, 2025 · c8f8afd · c8f8afd
1 parent 03ca435
commit c8f8afd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/AMDGPU.jl b/src/AMDGPU.jl
@@ -4,6 +4,7 @@ using Adapt
 using CEnum
 using GPUCompiler
 using GPUArrays
+using GPUArrays: allowscalar
 using Libdl
 using LLVM, LLVM.Interop
 using Preferences
@@ -116,8 +117,6 @@ include("kernels/accumulate.jl")
 include("kernels/sorting.jl")
 include("kernels/reverse.jl")
 
-allowscalar(x::Bool) = GPUArrays.allowscalar(x)
-
 include(joinpath("blas", "rocBLAS.jl"))
 include(joinpath("solver", "rocSOLVER.jl"))
 include(joinpath("sparse", "rocSPARSE.jl"))