pub struct XmxTileConfig {
pub m: u32,
pub n: u32,
pub k: u32,
}Expand description
Tile dimensions for XMX GEMM.
Intel’s XMX engines support the following sizes on Xe-HPC (Ponte Vecchio):
- FP16 / BF16 input, FP32 accumulation: 8 × 16, 8 × 32
- INT8 / INT4 input, INT32 accumulation: 8 × 32
On Arc (Alchemist) and later, additional sizes are available.
Fields§
§m: u32Rows of the A/C matrix tile.
n: u32Columns of the B/C matrix tile.
k: u32Inner dimension (columns of A, rows of B).
Implementations§
Source§impl XmxTileConfig
impl XmxTileConfig
Sourcepub const XE_HPC_FP16: Self
pub const XE_HPC_FP16: Self
Default XMX tile size for FP16 input targeting Xe-HPC.
Sourcepub const XE_DEFAULT: Self
pub const XE_DEFAULT: Self
XMX tile size for FP32 GEMM fallback (regular compute path).
Sourcepub fn accum_elements(&self) -> u32
pub fn accum_elements(&self) -> u32
Returns the number of accumulator elements per tile.
Trait Implementations§
Source§impl Clone for XmxTileConfig
impl Clone for XmxTileConfig
Source§fn clone(&self) -> XmxTileConfig
fn clone(&self) -> XmxTileConfig
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 moreSource§impl Debug for XmxTileConfig
impl Debug for XmxTileConfig
Source§impl Default for XmxTileConfig
impl Default for XmxTileConfig
Source§impl PartialEq for XmxTileConfig
impl PartialEq for XmxTileConfig
Source§fn eq(&self, other: &XmxTileConfig) -> bool
fn eq(&self, other: &XmxTileConfig) -> bool
Tests for
self and other values to be equal, and is used by ==.impl Copy for XmxTileConfig
impl Eq for XmxTileConfig
impl StructuralPartialEq for XmxTileConfig
Auto Trait Implementations§
impl Freeze for XmxTileConfig
impl RefUnwindSafe for XmxTileConfig
impl Send for XmxTileConfig
impl Sync for XmxTileConfig
impl Unpin for XmxTileConfig
impl UnsafeUnpin for XmxTileConfig
impl UnwindSafe for XmxTileConfig
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