Add support for AMDGPU 0.9

FluxML · May 18, 2024 · 03a1e09 · 03a1e09
1 parent f2a2674
commit 03a1e09
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 13 deletions.
diff --git a/Project.toml b/Project.toml
@@ -27,7 +27,7 @@ NNlibCUDAExt = "CUDA"
 NNlibEnzymeCoreExt = "EnzymeCore"
 
 [compat]
-AMDGPU = "0.7, 0.8"
+AMDGPU = "0.8, 0.9"
 Adapt = "3.2, 4"
 Atomix = "0.1"
 CUDA = "4, 5"

diff --git a/ext/NNlibAMDGPUExt/NNlibAMDGPUExt.jl b/ext/NNlibAMDGPUExt/NNlibAMDGPUExt.jl
@@ -53,7 +53,6 @@ include("batched_mul.jl")
 
     include("conv.jl")
     include("pool.jl")
-    include("softmax.jl")
     include("activations.jl")
 else
     @warn """

diff --git a/ext/NNlibAMDGPUExt/softmax.jl b/ext/NNlibAMDGPUExt/softmax.jl