AMDGPU.jlProgramming AMD GPUs with Julia

Install ​

Simply add the AMDGPU.jl package to your Julia environment:

using Pkg

Pkg.add("AMDGPU")

Requirements ​

• Julia 1.10+

• MI300X requires Julia 1.12+

• 64-bit Linux or Windows

• ROCm 6.0+

Required software ​

sudo dnf install rocminfo rocblas rocfft rocsparse rocsolver rocrand roctracer miopen rocm-hip-devel

Test ​

To ensure that everything works, you can run the test suite:

using AMDGPU
using Pkg

Pkg.test("AMDGPU")

Example ​

Element-wise addition via high-level interface & low-level kernel:

using AMDGPU

function vadd!(c, a, b)
   i = workitemIdx().x + (workgroupIdx().x - 1) * workgroupDim().x
   if i ≤ length(a)
       c[i] = a[i] + b[i]
   end
   return
end

a = AMDGPU.ones(Int, 1024)
b = AMDGPU.ones(Int, 1024)
c = AMDGPU.zeros(Int, 1024)

groupsize = 256
gridsize = cld(length(c), groupsize)
@roc groupsize=groupsize gridsize=gridsize vadd!(c, a, b)

@assert (a .+ b) ≈ c

Feature overview ​

Questions and Contributions ​

Usage questions can be posted on the Julia Discourse forum under the GPU domain and/or in the #gpu channel of the Julia Slack.

Contributions are very welcome, as are feature requests and suggestions. Please open an issue if you encounter any problems.

Acknowledgment ​

AMDGPU.jl would not have been possible without the work by Tim Besard and contributors to CUDA.jl and LLVM.jl.

License ​

AMDGPU.jl is licensed under the MIT License.