Skip to content

Commit e4c0de0

Browse files
committed
fix by comment
1 parent 840bd1f commit e4c0de0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

paddle/operators/send_op.cc

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ class SendOp : public framework::OperatorBase {
3737
auto ins = Inputs("X");
3838
auto outs = Outputs("Out");
3939
std::vector<std::string> epmap = Attr<std::vector<std::string>>("epmap");
40+
std::vector<std::string> endpoints =
41+
Attr<std::vector<std::string>>("endpoints");
4042

4143
platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance();
4244
auto& ctx = *pool.Get(place);
@@ -46,8 +48,7 @@ class SendOp : public framework::OperatorBase {
4648
}
4749
PADDLE_ENFORCE(client_.Wait());
4850

49-
std::set<std::string> epset(epmap.begin(), epmap.end());
50-
for (auto& ep : epset) {
51+
for (auto& ep : endpoints) {
5152
VLOG(3) << "batch barrier, ep: " << ep;
5253
client_.AsyncSendBatchBarrier(ep);
5354
}

0 commit comments

Comments
 (0)