AMD GPU (ROCm) programming in Julia
Documentation | Build Status |
---|---|
The AMDGPU.jl package requires Julia 1.9 or higher.
AMDGPU.jl currently requires ROCm 5.3-5.4, which means only GPUs that are supported by these versions will work.
Only 64-bit Linux is supported and working at this time.
This package is under active development and is reasonably complete, however not all features are up to par with CUDA.jl.
AMDGPU.jl can be installed with the Julia package manager.
From the Julia REPL, type ]
to enter the Pkg REPL mode and run:
pkg> add AMDGPU
Or, equivalently, via the Pkg
API:
julia> import Pkg; Pkg.add("AMDGPU")
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.
AMDGPU would not have been possible without the work by Tim Besard and contributors to CUDA.jl and LLVM.jl.
AMDGPU.jl is licensed under the MIT License.