-
-
Notifications
You must be signed in to change notification settings - Fork 11.2k
[EPLB] Expert histogram logging #27105
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
[EPLB] Expert histogram logging #27105
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
💡 Codex Review
Here are some automated review suggestions for this pull request.
ℹ️ About Codex in GitHub
Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you
- Open a pull request for review
- Mark a draft as ready
- Comment "@codex review".
If Codex has suggestions, it will comment; otherwise it will react with 👍.
When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR.
Signed-off-by: Patryk Saffer <patryk.saffer99@gmail.com>
44f77ec to
8328a52
Compare
|
This looks pretty cool! Please ping me when it’s ready for review. |
Purpose
Track expert metrics in MoEs.

This tracks experts usage per layer, rank usage per layer.
This adds ~1-2% overhead e2e.
Example results in grafana:
Test Plan
added test test_expert_usage_histogram.py
Test Result
passed