File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments