add sync in load selectedrows

7 years ago · 7c12c0f865
parent 9f11db4080
commit 7c12c0f865
1 changed files with 1 additions and 0 deletions
--- 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();
  }
 };