cubecl-convolution 0.8.1

CubeCL Convolution Kernels Engine
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13

use cubecl_core as cubecl;
use cubecl_core::{CubeType, prelude::*};
use cubecl_std::FastDivmod;

#[derive(CubeType, CubeLaunch, Clone)]
pub struct RuntimeArgs {
    pub shape_m: u32,
    pub shape_n: u32,
    pub shape_k: u32,
    pub padded_channels: FastDivmod,
    pub shape_out: Sequence<FastDivmod>,
    pub shape_channel: FastDivmod,
}