Import Self from typing_extensions (backport to 3.10).

iree-org · Dec 10, 2024 · b2dd271 · b2dd271
1 parent cc0f068
commit b2dd271
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/iree/turbine/kernel/lang/wave_types.py b/iree/turbine/kernel/lang/wave_types.py
@@ -4,7 +4,6 @@
     ClassVar,
     Iterable,
     Optional,
-    Self,
     Type,
     TypeAlias,
     TypeVar,
@@ -17,6 +16,7 @@
 
 from sympy import Symbol
 from sympy.core.expr import Expr
+from typing_extensions import Self
 
 from itertools import chain
 

diff --git a/iree/turbine/kernel/ops/wave_ops.py b/iree/turbine/kernel/ops/wave_ops.py
@@ -9,12 +9,12 @@
     Any,
     Callable,
     Optional,
-    Self,
     Sequence,
     Type,
     TypeVar,
     final,
 )
+from typing_extensions import Self
 import torch.fx as fx
 
 from ..lang.wave_types import Memory, Register, IndexMapping

diff --git a/requirements.txt b/requirements.txt
@@ -9,6 +9,7 @@ lit==18.1.7
 mypy==1.8.0
 ml_dtypes==0.5.0
 setuptools
+typing_extensions
 wheel
 
 # It is expected that you have installed a PyTorch version/variant specific

diff --git a/setup.py b/setup.py
@@ -110,6 +110,7 @@ def initialize_options(self):
         "torch>=2.3.0",
         f"Jinja2{get_version_spec('Jinja2')}",
         f"ml_dtypes{get_version_spec('ml_dtypes')}",
+        f"typing_extensions{get_version_spec('typing_extensions')}",
     ],
     extras_require={
         "testing": [