From 0c8b2e3bb9704ed83c6af9d811b3952e3cb912f1 Mon Sep 17 00:00:00 2001 From: Zheng-Ning Liu Date: Wed, 23 Feb 2022 21:32:32 +0800 Subject: [PATCH] Update compiler.py fix-bug: explicitly set log_v when querying cuda to avoid wrong output of jittor_utils. --- python/jittor/compiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/jittor/compiler.py b/python/jittor/compiler.py index f1f2557f..054b2d18 100644 --- 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)