Skip to content

Commit 5ae0c66

Browse files
author
yi.wu
committed
fix build and merge develop
1 parent 6387a15 commit 5ae0c66

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

paddle/fluid/operators/test_send_nccl_id.cc

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ limitations under the License. */
2424
#include "paddle/fluid/operators/listen_and_serv_op.h"
2525
#include "paddle/fluid/operators/math/math_function.h"
2626
#include "paddle/fluid/operators/math/selected_rows_functor.h"
27+
#include "paddle/fluid/platform/nccl_helper.h"
2728
#include "paddle/fluid/string/printf.h"
2829

2930
USE_NO_KERNEL_OP(listen_and_serv);
@@ -36,7 +37,7 @@ namespace string = paddle::string;
3637

3738
std::unique_ptr<detail::AsyncGRPCServer> rpc_service;
3839

39-
void StartServer() {
40+
void StartServer(std::atomic<bool>* initialized) {
4041
f::Scope scope;
4142
p::CPUPlace place;
4243
scope.Var(NCCL_ID_VARNAME);
@@ -54,6 +55,7 @@ void StartServer() {
5455

5556
std::thread server_thread(
5657
std::bind(&detail::AsyncGRPCServer::RunSyncUpdate, rpc_service.get()));
58+
*initialized = true;
5759
rpc_service->SetCond(0);
5860
auto recv = rpc_service->Get();
5961
LOG(INFO) << "got nccl id and stop server...";
@@ -62,9 +64,13 @@ void StartServer() {
6264
}
6365

6466
TEST(SendNcclId, Normal) {
65-
std::thread server_thread(StartServer);
67+
std::atomic<bool> initialized{false};
68+
std::thread server_thread(StartServer, &initialized);
69+
while (!initialized) {
70+
}
6671
// wait server to start
67-
rpc_service.WaitServerReady();
72+
// sleep(2);
73+
rpc_service->WaitServerReady();
6874

6975
f::Scope scope;
7076
p::CPUPlace place;

0 commit comments

Comments
 (0)