[tvm] branch main updated: [TIR] Correct type annotation for `rfactor` (#13485)

[ruihangl]

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

ruihangl 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 9098b497bb [TIR] Correct type annotation for `rfactor` (#13485)
9098b497bb is described below

commit 9098b497bbbe20fb6c3b3ea6e062d2372cb0c8c9
Author: Junru Shao <junrushao1...@gmail.com>
AuthorDate: Fri Nov 25 14:49:09 2022 -0800

    [TIR] Correct type annotation for `rfactor` (#13485)
---
 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 170179d0d4..69feaff53a 100644
--- a/python/tvm/tir/schedule/schedule.py
+++ b/python/tvm/tir/schedule/schedule.py
@@ -1875,7 +1875,7 @@ class Schedule(Object):
         return _ffi_api.ScheduleDecomposeReduction(self, block, loop)  # type: 
ignore # pylint: disable=no-member
 
     @type_checked
-    def rfactor(self, loop: LoopRV, factor_axis: int) -> LoopRV:
+    def rfactor(self, loop: LoopRV, factor_axis: int) -> BlockRV:
         """Factorize an associative reduction block by the specified loop.
 
         An associative reduction cannot be parallelized directly,