refine feed_op

7 years ago · e49b8b9c55
parent 4326260844
commit e49b8b9c55
1 changed files with 5 additions and 1 deletions
--- a/paddle/operators/feed_op.cc
+++ b/paddle/operators/feed_op.cc
@ -52,7 +52,11 @@ class FeedOp : public framework::OperatorBase {
    platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance();
    auto &dev_ctx = *pool.Get(place);

-    framework::Copy(feed_item, place, dev_ctx, out_item);
+    if (platform::is_same_place(feed_item.place(), place)) {
+      out_item->ShareDataWith(feed_item);
+    } else {
+      framework::Copy(feed_item, place, dev_ctx, out_item);
+    }
    out_item->set_lod(feed_item.lod());
  }
 };