Skip to main content

cubek_convolution/components/global/
args.rs

1use cubecl::{prelude::*, std::FastDivmod};
2
3use crate::components::ConvolutionOperation;
4
5#[derive(CubeType, CubeLaunch, Clone)]
6#[expand(derive(Clone))]
7pub struct RuntimeArgs {
8    pub shape_k: u32,
9    pub channels: u32,
10    pub padded_channels: FastDivmod<u32>,
11    #[cube(comptime)]
12    pub operation: ConvolutionOperation,
13}