Skip to content

GitLab

"library/vscode:/vscode.git/clone" did not exist on "c77ae65d40b1316dac02c4decf02d8517c840be2"
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 typing import Callable, Optional, Union
from typing import Callable, Optional, Union, List
import torch
from tvm import tir
......@@ -14,8 +14,8 @@ class MetalKernelAdapter(BaseKernelAdapter):
def __init__(
self,
params: list[KernelParam],
result_idx: list[int],
params: List[KernelParam],
result_idx: List[int],
# target: Union[str, Target],
func_or_mod: Union[tir.PrimFunc, tvm.IRModule],
# 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