X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/237fd22b56db7d1c67360c37559ce3aab16a002d..ea74f5d95928a521a588737e81f1de94eef25d19:/src/smpi/internals/smpi_deployment.cpp diff --git a/src/smpi/internals/smpi_deployment.cpp b/src/smpi/internals/smpi_deployment.cpp index a9d4450b5f..ff50710791 100644 --- a/src/smpi/internals/smpi_deployment.cpp +++ b/src/smpi/internals/smpi_deployment.cpp @@ -1,4 +1,4 @@ -/* Copyright (c) 2004-2021. The SimGrid Team. +/* Copyright (c) 2004-2022. The SimGrid Team. * All rights reserved. */ /* This program is free software; you can redistribute it and/or modify it @@ -7,6 +7,7 @@ #include "smpi_host.hpp" #include "private.hpp" #include "simgrid/s4u/Engine.hpp" +#include "simgrid/s4u/Barrier.hpp" #include "smpi_comm.hpp" #include @@ -25,8 +26,9 @@ public: auto* group = new simgrid::smpi::Group(size_); comm_world_ = new simgrid::smpi::Comm(group, nullptr, false, -1); universe_size += max_no_processes; + bar_ = std::make_shared(size_); } - + std::shared_ptr bar_; unsigned int size_; unsigned int finalized_ranks_ = 0; MPI_Comm comm_world_; @@ -65,6 +67,12 @@ void smpi_deployment_register_process(const std::string& instance_id, int rank, instance.comm_world_->group()->set_mapping(actor->get_pid(), rank); } +void smpi_deployment_startup_barrier(const std::string& instance_id) +{ + const Instance& instance = smpi_instances.at(instance_id); + instance.bar_->wait(); +} + void smpi_deployment_unregister_process(const std::string& instance_id) { Instance& instance = smpi_instances.at(instance_id);