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

polish corex_compiler.py

This commit is contained in:
lzhengning 2022-12-17 23:33:50 +08:00
parent d50fe5e754
commit 6b7d1a27d4
2 changed files with 5 additions and 1 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.6.3'
__version__ = '1.3.6.4'
from jittor_utils import lock
with lock.lock_scope():
ori_int = int

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

@ -51,6 +51,10 @@ string process_acl(const string& src, const string& name, const map<string,strin
// default where kernel cannot handle 64 warp size, use cub_where instead
new_src = token_replace_all(new_src, "if (cub_where$1) {", "if (cub_where) {");
}
if (name == "loop_var_analyze_pass.cc") {
new_src = token_replace_all(new_src, "DEFINE_FLAG($1, para_opt_level,$2,$3);",
"DEFINE_FLAG($1, para_opt_level, 4,$3);");
}
return new_src;
}
}''', compiler.cc_flags + " " + " ".join(cc_files) + cc_flags)