#include "private.h"
#include "xbt/sysdep.h"
+#include "xbt/synchro_core.h"
#include "xbt/log.h"
#include "xbt/dict.h"
int present_processes;
int index;
MPI_Comm comm_world;
+ xbt_bar_t finalization_barrier;
} s_smpi_mpi_instance_t;
instance->present_processes = 0;
instance->index = process_count;
instance->comm_world = MPI_COMM_NULL;
+ instance->finalization_barrier=xbt_barrier_init(num_processes);
+
process_count+=num_processes;
if(!smpi_instances){
//get the index of the process in the process_data array
-MPI_Comm* smpi_deployment_register_process(const char* instance_id, int rank, int index){
+void smpi_deployment_register_process(const char* instance_id, int rank, int index,MPI_Comm** comm, xbt_bar_t* bar){
if(!smpi_instances){//no instance registered, we probably used smpirun.
- index_to_process_data[index]=index;
- return NULL;
+ index_to_process_data[index]=index;
+ *bar = NULL;
+ *comm = NULL;
+ return;
}
s_smpi_mpi_instance_t* instance = xbt_dict_get_or_null(smpi_instances, instance_id);
instance->present_processes++;
index_to_process_data[index]=instance->index+rank;
smpi_group_set_mapping(smpi_comm_group(instance->comm_world), index, rank);
- return & instance->comm_world;
+ *bar = instance->finalization_barrier;
+ *comm = &instance->comm_world;
+ return;
+}
+
+void smpi_deployment_cleanup_instances(){
+ xbt_dict_cursor_t cursor = NULL;
+ s_smpi_mpi_instance_t* instance = NULL;
+ char *name = NULL;
+ xbt_dict_foreach(smpi_instances, cursor, name, instance) {
+ while (smpi_group_unuse(smpi_comm_group(instance->comm_world)) > 0);
+ xbt_free(instance->comm_world);
+ xbt_barrier_destroy(instance->finalization_barrier);
+ }
+ xbt_dict_free(&smpi_instances);
}