OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add torch.cuda.comm to typechecking CI (#45350)

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45350

Reviewed By: walterddr

Differential Revision: D23935750

Pulled By: malfet

fbshipit-source-id: 5a7d2d4fbc976699d80bb5caf4727c19fa2c5bc8
This commit is contained in:
Nikita Shulga 2020-09-25 12:11:30 -07:00 committed by Facebook GitHub Bot
parent 22401b850b
commit c8166d4b58
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mypy.ini
View File

@ -186,8 +186,8 @@ ignore_errors = True
[mypy-torch.cuda.amp.*]
ignore_errors = True
[mypy-torch.cuda.comm]
ignore_errors = True
#[mypy-torch.cuda.comm]
#ignore_errors = True
[mypy-torch.cuda.nccl]
ignore_errors = True

2
torch/cuda/comm.py
View File

@ -2,4 +2,4 @@
from torch.nn.parallel.comm import broadcast, broadcast_coalesced, reduce_add, \
reduce_add_coalesced, scatter, gather
__all__ = [broadcast, broadcast_coalesced, reduce_add, reduce_add_coalesced, scatter, gather]
__all__ = ['broadcast', 'broadcast_coalesced', 'reduce_add', 'reduce_add_coalesced', 'scatter', 'gather']