Skip to content

Commit 18acf1e

Browse files
author
chengduo
authored
Merge pull request #8011 from chengduoZH/feature/refine_feed_op
Refine feed_op
2 parents 4326260 + e49b8b9 commit 18acf1e

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

paddle/operators/feed_op.cc

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,11 @@ class FeedOp : public framework::OperatorBase {
5252
platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance();
5353
auto &dev_ctx = *pool.Get(place);
5454

55-
framework::Copy(feed_item, place, dev_ctx, out_item);
55+
if (platform::is_same_place(feed_item.place(), place)) {
56+
out_item->ShareDataWith(feed_item);
57+
} else {
58+
framework::Copy(feed_item, place, dev_ctx, out_item);
59+
}
5660
out_item->set_lod(feed_item.lod());
5761
}
5862
};

0 commit comments

Comments
 (0)