fixed some python imports

Change required to enable ./tools/torchscript_e2e_test.sh --config=iree
llvm · Aug 27, 2021 · d7320f3 · d7320f3
1 parent 1c53424
commit d7320f3
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/python/npcomp/compiler/numpy/backend/iree.py b/python/npcomp/compiler/numpy/backend/iree.py
@@ -5,8 +5,8 @@
 import io
 import subprocess
 
-from mlir.ir import *
-from mlir.passmanager import *
+from npcomp.ir import *
+from npcomp.passmanager import *
 from npcomp.compiler.generic.backend import iree as iree_backend
 from npcomp.compiler.utils import logging
 

diff --git a/python/npcomp/compiler/numpy/backend/refjit.py b/python/npcomp/compiler/numpy/backend/refjit.py
@@ -4,8 +4,8 @@
 
 import os
 
-from mlir.ir import *
-from mlir.passmanager import *
+from npcomp.ir import *
+from npcomp.passmanager import *
 from npcomp.compiler.generic.backend import refjit as refjit_backend
 from npcomp.compiler.utils import logging
 

diff --git a/python/npcomp/compiler/pytorch/backend/frontend_lowering.py b/python/npcomp/compiler/pytorch/backend/frontend_lowering.py
@@ -6,8 +6,8 @@
 
 import torch
 
-from mlir.ir import *
-from mlir.passmanager import *
+from npcomp.ir import *
+from npcomp.passmanager import *
 from npcomp.compiler.utils import logging
 
 __all__ = [

diff --git a/python/npcomp/compiler/pytorch/backend/iree.py b/python/npcomp/compiler/pytorch/backend/iree.py
@@ -7,8 +7,8 @@
 import torch
 import numpy as np
 
-from mlir.ir import *
-from mlir.passmanager import *
+from npcomp.ir import *
+from npcomp.passmanager import *
 from npcomp.compiler.utils import logging
 import iree.runtime as ireert
 import iree.compiler as ireec