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

Update compiler.py 

fix-bug: explicitly set log_v when querying cuda to avoid wrong output of jittor_utils.
This commit is contained in:
Zheng-Ning Liu 2022-02-23 21:32:32 +08:00 committed by GitHub
parent 4383886feb
commit 0c8b2e3bb9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -987,7 +987,7 @@ if nvcc_path:
nvcc_version = list(map(int,v.split('.'))) nvcc_version = list(map(int,v.split('.')))
cu += v cu += v
try: try:
r, s = sp.getstatusoutput(f"{sys.executable} -m jittor_utils.query_cuda_cc") r, s = sp.getstatusoutput(f"log_v=0 {sys.executable} -m jittor_utils.query_cuda_cc")
if r==0: if r==0:
s = sorted(list(set(s.strip().split()))) s = sorted(list(set(s.strip().split())))
cu += "_sm_" + "_".join(s) cu += "_sm_" + "_".join(s)