X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/9df3f18b909012a944b85248cc2fff8988a8f83a..cfa37467445963796bfdafdec9bad8ec8566a60a:/include/simgrid/smpi/replay.hpp diff --git a/include/simgrid/smpi/replay.hpp b/include/simgrid/smpi/replay.hpp index 0b4d32f8d8..21e09f7634 100644 --- a/include/simgrid/smpi/replay.hpp +++ b/include/simgrid/smpi/replay.hpp @@ -2,27 +2,40 @@ /* This program is free software; you can redistribute it and/or modify it * under the terms of the license (GNU LGPL) which comes with this package. */ -#include "private.hpp" +#ifndef SMPI_REPLAY_HPP_ +#define SMPI_REPLAY_HPP_ + +#include "src/smpi/include/smpi_actor.hpp" + +#include #include +#include +#include #include -#define CHECK_ACTION_PARAMS(action, mandatory, optional) \ -{ \ - if (action.size() < static_cast(mandatory + 2)) { \ - std::stringstream ss; \ - for (const auto& elem : action) { \ - ss << elem << " "; \ - } \ - THROWF(arg_error, 0, "%s replay failed.\n" \ - "%zu items were given on the line. First two should be process_id and action. " \ - "This action needs after them %lu mandatory arguments, and accepts %lu optional ones. \n" \ - "The full line that was given is:\n %s\n" \ - "Please contact the Simgrid team if support is needed", \ - __func__, action.size(), static_cast(mandatory), static_cast(optional), \ - ss.str().c_str()); \ - } \ -} +#define CHECK_ACTION_PARAMS(action, mandatory, optional) \ + { \ + if (action.size() < static_cast(mandatory + 2)) { \ + std::stringstream ss; \ + ss << __func__ << " replay failed.\n" \ + << action.size() << " items were given on the line. First two should be process_id and action. " \ + << "This action needs after them " << mandatory << " mandatory arguments, and accepts " << optional \ + << " optional ones. \n" \ + << "The full line that was given is:\n "; \ + for (const auto& elem : action) { \ + ss << elem << " "; \ + } \ + ss << "\nPlease contact the Simgrid team if support is needed"; \ + throw std::invalid_argument(ss.str()); \ + } \ + } + +XBT_PRIVATE void* smpi_get_tmp_sendbuffer(int size); +XBT_PRIVATE void* smpi_get_tmp_recvbuffer(int size); +XBT_PRIVATE void smpi_free_tmp_buffer(void* buf); + +XBT_PRIVATE void log_timed_action(simgrid::xbt::ReplayAction& action, double clock); namespace simgrid { namespace smpi { @@ -78,8 +91,6 @@ public: int root = 0; MPI_Datatype datatype1 = MPI_DEFAULT_TYPE; MPI_Datatype datatype2 = MPI_DEFAULT_TYPE; - - virtual void parse(simgrid::xbt::ReplayAction& action, std::string name) = 0; }; class BcastArgParser : public CollCommParser { @@ -150,7 +161,167 @@ public: void parse(simgrid::xbt::ReplayAction& action, std::string name) override; }; +/** + * Base class for all ReplayActions. + * Note that this class actually implements the behavior of each action + * while the parsing of the replay arguments is done in the @ref ActionArgParser class. + * In other words: The logic goes here, the setup is done by the ActionArgParser. + */ +template class ReplayAction { +protected: + const std::string name; + const aid_t my_proc_id; + T args; + +public: + explicit ReplayAction(std::string name) : name(name), my_proc_id(simgrid::s4u::this_actor::get_pid()) {} + virtual ~ReplayAction() = default; + + void execute(simgrid::xbt::ReplayAction& action) + { + // Needs to be re-initialized for every action, hence here + double start_time = smpi_process()->simulated_elapsed(); + args.parse(action, name); + kernel(action); + if (name != "Init") + log_timed_action(action, start_time); + } + + virtual void kernel(simgrid::xbt::ReplayAction& action) = 0; + void* send_buffer(int size) { return smpi_get_tmp_sendbuffer(size); } + void* recv_buffer(int size) { return smpi_get_tmp_recvbuffer(size); } +}; + +class WaitAction : public ReplayAction { +private: + RequestStorage& req_storage; + +public: + explicit WaitAction(RequestStorage& storage) : ReplayAction("Wait"), req_storage(storage) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class SendAction : public ReplayAction { +private: + RequestStorage& req_storage; + +public: + explicit SendAction(std::string name, RequestStorage& storage) : ReplayAction(name), req_storage(storage) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class RecvAction : public ReplayAction { +private: + RequestStorage& req_storage; + +public: + explicit RecvAction(std::string name, RequestStorage& storage) : ReplayAction(name), req_storage(storage) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class ComputeAction : public ReplayAction { +public: + explicit ComputeAction() : ReplayAction("compute") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class TestAction : public ReplayAction { +private: + RequestStorage& req_storage; + +public: + explicit TestAction(RequestStorage& storage) : ReplayAction("Test"), req_storage(storage) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class InitAction : public ReplayAction { +public: + explicit InitAction() : ReplayAction("Init") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class CommunicatorAction : public ReplayAction { +public: + explicit CommunicatorAction() : ReplayAction("Comm") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; +class WaitAllAction : public ReplayAction { +private: + RequestStorage& req_storage; + +public: + explicit WaitAllAction(RequestStorage& storage) : ReplayAction("waitall"), req_storage(storage) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class BarrierAction : public ReplayAction { +public: + explicit BarrierAction() : ReplayAction("barrier") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class BcastAction : public ReplayAction { +public: + explicit BcastAction() : ReplayAction("bcast") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class ReduceAction : public ReplayAction { +public: + explicit ReduceAction() : ReplayAction("reduce") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class AllReduceAction : public ReplayAction { +public: + explicit AllReduceAction() : ReplayAction("allreduce") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class AllToAllAction : public ReplayAction { +public: + explicit AllToAllAction() : ReplayAction("alltoall") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class GatherAction : public ReplayAction { +public: + explicit GatherAction(std::string name) : ReplayAction(name) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class GatherVAction : public ReplayAction { +public: + explicit GatherVAction(std::string name) : ReplayAction(name) {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class ScatterAction : public ReplayAction { +public: + explicit ScatterAction() : ReplayAction("scatter") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class ScatterVAction : public ReplayAction { +public: + explicit ScatterVAction() : ReplayAction("scatterv") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class ReduceScatterAction : public ReplayAction { +public: + explicit ReduceScatterAction() : ReplayAction("reducescatter") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; + +class AllToAllVAction : public ReplayAction { +public: + explicit AllToAllVAction() : ReplayAction("alltoallv") {} + void kernel(simgrid::xbt::ReplayAction& action) override; +}; } } } + +#endif