[Relay] fixed to make TupleGetItem inherits the previous span (#16996)

Co-authored-by: wenjian.ma <wenjian.ma@denglin.ai>
apache · May 14, 2024 · b7467aa · b7467aa
1 parent d9dbbc9
commit b7467aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/expr_functor.py b/python/tvm/relay/expr_functor.py
@@ -251,7 +251,7 @@ def visit_tuple_getitem(self, op):
         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