Skip to content
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

Add support to multiple process groups by syncing across ranks #151

Closed
wants to merge 1 commit into from

Conversation

shengfukevin
Copy link
Contributor

Summary

Add support to multiple process groups by syncing across ranks.

Test Plan

$ comm_replay --trace-type et --trace-path /home/sanshang/021_debug/000_code/param/trace/traces_megatronlm_gpt_43B_32ranks_pytnightly0703/execution_trace

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 5, 2024
@facebook-github-bot
Copy link
Contributor

@shengfukevin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

Copy link
Contributor Author

@shengfukevin shengfukevin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

minor comments.

@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D60788539

shengfukevin pushed a commit that referenced this pull request Aug 5, 2024
Summary:
Add support to multiple process groups by syncing across ranks.

Pull Request resolved: #151

Test Plan: /usr/local/fbcode/platform010/bin/mpirun -np 2 path-to/comm_replay.par --trace-path param_bench/fb/integration_tests/resnet-2gpu --trace-type et

Differential Revision: D60788539

Pulled By: shengfukevin
shengfukevin pushed a commit that referenced this pull request Aug 6, 2024
Summary:
Add support to multiple process groups by syncing across ranks.

Pull Request resolved: #151

Test Plan: /usr/local/fbcode/platform010/bin/mpirun -np 2 path-to/comm_replay.par --trace-path param_bench/fb/integration_tests/resnet-2gpu --trace-type et

Differential Revision: D60788539

Pulled By: shengfukevin
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D60788539

@shengfukevin
Copy link
Contributor Author

@GSSBMW, I converted the dictionary (loaded from json) between "str" to "str" to a dictionary between "int" to "str", keep the sort code. Please review it. Thanks

Summary:
Add support to multiple process groups by syncing across ranks.

Pull Request resolved: #151

Test Plan: /usr/local/fbcode/platform010/bin/mpirun -np 2 path-to/comm_replay.par --trace-path param_bench/fb/integration_tests/resnet-2gpu --trace-type et

Reviewed By: briancoutinho

Differential Revision: D60788539

Pulled By: shengfukevin
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D60788539

@facebook-github-bot
Copy link
Contributor

@shengfukevin merged this pull request in c466b60.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported Merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants