From 0d14b3cb16c9e0a0087d2c2dc1b82cb6023d93a2 Mon Sep 17 00:00:00 2001 From: "wenjian.ma" Date: Tue, 14 May 2024 10:36:12 +0800 Subject: [PATCH] [Relay] fixed to make TupleGetItem inherits the previous span --- 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 48941b2b23b9..05e0feb0c354 100644 --- 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