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 = "cuda")]
14pub mod cuda;
15//#[cfg(feature = "hip")]
16pub mod hip;
17#[cfg(feature = "metal")]
18pub mod metal;
19
20#[cfg(feature = "metal")]
21pub type MslCompiler = shared::CppCompiler<metal::MslDialect>;