(tvm) branch main updated: [TIR] Update function signatures for decompose_reduction (#18505)

[mshr]

This is an automated email from the ASF dual-hosted git repository.

mshr pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new c3a52ea3f2 [TIR] Update function signatures for decompose_reduction 
(#18505)
c3a52ea3f2 is described below

commit c3a52ea3f253055a2f68f0317447817e5c8af0ca
Author: Guan-Ming (Wesley) Chiu <105915352+guan404m...@users.noreply.github.com>
AuthorDate: Wed Nov 26 21:54:12 2025 +0800

    [TIR] Update function signatures for decompose_reduction (#18505)
    
    ## Related Issue
    
    closes https://github.com/apache/tvm/issues/18215
---
 python/tvm/tir/schedule/schedule.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/tvm/tir/schedule/schedule.py 
b/python/tvm/tir/schedule/schedule.py
index 92d0822746..0d41ffe943 100644
--- a/python/tvm/tir/schedule/schedule.py
+++ b/python/tvm/tir/schedule/schedule.py
@@ -2411,7 +2411,7 @@ class Schedule(Object):
         .. code-block:: python
 
             @T.prim_func
-            def before_decompose(a: ty.handle, c: ty.handle) -> None:
+            def before_decompose(a: ty.handle, b: ty.handle, c: ty.handle) -> 
None:
                 A = tir.match_buffer(a, [128, 128])
                 B = tir.match_buffer(b, [128, 128])
                 C = tir.match_buffer(c, [128, 128])
@@ -2436,7 +2436,7 @@ class Schedule(Object):
         .. code-block:: python
 
             @T.prim_func
-            def after_decompose(a: ty.handle, c: ty.handle) -> None:
+            def after_decompose(a: ty.handle, b: ty.handle, c: ty.handle) -> 
None:
                 A = tir.match_buffer(a, [128, 128])
                 B = tir.match_buffer(b, [128, 128])
                 C = tir.match_buffer(c, [128, 128])