cubecl_convolution/kernels/layered/selector/
base.rs

1use cubecl_core as cubecl;
2use cubecl_core::{CubeType, prelude::*};
3use cubecl_std::FastDivmod;
4
5#[derive(CubeType, CubeLaunch, Clone)]
6pub struct RuntimeArgs {
7    pub shape_m: u32,
8    pub shape_n: u32,
9    pub shape_k: u32,
10    pub padded_channels: FastDivmod,
11    pub shape_out: Sequence<FastDivmod>,
12    pub shape_channel: FastDivmod,
13}