cubecl-convolution 0.8.1

CubeCL Convolution Kernels Engine
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 cubecl_matmul::components::tile::{accelerated::AcceleratedMatmul, io::Strided};
use cubecl_std::CubeOption;
pub use error::*;
pub use problem::*;
pub use selection::*;

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