1#[allow(unused_imports)]
2#[macro_use]
3extern crate derive_new;
4extern crate alloc;
5
6pub mod compute;
7pub mod device;
8pub mod runtime;
9pub use device::*;
10pub use runtime::HipRuntime;
11
12#[cfg(not(feature = "rocwmma"))]
13pub(crate) type HipWmmaCompiler = cubecl_cpp::hip::mma::WmmaIntrinsicCompiler;
14
15#[cfg(feature = "rocwmma")]
16pub(crate) type HipWmmaCompiler = cubecl_cpp::hip::mma::RocWmmaCompiler;
17
18#[cfg(test)]
19mod tests {
20 use half::f16;
21 pub type TestRuntime = crate::HipRuntime;
22
23 cubecl_std::testgen!();
24 cubecl_core::testgen_all!(f32: [f16, f32], i32: [i16, i32], u32: [u16, u32]);
25}