burn-rocm 0.20.1

ROCm HIP backend for the Burn framework
Documentation
1
2
3
4
5
6
7
# burn-rocm

Backend using ROCm HIP runtime.

To execute the tests for this backend set an environment variable called `ROCM_PATH` or `CUBECL_ROCM_PATH` to the installation path of ROCm. It is often `/opt/rocm`.

For now this backend requires the version `6.2.2` of ROCm or a compatible version.