From db81cc938f280bf4b2370479face9c2706c57f4a Mon Sep 17 00:00:00 2001 From: lzhengning Date: Mon, 17 Oct 2022 12:08:51 +0800 Subject: [PATCH] fix: default to use CUDA when both CUDA and ROCm are installed --- python/jittor/__init__.py | 2 +- python/jittor/compiler.py | 9 +++++---- 2 files changed, 6 insertions(+), 5 deletions(-) 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()