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