X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/9335baa2b23f940577448a84168eb0b50d3dd966..HEAD:/src/smpi/colls/allreduce/allreduce-smp-binomial-pipeline.cpp diff --git a/src/smpi/colls/allreduce/allreduce-smp-binomial-pipeline.cpp b/src/smpi/colls/allreduce/allreduce-smp-binomial-pipeline.cpp index 4750e238f4..70d57da0ad 100644 --- a/src/smpi/colls/allreduce/allreduce-smp-binomial-pipeline.cpp +++ b/src/smpi/colls/allreduce/allreduce-smp-binomial-pipeline.cpp @@ -1,20 +1,18 @@ -/* Copyright (c) 2013-2017. The SimGrid Team. +/* Copyright (c) 2013-2023. The SimGrid Team. * All rights reserved. */ /* 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 "../colls_private.h" -/* IMPLEMENTED BY PITCH PATARASUK - Non-topoloty-specific (however, number of cores/node need to be changed) +#include "../colls_private.hpp" +/* IMPLEMENTED BY PITCH PATARASUK + Non-topology-specific (however, number of cores/node need to be changed) all-reduce operation designed for smp clusters - It uses 2-layer communication: binomial for both intra-communication + It uses 2-layer communication: binomial for both intra-communication inter-communication The communication are done in a pipeline fashion */ - - -/* this is a default segment size for pipelining, +/* this is a default segment size for pipelining, but it is typically passed as a command line argument */ int allreduce_smp_binomial_pipeline_segment_size = 4096; @@ -32,21 +30,19 @@ int allreduce_smp_binomial_pipeline_segment_size = 4096; */ /* -This fucntion performs all-reduce operation as follow. ** in a pipeline fashion ** +This function performs all-reduce operation as follow. ** in a pipeline fashion ** 1) binomial_tree reduce inside each SMP node 2) binomial_tree reduce intra-communication between root of each SMP node 3) binomial_tree bcast intra-communication between root of each SMP node 4) binomial_tree bcast inside each SMP node */ -namespace simgrid{ -namespace smpi{ -int Coll_allreduce_smp_binomial_pipeline::allreduce(void *send_buf, - void *recv_buf, int count, - MPI_Datatype dtype, - MPI_Op op, MPI_Comm comm) +namespace simgrid::smpi { +int allreduce__smp_binomial_pipeline(const void *send_buf, + void *recv_buf, int count, + MPI_Datatype dtype, + MPI_Op op, MPI_Comm comm) { int comm_size, rank; - void *tmp_buf; int tag = COLL_TAG_ALLREDUCE; int mask, src, dst; MPI_Status status; @@ -62,7 +58,7 @@ int Coll_allreduce_smp_binomial_pipeline::allreduce(void *send_buf, rank = comm->rank(); MPI_Aint extent; extent = dtype->get_extent(); - tmp_buf = (void *) smpi_get_tmp_sendbuffer(count * extent); + unsigned char* tmp_buf = smpi_get_tmp_sendbuffer(count * extent); int intra_rank, inter_rank; intra_rank = rank % num_core; @@ -113,7 +109,7 @@ int Coll_allreduce_smp_binomial_pipeline::allreduce(void *send_buf, } } - /* start binomial reduce inter-communication between each SMP nodes: + /* start binomial reduce inter-communication between each SMP nodes: each node only have one process that can communicate to other nodes */ if ((phase > 0) && (phase < (pipelength + 1))) { if (intra_rank == 0) { @@ -138,7 +134,7 @@ int Coll_allreduce_smp_binomial_pipeline::allreduce(void *send_buf, } } - /* start binomial broadcast inter-communication between each SMP nodes: + /* start binomial broadcast inter-communication between each SMP nodes: each node only have one process that can communicate to other nodes */ if ((phase > 1) && (phase < (pipelength + 2))) { if (intra_rank == 0) { @@ -202,5 +198,4 @@ int Coll_allreduce_smp_binomial_pipeline::allreduce(void *send_buf, smpi_free_tmp_buffer(tmp_buf); return MPI_SUCCESS; } -} -} +} // namespace simgrid::smpi