Skip to content

Conversation

NicolasHug
Copy link
Member

This PR avoids calling torch.cuda.synchronize() if cuda is not available on the machine. This is to address #1204 (comment)

@prabhat00155
Copy link
Contributor

Thanks @NicolasHug, looks good to me.

@datumbox datumbox added bug and removed fix labels Apr 17, 2021
Copy link
Contributor

@datumbox datumbox left a comment

Choose a reason for hiding this comment

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

LGTM, thanks!

@NicolasHug I changed the tag from fix to bug because from an earlier discussion with @fmassa we said to use the same tag for marking Issues and PRs. Can't find it at the moment to link it. If there is a change on that let me know.

@datumbox datumbox merged commit dc42f93 into pytorch:master Apr 17, 2021
facebook-github-bot pushed a commit that referenced this pull request May 4, 2021
Summary: Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

Reviewed By: NicolasHug

Differential Revision: D28169155

fbshipit-source-id: 16236971b990960ca4df92f4d82c0713c5f6a30a
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.

4 participants