refine feed_op

e49b8b9c · chengduoZH · 43262608 · e49b8b9c
显示空白变更内容
内联并排

Showing with 5 addition and 1 deletion

paddle/operators/feed_op.cc paddle/operators/feed_op.cc +5 -1

未找到文件。
--- 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);
+    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());
  }
 };