【MPI高性能计算】用集合通信改进梯形求和积分
生活随笔
收集整理的這篇文章主要介紹了
【MPI高性能计算】用集合通信改进梯形求和积分
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
簡述
- 【MPI高性能計算】梯形面積積分計算
- 里面有梯形積分公式的具體寫法
這里采用MPI_Reduce的集合通信的方式來計算
API
int MPI_Reduce(const void *sendbuf, void *recvbuf, int count,MPI_Datatype datatype, MPI_Op op, int root,MPI_Comm comm)注意,在send和recv上,放的指針最好別指向同一個空間
#include <mpi.h> #include <stdio.h> #include <string> #include <string.h> #pragma warning(disable : 4996) const int MAX_STRING = 100;#define FUN(x) (x * x)int main(int argc, char **argv) {int comm_sz;int my_rank;if (argc == 1) return 0;int n = strtol(argv[1], NULL, 10);double a = 0, b = 1;MPI_Init(NULL, NULL);MPI_Comm_size(MPI_COMM_WORLD, &comm_sz);MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);int left = n % comm_sz;int localn = n / comm_sz + (my_rank < left);double h = (b - a) / n, locala, localb;if (my_rank < left) { locala = a + my_rank * localn * h; }else { locala = a + left * (localn + 1) * h + (my_rank - left) * localn * h; }localb = locala + localn * h;double x = locala; // initial xdouble localSum = ( FUN(locala) + FUN(localb)) / 2, totalSum=0;for (int i = 1; i < localn; ++i) {x += h;localSum += FUN(x);}localSum *= h;MPI_Reduce(&localSum, &totalSum, 1, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD);if (my_rank == 0) printf("comm_sz: %d, n: %d, ans: %.5f\n", comm_sz, n, totalSum);MPI_Finalize(); }- 一樣能算的正確,主要是會簡化,同時用這樣的方式也可以加速計算,畢竟求和的過程也被默認并行了。
總結(jié)
以上是生活随笔為你收集整理的【MPI高性能计算】用集合通信改进梯形求和积分的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【笔记】大数定理证明
- 下一篇: MPI派生数据类型发送接收,降低发送时间