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

[stellaraccident]

Summary

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

[gysit]

Thanks for fixing!

mlir/python/mlir/dialects/linalg/opdsl/lang/affine.py

Referencing a named symbol:
   >>> S.a is S.b
   False
   >>> S.a is S.a
   True
 """
 
 from typing import Callable, Dict, Optional, Tuple, Union
 
-from mlir import ir as _ir
+from ..... import ir as _ir
 
 __all__ = [
     "AffineBuildState",
     "AffineExprDef",
     "D",
     "DimDef",
     "S",
     "SymbolDef",

mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py

 #  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 #  See https://llvm.org/LICENSE.txt for license information.
 #  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 """Model classes representing a tensor comprehension.
 
 These classes model the language more at an AST level as evaluated. Reasoning
 about it typically involves processing this form into config objects that
 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 *
 from .yaml_helper import *
 
 # Type aliases.
 AffineDimList = Dict[str, _ir.AffineExpr]
 

mlir/python/mlir/dialects/linalg/opdsl/lang/config.py

 #  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 #  See https://llvm.org/LICENSE.txt for license information.
 #  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 """Represents configured ops as emitted for code generation.
 
 Classes in this module generally are directly serializable to YAML for use
 by the code generator.
 
 TODO: These should just be dumb containers or serialization code but they
 currently encode too many details of how the language is interpreted. Move this
 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 *
 
 __all__ = ["LinalgStructuredOpConfig", "LinalgOpConfig", "OperandDefConfig"]
 
 
 def _serialize_affine_map(affine_map: _ir.AffineMap) -> str:
   with affine_map.context:

mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py

 #  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 #  See https://llvm.org/LICENSE.txt for license information.
 #  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
 from typing import Dict, List
 
 from contextlib import contextmanager
 import functools
 import inspect
 import threading
 
-from mlir import ir
+from ..... import ir
 from .comprehension import *
 from .config import *
 from .emitter import *
 
 _CONTEXT = threading.local()
 
 
 @contextmanager

mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py

 #  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 #  See https://llvm.org/LICENSE.txt for license information.
 #  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
 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
 fill_builtin_region = _cext.dialects.linalg.fill_builtin_region
 
 from .scalar_expr import *
 from .config import *
 import numpy as np