Skip to content

feat(moe): support clamped SwiGLU expert paths - #5941

Closed
FDecaYed wants to merge 1 commit into
NVIDIA:pull-request/5940from
FDecaYed:codex/clamped-swiglu-moe
Closed

feat(moe): support clamped SwiGLU expert paths#5941
FDecaYed wants to merge 1 commit into
NVIDIA:pull-request/5940from
FDecaYed:codex/clamped-swiglu-moe

Conversation

@FDecaYed

Copy link
Copy Markdown
Contributor

What

Extend the core ClampedSwiGLU primitive into MoE-owned paths:

  • routed expert fallback and token-weighted fusion;
  • TE grouped MLP op-fuser using ScaledClampedQGeGLU when available;
  • shared-expert overlap fallback;
  • current-main FusedSharedExpertMLP grouped op-fuser;
  • explicit TE >= 2.17 and operator-availability gates;
  • focused construction, version-gating, fallback, and CUDA integration coverage.

Stack

Provenance

Testing

  • isort --check-only, ruff, Black --check, compileall, and git diff --check passed.
  • An independent read-only review closed non-positive/NaN validation and shared-expert op-fuser gaps.
  • Focused pytest collection is blocked locally because this host lacks Torch; execution is left to stack CI.

Route ClampedSwiGLU through routed and shared expert fallbacks plus Transformer Engine grouped op-fusers, including TE availability gates and focused coverage.

Signed-off-by: Deyu Fu <Deyu.Foo@gmail.com>
@copy-pr-bot

copy-pr-bot Bot commented Jul 21, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@FDecaYed

Copy link
Copy Markdown
Contributor Author

/ok to test 1412381

Copy link
Copy Markdown
Contributor Author

Closing this stacked boundary because it is absorbed into P5. The complete clamped-SwiGLU feature across primitive, routed/shared, fallback, and Transformer Engine paths will be published by updating #5940. Tracking PR: #5795.

@FDecaYed FDecaYed closed this Jul 23, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant