From 00014e20c3cc077727d467a67d3498260627e4e0 Mon Sep 17 00:00:00 2001
From: weiliangweiliang <617878975@qq.com>
Date: Fri, 10 Apr 2020 22:47:19 +0800
Subject: [PATCH] Update device_annotation.cc (#5291)

---
 src/relay/transforms/device_annotation.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/relay/transforms/device_annotation.cc b/src/relay/transforms/device_annotation.cc
index 908ba87..d5e1d2e 100644
--- a/src/relay/transforms/device_annotation.cc
+++ b/src/relay/transforms/device_annotation.cc
@@ -140,10 +140,10 @@ class RewriteAnnotation : public ExprMutator {
     }
   }
 
-  Expr VisitExp_(const TupleNode* op) {
+  Expr VisitExpr_(const TupleNode* op) {
     Array<Expr> fields;
     bool annotated = false;
-    for (const auto& field : fields) {
+    for (const auto& field : op->fields) {
       annotated |= NeedDeviceCopy(field.operator->(), op);
       fields.push_back(GetDeviceCopyExpr(field, op));
     }
-- 
2.7.4