Skip to main content

CudaComputeKernel

cubecl_cuda

Type Alias CudaComputeKernel 

Source 
pub type CudaComputeKernel = ComputeKernel<CudaDialect<CudaWmmaCompiler>>;

Aliased Type§

pub struct CudaComputeKernel {
    pub tensor_maps: Vec<Binding<CudaDialect<CudaWmmaCompiler>>>,
    pub buffers: Vec<Binding<CudaDialect<CudaWmmaCompiler>>>,
    pub scalars: Vec<(Elem<CudaDialect<CudaWmmaCompiler>>, usize)>,
    pub meta_static_len: usize,
    pub body: Body<CudaDialect<CudaWmmaCompiler>>,
    pub cube_dim: CubeDim,
    pub cluster_dim: Option<CubeDim>,
    pub extensions: Vec<Extension<CudaDialect<CudaWmmaCompiler>>>,
    pub flags: Flags<CudaDialect<CudaWmmaCompiler>>,
    pub items: HashSet<Item<CudaDialect<CudaWmmaCompiler>>>,
    pub kernel_name: String,
}

Fields§

§tensor_maps: Vec<Binding<CudaDialect<CudaWmmaCompiler>>>§buffers: Vec<Binding<CudaDialect<CudaWmmaCompiler>>>§scalars: Vec<(Elem<CudaDialect<CudaWmmaCompiler>>, usize)>§meta_static_len: usize§body: Body<CudaDialect<CudaWmmaCompiler>>§cube_dim: CubeDim§cluster_dim: Option<CubeDim>§extensions: Vec<Extension<CudaDialect<CudaWmmaCompiler>>>§flags: Flags<CudaDialect<CudaWmmaCompiler>>§items: HashSet<Item<CudaDialect<CudaWmmaCompiler>>>§kernel_name: String