Struct BatchedGemmDescriptor 

pub struct BatchedGemmDescriptor {
    pub m: i32,
    pub n: i32,
    pub k: i32,
    pub batch_count: i32,
    pub layout: LayoutSku,
    pub epilogue: EpilogueKind,
}

Problem shape and configuration handed to BatchedGemmPlan::select.

All batches share the same (M, N, K) and per-batch operands are addressed by adding i * stride_* (in elements) to the base pointer — see BatchedGemmArgs. For variable-shape grouped problems use GroupedGemmPlan instead.

Fields

m: i32

Output row count (per batch).

n: i32

Output column count (per batch).

k: i32

Reduction depth (per batch).

batch_count: i32

Number of batches launched in a single kernel invocation.

layout: LayoutSku

Layout SKU. v1 supports only LayoutSku::Rcr.

epilogue: EpilogueKind

Epilogue kind. v1 supports only EpilogueKind::Identity.

Trait Implementations

impl Clone for BatchedGemmDescriptor

fn clone(&self) -> BatchedGemmDescriptor

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for BatchedGemmDescriptor

impl Debug for BatchedGemmDescriptor

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.