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
Implementation of torch::cuda::synchronize #50072
Conversation
💊 CI failures summary and remediationsAs of commit c59c611 (more details on the Dr. CI page):
ci.pytorch.org: 1 failedThis 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 to the (internal) Dr. CI Users group. This comment has been revised 7 times. |
Codecov Report
@@ Coverage Diff @@
## master #50072 +/- ##
==========================================
- Coverage 80.67% 80.66% -0.01%
==========================================
Files 1899 1899
Lines 205951 205959 +8
==========================================
- Hits 166149 166146 -3
- Misses 39802 39813 +11 |
I think CUDAHooksInterface is the correct entry, no need to add a method to DeviceGuard. |
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.
Per discussion with the author, this is good to me now.
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.
@jbschlosser has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
@jbschlosser merged this pull request in 7d9eb6c. |
Summary: Adding `torch::cuda::synchronize()` to libtorch. Note that the implementation here adds a new method to the `CUDAHooksInterface`. An alternative that was suggested to me is to add a method to the `DeviceGuard` interface. Fixes pytorch#47722 Pull Request resolved: pytorch#50072 Reviewed By: H-Huang Differential Revision: D25804342 Pulled By: jbschlosser fbshipit-source-id: 45aa61d7c6fbfd3178caf2eb5ec053d6c01b5a43
Adding
torch::cuda::synchronize()
to libtorch. Note that the implementation here adds a new method to theCUDAHooksInterface
. An alternative that was suggested to me is to add a method to theDeviceGuard
interface.Fixes #47722