[Minor] Remove from __future__ import annotations for python 3.8 (#1273)

tilelang/language/annotations.py

 """Annotation helpers exposed on the TileLang language surface."""
-from __future__ import annotations
-
 from typing import Callable
 
 from tilelang.layout import Layout

tilelang/language/copy.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from typing import Literal
 from tilelang import language as T
 from tilelang.utils.language import (

tilelang/language/customize.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 import tilelang.language as T
 from tvm.tir import PrimExpr, Buffer, op
 from .atomic import atomic_max, atomic_min, atomic_add, atomic_addx2, atomic_addx4, atomic_load, atomic_store  # noqa: F401

tilelang/language/experimental/gemm_sp.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from tilelang.primitives.gemm.base import GemmWarpPolicy
 import tilelang.language as T
 from tvm import tir

tilelang/language/fill.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from tvm import tir
 from tilelang.language import has_let_value, get_let_value
 from tilelang.utils.language import get_buffer_region_from_load

tilelang/language/frame.py

 """Override the LetFrame to print a message when entering the frame."""
 from __future__ import annotations
-
 from tvm.ffi import register_object as _register_object
 from tvm.tir import Var, PrimExpr, BufferLoad, BufferRegion
 from tvm.ir import Range

tilelang/language/gemm.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from tilelang.primitives.gemm.base import GemmWarpPolicy
 import tilelang.language as T
 from tvm import tir

tilelang/language/kernel.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from collections import deque
 from tvm import tir
 from tvm.tir import Var

tilelang/language/loop.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from typing import Any
 from tvm import tir
 from tvm.tir import IntImm

tilelang/language/overrides/parser.py

 """TVMScript parser overrides tailored for TileLang."""
-from __future__ import annotations
-
 from functools import partial
 
 from tvm.script.ir_builder import tir as T

tilelang/language/parser/operation.py

@@ -17,8 +17,6 @@
 # This file is modified from the original version,
 # which is part of the TVM project (https://tvm.apache.org/).
 """The tir expression operation registration"""
-from __future__ import annotations
-
 from tvm import tir
 from tvm.ffi.runtime_ctypes import DataType, DataTypeCode
 from tvm.tir import IntImm

tilelang/language/proxy.py

 """The language interface for tl programs."""
-
 from __future__ import annotations
+
 from typing import Any, SupportsIndex, TYPE_CHECKING
 from collections.abc import Sequence
 from typing_extensions import Self

tilelang/language/reduce.py

 """The language interface for tl programs."""
 from __future__ import annotations
-
 from tvm import tir
 from tilelang.language import copy, macro, alloc_shared, alloc_fragment
 from tilelang.language.utils import buffer_to_tile_region

tilelang/language/tir/ir.py

-from __future__ import annotations
 import tvm.script.ir_builder.tir.ir as _ir
 from tvm.script.ir_builder.tir import frame
 from tvm.tir import PrimExpr

tilelang/language/utils.py

-from __future__ import annotations
 from tilelang import tvm as tvm
 from tvm import tir
 from tvm.tir import PrimExpr, Buffer, BufferLoad, op

tilelang/language/v2/builder.py

 from __future__ import annotations
-
 from contextlib import contextmanager, AbstractContextManager
 from dataclasses import dataclass
 import inspect

tilelang/language/warpgroup.py

 """The language interface for tl programs."""
-from __future__ import annotations
-
 from tvm.script.ir_builder.tir.frame import TIRFrame
 from tvm.ffi import register_object
 from tilelang import _ffi_api

tilelang/layout/fragment.py

 """Wrapping Layouts."""
 # pylint: disable=invalid-name, unsupported-binary-operation
-from __future__ import annotations
-
 import tvm
 import tvm_ffi
 from tvm.ir import Range
@@ -124,7 +122,7 @@ class Fragment(Layout):
     def repeat(self,
                repeats,
                repeat_on_thread: bool = False,
-               lower_dim_first: bool = True) -> Fragment:
+               lower_dim_first: bool = True) -> 'Fragment':
         """
         Returns a new Fragment that repeats the iteration space a given number of times.
 
@@ -144,7 +142,7 @@ class Fragment(Layout):
         """
         return _ffi_api.Fragment_repeat(self, repeats, repeat_on_thread, lower_dim_first)
 
-    def replicate(self, replicate: int) -> Fragment:
+    def replicate(self, replicate: int) -> 'Fragment':
         """
         Replicate the Fragment across a new thread dimension.
 
@@ -160,7 +158,7 @@ class Fragment(Layout):
         """
         return _ffi_api.Fragment_replicate(self, replicate)
 
-    def condense_rep_var(self) -> Fragment:
+    def condense_rep_var(self) -> 'Fragment':
         """
         Condense or fold the replicate variable into the existing iteration space.
         This operation may be used to reduce dimensionality if the replicate variable
@@ -207,7 +205,7 @@ class Fragment(Layout):
         """
         return f"Fragment<{self.get_input_shape()}->{self.get_output_shape()}, thread={self.thread}, index={self.index}>"
 
-    def is_equal(self, other: Fragment) -> bool:
+    def is_equal(self, other: 'Fragment') -> bool:
         """
         Check if the current fragment is equal to another fragment.
         """

tilelang/layout/gemm_sp.py

 """Wrapping Layouts."""
 # pylint: disable=invalid-name, unsupported-binary-operation
 from __future__ import annotations
-
 import tvm
 import tilelang.language as T
 import warnings

tilelang/layout/layout.py

 """Wrapping Layouts."""
 # pylint: disable=invalid-name, unsupported-binary-operation
-from __future__ import annotations
-
 import tvm_ffi
 from tvm.ir import Node, Range
 from tvm.tir import IterVar, Var, PrimExpr, IndexMap
@@ -122,7 +120,7 @@ class Layout(Node):
         # Map the provided indices using the constructed index mapping
         return index_map.map_indices(indices)
 
-    def inverse(self) -> Layout:
+    def inverse(self) -> 'Layout':
         """
         Compute the inverse of the current layout transformation.
 
@@ -133,7 +131,7 @@ class Layout(Node):
         """
         return _ffi_api.Layout_inverse(self)
 
-    def is_equal(self, other: Layout) -> bool:
+    def is_equal(self, other: 'Layout') -> bool:
         """
         Check if the current layout is equal to another layout.