Back to Burn

burn-rocm

crates/burn-rocm/README.md

0.20.1295 B
Original Source

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.