【MPI编程】MPI_Bcast广播讲解和使用

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/a19990412/article/details/84283205

函数范式

int MPI_Bcast(
	void * data_p;
	int count;
	MPI_Datatype datatype;
	int source_proc;
	MPI_Comm comm;
);

实例

• 功能描述： 下面实现了0件进程读取数据，并广播给所有其他线程。

• 注意，跟其他集合通信函数类似，每个函数的内容都是一样的。

• 还有，关于非0的进程，需要单独申请空间的。 我一开始以为不需要，所以还想看看是不是同一空间，所以加了那个输出指针的部分。

• 只有一个线程不操作，是我想通过vs来生成可执行文件，这样我就只需要加了个只有一个线程的时候不操作。

#include<stdio.h>
#include<string.h>
#include<mpi.h>
#pragma warning(disable : 4996)
#define MAX_STRING 100
using namespace std;
#include <fstream>
#include <iostream>

int main(void) {
	int comm_sz;
	int my_rank;
	MPI_Init(NULL, NULL);
	MPI_Comm_size(MPI_COMM_WORLD, &comm_sz);
	MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
	
	
	// 只有一个线程的时候不操作
	if (comm_sz <= 1) {
		MPI_Finalize();
		return 0;
	}
	ifstream cin("D:\\C++\\VS\\repo\\MPI-DEMO\\MPI-DEMO\\input.txt");
	
	int N = 0;
	double *a=NULL;
	if (my_rank == 0) {
		cin >> N;
		a = new double[N];
		for (int i = 0; i < N; ++i) {
			cin >> a[i];
		}
	}
	MPI_Bcast(&N, 1, MPI_INT, 0, MPI_COMM_WORLD);
	if (my_rank != 0) { a = new double[N]; }
	MPI_Bcast(a, N, MPI_DOUBLE, 0, MPI_COMM_WORLD);
	
	cout << my_rank << endl;
	for (int i = 0; i < N; ++i) {
		cout << a[i] << " ";
	}cout << endl;
	cout <<"ADDR: "<< a << endl;
	delete[] a;
	MPI_Finalize();

	return 0;
}