44 строки
1.0 KiB
C
44 строки
1.0 KiB
C
|
/*
|
||
|
|
* Copyright (c) 2013 Mellanox Technologies, Inc.
|
||
|
|
* All rights reserved.
|
||
|
|
* Copyright (c) 2015 Research Organization for Information Science
|
||
|
|
* and Technology (RIST). All rights reserved.
|
||
|
|
* $COPYRIGHT$
|
||
|
|
*
|
||
|
|
* Additional copyrights may follow
|
||
|
|
*
|
||
|
|
* $HEADER$
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "oshmem_config.h"
|
||
|
|
|
||
|
|
#include "oshmem/constants.h"
|
||
|
|
#include "oshmem/runtime/runtime.h"
|
||
|
|
#include "oshmem/runtime/params.h"
|
||
|
|
|
||
|
|
int oshmem_shmem_allgather(void *send_buf, void *rcv_buf, int elem_size)
|
||
|
|
{
|
||
|
|
int rc;
|
||
|
|
|
||
|
|
rc = PMPI_Allgather(send_buf, elem_size, MPI_BYTE,
|
||
|
|
rcv_buf, elem_size, MPI_BYTE, oshmem_comm_world);
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
int oshmem_shmem_allgatherv(void *send_buf, void* rcv_buf, int send_count,
|
||
|
|
int* rcv_size, int* displs)
|
||
|
|
{
|
||
|
|
int rc;
|
||
|
|
|
||
|
|
rc = PMPI_Allgatherv(send_buf, send_count, MPI_BYTE,
|
||
|
|
rcv_buf, rcv_size, displs, MPI_BYTE, oshmem_comm_world);
|
||
|
|
|
||
|
|
return rc;
|
||
|
|
}
|
||
|
|
|
||
|
|
void oshmem_shmem_barrier(void)
|
||
|
|
{
|
||
|
|
PMPI_Barrier(oshmem_comm_world);
|
||
|
|
}
|