From d7320f3bdaa35f3b6c6104d3949a06fcd5f50a46 Mon Sep 17 00:00:00 2001
From: dan <dan@nod-labs.com>
Date: Fri, 27 Aug 2021 18:17:32 +0000
Subject: [PATCH] fixed some python imports

Change required to enable
./tools/torchscript_e2e_test.sh --config=iree
---
 python/npcomp/compiler/numpy/backend/iree.py                | 4 ++--
 python/npcomp/compiler/numpy/backend/refjit.py              | 4 ++--
 python/npcomp/compiler/pytorch/backend/frontend_lowering.py | 4 ++--
 python/npcomp/compiler/pytorch/backend/iree.py              | 4 ++--
 4 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/python/npcomp/compiler/numpy/backend/iree.py b/python/npcomp/compiler/numpy/backend/iree.py
index 35efe388d7e4..467c65d64ac8 100644
--- 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
index 6e5a6c853a30..3f99488178ed 100644
--- 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
index 237a456d2f39..b9371d05716c 100644
--- 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
index ea711f023f92..b7a9abcc92d4 100644
--- 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