【MPI编程】任意数节点的树形求和（高性能计算）

以你之姓@ 2022-04-18 00:43 363阅读 0赞

## 简述 ##

之前写过一个只能在2幂次数的节点环境下使用的树形结构求和  
[【MPI编程】2幂节点的树形求和（高性能计算）][MPI_2]  
稍微修改了一下对应代码，就可以使用了。其实就是在逻辑上扩充一下为2的幂数。也就是说接受节点上的只要不越界就是任然要接受。如果越界了，就说明对应的在后面的节点是不存在的。那么就保留当前节点就好了。

## 运行效果 ##

输入为一个文件：  
内容为：

19
    1 2 3 4 5 6 7 8 9 10
    2 3 4 5 6 7 0 0 2

![在这里插入图片描述][20181106093734582.png]

## 代码 ##

#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 len;
    	double scalar, local_sum = 0;
    	int comm_sz;
    	int my_rank;
    	int divided_len;
    	int last_divided;
    	int begin_i;
    	int oper_N = 2;
    	MPI_Init(NULL, NULL);
    	MPI_Comm_size(MPI_COMM_WORLD, &comm_sz);
    	MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
    
    	// 假设 n % comm_sz = 0 整除
    	// 只有一个线程的时候不操作
    	if (comm_sz <= 1) { 
    		MPI_Finalize();
    		return 0;
    	}
    	ifstream cin("D:\\C++\\VS\\repo\\MPI-DEMO\\MPI-DEMO\\input.txt");
    	cin >> len; // 输入数据长度
    
    	divided_len = len / comm_sz;
    	last_divided = len % comm_sz;
    
    	if (my_rank < last_divided) { 
    		divided_len++;
    		begin_i = my_rank * divided_len;
    	}
    	else { 
    		begin_i = (my_rank - last_divided) * divided_len + ((divided_len + 1)* last_divided);
    	}
    	// 局部和
    	for (int i = 0; i < len; ++i) { 
    		cin >> scalar;
    		if (i < begin_i || i >= begin_i + divided_len) continue;
    		else { 
    			local_sum += scalar;
    		}
    	}
    	// 假设comm_sz不一定是2的幂
    	// 树形求和
    	while (oper_N <= comm_sz || (oper_N > comm_sz && oper_N / 2 < comm_sz)) { 
    		// recieve 
    		if (my_rank % oper_N == 0) { 
    			// 只需要控制有些收，有些不收即可
    			if (my_rank + oper_N / 2 < comm_sz) { 
    				MPI_Recv(&scalar, 1, MPI_DOUBLE, my_rank + oper_N / 2, 0, MPI_COMM_WORLD, MPI_STATUSES_IGNORE);
    				local_sum += scalar;
    			}
    		}// send
    		else if (my_rank % oper_N == oper_N / 2) { 
    			MPI_Send(&local_sum, 1, MPI_DOUBLE, my_rank - oper_N / 2, 0, MPI_COMM_WORLD);
    		}
    		else {  break; }
    		oper_N *= 2;
    	}
    	if (my_rank == 0) { 
    		cout << local_sum << endl;
    	}
    	MPI_Finalize();
    
    	return 0;
    }

[MPI_2]: https://blog.csdn.net/a19990412/article/details/83780669
[20181106093734582.png]: /images/20220418/eb25ee37dfb74c538fd765573a2e4ec0.png