AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

provide copy function for MPI_Comm_create_keyval for OpenMPI (#10)

Browse Source
This commit is contained in:
Jonathan R. Madsen
2021-09-09 16:42:20 -05:00
committed by GitHub
parent e70a9e994d
commit ea60acc4b6
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/libmisc.cpp
+9 -4
View File
@@ -15,7 +15,10 @@ void
hosttrace_mpi_set_attr()
{
#if defined(TIMEMORY_USE_MPI)
static auto _mpi_finalize = [](MPI_Comm, int, void*, void*) {
static auto _mpi_copy = [](MPI_Comm, int, void*, void*, void*, int*) {
return MPI_SUCCESS;
};
static auto _mpi_fini = [](MPI_Comm, int, void*, void*) {
if(mpip_index != std::numeric_limits<uint64_t>::max())
comp::deactivate_mpip<tim::component_tuple<hosttrace_component>, hosttrace>(
mpip_index);
@@ -23,9 +26,11 @@ hosttrace_mpi_set_attr()
hosttrace_trace_finalize();
return MPI_SUCCESS;
};
using func_t = int (*)(MPI_Comm, int, void*, void*);
int _comm_key = -1;
if(PMPI_Comm_create_keyval(nullptr, static_cast<func_t>(_mpi_finalize), &_comm_key,
using copy_func_t = int (*)(MPI_Comm, int, void*, void*, void*, int*);
using fini_func_t = int (*)(MPI_Comm, int, void*, void*);
int _comm_key = -1;
if(PMPI_Comm_create_keyval(static_cast<copy_func_t>(_mpi_copy),
static_cast<fini_func_t>(_mpi_fini), &_comm_key,
nullptr) == MPI_SUCCESS)
PMPI_Comm_set_attr(MPI_COMM_SELF, _comm_key, nullptr);
#endif