cubecl_cpp/
lib.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#[macro_use]
extern crate derive_new;

mod shared;

pub use shared::register_supported_types;

/// Format CPP code.
pub mod formatter;

#[cfg(feature = "hip")]
mod hip;
#[cfg(feature = "hip")]
pub type HipCompiler = shared::CppCompiler<hip::Hip>;

#[cfg(feature = "cuda")]
mod cuda;

#[cfg(feature = "cuda")]
pub type CudaCompiler = shared::CppCompiler<cuda::Cuda>;