未验证 提交 3493c46e 编写于 作者: Y yaoxuefeng 提交者: GitHub

[HETERPS]fix make error when pscore equals on

上级 0cfedb73
...@@ -235,13 +235,15 @@ void PSGPUWrapper::BuildTask(std::shared_ptr<HeterContext> gpu_task) { ...@@ -235,13 +235,15 @@ void PSGPUWrapper::BuildTask(std::shared_ptr<HeterContext> gpu_task) {
timeline.Start(); timeline.Start();
std::vector<std::vector<std::pair<uint64_t, char*>>> pass_values; std::vector<std::vector<std::pair<uint64_t, char*>>> pass_values;
uint16_t pass_id = 0;
bool record_status = false; bool record_status = false;
#ifdef PADDLE_WITH_PSLIB
uint16_t pass_id = 0;
if (multi_node_) { if (multi_node_) {
record_status = fleet_ptr->pslib_ptr_->_worker_ptr->take_sparse_record( record_status = fleet_ptr->pslib_ptr_->_worker_ptr->take_sparse_record(
table_id_, pass_id, pass_values); table_id_, pass_id, pass_values);
} }
#endif
auto build_func = [device_num, record_status, &pass_values, &local_keys, auto build_func = [device_num, record_status, &pass_values, &local_keys,
&local_ptr, &device_keys, &device_vals, &local_ptr, &device_keys, &device_vals,
&device_mutex](int i) { &device_mutex](int i) {
...@@ -260,6 +262,7 @@ void PSGPUWrapper::BuildTask(std::shared_ptr<HeterContext> gpu_task) { ...@@ -260,6 +262,7 @@ void PSGPUWrapper::BuildTask(std::shared_ptr<HeterContext> gpu_task) {
task_keys[shard].push_back(local_keys[i][j]); task_keys[shard].push_back(local_keys[i][j]);
task_ptrs[shard].push_back(local_ptr[i][j]); task_ptrs[shard].push_back(local_ptr[i][j]);
} }
#ifdef PADDLE_WITH_PSLIB
if (record_status) { if (record_status) {
size_t local_keys_size = local_keys.size(); size_t local_keys_size = local_keys.size();
size_t pass_values_size = pass_values.size(); size_t pass_values_size = pass_values.size();
...@@ -275,6 +278,7 @@ void PSGPUWrapper::BuildTask(std::shared_ptr<HeterContext> gpu_task) { ...@@ -275,6 +278,7 @@ void PSGPUWrapper::BuildTask(std::shared_ptr<HeterContext> gpu_task) {
} }
} }
} }
#endif
for (int dev = 0; dev < device_num; dev++) { for (int dev = 0; dev < device_num; dev++) {
device_mutex[dev]->lock(); device_mutex[dev]->lock();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册