Add AMD section to memcopy.jl example

Author: ali-ramadhan

examples/memcopy.jl

@@ -1,6 +1,8 @@
 using KernelAbstractions
-using CUDAKernels
 using CUDA
+using CUDAKernels
+using AMDGPU
+using ROCKernels
 using Test
 
 @kernel function copy_kernel!(A, @Const(B))
@@ -34,3 +36,26 @@ if has_cuda_gpu()
     wait(event)
     @test A == B
 end
+
+function has_rocm_gpu()
+    for agent in AMDGPU.get_agents()
+        if agent.type == :gpu
+            return true
+        end
+    end
+    return false
+end
+
+if has_rocm_gpu()
+
+    function mycopy!(A::ROCArray, B::ROCArray)
+        @assert size(A) == size(B)
+        copy_kernel!(ROCDevice(), 256)(A, B, ndrange=length(A))
+    end
+
+    A = zeros(Float32, 1024) |> ROCArray
+    B = ones(Float32, 1024) |> ROCArray
+    event = mycopy!(A, B)
+    wait(event)
+    @test A == B
+end