Homec4science
Diffusion akantu (bugfixes/pypi-py311) Browse

rAKA/src/synchronizer/communicator_inline_impl.hh
b35bce9f3386bugfixes/pypi-py311

Last Change
Raw File

communicator_inline_impl.hh
View Options

/**
* Copyright (©) 2016-2023 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* This file is part of Akantu
*
* Akantu is free software: you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* Akantu is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
*/
/* -------------------------------------------------------------------------- */
#include "communicator.hh"
/* -------------------------------------------------------------------------- */
#ifndef AKANTU_STATIC_COMMUNICATOR_INLINE_IMPL_HH_
#define AKANTU_STATIC_COMMUNICATOR_INLINE_IMPL_HH_
namespace akantu {
/* -------------------------------------------------------------------------- */
inline void
Communicator::freeCommunicationRequest(CommunicationRequest & request) {
request.free();
}
/* -------------------------------------------------------------------------- */
inline void Communicator::freeCommunicationRequest(
std::vector<CommunicationRequest> & requests) {
std::vector<CommunicationRequest>::iterator it;
for (it = requests.begin(); it != requests.end(); ++it) {
it->free();
}
}
/* -------------------------------------------------------------------------- */
template <typename T, typename MsgProcessor>
inline void Communicator::receiveAnyNumber(
std::vector<CommunicationRequest> & send_requests,
MsgProcessor && processor, Int tag) const {
CommunicationRequest barrier_request;
bool got_all{false};
bool are_send_finished{false};
if (getNbProc() == 1) {
return;
}
AKANTU_DEBUG_INFO("Sending " << send_requests.size()
<< " messages and checking for receives TAG["
<< tag << "]");
while (not got_all) {
bool are_receives_ready = true;
while (are_receives_ready) {
CommunicationStatus status;
are_receives_ready = asyncProbe<T>(_any_source, tag, status);
if (are_receives_ready) {
AKANTU_DEBUG_INFO("Receiving message from " << status.getSource());
Array<T> receive_buffer(status.size(), 1);
receive(receive_buffer, status.getSource(), tag);
std::forward<MsgProcessor>(processor)(status.getSource(),
receive_buffer);
}
}
if (not are_send_finished) {
are_send_finished = testAll(send_requests);
if (are_send_finished) {
AKANTU_DEBUG_INFO("All messages send, checking for more receives");
barrier_request = asyncBarrier();
}
}
if (are_send_finished) {
got_all = test(barrier_request);
}
}
AKANTU_DEBUG_INFO("Finished receiving");
}
} // namespace akantu
#endif /* AKANTU_STATIC_COMMUNICATOR_INLINE_IMPL_HH_ */
c4science · Help