Update device_annotation.cc (#5291)

00014e20 · weiliangweiliang · GitHub · 5da361d3 · 00014e20
Unverified Commit 00014e20 authored Apr 10, 2020 by weiliangweiliang Committed by GitHub Apr 10, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/relay/transforms/device_annotation.cc
+2 -2

No files found.
--- 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));
    }