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

Implementation of torch::cuda::synchronize #50072

Closed
wants to merge 2 commits into from

Conversation

jbschlosser
Copy link
Contributor

@jbschlosser jbschlosser commented Jan 4, 2021

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 #47722

@facebook-github-bot
Copy link
Contributor

facebook-github-bot commented Jan 4, 2021

💊 CI failures summary and remediations

As of commit c59c611 (more details on the Dr. CI page):


  • 2/2 failures possibly* introduced in this PR
    • 2/2 non-CircleCI failure(s)

ci.pytorch.org: 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 to the (internal) Dr. CI Users group.

This comment has been revised 7 times.

@codecov
Copy link

codecov bot commented Jan 5, 2021

Codecov Report

Merging #50072 (c59c611) into master (74dcb6d) will decrease coverage by 0.00%.
The diff coverage is 0.00%.

@@            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     

@glaringlee
Copy link
Contributor

I think CUDAHooksInterface is the correct entry, no need to add a method to DeviceGuard.

@jbschlosser jbschlosser changed the title [WIP] Implementation of torch::cuda::synchronize Implementation of torch::cuda::synchronize Jan 6, 2021
Copy link
Contributor

@glaringlee glaringlee left a 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.

Copy link
Contributor

@facebook-github-bot facebook-github-bot left a 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.

@facebook-github-bot
Copy link
Contributor

@jbschlosser merged this pull request in 7d9eb6c.

hwangdeyu pushed a commit to hwangdeyu/pytorch that referenced this pull request Jan 14, 2021
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Implement torch::cuda::synchronize()
3 participants