From 2103719a8c704e787285cc4566efcc242fa2866d Mon Sep 17 00:00:00 2001 From: Anirudh Sundar Date: Sat, 23 Mar 2024 15:40:05 +0530 Subject: [PATCH] [Doc] Fix set_axis_separator example Minor fix to update the `set_axis_separator` example to match the definition --- 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 b871c91987df..c2a538b39b25 100644 --- a/python/tvm/tir/schedule/schedule.py +++ b/python/tvm/tir/schedule/schedule.py @@ -3542,7 +3542,7 @@ def before_set_axis_separator( .. code-block:: python sch = tir.Schedule(before_set_axis_separator) - sch.set_axis_separators(sch.get_block("B"), buffer_index=0, buffer_index_type="write", + sch.set_axis_separators(sch.get_block("B"), buffer=("write", 0), axis_separators=[1]) print(sch.mod["main"].script())