Skip to main content

rudnn/convolution/components/global/
args.rs

1use ruda_kernel::dsl as kernel_dsl;
2use ruda_kernel::dsl::prelude::*;
3use ruda_kernel::library::FastDivmod;
4
5use crate::convolution::components::ConvolutionOperation;
6
7#[derive(RudaType, RudaLaunch, Clone)]
8pub struct RuntimeArgs {
9    pub shape_k: u32,
10    pub channels: u32,
11    pub padded_channels: FastDivmod<u32>,
12    #[ruda(comptime)]
13    pub operation: ConvolutionOperation,
14}