Skip to content

Comments

Add MORI-EP support for Expert Parallelism in MoE layers#2

Open
mpashkovskii wants to merge 1 commit intomainfrom
fix/mori-ep
Open

Add MORI-EP support for Expert Parallelism in MoE layers#2
mpashkovskii wants to merge 1 commit intomainfrom
fix/mori-ep

Conversation

@mpashkovskii
Copy link
Owner

@mpashkovskii mpashkovskii commented Jan 23, 2026

  • Implemented MoriPrepareAndFinalize class for efficient dispatch and combine operations using MORI-EP.
  • Introduced utilities for MORI buffer initialization and management in mori_utils.py.
  • Added support for FP8 and BF16 configurations, along with intra-node and inter-node communication.
  • Enhanced logging for better traceability of MORI-EP initialization and operations.
  • Included detailed documentation and comments for clarity on usage and configuration.

Purpose

Test Plan

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.
  • (Optional) Release notes update. If your change is user facing, please update the release notes draft in the Google Doc.

- Implemented `MoriPrepareAndFinalize` class for efficient dispatch and combine operations using MORI-EP.
- Introduced utilities for MORI buffer initialization and management in `mori_utils.py`.
- Added support for FP8 and BF16 configurations, along with intra-node and inter-node communication.
- Enhanced logging for better traceability of MORI-EP initialization and operations.
- Included detailed documentation and comments for clarity on usage and configuration.
mpashkovskii pushed a commit that referenced this pull request Feb 6, 2026
…3058)

Signed-off-by: ramos <49182011+nemoramo@users.noreply.github.com>
Signed-off-by: mayufeng <mayufeng@example.com>
Co-authored-by: mayufeng <mayufeng@example.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant