(tvm) branch main updated: [Relay] fixed to make TupleGetItem inherits the previous span (#16996)

[wuwei]

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

wuwei 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 b7467aa27b [Relay] fixed to make TupleGetItem inherits the previous 
span (#16996)
b7467aa27b is described below

commit b7467aa27b289c147033b15c59cf660a25cd85f3
Author: mawnja <190936...@qq.com>
AuthorDate: Wed May 15 03:13:33 2024 +0800

    [Relay] fixed to make TupleGetItem inherits the previous span (#16996)
    
    Co-authored-by: wenjian.ma <wenjian...@denglin.ai>
---
 python/tvm/relay/expr_functor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/relay/expr_functor.py b/python/tvm/relay/expr_functor.py
index 48941b2b23..05e0feb0c3 100644
--- a/python/tvm/relay/expr_functor.py
+++ b/python/tvm/relay/expr_functor.py
@@ -251,7 +251,7 @@ class ExprMutator(ExprFunctor):
         new_tuple_value = self.visit(op.tuple_value)
         if new_tuple_value == op.tuple_value:
             return op
-        return TupleGetItem(new_tuple_value, op.index)
+        return TupleGetItem(new_tuple_value, op.index, span=op.span)
 
     def visit_global_var(self, gvar):
         return gvar