MPI学习笔记：MPI_Type_vector与打包解包

最新推荐文章于 2023-09-02 22:21:28 发布

精神科胡主任

最新推荐文章于 2023-09-02 22:21:28 发布

阅读量4.1k

点赞数

分类专栏：学习笔记

本文链接：https://blog.csdn.net/weixin_42742438/article/details/84306221

版权

MPI学习笔记：MPI_Type_vector与打包解包

用两种方式实现了进程间三维矩阵的通信。
MPI_Type_vector没什么好说的。而关于打包解包操作，虽然会占用额外的内存，但是在某些情况下还是比较方便的，例如需要发送一个矩阵中的子块。在这里，我使用两层循环来将一个三维矩阵打包到一个连续的缓冲区并发送，接受进程接收后将接收缓冲区中的数据解包到自身的矩阵中。
在使用打包解包操作时需要注意C语言中多维数组按行优先存储，而在fortran中则是列优先。

#include<stdio.h>
#include"mpi.h"
#include"stdlib.h"

#define mat_size 10

int main(int argc,char **argv)
{
   
  //定义矩阵，连续存放，使用mpi_type_vector生成新的数据类型来发送。
  struct vect{
   
    int vec[mat_size];
  };
  struct matrix{
   
    vect vecs[mat_size];
  };
  struct matrix3d{
   
    matrix matrixs[mat_size];
  };
  MPI_Datatype mpi_vec_type;
  MPI_Datatype mpi_3d_type;
  MPI_Aint extent[2];
  //使用mpi_pack打包的矩阵
  int matrix_for_pack[mat_size][mat_size][mat_size];
  int *buffer_for_packed;
  int buff_size;
  int position;//打包位置偏移

  int myrank,proc_nums;
  MPI_Status status;
  MPI_Init(&argc,&argv);
  MPI_Comm_rank(MPI_COMM_WORLD,&myrank);
  MPI_Comm_size(MPI_COMM_WORLD,&proc_nums);
//定义并提交三维矩阵数据类型
  MPI_Type_contiguous(mat_size,MPI_INT,&mpi_vec_type);
  MPI