cubek_convolution/lib.rs
1pub mod components;
2pub mod definition;
3pub mod kernels;
4pub mod launch;
5pub mod routines;
6
7#[cfg(feature = "cpu-reference")]
8pub mod cpu_reference;
9
10// Re-export per-operation modules at the crate root for internal paths
11// (`crate::forward`, etc.) and for downstream users that previously relied on
12// `cubek_convolution::*`.
13pub use kernels::{backward_data, backward_weight, forward};
14
15// Top-level launcher: the single public entry point.
16pub use launch::{
17 AcceleratedTileKind, ConvAlgorithm, ConvolutionArgs, ConvolutionInputs, Strategy, launch_ref,
18};