Skip to content

vllm.models.muse_spark.nvidia.specialized_decode.cutedsl.common

Modules:

  • draft_boundary –

    The DSpark draft's residual boundary, bitwise equal to the eager chain.

  • dual_norm –

    The dual norm at every Muse Spark residual boundary.

  • host –

    Host-side helpers for the CuTe providers of the specialized decode.

  • mailbox –

    Lamport mailboxes in symmetric memory.

  • policy –

    Per-batch tables and per-TP-size choices of the specialized decode.

  • primitives –

    PTX building blocks and launch helpers of the Muse Spark specialized decode.

  • row_norm –

    Single-tile RMS norm of one row, bitwise equal to the stock Triton norm