KEMBAR78
Restore current streams on dst device after switching streams by mrshenli · Pull Request #17439 · pytorch/pytorch · GitHub
Skip to content

Conversation

@mrshenli
Copy link
Contributor

When switching back to d0 from a stream on a different device d1, we need to restore the current streams on both d0 and d1. The current implementation only does that for d0.

@mrshenli
Copy link
Contributor Author

The failed test passed in rerun.

Copy link
Member

@soumith soumith left a comment

Choose a reason for hiding this comment

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

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.

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants