Support Dagger 0.13.3

Author: jpsamaroo

Project.toml

@@ -14,7 +14,7 @@ UUIDs = "cf7118a7-6976-5b1a-9a39-7adc72f591a4"
 
 [compat]
 Adapt = "1, 2, 3"
-Dagger = "0.12.1"
+Dagger = "0.13.3"
 KernelAbstractions = "0.5, 0.6, 0.7"
 Requires = "1"
 julia = "1"

src/cu.jl

@@ -97,8 +97,8 @@ kernel_backend(::CuArrayDeviceProc) = CUDADevice()
 
 if CUDA.has_cuda()
     for dev in devices()
-        Dagger.add_callback!(() -> begin
-            return CuArrayDeviceProc(Distributed.myid(), dev.handle, CUDA.uuid(dev))
-        end)
+        Dagger.add_processor_callback!("cuarray_device_$(dev.handle)") do
+            CuArrayDeviceProc(Distributed.myid(), dev.handle, CUDA.uuid(dev))
+        end
     end
 end

src/roc.jl

@@ -13,7 +13,7 @@ cancompute(::Val{:ROC}) = AMDGPU.configured
 kernel_backend(::ROCDevice) = ROCArrayProc
 
 if AMDGPU.configured
-    Dagger.add_callback!(proc -> begin
-        return ROCArrayProc(AMDGPU.get_default_agent())
-    end)
+    Dagger.add_processor_callback!("rocarray_device_default") do
+        ROCArrayProc(AMDGPU.get_default_agent())
+    end
 end