pub enum CutlassEpilogue {
LinearCombination,
Relu,
}Expand description
A fused pointwise epilogue our compiler folds INTO the GEMM — the thing cuBLAS cannot do in-kernel. cuBLAS must write the GEMM result to global memory and run a SECOND kernel for the activation (an extra full-output read+write); fusing it here removes that round-trip and BEATS a cuBLAS+elementwise baseline.
Variants§
LinearCombination
alphaacc + betaC — a plain GEMM (parity with cuBLAS).
Relu
relu(alphaacc + betaC) — GEMM + activation fused in one kernel (the KernelBench Level-2 win).
Trait Implementations§
Source§impl Clone for CutlassEpilogue
impl Clone for CutlassEpilogue
Source§fn clone(&self) -> CutlassEpilogue
fn clone(&self) -> CutlassEpilogue
Returns a duplicate of the value. Read more
1.0.0 · 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 CutlassEpilogue
impl Debug for CutlassEpilogue
Source§impl PartialEq for CutlassEpilogue
impl PartialEq for CutlassEpilogue
impl Copy for CutlassEpilogue
impl Eq for CutlassEpilogue
impl StructuralPartialEq for CutlassEpilogue
Auto Trait Implementations§
impl Freeze for CutlassEpilogue
impl RefUnwindSafe for CutlassEpilogue
impl Send for CutlassEpilogue
impl Sync for CutlassEpilogue
impl Unpin for CutlassEpilogue
impl UnsafeUnpin for CutlassEpilogue
impl UnwindSafe for CutlassEpilogue
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,
§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
§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.§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.§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.§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.§impl<T> DowncastSync for T
impl<T> DowncastSync for T
§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Checks if this value is equivalent to the given key. Read more
§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.