There was an error while loading. Please reload this page.
2 parents 4326260 + e49b8b9 commit 18acf1eCopy full SHA for 18acf1e
paddle/operators/feed_op.cc
@@ -52,7 +52,11 @@ class FeedOp : public framework::OperatorBase {
52
platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance();
53
auto &dev_ctx = *pool.Get(place);
54
55
- framework::Copy(feed_item, place, dev_ctx, out_item);
+ 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
+ }
60
out_item->set_lod(feed_item.lod());
61
}
62
};
0 commit comments