}

phi::CustomContext* dev_context() const override { return dev_ctx_.get(); }

return compute_event_;

}

return comm_event_;

}

compute_event_ = std::move(compute_event);

}

comm_event_ = std::move(comm_event);

}

std::unique_ptr<phi::CustomContext> dev_ctx_;

// used for comm wait compute, compute_stream-->event-->comm_stream

// used for compute wait comm, comm_stream-->event-->compute_stream

};

static std::unordered_map<std::string, std::unique_ptr<XCCLCommContext>>

c->set_rank(rank);

c->set_comm(comm);

c->set_dev_ctx(std::move(dev_ctx));

comm_map_mutex_.lock();

if (comm_map_.count(ring_id) == 0) {

virtual phi::ccl::CCLComm comm() const = 0;

virtual phi::stream::stream_t stream() const = 0;

virtual std::shared_ptr<phi::stream::Stream> GetStream() const = 0;

virtual phi::CustomContext* dev_context() const = 0;

virtual ~XCCLComm() = default;

};