猿代码 — 科研/AI模型/高性能计算
0

mpi测试程序2, 输入数据<0退出

摘要: #include stdio.h#include "mpi.h"int main(int argc, char *argv){ int rank, value, size; MPI_Status status; MPI_Init(argc, argv); MPI_Comm_rank(MPI_COMM_WORLD, rank); /*当前进程在MPI_COMM_WORLD这个 ...

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

int main(int argc, char *argv[])
{
    int rank, value, size;
    MPI_Status status;
    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank); /*当前进程在MPI_COMM_WORLD这个通信组下面 编号是多少*/
    MPI_Comm_size(MPI_COMM_WORLD, &size); /*MPI_COMM_WORLD这个通信组下面 有多少个进程*/
    do {
        if (rank==0) {
            fprintf(stderr, "\nPlease give new value=, <0 break\n");
            scanf("%d",&value);
            fprintf(stderr, "%d read <-<- (%d)\n",rank,value);
            /*必须至少有两个进程的时候 才能进行数据传递*/
            if (size>1) {
                MPI_Send(&value, 1, MPI_INT, rank+1, 0, MPI_COMM_WORLD);
                fprintf(stderr, "%d send (%d)->-> %d\n", rank,value,rank+1);
            }
        }
        else {
            MPI_Recv(&value, 1, MPI_INT, rank-1, 0, MPI_COMM_WORLD, &status);
            fprintf(stderr, "%d receive(%d)<-<- %d\n",rank, value, rank-1);
            if (rank<size-1) {
                MPI_Send(&value, 1, MPI_INT, rank+1, 0, MPI_COMM_WORLD);
                fprintf(stderr, "%d send (%d)->-> %d\n", rank, value, rank+1);
            }
        }
        MPI_Barrier(MPI_COMM_WORLD);
    }while(value>=0);
    MPI_Finalize();
}

说点什么...

已有0条评论

最新评论...

本文作者
2024-2-27 09:47
  • 0
    粉丝
  • 334
    阅读
  • 0
    回复
资讯幻灯片
热门评论
热门专题
排行榜
Copyright   ©2015-2023   猿代码-超算人才智造局 高性能计算|并行计算|人工智能      ( 京ICP备2021026424号-2 )