[TIR] Fix the error example in the documentation for pad_einsum (#15906)

fix the error example in pad_einsum documentation
apache · Oct 11, 2023 · 4e9866f · 4e9866f
1 parent eb2a4bc
commit 4e9866f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/tir/schedule/schedule.py b/python/tvm/tir/schedule/schedule.py
@@ -3592,7 +3592,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: