lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update corex backend

This commit is contained in:
Dun Liang 2022-11-04 14:45:02 +08:00
parent 7e40d7831c
commit 20357caf42
2 changed files with 7 additions and 2 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.25'
__version__ = '1.3.5.26'
from jittor_utils import lock
with lock.lock_scope():
ori_int = int

7
python/jittor/extern/corex/corex_compiler.py vendored
View File

@ -38,6 +38,11 @@ string process_acl(const string& src, const string& name, const map<string,strin
if (name == "jit_compiler.cc") {
// remove asm tuner
new_src = token_replace_all(new_src, "cmd = python_path$1;", "");
new_src = token_replace_all(new_src, "JPU(op_compiler($1));",
R"(JPU(op_compiler($1));
*extra_flags2 = replace(*extra_flags2, "--extended-lambda", "");
*extra_flags2 = replace(*extra_flags2, "--expt-relaxed-constexpr", "");
)");
new_src = token_replace_all(new_src,
"if (is_cuda_op && $1 != string::npos)",
"if (is_cuda_op)");
@ -57,7 +62,7 @@ string process_acl(const string& src, const string& name, const map<string,strin
compiler.cc_path = compiler.nvcc_path
compiler.cc_flags = compiler.cc_flags.replace("-fopenmp", "")
# compiler.nvcc_flags = cc_flags_to_corex(compiler.cc_flags)
compiler.nvcc_flags = compiler.cc_flags + " -x cu -Ofast -DNO_ATOMIC64 "
compiler.nvcc_flags = compiler.cc_flags + " -x cu -Ofast -DNO_ATOMIC64 -Wno-c++11-narrowing "
compiler.convert_nvcc_flags = lambda x:x
compiler.is_cuda = 0
os.environ["use_cutt"] = "0"