wanjia9506
/
jittor
forked from maxjhandsome/jittor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

polish backend compiler

This commit is contained in:
Dun Liang 2022-10-26 20:25:14 +08:00
parent 344e13948c
commit bc7467dbff
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

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

8
python/jittor/compiler.py
View File

@ -1231,16 +1231,14 @@ if has_cuda:
# from .acl_compiler import check_acl
from .extern.acl import acl_compiler
jit_utils.add_backend(acl_compiler)
has_acl = False
from .extern.rocm import rocm_compiler
jit_utils.add_backend(rocm_compiler)
from .extern.corex import corex_compiler
jit_utils.add_backend(corex_compiler)
if not has_cuda:
for mod in jit_utils.backends:
if mod.check():
break
for mod in jit_utils.backends:
if mod.check():
break
# build core
gen_jit_flags()