c4c69b4ddf
Add support for MPI_Count type and MPI_COUNT datatype and add the required MPI-3 functions MPI_Get_elements_x, MPI_Status_set_elements_x, MPI_Type_get_extent_x, MPI_Type_get_true_extent_x, and MPI_Type_size_x. This commit adds only the C bindings. Fortran bindins will be added in another commit. For now the MPI_Count type is define to have the same size as MPI_Offset. The type is required to be at least as large as MPI_Offset and MPI_Aint. The type was initially intended to be a ssize_t (if it was the same size as a long long) but there were issues compiling romio with that definition (despite the inclusion of stddef.h). I updated the datatype engine to use size_t instead of uint32_t to support large datatypes. This will require some review to make sure that 1) the changes are beneficial, 2) nothing was broken by the change (I doubt anything was), and 3) there are no performance regressions due to this change. Increase the maximum number of predifined datatypes to support MPI_Count Put common get_elements code to ompi/datatype/ompi_datatype_get_elements.c Update MPI_Get_count to reflect changes in MPI-3 (return MPI_UNDEFINED when the count is too large for an int) This commit was SVN r28932.
91 строка
3.1 KiB
C
91 строка
3.1 KiB
C
/*
|
|
* Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2010 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2008 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2013 Los Alamos National Security, LLC. All rights
|
|
* reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
#include <stdio.h>
|
|
#include <limits.h>
|
|
|
|
#include "ompi/mpi/c/bindings.h"
|
|
#include "ompi/runtime/params.h"
|
|
#include "ompi/communicator/communicator.h"
|
|
#include "ompi/errhandler/errhandler.h"
|
|
#include "ompi/datatype/ompi_datatype.h"
|
|
#include "ompi/memchecker.h"
|
|
|
|
#if OPAL_HAVE_WEAK_SYMBOLS && OMPI_PROFILING_DEFINES
|
|
#pragma weak MPI_Get_elements = PMPI_Get_elements
|
|
#endif
|
|
|
|
#if OMPI_PROFILING_DEFINES
|
|
#include "ompi/mpi/c/profile/defines.h"
|
|
#endif
|
|
|
|
static const char FUNC_NAME[] = "MPI_Get_elements";
|
|
|
|
int MPI_Get_elements(MPI_Status *status, MPI_Datatype datatype, int *count)
|
|
{
|
|
size_t internal_count;
|
|
int ret;
|
|
|
|
OPAL_CR_NOOP_PROGRESS();
|
|
|
|
MEMCHECKER(
|
|
if (status != MPI_STATUSES_IGNORE) {
|
|
/*
|
|
* Before checking the complete status, we need to reset the definedness
|
|
* of the MPI_ERROR-field (single-completion calls wait/test).
|
|
*/
|
|
opal_memchecker_base_mem_defined(&status->MPI_ERROR, sizeof(int));
|
|
memchecker_status(status);
|
|
memchecker_datatype(datatype);
|
|
}
|
|
);
|
|
|
|
if (MPI_PARAM_CHECK) {
|
|
int err = MPI_SUCCESS;
|
|
OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
|
|
if (NULL == status || MPI_STATUSES_IGNORE == status ||
|
|
MPI_STATUS_IGNORE == status || NULL == count) {
|
|
err = MPI_ERR_ARG;
|
|
} else if (NULL == datatype || MPI_DATATYPE_NULL == datatype) {
|
|
err = MPI_ERR_TYPE;
|
|
} else {
|
|
OMPI_CHECK_DATATYPE_FOR_RECV(err, datatype, 1);
|
|
}
|
|
OMPI_ERRHANDLER_CHECK(err, MPI_COMM_WORLD, err, FUNC_NAME);
|
|
}
|
|
|
|
ret = ompi_datatype_get_elements (datatype, status->_ucount, &internal_count);
|
|
if (OMPI_SUCCESS == ret || OMPI_ERR_VALUE_OUT_OF_BOUNDS == ret) {
|
|
if (OMPI_SUCCESS == ret && internal_count <= INT_MAX) {
|
|
*count = internal_count;
|
|
} else {
|
|
/* If we have more elements that we can represent with a signed int then we must
|
|
* set count to MPI_UNDEFINED (MPI 3.0).
|
|
*/
|
|
*count = MPI_UNDEFINED;
|
|
}
|
|
|
|
return MPI_SUCCESS;
|
|
}
|
|
|
|
return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_ARG, FUNC_NAME);
|
|
}
|