cubecl_cpp/
lib.rs

1#[macro_use]
2extern crate derive_new;
3
4pub mod shared;
5
6pub use shared::register_supported_types;
7pub use shared::{Dialect, WmmaCompiler};
8
9/// Format CPP code.
10pub mod formatter;
11
12#[cfg(feature = "hip")]
13pub mod hip;
14#[cfg(feature = "hip")]
15pub type HipDialectIntrinsic = hip::HipDialect<hip::mma::WmmaIntrinsicCompiler>;
16#[cfg(feature = "hip")]
17pub type HipDialectRocWmma = hip::HipDialect<hip::mma::RocWmmaCompiler>;
18
19#[cfg(feature = "cuda")]
20pub mod cuda;
21#[cfg(feature = "cuda")]
22pub type CudaCompiler = shared::CppCompiler<cuda::CudaDialect<cuda::mma::CudaWmmaCompiler>>;