cubecl_cpp/lib.rs
1#[macro_use]
2extern crate derive_new;
3
4pub mod shared;
5
6pub use shared::ComputeKernel;
7pub use shared::register_supported_types;
8pub use shared::{Dialect, DialectWmmaCompiler};
9
10/// Format CPP code.
11pub mod formatter;
12
13#[cfg(feature = "hip")]
14pub mod hip;
15
16// The hip dialects use the cuda dialect sometimes this is why we need it for hip feature as well
17#[cfg(any(feature = "cuda", feature = "hip"))]
18pub mod cuda;
19
20#[cfg(feature = "metal")]
21pub mod metal;
22#[cfg(feature = "metal")]
23pub type MslCompiler = shared::CppCompiler<metal::MslDialect>;