“c3158527717014f7875d61a4151ebdfaf3d30bd0”上不存在“paddle/phi/kernels/gpu/kron_kernel.cu”
提交 1360335f 编写于 作者: M malin10

test=develop, bug fix

上级 3fea70a3
...@@ -528,9 +528,13 @@ void GeoCommunicator::SendByCommunicator(int batches) { ...@@ -528,9 +528,13 @@ void GeoCommunicator::SendByCommunicator(int batches) {
tasks.reserve(send_varname_to_ctx_.size()); tasks.reserve(send_varname_to_ctx_.size());
for (auto &iter : send_varname_to_ctx_) { for (auto &iter : send_varname_to_ctx_) {
VLOG(1) << "debug " << iter.first;
auto &var_name = iter.first; auto &var_name = iter.first;
auto &send_ctx = iter.second; auto &send_ctx = iter.second;
int pserver_num = static_cast<int>(send_ctx.epmap.size()); int pserver_num = static_cast<int>(send_ctx.epmap.size());
if (var_name == STEP_COUNTER) {
continue;
}
auto &ids_queue = send_ids_to_queue_.at(var_name); auto &ids_queue = send_ids_to_queue_.at(var_name);
splited_ids_vec_[var_name].clear(); splited_ids_vec_[var_name].clear();
...@@ -549,7 +553,7 @@ void GeoCommunicator::SendByCommunicator(int batches) { ...@@ -549,7 +553,7 @@ void GeoCommunicator::SendByCommunicator(int batches) {
}; };
tasks.emplace_back( tasks.emplace_back(
send_threadpool_->enqueue(std::move(send_recv_task))); send_threadpool_->enqueue(std::move(send_recv_task)));
tasks[tasks.size() - 1].wait(); // tasks[tasks.size() - 1].wait();
} }
} else { } else {
auto send_recv_task = [this, &var_name, &send_ctx] { auto send_recv_task = [this, &var_name, &send_ctx] {
...@@ -567,9 +571,10 @@ void GeoCommunicator::SendByCommunicator(int batches) { ...@@ -567,9 +571,10 @@ void GeoCommunicator::SendByCommunicator(int batches) {
} }
} }
// for (auto &task : tasks) { for (auto &task : tasks) {
// task.wait(); task.wait();
// } }
VLOG(1) << "Finish SendByCommunicator";
} }
void GeoCommunicator::SendSparse(const std::string &varname, int ep_idx) { void GeoCommunicator::SendSparse(const std::string &varname, int ep_idx) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册