vllm.models.muse_spark.nvidia.ops.cute_dsl
¶
Modules:
-
moe_finalize_rmsnorm–CuTeDSL fused MoE finalize with MuseSpark1x's post-expert RMSNorm.
-
qk_norm–CuTeDSL fused QK-norm for MuseSpark1x attention.