diff --git a/src/relay/transforms/device_annotation.cc b/src/relay/transforms/device_annotation.cc index b4d61f110832..386a844797d1 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 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)); }