close socket connect (#17862)

test=develop

close socket connect (#17862)
test=develop
5436d666 · chengduo · GitHub · 3556e8ee · 5436d666
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/imperative/nccl_context.cc paddle/fluid/imperative/nccl_context.cc +1 -0

未找到文件。
--- a/paddle/fluid/imperative/nccl_context.cc
+++ b/paddle/fluid/imperative/nccl_context.cc
@@ -93,6 +93,7 @@ void NCCLParallelContext::SendNCCLID(const std::string &ep,
    send(sock, buffer, NCCL_UNIQUE_ID_BYTES, 0);
    break;
  }
+  close(sock);
 }

 void NCCLParallelContext::BcastNCCLId(ncclUniqueId *nccl_id, int root) {