Support Adapt.AbstractGPUDevice

Author: oschulz

src/device/array.jl

@@ -1,6 +1,23 @@
 # Contiguous on-device arrays
 
-export ROCDeviceArray, ROCDeviceVector, ROCDeviceMatrix, ROCBoundsError
+export ROCDevice, ROCDeviceArray, ROCDeviceVector, ROCDeviceMatrix, ROCBoundsError
+
+
+# ToDo: This is not sufficient/general enough (doesn't support multiple devices)
+# What's the AMDGPU equivalent of CUDA.CuDevice?
+"""
+    struct ROCDevice <: AbstractGPUDevice
+
+A ROC device.
+"""
+struct ROCDevice <: AbstractGPUDevice end
+
+# ToDo: This is not sufficient/general enough:
+Adapt.adapt_storage(::ROCDevice, x) = adapt_storage(ROCArray, x)
+
+# ToDo: implement Sys.total_memory(dev::ROCDevice)
+# ToDo: implement Sys.free_memory(dev::ROCDevice)
+
 
 # construction