lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix nccl

This commit is contained in:
Dun Liang 2020-04-02 16:09:09 +08:00
parent 2dd8e911df
commit 48aeff91e7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/jittor/compile_extern.py
View File

@ -263,11 +263,11 @@ def install_nccl(root_folder):
def setup_nccl():
global nccl_ops, use_nccl
use_nccl = os.environ.get("use_nccl", "1")=="1"
nccl_ops = None
if not has_cuda:
use_nccl = False
return
use_nccl = os.environ.get("use_nccl", "1")=="1"
nccl_ops = None
if not use_nccl: return
nccl_include_path = os.environ.get("nccl_include_path")
nccl_lib_path = os.environ.get("nccl_lib_path")