内置类型
MPI_CHAR: 字符型
- MPI_UNSIGNED_CHAR: 无符号字符型
- MPI_BYTE: 字节型
- MPI_SHORT: 短整型
- MPI_UNSIGNED_SHORT: 无符号短整型
- MPI_INT: 整型
- MPI_UNSIGNED: 无符号整型
- MPI_LONG: 长整型
- MPI_UNSIGNED_LONG: 无符号长整型
- MPI_FLOAT: 单精度浮点型
- MPI_DOUBLE: 双精度浮点型
- MPI_LONG_DOUBLE: 长双精度浮点型
自定义类型
MPI_Type_contiguous: 创建一个由相同大小的元素组成的类型
函数原型
int MPI_Type_contiguous(int count, MPI_Datatype oldtype,
MPI_Datatype *newtype)
参数详解
- count:新类型中元素的数量。
- oldtype:待复制元素的类型。
- newtype:返回一个新类型。
MPI_Type_vector: 创建一个由相同大小、位于相隔固定间距的元素组成的类型
函数原型
int MPI_Type_vector(int count,
int blocklength, int stride, MPI_Datatype oldtype,
MPI_Datatype *newtype)
参数详解
- count:向量中连续元素的数量。
- blocklength:向量中相邻元素之间的间距。
- stride:元素之间的间距(读取到该元素后,要跳过多少个元素才能读取下一个元素)。
- oldtype:待复制元素的类型。
- newtype:返回一个新类型。
MPI_Type_create_struct: 创建一个由不同类型的元素组成的类型
函数原型
int MPI_Type_create_struct(int count, const int* array_of_blocklengths,
const MPI_Aint* array_of_displacements,
const MPI_Datatype* array_of_types, MPI_Datatype* newtype)
参数详解
- count:新类型中元素的数量。
- array_of_blocklengths:指定每个元素的长度。
- array_of_displacements:指定每个元素的偏移量。需要注意的是,对于数组类型,偏移量必须是 MPI_Aint 类型。
- array_of_types:指定每个元素的类型。
- newtype:返回一个新类型。
代码实例
#include <stdio.h>
#include <mpi.h>
typedef struct {
int x, y;
} Vector2D;
int main(int argc, char** argv) {
int size, rank;
MPI_Init(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD, &size);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Datatype Vector2D_type;
MPI_Type_vector(1, 2, 3, MPI_INT, &Vector2D_type);
const MPI_Aint displacements[] = {0, offsetof(Vector2D, y)};
const int blocklengths[] = {1, 1};
MPI_Datatype types[] = {MPI_INT, MPI_INT};
MPI_Type_create_struct(2, blocklengths, displacements, types, &Vector2D_type);
MPI_Type_commit(&Vector2D_type);
if (rank == 0) {
Vector2D v = {1, 2};
MPI_Send(&v, 1, Vector2D_type, 1, 0, MPI_COMM_WORLD);
printf("Process 0 sent vector [%d, %d] to process 1\n", v.x, v.y);
} else if (rank == 1) {
Vector2D v_recv;
MPI_Recv(&v_recv, 1, Vector2D_type, 0, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
printf("Process 1 received vector [%d, %d] from process 0\n", v_recv.x, v_recv.y);
}
MPI_Type_free(&Vector2D_type);
MPI_Finalize();
return 0;
}
MPI_Type_indexed函数创建一个由相同大小的元素组成的类型,但这些元素并不连续,而是位于一个数组的不同位置
函数原型
int MPI_Type_indexed(int count, const int* array_of_blocklengths,
const int* array_of_displacements,
MPI_Datatype oldtype, MPI_Datatype* newtype)
参数详解
- count:新类型中元素的数量。
- array_of_blocklengths:一个整数数组,指定每个块中连续元素的数量。
- array_of_displacements:一个整数数组,指定每个块的起始位置。
- oldtype:待复制元素的类型。
- newtype:返回一个新类型。
代码实例
#include <stdio.h>
#include <mpi.h>
int main(int argc, char** argv) {
int size, rank;
MPI_Init(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD, &size);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
int block_lengths[3] = {2, 3, 2};
int displacements[3] = {0, 4, 12};
int data[7] = {1, 2, 3, 4, 5, 6, 7};
MPI_Datatype Complex_type;
MPI_Type_indexed(3, block_lengths, displacements, MPI_INT, &Complex_type);
MPI_Type_commit(&Complex_type);
if (rank == 0) {
printf("Sending complex data...\n");
MPI_Send(data, 1, Complex_type, 1, 0, MPI_COMM_WORLD);
} else if (rank == 1) {
int recv_data[7];
MPI_Recv(recv_data, 7, MPI_INT, 0, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
printf("Received complex data: [");
for (int i = 0; i < 7; i++) {
printf("%d ", recv_data[i]);
}
printf("]\n");
}
MPI_Type_free(&Complex_type);
MPI_Finalize();
return 0;
}
MPI_Type_hvector 创建一个向量数据类型,但与 MPI_Type_vector 不同的是,所有的数据元素不需要具有相同的大小和类型。具体来说,MPI_Type_hvector 允许用户按照任意的字节长距离来描述向量的结构
函数原型
int MPI_Type_hvector(int count, int blocklength,
MPI_Aint stride, MPI_Datatype oldtype, MPI_Datatype *newtype)
参数详解文章来源:https://www.toymoban.com/news/detail-692589.html
- count:向量中元素的数量。
- blocklength:向量中每个元素的个数。
- stride:相邻元素之间的偏移(以字节为单位)。
- oldtype:要重复的原始数据类型。
- newtype:输出的新数据类型。
代码实例
假设有一个数组 a,它的每个元素的大小是 4 字节,我们想要创建一个新的 MPI 类型,每 2 个元素组合在一起,组成一个长度为 8 字节的结构体。在这种情况下,我们可以使用 MPI_Type_hvector 来创建新的数据类型:文章来源地址https://www.toymoban.com/news/detail-692589.html
MPI_Datatype struct_type, temp_type;
MPI_Type_contiguous(2, MPI_INT, &temp_type);
MPI_Type_create_resized(temp_type, 0, 8, &struct_type);
MPI_Type_commit(&struct_type);
MPI_Type_free(&temp_type);
MPI_Datatype vector_type;
MPI_Type_hvector(4, 1, 8, struct_type, &vector_type);
MPI_Type_commit(&vector_type);
到了这里,关于MPI内置类型与自定义类型的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!