[mlir][python] Make a number of imports relative.

Avoiding absolute imports allows the code to be relocatable (which is used for out of tree integrations).

Differential Revision: https://reviews.llvm.org/D107617

diff --git a/mlir/python/mlir/dialects/linalg/opdsl/lang/affine.py b/mlir/python/mlir/dialects/linalg/opdsl/lang/affine.py
index 6db3bcf..9c1bb33 100644 (file)
--- a/mlir/python/mlir/dialects/linalg/opdsl/lang/affine.py
+++ b/mlir/python/mlir/dialects/linalg/opdsl/lang/affine.py
@@ -53,7 +53,7 @@ Referencing a named symbol:
 
 from typing import Callable, Dict, Optional, Tuple, Union
 
-from mlir import ir as _ir
+from ..... import ir as _ir
 
 __all__ = [
     "AffineBuildState",

diff --git a/mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py b/mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py
index 66d7510..f7bfa81 100644 (file)
--- a/mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py
+++ b/mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py
@@ -11,8 +11,7 @@ represent actual op definitions (i.e. YAML).
 from typing import Any, Dict, List, Optional, Sequence, Set, Tuple
 from enum import Enum
 
-from mlir import ir as _ir
-
+from ..... import ir as _ir
 from .affine import *
 from .scalar_expr import *
 from .types import *

diff --git a/mlir/python/mlir/dialects/linalg/opdsl/lang/config.py b/mlir/python/mlir/dialects/linalg/opdsl/lang/config.py
index f6d5248..fec41de 100644 (file)
--- a/mlir/python/mlir/dialects/linalg/opdsl/lang/config.py
+++ b/mlir/python/mlir/dialects/linalg/opdsl/lang/config.py
@@ -13,8 +13,7 @@ to helpers on the comprehension objects themselves.
 
 from typing import Dict, Optional
 
-from mlir import ir as _ir
-
+from ..... import ir as _ir
 from .comprehension import *
 from .yaml_helper import *
 

diff --git a/mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py b/mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py
index 1b42b57..047bde2 100644 (file)
--- a/mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py
+++ b/mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py
@@ -9,7 +9,7 @@ import functools
 import inspect
 import threading
 
-from mlir import ir
+from ..... import ir
 from .comprehension import *
 from .config import *
 from .emitter import *

diff --git a/mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py b/mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py
index 4568298..ea2da71 100644 (file)
--- a/mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py
+++ b/mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py
@@ -4,10 +4,10 @@
 
 from typing import Dict, Sequence
 
-from mlir.ir import *
-from mlir.dialects import linalg
-from mlir.dialects import std
-from mlir.dialects import math
+from .....ir import *
+from .... import linalg
+from .... import std
+from .... import math
 # TODO: resolve name collision for Linalg functionality that is injected inside
 # the _mlir.dialects.linalg directly via pybind.
 from ....._cext_loader import _cext