pub enum ClassId {
Show 86 variants
NormWeight,
AddNorm,
NormThenAdd,
NormScale,
NormWeightOffset,
NormQuant,
AddNormQuant,
DualNorm,
GatedNorm,
EmbedNorm,
EmbedScale,
GemvNormEpilogue,
PrefillNormEpilogues,
AttnResidualNorm,
CompensatedResidual,
RecomputeFanout,
QkNormRope,
QkNormRopeKvAppend,
RopeKvAppend,
RopeQkOneLaunch,
QkNormRopeGateSplit,
QkvHeadEpilogue,
HorizontalNormRopeInsert,
KvCacheQuant,
HadamardRotation,
CrossLayerLaunch,
HostResidentKv,
Mla,
MlaPrologue,
FlashDecode,
FlashPrefill,
AttentionVariantsInline,
Softcap,
AttnOutputQuant,
AttnOutputGate,
RowWriteQuant,
AttnGqaRowPacking,
AttnLseMerge,
MlaUnabsorbInFold,
MixedBatchAttention,
SparseAttention,
GluActMul,
GluActMulQuant,
GemvGateUpGlu,
ElementwiseChains,
ElementwiseRegions,
ReluSqr,
MaskedExpertActivation,
GemmBias,
GemmScale,
MergedQkv,
MergedGateUp,
GemmDequantInkernel,
AffineInputFold,
RouterTopk,
GroupedTopk,
RouterSharedGateMerge,
MoeTokenSort,
MoeDirectPairs,
MoeGateUpAct,
MoeGatherInOperand,
MoeDownCombine,
MoeDeferredFinalize,
SharedExpertFusion,
GdnStep,
GdnInputEpilogue,
GdnDecodeWhole,
GdnPostConvPrep,
ConvBiasSilu,
ConvParts,
GdnGateProjections,
GdnGating,
FusedSampling,
Sampling,
LastLayerRows,
StridedRead,
RowWrite,
JoinInPlace,
StridedValues,
CudaGraphs,
PersistentLayerMegakernel,
AllreduceNorm,
ConvSiluL2norm,
GdnOutputQuant,
GemvQuant,
ElementwiseQuant,
}Expand description
A registered class, by name: a fused op’s class, a rule’s product, a shim’s answer. A class added to the registry is a variant every exhaustive match over it must answer.
Variants§
NormWeight
AddNorm
NormThenAdd
NormScale
NormWeightOffset
NormQuant
AddNormQuant
DualNorm
GatedNorm
EmbedNorm
EmbedScale
GemvNormEpilogue
PrefillNormEpilogues
AttnResidualNorm
CompensatedResidual
RecomputeFanout
QkNormRope
QkNormRopeKvAppend
RopeKvAppend
RopeQkOneLaunch
QkNormRopeGateSplit
QkvHeadEpilogue
HorizontalNormRopeInsert
KvCacheQuant
HadamardRotation
CrossLayerLaunch
HostResidentKv
Mla
MlaPrologue
FlashDecode
FlashPrefill
AttentionVariantsInline
Softcap
AttnOutputQuant
AttnOutputGate
RowWriteQuant
AttnGqaRowPacking
AttnLseMerge
MlaUnabsorbInFold
MixedBatchAttention
SparseAttention
GluActMul
GluActMulQuant
GemvGateUpGlu
ElementwiseChains
ElementwiseRegions
ReluSqr
MaskedExpertActivation
GemmBias
GemmScale
MergedQkv
MergedGateUp
GemmDequantInkernel
AffineInputFold
RouterTopk
GroupedTopk
MoeTokenSort
MoeDirectPairs
MoeGateUpAct
MoeGatherInOperand
MoeDownCombine
MoeDeferredFinalize
GdnStep
GdnInputEpilogue
GdnDecodeWhole
GdnPostConvPrep
ConvBiasSilu
ConvParts
GdnGateProjections
GdnGating
FusedSampling
Sampling
LastLayerRows
StridedRead
RowWrite
JoinInPlace
StridedValues
CudaGraphs
PersistentLayerMegakernel
AllreduceNorm
ConvSiluL2norm
GdnOutputQuant
GemvQuant
ElementwiseQuant
Implementations§
Trait Implementations§
Source§impl Ord for ClassId
impl Ord for ClassId
Source§impl PartialOrd for ClassId
impl PartialOrd for ClassId
impl Copy for ClassId
impl Eq for ClassId
impl StructuralPartialEq for ClassId
Auto Trait Implementations§
impl Freeze for ClassId
impl RefUnwindSafe for ClassId
impl Send for ClassId
impl Sync for ClassId
impl Unpin for ClassId
impl UnsafeUnpin for ClassId
impl UnwindSafe for ClassId
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<Q, K> Comparable<K> for Q
impl<Q, K> Comparable<K> for Q
§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.