From 87e4daf04a530480c291f425c2574c1dacf680d9 Mon Sep 17 00:00:00 2001
From: James Schloss <jrs.schloss@gmail.com>
Date: Wed, 10 Apr 2024 18:54:31 +0200
Subject: [PATCH] adding necessary changes for KA transition with gpuarrays

---
 src/gpuarrays.jl | 46 +---------------------------------------------
 1 file changed, 1 insertion(+), 45 deletions(-)

diff --git a/src/gpuarrays.jl b/src/gpuarrays.jl
index af87b5912..94e2e73d6 100644
--- a/src/gpuarrays.jl
+++ b/src/gpuarrays.jl
@@ -1,48 +1,4 @@
-struct ROCArrayBackend <: AbstractGPUBackend end
-
-struct ROCKernelContext <: AbstractKernelContext end
-
-@inline function GPUArrays.gpu_call(
-    ::ROCArrayBackend, f, args, threads::Int, blocks::Int; name::Maybe{String},
-)
-    @roc gridsize=blocks groupsize=threads name=name f(ROCKernelContext(), args...)
-end
-
-# indexing
-
-for (f, froc) in (
-    (:blockidx, :blockIdx),
-    (:blockdim, :blockDim),
-    (:threadidx, :threadIdx),
-    (:griddim, :gridGroupDim)
-)
-    @eval @inline GPUArrays.$f(::ROCKernelContext) = AMDGPU.$froc().x
-end
-
-# math
-
-@inline GPUArrays.cos(::ROCKernelContext, x) = cos(x)
-@inline GPUArrays.sin(::ROCKernelContext, x) = sin(x)
-@inline GPUArrays.sqrt(::ROCKernelContext, x) = sqrt(x)
-@inline GPUArrays.log(::ROCKernelContext, x) = log(x)
-
-# memory
-
-@inline function GPUArrays.LocalMemory(::ROCKernelContext, ::Type{T}, ::Val{dims}, ::Val{id}) where {T,dims,id}
-    ptr = AMDGPU.Device.alloc_special(Val{id}(), T, Val{AMDGPU.AS.Local}(), Val{prod(dims)}())
-    ROCDeviceArray(dims, ptr)
-end
-
-# synchronization
-
-@inline function GPUArrays.synchronize_threads(::ROCKernelContext)
-    sync_workgroup()
-    return
-end
-
-GPUArrays.device(x::ROCArray) = x.buf[].device
-
-GPUArrays.backend(::Type{<:ROCArray}) = ROCArrayBackend()
+import KernelAbstractions
 
 function GPUArrays.derive(
     ::Type{T}, x::ROCArray, dims::Dims{N}, offset::Int,