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

polish cu key

This commit is contained in:
Dun Liang 2021-11-04 20:51:58 +08:00
parent 0274c22afd
commit b3ee4a54e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/compiler.py
View File

@ -991,7 +991,7 @@ if nvcc_path:
cu += "_sm_" + "_".join(s) cu += "_sm_" + "_".join(s)
if "cuda_arch" not in os.environ: if "cuda_arch" not in os.environ:
os.environ["cuda_arch"] = " ".join(cu) os.environ["cuda_arch"] = " ".join(cu)
cu = cu.replace(":", "") cu = cu.replace(":", "").replace(" ", "")
except: except:
pass pass
LOG.i("cuda key:", cu) LOG.i("cuda key:", cu)