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, UnitStrategy,
18    UnitTiling, launch_ref,
19};
20
21// The depthwise routine is its own entry point rather than a `ConvAlgorithm`: it shares no
22// blueprint with the accelerated routines, so there is nothing for `Strategy` to select.
23pub use kernels::forward::{
24    DepthwiseStrategy, DepthwiseTensors, DepthwiseTiling, launch_depthwise,
25};
26
27pub use kernels::forward::{DirectTensors, launch_direct};
28
29pub use kernels::im2col::{Im2colTensors, launch_im2col};
30
31pub use kernels::backward_weight::depthwise::{
32    DepthwiseWeightBackwardTensors, launch_depthwise_weight_backward,
33};
cubek_convolution/lib.rs

cubek_convolution/
lib.rs
