pub struct Config { /* private fields */ }
Expand description
Configuration for Accelerated instruction
Implementations§
Source§impl Config
impl Config
pub fn new( size: MatmulSize, plane_dim: u32, lhs_layout: MatrixLayout, rhs_layout: MatrixLayout, stage_dynamic_line_size: bool, lhs_line_size: u32, rhs_line_size: u32, out_line_size: u32, ) -> Self
Trait Implementations§
Source§impl CubeType for Config
impl CubeType for Config
type ExpandType = ConfigExpand
Source§fn init(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType
fn init(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType
Wrapper around the init method, necessary to type inference.
Source§impl TileConfig for Config
impl TileConfig for Config
Source§fn matrix_layout(&self, ident: Ident) -> MatrixLayout
fn matrix_layout(&self, ident: Ident) -> MatrixLayout
Returns the MatrixLayout for the given ident
Source§fn stage_line_size(&self, ident: Ident) -> u32
fn stage_line_size(&self, ident: Ident) -> u32
Returns the line size for the given ident
Source§fn tile_shape(&self) -> &MatmulSize
fn tile_shape(&self) -> &MatmulSize
Returns the shape of the tiles in the three axes m, k and n.
impl Copy for Config
impl Eq for Config
impl MatmulConfig for Config
impl StructuralPartialEq for Config
Auto Trait Implementations§
impl Freeze for Config
impl RefUnwindSafe for Config
impl Send for Config
impl Sync for Config
impl Unpin for Config
impl UnwindSafe for Config
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