dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/crates/burn-rocm
History
Nathaniel Simard fb6be2be8f
Perf/fusion deadcode (#3263) 		2025-06-10 09:00:55 -04:00
..
src Perf/fusion deadcode (#3263) 2025-06-10 09:00:55 -04:00
Cargo.toml Fix rocm (#3240) 2025-05-29 09:32:30 -04:00
README.md [hip] Move burn-hip to burn-rocm and rename backend to ROCm (#3062) 2025-04-23 11:05:34 -04:00

README.md

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.