Skip to content

Commit dce4cfd

Browse files
committed
lint fix
1 parent a78e640 commit dce4cfd

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed
Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,23 +1,23 @@
11
import tilelang.testing
22
from tilelang import tvm as tvm
3-
from tvm import IRModule
43
from tilelang import language as T
5-
from tilelang.utils.tensor import map_torch_type
4+
65

76
def test_let_vectorize_load():
7+
88
@T.prim_func
99
def main(A_ptr: T.handle):
1010
A = T.match_buffer(A_ptr, (16, 16), dtype="float32", align=16)
1111

12-
for blockIdx in T.thread_binding(1, thread="blockIdx.x"):
13-
for threadIdx in T.thread_binding(128, thread="threadIdx.x"):
12+
for _blockIdx in T.thread_binding(1, thread="blockIdx.x"):
13+
for _threadIdx in T.thread_binding(128, thread="threadIdx.x"):
1414
b: T.float32x4 = A[0, 0:4]
1515
A[0, 4:8] = b
1616

1717
mod = tvm.IRModule({"main": main})
1818
mod = tvm.compile(mod, target="cuda")
1919
assert "float4 b" in mod.mod.imported_modules[0].get_source()
20-
20+
2121

2222
if __name__ == "__main__":
2323
tilelang.testing.main()

0 commit comments

Comments
 (0)