alltoallo.java

来自「MPI for java for Distributed Programming」· Java 代码 · 共 55 行

package mpi.ccl_ObjSer;
/****************************************************************************

 Object version :
    Sang Lim([e-mail])
    Northeast Parallel Architectures Center at Syracuse University
    11/28/98
****************************************************************************/

import mpi.*;

public class alltoallO {
  static public void main(String[] args) throws MPIException {
  }

  public alltoallO() {
  }

  public alltoallO(String[] args) throws Exception {

    final int MAXLEN = 10;

    int i,j,k,l;
    int myself,tasks;
 
    MPI.Init(args);
    myself = MPI.COMM_WORLD.Rank();
    tasks = MPI.COMM_WORLD.Size();

    int out[][] = new int[MAXLEN*tasks][MAXLEN]; 
    int in[][]  = new int[MAXLEN*tasks][];

    for(k=0;k<MAXLEN;k++)
      for(i=0;i<MAXLEN*tasks;i++)  
        out[i][k] = k;   
    
    MPI.COMM_WORLD.Alltoall(out,0,MAXLEN,MPI.OBJECT,in,0,MAXLEN,MPI.OBJECT);

    for(k=0;k<MAXLEN*tasks;k++) 
      for(l=0; l<MAXLEN; l++){
 	if(in[k][l] != l) {
	  System.out.println("bad answer in["+k+"]["+l+"] = "+
		  	       in[k][l]+" should be "+l);
          break; 
	}
      }

    MPI.COMM_WORLD.Barrier();
    if(myself==0)  System.out.println("AllToAllO TEST COMPLETE");
    MPI.Finalize();
  }
}