diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py index b677fa9c..826dcc7d 100644 --- a/python/jittor/__init__.py +++ b/python/jittor/__init__.py @@ -9,7 +9,7 @@ # file 'LICENSE.txt', which is part of this source code package. # *************************************************************** -__version__ = '1.3.5.19' +__version__ = '1.3.5.20' from jittor_utils import lock with lock.lock_scope(): ori_int = int diff --git a/python/jittor/compiler.py b/python/jittor/compiler.py index c1a6a002..cd2c565a 100644 --- a/python/jittor/compiler.py +++ b/python/jittor/compiler.py @@ -1234,10 +1234,11 @@ jit_utils.add_backend(acl_compiler) from .extern.rocm import rocm_compiler jit_utils.add_backend(rocm_compiler) -for mod in jit_utils.backends: - mod.check() - # if mod.check(): - # break +if not has_cuda: + for mod in jit_utils.backends: + mod.check() + # if mod.check(): + # break # build core gen_jit_flags()