Update generic move implementations

Author: jpsamaroo

src/DaggerGPU.jl

@@ -4,6 +4,8 @@ using Dagger, MemPool, Requires, Adapt
 using Distributed
 using KernelAbstractions
 
+import Dagger: Chunk
+
 macro gpuproc(PROC, T)
     quote
         # Assume that we can run anything
@@ -14,8 +16,22 @@ macro gpuproc(PROC, T)
         Dagger.iscompatible_arg(proc::Dagger.ThreadProc, opts, x::$T) = false
 
         # Adapt to/from the appropriate type
-        Dagger.move(from_proc::OSProc, to_proc::$PROC, x) = adapt($T, x)
-        Dagger.move(from_proc::$PROC, to_proc::OSProc, x) = adapt(Array, x)
+        function Dagger.move(from_proc::OSProc, to_proc::$PROC, x::Chunk)
+            from_pid = from_proc.pid
+            to_pid = Dagger.get_parent(to_proc).pid
+            @assert myid() == to_pid
+            adapt($T, remotecall_fetch(from_pid, x) do x
+                poolget(x.handle)
+            end)
+        end
+        function Dagger.move(from_proc::$PROC, to_proc::OSProc, x::Chunk)
+            from_pid = Dagger.get_parent(from_proc).pid
+            to_pid = to_proc.pid
+            @assert myid() == to_pid
+            remotecall_fetch(from_pid, x) do x
+                adapt(Array, poolget(x.handle))
+            end
+        end
     end
 end