Updates for HIP-based AMDGPU

SciML · Jan 10, 2024 · c528f76 · c528f76
1 parent 7971f09
commit c528f76
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/Project.toml b/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"

diff --git a/ext/AMDGPUExt.jl b/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()