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
[RPC Framework] Support remote device format "<workername>/<device>" #46773
Conversation
Changed the constructor of RemoteModule to accept a `remote_device` arg in the following format: "<workername>/<device>" (e.g., "trainer0/cpu", "ps0/cuda:0") This arg merges the original `on` and `device` arg. Differential Revision: [D24482562](https://our.internmc.facebook.com/intern/diff/D24482562/) [ghstack-poisoned]
Changed the constructor of RemoteModule to accept a `remote_device` arg in the following format: "<workername>/<device>" (e.g., "trainer0/cpu", "ps0/cuda:0") This arg merges the original `on` and `device` arg. Differential Revision: [D24482562](https://our.internmc.facebook.com/intern/diff/D24482562/) ghstack-source-id: 115042926 Pull Request resolved: #46773
Codecov Report
@@ Coverage Diff @@
## gh/SciPioneer/18/base #46773 +/- ##
=========================================================
- Coverage 68.44% 68.43% -0.01%
=========================================================
Files 413 413
Lines 54366 54368 +2
=========================================================
- Hits 37210 37209 -1
- Misses 17156 17159 +3 |
💊 CI failures summary and remediationsAs of commit 3a89ecc (more details on the Dr. CI page):
codecov.io: 1 failed
This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions on the GitHub issue tracker or post in the (internal) Dr. CI Users group. This comment has been revised 9 times. |
Would be nice to mention the original issue in the PR summary: #46554 |
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.
Thanks for getting this PR up quickly!
…/<device>"" Changed the constructor of RemoteModule to accept a `remote_device` arg in the following format: "<workername>/<device>" (e.g., "trainer0/cpu", "ps0/cuda:0") This arg merges the original `on` and `device` arg. Differential Revision: [D24482562](https://our.internmc.facebook.com/intern/diff/D24482562/) [ghstack-poisoned]
Pull Request resolved: #46773 Changed the constructor of RemoteModule to accept a `remote_device` arg in the following format: "<workername>/<device>" (e.g., "trainer0/cpu", "ps0/cuda:0") This arg merges the original `on` and `device` arg. ghstack-source-id: 115340442 Differential Revision: [D24482562](https://our.internmc.facebook.com/intern/diff/D24482562/)
…/<device>"" Changed the constructor of RemoteModule to accept a `remote_device` arg in the following format: "<workername>/<device>" (e.g., "trainer0/cpu", "ps0/cuda:0") This arg merges the original `on` and `device` arg. Differential Revision: [D24482562](https://our.internmc.facebook.com/intern/diff/D24482562/) [ghstack-poisoned]
Pull Request resolved: #46773 Changed the constructor of RemoteModule to accept a `remote_device` arg in the following format: "<workername>/<device>" (e.g., "trainer0/cpu", "ps0/cuda:0") This arg merges the original `on` and `device` arg. Original PR issue: RemoteDevice Format #46554 ghstack-source-id: 115448051 Differential Revision: [D24482562](https://our.internmc.facebook.com/intern/diff/D24482562/)
Thanks for the reminder! |
This pull request has been merged in cab32d9. |
Stack from ghstack:
Changed the constructor of RemoteModule to accept a
remote_device
arg in the following format:"/" (e.g., "trainer0/cpu", "ps0/cuda:0")
This arg merges the original
on
anddevice
arg.Original PR issue: RemoteDevice Format #46554
Differential Revision: D24482562