matrix6

//http://blog.csdn.net/pouloghost/article/details/7913342
//oct 9 2016
//jaccobi
#include "mpi.h"  
#include <stdio.h>  
#include <stdlib.h>  
  
//broadcast x  
void bcastx(float *xs,int size) {  
    for(int i=0;i<size;++i) {  
        MPI_Bcast(&xs[i],1,MPI_FLOAT,i,MPI_COMM_WORLD);  
    }  
}  
//while condition  
static inline int condition(int send) {  
    int recv=0;  
    MPI_Allreduce(&send,&recv,1,MPI_INT,MPI_SUM,MPI_COMM_WORLD);  
    return recv;  
}  
int main(int argc,char *argv[]) {  
    float equs[][3]={{2,1,3},  
                {3,2,4}};  
    //threshold  
    float thd=0.1;  
    int self,size;  
    MPI_Init(&argc,&argv);  
    MPI_Comm_rank(MPI_COMM_WORLD,&self);  
    MPI_Comm_size(MPI_COMM_WORLD,&size);  
    //store every x  
    float *xs=(float*)malloc(size*sizeof(float));  
    float nx;  
    //init  
    for(int i=0;i<size;++i) {  
        xs[i]=0;  
    }  
    int con=1;  
    while(0!=condition(con)) {  
        nx=equs[self][size];  
        //calculate only use the correspond equivalent  
        for(int i=0;i<size;++i) {  
            if(self!=i) {  
                nx-=equs[self][i]*xs[i];  
            }             
        }  
        nx/=equs[self][self];  
        //get the positive different  
        float diff=nx-xs[self];  
        if(diff<0) {  
            diff*=-1;  
        }  
        //update local x  
        xs[self]=nx;  
        //loop condition  
        con=diff>thd?1:0;  
        //update all xs  
        bcastx(xs,size);  
    }  
    printf("%d,%f \n",self,nx);  
    free(xs);  
    MPI_Finalize();  
    return 0;  
}  

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值