cubek-convolution 0.4.0-pre.1

CubeK: Convolution Kernels
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

pub mod components;
pub mod definition;
pub mod kernels;
pub mod launch;
pub mod routines;

#[cfg(any(feature = "cpu-reference", feature = "benchmarks"))]
pub mod eval;

// Re-export per-operation modules at the crate root for internal paths
// (`crate::forward`, etc.) and for downstream users that previously relied on
// `cubek_convolution::*`.
pub use kernels::{backward_data, backward_weight, forward};

// Top-level launcher: the single public entry point.
pub use launch::{
    AcceleratedTileKind, ConvAlgorithm, ConvolutionArgs, ConvolutionInputs, Strategy, UnitStrategy,
    UnitTiling, launch_ref,
};

// The depthwise routine is its own entry point rather than a `ConvAlgorithm`: it shares no
// blueprint with the accelerated routines, so there is nothing for `Strategy` to select.
pub use kernels::forward::{
    DepthwiseStrategy, DepthwiseTensors, DepthwiseTiling, launch_depthwise,
};

pub use kernels::forward::{DirectTensors, launch_direct};

pub use kernels::im2col::{Im2colTensors, launch_im2col};

pub use kernels::backward_weight::depthwise::{
    DepthwiseWeightBackwardTensors, launch_depthwise_weight_backward,
};