cubek-convolution 0.2.0-pre.3

CubeK: Convolution Kernels
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

pub mod global;
pub mod stage;

mod config;
mod error;
mod problem;
mod selection;

pub use config::*;
use cubek_matmul::components::tile::cmma::CmmaMatmul;
use cubek_std::tile::Strided;
pub use error::*;
pub use problem::*;
pub use selection::*;

/// Convolution using `AcceleratedMatmul`
pub type AcceleratedConv = CmmaMatmul<Option<Strided>>;