From a99b208cb3c01ef5b71c1175e52ff01fad3188e6 Mon Sep 17 00:00:00 2001 From: Luke Hutton Date: Fri, 23 Jun 2023 16:24:52 +0000 Subject: [PATCH] [TIR] Fix typo in code example Change-Id: If241561fc85c6737ff726a1e7150a6a496d88b30 --- python/tvm/te/operation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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():