Update compiler.py

fix-bug: explicitly set log_v when querying cuda to avoid wrong output of jittor_utils.
2022-02-23 21:32:32 +08:00 · 2022-02-23 21:32:32 +08:00 · 0c8b2e3bb9
parent 4383886feb
commit 0c8b2e3bb9
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/compiler.py
+++ b/python/jittor/compiler.py
@ -987,7 +987,7 @@ if nvcc_path:
    nvcc_version = list(map(int,v.split('.')))
    cu += v
    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:
            s = sorted(list(set(s.strip().split())))
            cu += "_sm_" + "_".join(s)