polish import message

python/jittor/__init__.py

@@ -9,7 +9,7 @@
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
 
-__version__ = '1.3.5.24'
+__version__ = '1.3.5.25'
 from jittor_utils import lock
 with lock.lock_scope():
     ori_int = int

python/jittor/compile_extern.py

@@ -27,7 +27,7 @@ def search_file(dirs, name, prefer_version=()):
         for i in range(len(prefer_version),-1,-1):
             vname = ".".join((fname,)+prefer_version[:i])
             if os.path.isfile(vname):
-                LOG.i(f"found {vname}")
+                LOG.v(f"found {vname}")
                 return vname
     LOG.f(f"file {name} not found in {dirs}")
 

python/jittor/compiler.py

@@ -842,7 +842,7 @@ def compile_extern():
 def check_cuda():
     if not nvcc_path:
         return
-    global cc_flags, has_cuda, core_link_flags, cuda_dir, cuda_lib, cuda_include, cuda_home, cuda_bin
+    global cc_flags, has_cuda, is_cuda, core_link_flags, cuda_dir, cuda_lib, cuda_include, cuda_home, cuda_bin
     cuda_dir = os.path.dirname(get_full_path_of_executable(nvcc_path))
     cuda_bin = cuda_dir
     cuda_home = os.path.abspath(os.path.join(cuda_dir, ".."))

python/jittor/src/misc/cuda_flags.cc

@@ -24,6 +24,7 @@ DEFINE_FLAG_WITH_SETTER(int, device_id, -1,
 EXTERN_LIB void sync_all(bool device_sync);
 
 void setter_use_cuda(int value) {
+    if (use_cuda == value) return;
 #ifdef HAS_CUDA
     if (value) {
         int count=0;