From c8166d4b58d877e4b6f9b6160fd83d6775fd1e04 Mon Sep 17 00:00:00 2001 From: Nikita Shulga Date: Fri, 25 Sep 2020 12:11:30 -0700 Subject: [PATCH] Add `torch.cuda.comm` to typechecking CI (#45350) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45350 Reviewed By: walterddr Differential Revision: D23935750 Pulled By: malfet fbshipit-source-id: 5a7d2d4fbc976699d80bb5caf4727c19fa2c5bc8 --- mypy.ini | 4 ++-- torch/cuda/comm.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mypy.ini b/mypy.ini index a7c82cb6935..50287b8ee99 100644 --- a/mypy.ini +++ b/mypy.ini @@ -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 diff --git a/torch/cuda/comm.py b/torch/cuda/comm.py index f9856eda380..557ffb0c0de 100644 --- a/torch/cuda/comm.py +++ b/torch/cuda/comm.py @@ -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']