frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

wrap cudaStreamSynchronize calls (#61889)

Browse Source 
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
This commit is contained in:
Natalia Gimelshein
2021-07-21 19:29:19 -07:00
committed by Facebook GitHub Bot
parent 3d6aa3a2f6
commit 6284d2a82b
20 changed files with 89 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Makefile
View File

@ -87,7 +87,8 @@ quick_checks:
--step 'Ensure no unqualified noqa' \
--step 'Ensure no unqualified type ignore' \
--step 'Ensure no direct cub include' \
--step 'Ensure correct trailing newlines'
--step 'Ensure correct trailing newlines' \
--step 'Ensure no raw cuda api calls'
flake8:
@$(PYTHON) tools/actions_local_runner.py \