Update device_annotation.cc (#5291)

diff --git a/src/relay/transforms/device_annotation.cc b/src/relay/transforms/device_annotation.cc
index 908ba87..d5e1d2e 100644 (file)
--- 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));
     }