你几乎做对了,除了应该从结构的开头indices
给出每个结构字段的偏移量(以字节为单位)。构造这种类型的正确方法是使用运算符,定义在:offsetof
stddef.h
#include <stddef.h> // or <cstddef> for C++
struct Residence
{
double x;
double y;
};
MPI_Datatype createRecType()
{
// Set-up the arguments for the type constructor
MPI_Datatype new_type;
int count = 2;
int blocklens[] = { 1,1 };
MPI_Aint indices[2];
indices[0] = (MPI_Aint)offsetof(struct Residence, x);
indices[1] = (MPI_Aint)offsetof(struct Residence, y);
MPI_Datatype old_types[] = {MPI_DOUBLE,MPI_DOUBLE};
MPI_Type_struct(count,blocklens,indices,old_types,&new_type);
MPI_Type_commit(&new_type);
return new_type;
}
虽然这对于该特定结构就足够了,但通常必须调整结构化类型长度以考虑编译器可能在结构末尾插入的任何尾随填充。仅当一个人想要发送该结构化类型的多个项目(即结构元素数组)时才需要这样做。旧的方法是在类型的结构中添加第三个成员MPI_UB
(UB 来自 Upper Bound)并将该成员的偏移量设置为等于sizeof(struct Residence)
(填充计入由 返回的结构大小sizeof
)。现代方法是使用MPI_Type_create_resized
,它创建一个新的 MPI 类型,其类型签名与原始类型签名相同,但程度不同:
MPI_Type_struct(count,blocklens,indices,old_types,&new_type);
// Create a resized type
MPI_Type resized_new_type;
MPI_Type_create_resized(new_type,
// lower bound == min(indices) == indices[0]
indices[0],
(MPI_Aint)sizeof(struct Residence),
&resized_new_type);
MPI_Type_commit(&resized_new_type);
// Free new_type as it is no longer needed
MPI_Type_free(&new_type);
return resized_new_type;
仅显示相关的代码行。上面的代码假设indices[0]
给出了第一个结构元素的偏移量。可以改为使用MPI_Type_get_extent
来获得真正的下限,这适用于具有负偏移量的结构类型。没有必要提交new_type
,因为它仅用于构造调整大小的类型。也没有必要保留它,这就是它resized_new_type
在创建后被释放的原因。