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

pytorch make jittor segfault, wtf

This commit is contained in:
Dun Liang 2021-06-23 20:59:03 +08:00
parent 69550e3efd
commit 1d0df10f13
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
python/jittor/compile_extern.py
View File

@ -64,13 +64,12 @@ def setup_mkl():
# yield error "free: invalide size" or # yield error "free: invalide size" or
# "mmap error" # "mmap error"
# import pytorch(>1.8) first can fix this problem # import pytorch(>1.8) first can fix this problem
# try:
try: # # jt.dirty_fix_pytorch_runtime_error()
# jt.dirty_fix_pytorch_runtime_error() # import torch
import torch # from torch import nn
from torch import nn # except:
except: # torch = None
torch = None
mkl_include_path = os.environ.get("mkl_include_path") mkl_include_path = os.environ.get("mkl_include_path")
mkl_lib_path = os.environ.get("mkl_lib_path") mkl_lib_path = os.environ.get("mkl_lib_path")