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