Update device_annotation.cc (apache#5291)

trevor-m · Apr 16, 2020 · 70873dc · 70873dc
1 parent fbffeb8
commit 70873dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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));
     }