Kernels Module#
API documentation for the kernels module.
- Nv Fp4 Mo E Utils
- Apply Rope Write KV
- Batch Evict Kernels
- Build Layout
- Causal Conv1d
- Common
- Context FMHA Runner
- Conversion
- Cute Dsl Ffpa Runner
- Cute Dsl FMHA Runner
- Cute Dsl Gdn Runner
- Cute Dsl GEMM Runner
- Cute Dsl Nvfp4 Moe Runner
- Cute Dsl Nvfp4 Moe Sm110 Runner
- Cute Dsl Ssd Runner
- Decoder XQA Runner
- Dequant
- Dequantize
- Dflash Accept Kernels
- Dflash Runtime Kernels
- EAGLE Accept Kernels
- EAGLE Util Kernels
- Embedding Kernels
- FMHA Params V2
- Fp4 Quantize
- Gdn Kernel Utils
- Gemma4 Audio Attention
- Image Util Kernels
- Initialize Cos Sin Cache
- Int4 Groupwise GEMM
- Kernel
- Kernel Selector
- KV Cache Utils Kernels
- Marlin
- Marlin Dtypes
- Marlin Mma
- Marlin Template
- Moe Activation Kernels
- Moe Align Sum Kernels
- Moe Marlin
- Moe Marlin Indices Kernels
- Moe Sigmoid Group Topk Kernels
- Moe Topk Softmax Kernels
- Mtp State Scatter Kernels
- Nvfp4 Moe Types
- Nvfp4 Dequant
- Nvfp4 Tensor
- Selective State Update
- Ssd Varlen Metadata
- Talker Mlp Kernels
- Util Kernels
- Vectorized Types