From 7c12c0f865ba11186234ee4e0b922050978be4f4 Mon Sep 17 00:00:00 2001 From: tangwei12 Date: Fri, 17 Aug 2018 14:45:56 +0800 Subject: [PATCH] add sync in load selectedrows --- paddle/fluid/operators/load_op.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/operators/load_op.cc b/paddle/fluid/operators/load_op.cc index 27e26cb1b5c..51219504ffa 100644 --- 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(); } }; -- GitLab