[Fix] Remove unsupported type params (#1186)

* [Fix] Remove type params

* fix lint error

* [Fix] fix dtype new error

.github/workflows/ci.yml

@@ -56,10 +56,10 @@ jobs:
         run: |
           "${{ steps.setup-pylowest.outputs.python-path }}" -m compileall -q -f tilelang
 
-      - name: Setup Python 3.12
+      - name: Setup Python 3.9
         uses: actions/setup-python@v6
         with:
-          python-version: "3.12"
+          python-version: "3.9"
           update-environment: true
           cache: pip
           cache-dependency-path: |

tilelang/language/v2/builder.py

@@ -536,7 +536,8 @@ def get_type_hints(func):
     if annot is None:
         raise TypeError(f'Failed to get function type hints, {func} is not a function')
     hints = {}
-    type_params = getattr(func, "__type_params__", ())
+    # type params are not used currently, it is support since python 3.12.4
+    # type_params = getattr(func, "__type_params__", ())
     globalns = getattr(func, '__globals__', {})
     localns = globalns
     for name, value in annot.items():
@@ -559,7 +560,8 @@ def get_type_hints(func):
                 except Exception:
                     pass
             value = ForwardRef(value, is_argument=True, is_class=False)
-        hints[name] = _eval_type(value, globalns=globalns, localns=localns, type_params=type_params)
+        hints[name] = _eval_type(
+            value, globalns=globalns, localns=localns)  #, type_params=type_params)
     return hints
 
 

tilelang/language/v2/dtypes.py

 from tilelang import tvm
 from tvm import ir
-import tvm_ffi
 import torch
 import ctypes
 from typing import TYPE_CHECKING
@@ -100,16 +99,17 @@ def __dtype_call__(self: dtype, expr=None, is_size_var: bool = False) -> tir.Var
     return call(expr, is_size_var)
 
 
+__orig_dtype_new = dtype.__new__
+
+
 def __dtype_new__(cls, value: AnyDType) -> dtype:
     if isinstance(value, str):
-        val = str.__new__(cls, value)
+        return __orig_dtype_new(cls, value)
     elif value in _dtype_py2tvmstr:
-        val = str.__new__(cls, _dtype_py2tvmstr[value])
+        return __orig_dtype_new(cls, _dtype_py2tvmstr[value])
     else:
         expected = set(list(_dtype_py2tvmstr.keys()) + list(_dtype_tvmstr2fficall.values()))
         raise TypeError(f"Invalid DataType {value}({type(value)}), expect one of {expected}")
-    val.__tvm_ffi_dtype__ = tvm_ffi.core.DataType(val)
-    return val
 
 
 dtype.__eq__ = __dtype_eq__