frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix lint (#8567)

Browse Source
This commit is contained in:
Tongzhou Wang
2018-06-15 17:34:39 -04:00
committed by Edward Z. Yang
parent c457fc994d
commit c537fd7432
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -113,6 +113,7 @@ from tools.setup_helpers.env import check_env_flag, check_negative_env_flag
config_env_vars = ['CUDA', 'CUDNN', 'MKLDNN', 'NNPACK', 'DISTRIBUTED', 'DISTRIBUTED_MW',
'SYSTEM_NCCL', 'GLOO_IBVERBS']
def hotpatch_var(var):
if check_env_flag('NO_' + var):
os.environ['USE_' + var] = '0'