cubek_convolution/components/global/
args.rs

1use cubecl::prelude::*;
2use cubecl::std::FastDivmod;
3
4use crate::components::ConvolutionOperation;
5
6#[derive(CubeType, CubeLaunch, 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}