Skip to content

refactor: Remove separate moe_mesh references - #2123

Closed
edjson wants to merge 25 commits into
NVIDIA-NeMo:mainfrom
edjson:feat/remove-moe-mesh-user-api
Closed

refactor: Remove separate moe_mesh references#2123
edjson wants to merge 25 commits into
NVIDIA-NeMo:mainfrom
edjson:feat/remove-moe-mesh-user-api

Conversation

@edjson

@edjson edjson commented May 4, 2026

Copy link
Copy Markdown
Contributor

What does this PR do ?

Removes moe_mesh from the user-facing API

Changelog

  • auto_model.py: removed moe_mesh from signature, docstrings, _retry forwarding, and MeshContext.from_meshes calls
  • extract_nemo_activations.py: removed stale moe_mesh=
  • test_auto_model.py: added TestMoeMeshRemovedFromAPI to guard against regressions

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?

If you haven't finished some of the above items you can still open "Draft" PR.

Additional Information

Signed-off-by: Edison <edisonggacc@gmail.com>
@copy-pr-bot

copy-pr-bot Bot commented May 4, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@akoumpa

akoumpa commented May 4, 2026

Copy link
Copy Markdown
Contributor

/ok to test f2e9211

@akoumpa

akoumpa commented May 4, 2026

Copy link
Copy Markdown
Contributor

/ok to test 06f1dab

@akoumpa

akoumpa commented May 13, 2026

Copy link
Copy Markdown
Contributor

/ok to test cfeaacf

@akoumpa

akoumpa commented May 13, 2026

Copy link
Copy Markdown
Contributor

/ok to test dadce47

@akoumpa

akoumpa commented May 14, 2026

Copy link
Copy Markdown
Contributor

/ok to test b4a2d03

@akoumpa

akoumpa commented May 14, 2026

Copy link
Copy Markdown
Contributor

/ok to test 0998758

@akoumpa

akoumpa commented May 18, 2026

Copy link
Copy Markdown
Contributor

/ok to test eaef478

@edjson

edjson commented May 28, 2026

Copy link
Copy Markdown
Contributor Author

@akoumpa could you test this when you get the chance please?

@akoumpa

akoumpa commented May 28, 2026

Copy link
Copy Markdown
Contributor

/ok to test 596b60b

@edjson

edjson commented Jun 11, 2026

Copy link
Copy Markdown
Contributor Author

Closing in favor of #2266

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.

4 participants