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

polish win space support

This commit is contained in:
Dun Liang 2022-08-13 13:46:01 +08:00
parent 93c00f884e
commit 268bdbea80
2 changed files with 3 additions and 3 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. # file 'LICENSE.txt', which is part of this source code package.
# *************************************************************** # ***************************************************************
__version__ = '1.3.5.4' __version__ = '1.3.5.5'
from jittor_utils import lock from jittor_utils import lock
with lock.lock_scope(): with lock.lock_scope():
ori_int = int ori_int = int

4
python/jittor/compiler.py
View File

@ -1111,7 +1111,7 @@ if os.name == 'nt':
cmd = cmd.replace(" -o ", " -Fe: ") cmd = cmd.replace(" -o ", " -Fe: ")
output = shsplit(cmd.split("-Fe:")[1].strip())[0] output = shsplit(cmd.split("-Fe:")[1].strip())[0]
base_output = os.path.basename(output).split('.')[0] base_output = os.path.basename(output).split('.')[0]
cmd += f" -DEF:\"{output}.def\" -IGNORE:4102 -IGNORE:4197 -IGNORE:4217 " cmd += f" -DEF:{output}.def -IGNORE:4102 -IGNORE:4197 -IGNORE:4217 "
elif " -c -o " in cmd: elif " -c -o " in cmd:
cmd = cmd.replace(" -c -o ", " -c -Fo: ") cmd = cmd.replace(" -c -o ", " -c -Fo: ")
@ -1323,7 +1323,7 @@ if use_data_gz:
.replace("-Werror", "") \ .replace("-Werror", "") \
.replace("-shared", "") .replace("-shared", "")
vdp = os.path.join(jittor_path, "src", "utils", "vdp") vdp = os.path.join(jittor_path, "src", "utils", "vdp")
run_cmd(fix_cl_flags(f"{cc_path} {dflags} -include \"{vdp}\" \"{data_s_path}\" -c -o \"{data_o_path}\"")) run_cmd(fix_cl_flags(f"\"{cc_path}\" {dflags} -include \"{vdp}\" \"{data_s_path}\" -c -o \"{data_o_path}\""))
os.remove(data_s_path) os.remove(data_s_path)
with open(data_gz_md5_path, 'w') as f: with open(data_gz_md5_path, 'w') as f:
f.write(md5) f.write(md5)