Skip to content

GitLab

Unverified Commit 6b2bb310 authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

[Bugfix] Fix type object is not subscriptable in py38 (#959)

parent 9a7cda42
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tilelang/jit/adapter/torch/metal.py
View file @ 6b2bb310
from functools import wraps from functools import wraps
from typing import Callable, Optional, Union from typing import Callable, Optional, Union, List
import torch import torch
from tvm import tir from tvm import tir
...@@ -14,8 +14,8 @@ class MetalKernelAdapter(BaseKernelAdapter): ...@@ -14,8 +14,8 @@ class MetalKernelAdapter(BaseKernelAdapter):
def __init__( def __init__(
self, self,
params: list[KernelParam], params: List[KernelParam],
result_idx: list[int], result_idx: List[int],
# target: Union[str, Target], # target: Union[str, Target],
func_or_mod: Union[tir.PrimFunc, tvm.IRModule], func_or_mod: Union[tir.PrimFunc, tvm.IRModule],
# host_mod: Optional[tvm.IRModule] = None, # host_mod: Optional[tvm.IRModule] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment