pub struct DynKernel<const MR: usize, const NR: usize, Acc>where
Acc: LADatum,{
pub name: String,
pub kernel: unsafe fn(&[FusedKerSpec<Acc>]) -> isize,
pub quality: ImplementationQuality,
pub packings: Vec<(Box<dyn MMMInputFormat>, Box<dyn MMMInputFormat>)>,
pub stores: Vec<DatumType>,
pub supported_predicate: fn() -> bool,
pub boost: fn() -> isize,
pub can_fuse: fn(&FusedSpec<'_>) -> bool,
}Fields§
§name: String§kernel: unsafe fn(&[FusedKerSpec<Acc>]) -> isize§quality: ImplementationQuality§packings: Vec<(Box<dyn MMMInputFormat>, Box<dyn MMMInputFormat>)>§stores: Vec<DatumType>§supported_predicate: fn() -> bool§boost: fn() -> isize§can_fuse: fn(&FusedSpec<'_>) -> boolImplementations§
Source§impl<const MR: usize, const NR: usize, Acc> DynKernel<MR, NR, Acc>where
Acc: LADatum,
impl<const MR: usize, const NR: usize, Acc> DynKernel<MR, NR, Acc>where
Acc: LADatum,
pub fn new( name: &str, kernel: unsafe fn(&[FusedKerSpec<Acc>]) -> isize, packing_a: PackedFormat, packing_b: PackedFormat, quality: ImplementationQuality, ) -> DynKernel<MR, NR, Acc>
pub fn with_platform_condition(self, f: fn() -> bool) -> DynKernel<MR, NR, Acc>
pub fn with_boost(self, f: fn() -> isize) -> DynKernel<MR, NR, Acc>
pub fn with_packing( self, a: impl MMMInputFormat, b: impl MMMInputFormat, ) -> DynKernel<MR, NR, Acc>
pub fn with_packing_a(self, a: impl MMMInputFormat) -> DynKernel<MR, NR, Acc>
pub fn regular_pack_a(&self) -> PackedFormat
pub fn regular_pack_b(&self) -> PackedFormat
pub fn with_can_fuse( self, can_fuse: fn(&FusedSpec<'_>) -> bool, ) -> DynKernel<MR, NR, Acc>
pub fn with_store<D>(self) -> DynKernel<MR, NR, Acc>where
D: LADatum,
pub fn mmm(&self) -> Box<dyn MatMatMul>
Trait Implementations§
Source§impl<const MR: usize, const NR: usize, Acc> Debug for DynKernel<MR, NR, Acc>where
Acc: LADatum,
impl<const MR: usize, const NR: usize, Acc> Debug for DynKernel<MR, NR, Acc>where
Acc: LADatum,
Source§impl<const MR: usize, const NR: usize, Acc> MatMatMulKer for DynKernel<MR, NR, Acc>where
Acc: LADatum,
impl<const MR: usize, const NR: usize, Acc> MatMatMulKer for DynKernel<MR, NR, Acc>where
Acc: LADatum,
type Acc = Acc
fn name(&self) -> &str
fn mr(&self) -> usize
fn nr(&self) -> usize
fn quality(&self) -> ImplementationQuality
fn is_supported_here(&self) -> bool
fn can_fuse(&self, spec: &FusedSpec<'_>) -> bool
fn kernel( &self, op: &[FusedKerSpec<<DynKernel<MR, NR, Acc> as MatMatMulKer>::Acc>], ) -> isize
fn packings(&self) -> &[(Box<dyn MMMInputFormat>, Box<dyn MMMInputFormat>)]
fn stores(&self) -> Cow<'_, [DatumType]>
fn dynamic_boost(&self) -> isize
Auto Trait Implementations§
impl<const MR: usize, const NR: usize, Acc> Freeze for DynKernel<MR, NR, Acc>
impl<const MR: usize, const NR: usize, Acc> !RefUnwindSafe for DynKernel<MR, NR, Acc>
impl<const MR: usize, const NR: usize, Acc> Send for DynKernel<MR, NR, Acc>
impl<const MR: usize, const NR: usize, Acc> Sync for DynKernel<MR, NR, Acc>
impl<const MR: usize, const NR: usize, Acc> Unpin for DynKernel<MR, NR, Acc>
impl<const MR: usize, const NR: usize, Acc> !UnwindSafe for DynKernel<MR, NR, Acc>
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
Source§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Convert
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can
then be further downcast into Box<ConcreteType> where ConcreteType implements Trait.Source§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Convert
Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be
further downcast into Rc<ConcreteType> where ConcreteType implements Trait.Source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.Source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Convert
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.Source§impl<T> DowncastSync for T
impl<T> DowncastSync for T
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more