add sync in load selectedrows

7c12c0f8 · tangwei12 · 9f11db40 · 7c12c0f8
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/load_op.cc paddle/fluid/operators/load_op.cc +1 -0

未找到文件。
--- a/paddle/fluid/operators/load_op.cc
+++ b/paddle/fluid/operators/load_op.cc
@@ -92,6 +92,7 @@ class LoadOp : public framework::OperatorBase {
    platform::DeviceContextPool &pool = platform::DeviceContextPool::Instance();
    auto &dev_ctx = *pool.Get(place);
    framework::DeserializeFromStream(fin, selectedRows, dev_ctx);
+    selectedRows->SyncIndex();
  }
 };