1. 程式人生 > >【MPI高效能運算】蒙特卡洛方法計算pi值

【MPI高效能運算】蒙特卡洛方法計算pi值

蒙特卡洛方法

就是通過概率模擬來近似計算。 其實演算法進度不是很高。

程式碼

在下面程式碼中的input檔案中的內容是

10000000

執行效果:下面用四個核來做計算

PS D:\C++\VS\repo\MPI-DEMO\Debug> mpiexec -n 4 ./MPI-DEMO.exe
3.14111
#include<stdio.h>
#include<string.h>
#include<mpi.h>
#pragma warning(disable : 4996)
#define MAX_STRING 100
using namespace std;
#include <fstream>
#include <ctime>
#include <iostream>

#define random() (rand() / double (RAND_MAX) * 2 - 1)

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;
	}
	long long int count_time; // 總的計算次數
	long long int Cal[2] = { 0 };
	if (my_rank == 0) {
		ifstream cin("D:\\C++\\VS\\repo\\MPI-DEMO\\MPI-DEMO\\input.txt");
		
		cin >> count_time;
		long long int tempCal[2] = { 0 };

		long long int width = count_time / (comm_sz - 1);
		long long int reminder = count_time % (comm_sz - 1);
		long long int temp_count_time;
		for (int i = 1; i < comm_sz; ++i)
		{
			temp_count_time = width;
			if (i <= reminder) temp_count_time++;
			MPI_Send(&temp_count_time, 1, MPI_LONG_LONG_INT, i, 0, MPI_COMM_WORLD);
		}
		for (int i = 1; i < comm_sz; ++i)
		{
			MPI_Recv(tempCal, 2, MPI_LONG_LONG_INT, i, 0, MPI_COMM_WORLD, MPI_STATUSES_IGNORE);
			for (int j = 0; j < 2; ++j) { Cal[j] += tempCal[j]; }
		}

		std::cout << 4.0 * Cal[0] / count_time << endl;
	} // 運算的子節點
	else {
		srand((unsigned)time(NULL));
		MPI_Recv(&count_time, 1, MPI_LONG_LONG_INT, 0, 0, MPI_COMM_WORLD, MPI_STATUSES_IGNORE);
		Cal[1] = count_time;
		Cal[0] = 0;
		double x, y, dst;
		for (int i = 0; i < count_time; ++i) {
			x = random();
			y = random();
			dst = x * x + y * y;
			if (dst <= 1) Cal[0] += 1;
		}
		MPI_Send(Cal, 2, MPI_LONG_LONG_INT, 0, 0, MPI_COMM_WORLD);
	}
	MPI_Finalize();
	
	return 0;
}