pytorch
6284d2a8 - wrap cudaStreamSynchronize calls (#61889)

Commit
3 years ago
wrap cudaStreamSynchronize calls (#61889) Summary: This is a first step towards creating context manager that errors out on synchronizing calls. Pull Request resolved: https://github.com/pytorch/pytorch/pull/61889 Reviewed By: albanD Differential Revision: D29805280 Pulled By: ngimel fbshipit-source-id: b66400fbe0941b7daa51e6b30abe27b9cccd4e8a
Author
Natalia Gimelshein
Parents
Loading