Skip to content

vllm.models.kimi_k3.nvidia.ops.cute_dsl

Modules:

  • gemm_rs

    SM100 BF16 GEMM with a fused tensor-parallel reduce-scatter.

  • latent_moe_tail

    CuTe DSL kernels for KimiK3LatentMoETailOp.