Merge pull request #585 from FluxML/pxl-th/amdgpu-0.9

darsnack · web-flow · commit ca600b16334a · 2024-05-19T14:15:40.000-04:00
Add support for AMDGPU 0.9
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
