diff --git a/python/tvm/te/operation.py b/python/tvm/te/operation.py index 1a28f9bb3d71..b6da01b55800 100644 --- a/python/tvm/te/operation.py +++ b/python/tvm/te/operation.py @@ -606,7 +606,7 @@ def tir_matmul(a: T.handle, b: T.handle, c: T.handle) -> None: B = T.match_buffer(b, (128, 128)) C = T.match_buffer(c, (128, 128)) - for i, j, k in T.grip(128, 128, 128): + for i, j, k in T.grid(128, 128, 128): with T.block(): vi, vj, vk = T.axis.remap("SSR", [i, j, k]) with T.init():