#include "private.hpp"
#include "smpi_datatype_derived.hpp"
+#include "smpi_comm.hpp"
XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_pmpi);
if (*datatype == MPI_DATATYPE_NULL || (*datatype)->flags() & DT_FLAG_PREDEFINED) {
return MPI_ERR_TYPE;
} else {
+ (*datatype)->mark_as_deleted();
simgrid::smpi::Datatype::unref(*datatype);
*datatype=MPI_DATATYPE_NULL;
return MPI_SUCCESS;
return datatype->extent(lb, extent);
}
+int PMPI_Type_get_extent_x(MPI_Datatype datatype, MPI_Count * lb, MPI_Count * extent)
+{
+ MPI_Aint tmplb;
+ MPI_Aint tmpext;
+ int ret = PMPI_Type_get_extent(datatype, &tmplb, &tmpext);
+ if(ret == MPI_SUCCESS){
+ *lb = static_cast<MPI_Count>(tmplb);
+ *extent = static_cast<MPI_Count>(tmpext);
+ }
+ return ret;
+}
+
int PMPI_Type_get_true_extent(MPI_Datatype datatype, MPI_Aint * lb, MPI_Aint * extent)
{
return PMPI_Type_get_extent(datatype, lb, extent);
}
+int PMPI_Type_get_true_extent_x(MPI_Datatype datatype, MPI_Count * lb, MPI_Count * extent)
+{
+ return PMPI_Type_get_extent_x(datatype, lb, extent);
+}
+
int PMPI_Type_extent(MPI_Datatype datatype, MPI_Aint * extent)
{
CHECK_MPI_NULL(1, MPI_DATATYPE_NULL, MPI_ERR_TYPE, datatype)
if (ndims==0){
*newtype = MPI_DATATYPE_NULL;
return MPI_SUCCESS;
- } else if (ndims==1){
+ }
+ CHECK_NULL(2, MPI_ERR_ARG, array_of_sizes)
+ CHECK_NULL(3, MPI_ERR_ARG, array_of_subsizes)
+ CHECK_NULL(4, MPI_ERR_ARG, array_of_starts)
+ for (int i = 0; i < ndims; i++) {
+ CHECK_NEGATIVE_OR_ZERO(2, MPI_ERR_COUNT, array_of_sizes[i])
+ CHECK_NEGATIVE(3, MPI_ERR_COUNT, array_of_subsizes[i])
+ CHECK_NEGATIVE(4, MPI_ERR_COUNT, array_of_starts[i])
+ }
+ if (ndims==1){
simgrid::smpi::Datatype::create_contiguous( array_of_subsizes[0], oldtype, array_of_starts[0]*oldtype->get_extent(), newtype);
return MPI_SUCCESS;
} else if (oldtype == MPI_DATATYPE_NULL || not oldtype->is_valid() ) {
CHECK_NEGATIVE(1, MPI_ERR_COUNT, incount)
CHECK_TYPE(2, datatype)
CHECK_COMM(3)
- *size=incount*datatype->size();
+ *size = incount * std::max<long>(datatype->size(), datatype->get_extent());
return MPI_SUCCESS;
}