Skip to content

Commit 4def1e9

Browse files
committed
ci fix
1 parent a51569d commit 4def1e9

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

tilelang/language/print.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
from tvm import tir
77
from typing import Any
88
from tilelang.language.kernel import get_thread_bindings
9-
from tilelang.language import copy, macro, serial, alloc_shared, index_to_coordinates
9+
from tilelang.language import copy, macro, serial, alloc_shared
10+
from tilelang.language.utils import index_to_coordinates
1011

1112

1213
@macro

tilelang/language/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from tilelang import tvm as tvm
2-
from tvm import PrimExpr
2+
from tvm.tir import PrimExpr
33

44

55
def index_to_coordinates(index, shape) -> list[PrimExpr]:

0 commit comments

Comments
 (0)