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

polish

This commit is contained in:
lzhengning 2021-06-11 20:34:07 +08:00
parent ebb07a1efa
commit 8bf235bb85
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/jittor/utils/polish.py
View File

@ -63,7 +63,7 @@ os_name_system_dict = {
for os_name, os_type in os_name_system_dict.items():
if platform.system() != os_type:
continue
os_arch = platform.machine() if os_type == 'macos' else ''
os_arch = platform.machine() if os_type == 'Darwin' else ''
for cc_type in ["g++"]:
for device in ["cpu"]:
@ -76,7 +76,7 @@ for os_name, os_type in os_name_system_dict.items():
if platform.machine() == "x86_64":
env += " cc_flags='-march=core2' "
if device == "cpu":
env += "nvcc_path='' "
env += " nvcc_path='' "
elif jt.flags.nvcc_path == "":
env = "unset nvcc_path && " + env
cmd = f"{env} {sys.executable} -c 'import jittor'"