MPI（一）常见函数

MPI简介

头文件
相关变量声明 MPI_Init(&argc, &argv)
程序开始
程序体 计算与通信
程序结束 MPI_Finalize()

MPI常见函数

// MPI_Init
MPI_Init(&argc,&argv)  // 也可以是NULL，NULL
// MPI_Finalize
MPI_Finalize

// 给rank赋值为进程号
int MPI_Comm_rank(MPI_Comm comm, int *rank)
MPI_Comm_rank(MPI_COMM_WORLD, &myrank);

// 给size赋值通信域内总进程数
int MPI_Comm_size(MPI_Comm, int *size)
MPI_Comm_size(MPI_COMM_WORLD, &size);

// 发送函数，进程间发送消息。其中dest是接收的进程号。
// tag为信息标志，同为整型变量，发送和接收需要tag一致，这将可以区分同一目的地的不同消息。比如进程0给进程1分别发送了数据A和数据B，tag可分别定义成0和1，这样在进程1接收时同样设置tag0和1去接收，避免接收混乱。
// 缓冲区起始地址 发送数据个数 数据类型 目的进程标识号 消息标志 通信域
// 注意：是以数据类型为单位指定消息的长度
int MPI_Send(type* buf, int count, MPI_Datatype datatype, int dest, int tag, MPI_Comm comm)
    
// MPI_RECV 需要和MPI_SEND成对出现，
/**编程者要保证接收缓存区的长度不小于发送数据的长度
注意：最后有个status**/
int MPI_Recv(type* buf, int count, MPI_Datatype datatype, int source, int tag, MPI_Comm comm, MPI_Status *status) 

/**返回状态status。  也可以对status变量执行MPI_GET_COUNT调用可以得到接收到的消息的长度信息**/
MPI_Status status;
cout<<status.MPI_SOURSE<<status.MPI_TAG<<status.ERROR<<endl;
    
// scatter 对比, 是给很多进程分别发送count个消息。
int MPI_Scatter(type* send_buf, int count, MPI_Datatype, type *rec_buf, int count, MPI_Datatype, int tag, MPI_Comm comm)

简单的发送接收程序

#include<stdio.h>
#include<mpi.h>
#include<string.h>

int main(int argc, char **argv){
	char message[20];
	int myrank;
	MPI_Init(&argc,&argv); // MPI初始化 
	MPI_Comm_rank(MPI_COMM_WORLD,&myrank);  // 得到当前进程的标示 
	if(myrank == 0){
		/**
		发送数据，发送方和接收方必须在同一个通信域中，由通信域来统一协调和控制消息的发送和接收 
		**/
		strcpy(message, "Hello, process 1");
		MPI_Send(message, strlen(message),MPI_CHAR, 1, 99, MPI_COMM_WORLD);
	}
	else if(myrank == 1){
		/**
		进程1接收消息。使用message作为接收缓冲区。
		同一个变量在发送进程和接收进程中的作用是不同的。
		指定接收消息的最大长度 为20， 
		**/
		MPI_Status status;
		MPI_Recv(message, 20, MPI_CHAR, 0, 99, MPI_COMM_WORLD, &status);
		printf("received :%send",message);
	}
	MPI_Finalize();
}