diff --git a/src/context.cc b/src/context.cc index 12b52b78a..7f8ed57be 100644 --- a/src/context.cc +++ b/src/context.cc @@ -43,10 +43,7 @@ MSCCLPP_API_CPP std::shared_ptr Context::connect(Endpoint localEndpo if (remoteEndpoint.transport() != Transport::CudaIpc) { throw mscclpp::Error("Local transport is CudaIpc but remote is not", ErrorCode::InvalidUsage); } - if (!(pimpl_->ipcStream_)) { - pimpl_->ipcStream_ = std::make_shared(cudaStreamNonBlocking); - } - conn = std::make_shared(localEndpoint, remoteEndpoint, cudaStream_t(*(pimpl_->ipcStream_))); + conn = std::make_shared(localEndpoint, remoteEndpoint, cudaStream_t(0)); } else if (AllIBTransports.has(localEndpoint.transport())) { if (!AllIBTransports.has(remoteEndpoint.transport())) { throw mscclpp::Error("Local transport is IB but remote is not", ErrorCode::InvalidUsage);