Skip to main content

cubek_convolution/
lib.rs

1pub mod components;
2pub mod definition;
3pub mod kernels;
4pub mod launch;
5pub mod routines;
6
7#[cfg(any(feature = "cpu-reference", feature = "benchmarks"))]
8pub mod eval;
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};
19
20// The depthwise routine is its own entry point rather than a `ConvAlgorithm`: it shares no
21// blueprint with the accelerated routines, so there is nothing for `Strategy` to select.
22pub use kernels::forward::{
23    DepthwiseStrategy, DepthwiseTensors, DepthwiseTiling, launch_depthwise,
24};
25
26pub use kernels::forward::{DirectTensors, launch_direct};