From f43ce0a77f0a484d79ffd8f7e1e3a367d706c9da Mon Sep 17 00:00:00 2001
From: HongHongHongL <11612425@mail.sustech.edu.cn>
Date: Mon, 9 Oct 2023 20:05:32 +0800
Subject: [PATCH] fix the error example in pad_einsum documentation

---
 python/tvm/tir/schedule/schedule.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py
index 85d1f19bba00..638e1751b1d4 100644
--- a/python/tvm/tir/schedule/schedule.py
+++ b/python/tvm/tir/schedule/schedule.py
@@ -3593,7 +3593,7 @@ def before_pad_einsum(
 
             sch = tir.Schedule(before_pad_einsum, debug_mask="all")
             block = sch.get_block("C_shared")
-            sch.pad_einsum(block, [0, 1, 1])
+            sch.pad_einsum(block, [32, 32, 32])
             print(sch.mod["main"].script())
 
         After applying decompose-padding, the IR becomes: