cubecl-hip 0.11.0-pre.4

AMD ROCm HIP runtime for CubeCL
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

mod server;

pub(crate) mod capture;
pub(crate) mod context;
pub(crate) mod driver;
pub(crate) mod events;
pub(crate) mod storage;
pub(crate) mod stream;

pub use server::*;
pub use storage::*;

/// One unit of work against the device — the shared
/// [`Command`](cubecl_server::command::Command), driven by [`driver::Hip`].
pub(crate) type Command<'a> = cubecl_server::command::Command<'a, driver::Hip>;

/// The graphs this device has captured, driven by [`driver::Hip`].
pub(crate) type Captures = cubecl_server::command::Captures<driver::Hip>;

/// A capture window on one stream, driven by [`driver::Hip`].
pub(crate) type Window<'a> = cubecl_server::command::Window<'a, driver::Hip>;