Rust bindings for HIP
- A ROCm platform ie a compatible AMD GPU
hipblas
can be accessed with theblas
feature- Specify the path to
hip
withHIP_PATH
. If not provided,/opt/rocm/hip
is assumed - Specify the path to
hipblas
withHIP_BLAS_PATH
. If not provided,/opt/rocm/hipblas
is assumed - Native bindings can be generated with the
bindgen
feature - Will compile without an AMD GPU, but device functions will fail
- Currently CUDA platform support is not available (see cuda-sys https://github.com/rust-cuda/cuda-sys)
- Ubuntu: See https://rocmdocs.amd.com/en/latest/deploy/linux/quick_start.html
- Works with the hcc backend
sudo apt install hip-hcc
(from ROCm package registry)
- You may need to install rocsolver as well
sudo apt install rocsolver
- Works with the hcc backend
- Arch: See https://github.com/arch4edu/arch4edu
sudo pacman -S rocm-hip-runtime
If you have any problems please post an issue.
Run the tests with:
cargo test
Open the documentation with:
cargo doc --open
This repo is derived from one created here; permission has been gained to publish this work with local modifications (see issue #1).