[Py38] Revert typing and parser updates for Python 3.8 compatibility (#850)

* Update submodule TVM to commit 872e32c1 and adjust type hints in nvcc.py and utils.py for compatibility with Python typing standards.

* Update requirements.txt to specify ml_dtypes without a version constraint, indicating that versions greater than 0.5.1 are needed for fp4 support.

tvm @ 872e32c1

-Subproject commit 6051f6dbdd741be340f47f944cd433f04ed18a8d
+Subproject commit 872e32c16d5bd0826b60f73f55af9e694d86a5a1

requirements.txt

@@ -4,6 +4,8 @@ numpy>=1.23.5
 tqdm>=4.62.3
 typing_extensions>=4.10.0
 cloudpickle
-ml_dtypes>=0.5.3
+# mldtypes should be greater than 0.5.1
+# if you want to enable fp4
+ml_dtypes
 psutil
 torch

tilelang/contrib/nvcc.py

@@ -6,6 +6,7 @@ from __future__ import absolute_import as _abs
 import os
 import subprocess
 import warnings
+from typing import Tuple
 from tilelang.env import CUDA_HOME
 
 import tvm.ffi
@@ -298,7 +299,7 @@ def get_target_compute_version(target=None):
                      "Try specifying it by adding '-arch=sm_xx' to your target.")
 
 
-def parse_compute_version(compute_version) -> tuple[int, int]:
+def parse_compute_version(compute_version) -> Tuple[int, int]:
     """Parse compute capability string to divide major and minor version
 
     Parameters

tilelang/language/utils.py

 from tilelang import tvm as tvm
+from typing import List
 from tvm.tir import PrimExpr
 
 
-def index_to_coordinates(index, shape) -> list[PrimExpr]:
+def index_to_coordinates(index, shape) -> List[PrimExpr]:
     """
     Convert a flat (linear) index into multi-dimensional coordinates for a given shape.
 
@@ -13,7 +14,7 @@ def index_to_coordinates(index, shape) -> list[PrimExpr]:
         shape (Sequence[int]): The extents of each dimension (length >= 1).
 
     Returns:
-        list[PrimExpr]: Coordinates for each dimension in the same order as `shape`.
+        List[PrimExpr]: Coordinates for each dimension in the same order as `shape`.
     """
     coordinates = []
     dims = len(shape)