pytorch make jittor segfault, wtf

2021-06-23 20:59:03 +08:00 · 2021-06-23 20:59:03 +08:00 · 1d0df10f13
parent 69550e3efd
commit 1d0df10f13
1 changed files with 6 additions and 7 deletions
--- a/python/jittor/compile_extern.py
+++ b/python/jittor/compile_extern.py
@ -64,13 +64,12 @@ def setup_mkl():
    # yield error "free: invalide size" or
    # "mmap error"
    # import pytorch(>1.8) first can fix this problem
-
-    try:
-        # jt.dirty_fix_pytorch_runtime_error()
-        import torch
-        from torch import nn
-    except:
-        torch = None
+    # try:
+    #     # jt.dirty_fix_pytorch_runtime_error()
+    #     import torch
+    #     from torch import nn
+    # except:
+    #     torch = None

    mkl_include_path = os.environ.get("mkl_include_path")
    mkl_lib_path = os.environ.get("mkl_lib_path")