85 строки
2.4 KiB
C
85 строки
2.4 KiB
C
/*
|
|
* Copyright (c) 2013 Mellanox Technologies, Inc.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "oshmem_config.h"
|
|
|
|
#include "orte/runtime/orte_globals.h"
|
|
|
|
#include "ompi/communicator/communicator.h" /*TODO: ompi_communicator_t */
|
|
#include "ompi/patterns/comm/coll_ops.h" /*TODO: comm_bcast_pml */
|
|
|
|
#include "oshmem/constants.h"
|
|
#include "oshmem/runtime/runtime.h"
|
|
#include "oshmem/runtime/params.h"
|
|
|
|
OSHMEM_DECLSPEC int oshmem_shmem_exchange_allgather(void *buf,
|
|
int buf_size)
|
|
{
|
|
int rc = OSHMEM_SUCCESS;
|
|
int i = 0;
|
|
int *ranks_in_comm = NULL;
|
|
|
|
ranks_in_comm = (int *) malloc(orte_process_info.num_procs * sizeof(int));
|
|
if (NULL == ranks_in_comm) {
|
|
return OSHMEM_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
for (i = 0; i < (int) orte_process_info.num_procs; ++i) {
|
|
ranks_in_comm[i] = i;
|
|
}
|
|
void* buf_temp = malloc(buf_size);
|
|
memcpy(buf_temp, buf + buf_size * ORTE_PROC_MY_NAME->vpid, buf_size);
|
|
|
|
rc = comm_allgather_pml( buf_temp,
|
|
buf,
|
|
buf_size,
|
|
MPI_BYTE,
|
|
ORTE_PROC_MY_NAME->vpid,
|
|
orte_process_info.num_procs,
|
|
ranks_in_comm,
|
|
(ompi_communicator_t *) &ompi_mpi_comm_world);
|
|
|
|
if (ranks_in_comm)
|
|
free(ranks_in_comm);
|
|
if (buf_temp)
|
|
free(buf_temp);
|
|
return rc;
|
|
}
|
|
|
|
OSHMEM_DECLSPEC int oshmem_shmem_exchange_bcast(void *buf,
|
|
int buf_size,
|
|
int peer)
|
|
{
|
|
int rc = OSHMEM_SUCCESS;
|
|
int i = 0;
|
|
int *ranks_in_comm = NULL;
|
|
|
|
ranks_in_comm = (int *) malloc(orte_process_info.num_procs * sizeof(int));
|
|
if (NULL == ranks_in_comm) {
|
|
return OSHMEM_ERR_OUT_OF_RESOURCE;
|
|
}
|
|
|
|
for (i = 0; i < (int) orte_process_info.num_procs; ++i) {
|
|
ranks_in_comm[i] = i;
|
|
}
|
|
rc = comm_bcast_pml((void *) buf,
|
|
peer,
|
|
buf_size,
|
|
MPI_BYTE,
|
|
ORTE_PROC_MY_NAME->vpid,
|
|
orte_process_info.num_procs,
|
|
ranks_in_comm,
|
|
(ompi_communicator_t *) &ompi_mpi_comm_world);
|
|
if (ranks_in_comm)
|
|
free(ranks_in_comm);
|
|
|
|
return rc;
|
|
}
|