burn_rocm/lib.rs
1#![cfg_attr(docsrs, feature(doc_cfg))]
2extern crate alloc;
3
4use burn_cubecl::CubeBackend;
5
6pub use cubecl::hip::AmdDevice as RocmDevice;
7
8use cubecl::hip::HipRuntime;
9
10#[cfg(not(feature = "fusion"))]
11pub type Rocm<F = f32, I = i32, B = u8> = CubeBackend<HipRuntime, F, I, B>;
12
13#[cfg(feature = "fusion")]
14pub type Rocm<F = f32, I = i32, B = u8> = burn_fusion::Fusion<CubeBackend<HipRuntime, F, I, B>>;