pub struct IntGemmDescriptor {
pub m: i32,
pub n: i32,
pub k: i32,
pub layout: LayoutSku,
pub epilogue: EpilogueKind,
}Expand description
Problem shape and configuration handed to
IntGemmPlan::select.
Parallel to GemmDescriptor for the integer GEMM family.
LayoutSku and EpilogueKind are shared with the float family,
but coverage on int8 is limited to LayoutSku::Rcr in this
release — selecting LayoutSku::Rrr returns
Error::Unsupported. The
RowMajor × RowMajor integer SKU lives in the bespoke
baracuda-kernels-sys kernel family (lands in workspace alpha.16).
Fields§
§m: i32Output row count.
n: i32Output column count.
k: i32Reduction depth.
layout: LayoutSkuLayout SKU. Today’s int8 CUTLASS SKUs require LayoutSku::Rcr.
epilogue: EpilogueKindEpilogue kind. All five variants are supported on int8 RCR.
Trait Implementations§
Source§impl Clone for IntGemmDescriptor
impl Clone for IntGemmDescriptor
Source§fn clone(&self) -> IntGemmDescriptor
fn clone(&self) -> IntGemmDescriptor
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreimpl Copy for IntGemmDescriptor
Auto Trait Implementations§
impl Freeze for IntGemmDescriptor
impl RefUnwindSafe for IntGemmDescriptor
impl Send for IntGemmDescriptor
impl Sync for IntGemmDescriptor
impl Unpin for IntGemmDescriptor
impl UnsafeUnpin for IntGemmDescriptor
impl UnwindSafe for IntGemmDescriptor
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more