Updates for HIP-based AMDGPU

Author: jpsamaroo

Project.toml

@@ -38,7 +38,7 @@ MetalExt = ["Metal"]
 oneAPIExt = ["oneAPI"]
 
 [compat]
-AMDGPU = "0.4.9"
+AMDGPU = "0.5, 0.6, 0.7, 0.8"
 Adapt = "3, 4"
 CUDA = "4.1.0, 5"
 ChainRulesCore = "1"

ext/AMDGPUExt.jl

@@ -5,6 +5,9 @@ import DiffEqGPU
 using .AMDGPU
 import .AMDGPU: ROCBackend
 
+function DiffEqGPU.EnsembleGPUArray(cpu_offload::Float64)
+    DiffEqGPU.EnsembleGPUArray(ROCBackend(), cpu_offload)
+end
 DiffEqGPU.maxthreads(::ROCBackend) = 256
 DiffEqGPU.maybe_prefer_blocks(::ROCBackend) = ROCBackend()