Skip to main content

HipComputeKernel

cubecl_hip::runtime

Type Alias HipComputeKernel 

Source 
pub type HipComputeKernel = ComputeKernel<HipDialect<WmmaIntrinsicCompiler>>;

Aliased Type§

pub struct HipComputeKernel {
    pub tensor_maps: Vec<KernelArg<HipDialect<WmmaIntrinsicCompiler>>>,
    pub buffers: Vec<KernelArg<HipDialect<WmmaIntrinsicCompiler>>>,
    pub scalars: Vec<(Elem<HipDialect<WmmaIntrinsicCompiler>>, usize)>,
    pub info: Info,
    pub meta_static_len: usize,
    pub body: Body<HipDialect<WmmaIntrinsicCompiler>>,
    pub cube_dim: CubeDim,
    pub cluster_dim: Option<CubeDim>,
    pub extensions: Vec<Extension<HipDialect<WmmaIntrinsicCompiler>>>,
    pub flags: Flags<HipDialect<WmmaIntrinsicCompiler>>,
    pub items: HashSet<Item<HipDialect<WmmaIntrinsicCompiler>>>,
    pub kernel_name: String,
}

Fields§

§tensor_maps: Vec<KernelArg<HipDialect<WmmaIntrinsicCompiler>>>§buffers: Vec<KernelArg<HipDialect<WmmaIntrinsicCompiler>>>§scalars: Vec<(Elem<HipDialect<WmmaIntrinsicCompiler>>, usize)>§info: Info§meta_static_len: usize§body: Body<HipDialect<WmmaIntrinsicCompiler>>§cube_dim: CubeDim§cluster_dim: Option<CubeDim>§extensions: Vec<Extension<HipDialect<WmmaIntrinsicCompiler>>>§flags: Flags<HipDialect<WmmaIntrinsicCompiler>>§items: HashSet<Item<HipDialect<WmmaIntrinsicCompiler>>>§kernel_name: String