diff --git a/src/gpuarrays.jl b/src/gpuarrays.jl index 0330464..1cc3949 100644 --- a/src/gpuarrays.jl +++ b/src/gpuarrays.jl @@ -11,7 +11,7 @@ import KernelAbstractions: Backend ## execution @inline function GPUArrays.launch_heuristic(::oneAPIBackend, obj::O, args::Vararg{Any,N}; - elements::Int, elements_per_thread::Int) where {O,N} + elements::Int, elements_per_thread::Int) where {O,N} ndrange = ceil(Int, elements / elements_per_thread) ndrange, workgroupsize, iterspace, dynamic = KA.launch_config(obj, ndrange, nothing) diff --git a/src/oneAPI.jl b/src/oneAPI.jl index 7dfa053..00d8359 100644 --- a/src/oneAPI.jl +++ b/src/oneAPI.jl @@ -70,8 +70,8 @@ end include("broadcast.jl") include("mapreduce.jl") include("oneAPIKernels.jl") -import .oneAPIKernels: oneAPIBackend -export oneAPIBackend, KA.launch_config +import .oneAPIKernels: oneAPIBackend, KA.launch_config +export oneAPIBackend include("gpuarrays.jl") include("random.jl")