Kernels Module#
API documentation for the kernels module.
- Nv Fp4 Mo E Utils
- Apply Rope Write KV
- Argmax Kernel
- Attention Scale Utils
- Audio Fbank Kernels
- Batch Evict Kernels
- Build Layout
- Causal Conv1d
- Checkpoint Source Batch
- Common
- Conversion
- Cute Dsl F16 Moe Runner
- Cute Dsl FMHA Runner
- Cute Dsl Fmhav2 Runner
- Cute Dsl FMHA Params
- Cute Dsl Gdn Runner
- Cute Dsl GEMM Nv Fp4 Runner
- Cute Dsl GEMM Runner
- Cute Dsl Module Loader
- Cute Dsl Nvfp4 Moe Runner
- Cute Dsl Nvfp4 Moe Sm110 Runner
- Cute Dsl Ssd Runner
- Cute Dsl Tensor Descriptors
- Ddtree Kernels
- Decoder XQA Runner
- Dequant
- Dequantize
- Dflash Runtime Kernels
- Dspark Kernels
- EAGLE Accept Kernels
- EAGLE Util Kernels
- Embedding Kernels
- F16 Moe Support Kernels
- Fp16 Layout Convert
- Fp16 Moe Assemble
- Fp4 Quantize
- Gdn Kernel Utils
- Gdn Tree Chunk Kernels
- Gemma4 Audio Attention
- Gemma4 Mtp Runtime Kernels
- Gptq Activation Permutation
- Gptq Marlin Repack
- Image Util Kernels
- Initialize Cos Sin Cache
- Int4 Cute Dsl Repack
- Int4 Groupwise GEMM
- Int4 Moe Scale Repack
- Int4 Plugin V1 Repack
- Int4 Qkv Scale Concat
- Int4 Source Accessors
- Kernel
- Kernel Selector
- KV Cache Utils Kernels
- Marlin
- Marlin Repack Index
- Marlin Dtypes
- Marlin Mma
- Marlin Template
- Moe Activation Kernels
- Moe Align Sum Kernels
- Moe Marlin
- Moe Marlin Indices Kernels
- Moe Per Expert Scale Kernels
- Moe Sigmoid Group Topk Kernels
- Moe Topk Softmax Kernels
- Mtp State Scatter Kernels
- Nvfp4 Moe Layout
- Nvfp4 Moe Normalize
- Nvfp4 Moe Plugin Geforce Transform
- Nvfp4 Moe Plugin Transform
- Nvfp4 Moe Scale Transform
- Nvfp4 Moe Types
- Nvfp4 Moe Weight Transform
- Nvfp4 Dequant
- Nvfp4 Scale
- Nvfp4 Tensor
- Scale Convert
- Selective State Update
- Speculative Kernels Utils
- Ssd Varlen Metadata
- Talker Mlp Kernels
- Util Kernels
- Vectorized Types