mixed_precision_init
Qiao Longfei 6 years ago
parent 103c9bb376
commit c60f312d1b

@ -57,6 +57,8 @@ class ConcurrentSet {
std::future<void> GetAndClear(std::vector<int64_t>* result) {
auto task = [this, &result] {
result->clear();
// FIXME(qiao): use a trick to avoid the bug of recv an selected rows
result->push_back(0);
for (auto& id : set_) {
result->push_back(id);
}

@ -108,7 +108,9 @@ void ParameterRecv<T>::operator()(const RpcContext &rpc_ctx,
VLOG(3) << "recv_slr size: " << recv_slr.rows().size() << " "
<< sstream.str();
}
for (auto i = 0; i < recv_slr.rows().size(); ++i) {
// FIXME(qiao): use a trick to avoid the bug of recv an selected rows
for (auto i = 1; i < recv_slr.rows().size(); ++i) {
auto row_id = recv_slr.rows()[i] + row_offset;
PADDLE_ENFORCE_LT(row_id, recv_dims[1]);
memcpy(recv_tensor->data<T>() + row_id * width,

Loading…
Cancel
Save