Skip to content

T
tic
Unverified Commit 00014e20 by weiliangweiliang Committed by GitHub
Options

Update device_annotation.cc (#5291)

parent 5da361d3
Showing with 2 additions and 2 deletions
src/relay/transforms/device_annotation.cc
...@@ -140,10 +140,10 @@ class RewriteAnnotation : public ExprMutator { ...@@ -140,10 +140,10 @@ class RewriteAnnotation : public ExprMutator {
} }
} }
Expr VisitExp_(const TupleNode* op) { Expr VisitExpr_(const TupleNode* op) {
Array<Expr> fields; Array<Expr> fields;
bool annotated = false; bool annotated = false;
for (const auto& field : fields) { for (const auto& field : op->fields) {
annotated |= NeedDeviceCopy(field.operator->(), op); annotated |= NeedDeviceCopy(field.operator->(), op);
fields.push_back(GetDeviceCopyExpr(field, op)); fields.push_back(GetDeviceCopyExpr(field, op));
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment