Skip to content

Commit f655eca

Browse files
committed
fix typo;test=develop
1 parent cb052da commit f655eca

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

paddle/fluid/framework/fleet/heter_ps/hashtable.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@ limitations under the License. */
2323
#ifdef PADDLE_WITH_PSCORE
2424
#include "paddle/fluid/distributed/table/depends/large_scale_kv.h"
2525
#endif
26-
#include "thrust/pair.h"
2726
#include "paddle/fluid/framework/rw_lock.h"
28-
//#include "cudf/concurrent_unordered_map.cuh.h"
27+
#include "thrust/pair.h"
28+
// #include "cudf/concurrent_unordered_map.cuh.h"
2929
#include "paddle/fluid/framework/fleet/heter_ps/cudf/concurrent_unordered_map.cuh.h"
3030
#ifdef PADDLE_WITH_HETERPS
3131
#include "paddle/fluid/platform/type_defs.h"
@@ -64,13 +64,13 @@ class HashTable {
6464

6565
int size() { return container_->size(); }
6666

67-
std::unique_ptr<RWLock> rwlock_{nullptr};
67+
std::unique_ptr<RWLock> rwlock_{nullptr};
68+
6869
private:
6970
TableContainer<KeyType, ValType>* container_;
7071
int BLOCK_SIZE_{256};
7172
float LOAD_FACTOR{0.75f};
7273
size_t capacity_;
73-
7474
};
7575
} // end namespace framework
7676
} // end namespace paddle

paddle/fluid/framework/fleet/heter_ps/heter_comm_inl.h

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -528,7 +528,8 @@ void HeterComm<KeyType, ValType, GradType>::pull_sparse(int num,
528528
tables_[i]->rwlock_->RDLock();
529529
tables_[i]->get(reinterpret_cast<KeyType*>(node.key_storage),
530530
reinterpret_cast<ValType*>(node.val_storage),
531-
h_right[i] - h_left[i] + 1, resource_->remote_stream(i, num));
531+
h_right[i] - h_left[i] + 1,
532+
resource_->remote_stream(i, num));
532533
}
533534
for (int i = 0; i < total_gpu; ++i) {
534535
cudaStreamSynchronize(resource_->remote_stream(i, num));

0 commit comments

Comments
 (0)