1 2 static char help[] = "AO test contributed by Sebastian Steiger <steiger@purdue.edu>, March 2011\n\n"; 3 4 /* 5 Example of usage: 6 mpiexec -n 12 ./ex3 7 mpiexec -n 30 ./ex3 -ao_type basic 8 */ 9 #define PETSC_SKIP_CXX_COMPLEX_FIX 10 11 #include <iostream> 12 #include <fstream> 13 #include <vector> 14 #include <petscvec.h> 15 #include <petscao.h> 16 17 using namespace std; 18 19 int main(int argc, char** argv) 20 { 21 PetscErrorCode ierr; 22 AO ao; 23 IS isapp; 24 char infile[PETSC_MAX_PATH_LEN],datafiles[PETSC_MAX_PATH_LEN]; 25 PetscBool flg; 26 PetscMPIInt size,rank; 27 28 ierr = PetscInitialize(&argc, &argv, (char*)0, help);if (ierr) return ierr; 29 ierr = MPI_Comm_size(PETSC_COMM_WORLD, &size);CHKERRQ(ierr); 30 ierr = MPI_Comm_rank(PETSC_COMM_WORLD, &rank);CHKERRQ(ierr); 31 32 ierr = PetscOptionsGetString(NULL,NULL,"-datafiles",datafiles,sizeof(datafiles),&flg);CHKERRQ(ierr); 33 if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_USER,"Must specify -datafiles ${DATAFILESPATH}/ao"); 34 35 // read in application indices 36 ierr = PetscSNPrintf(infile,sizeof(infile),"%s/AO%dCPUs/ao_p%d_appindices.txt",datafiles,size,rank);CHKERRQ(ierr); 37 ifstream fin(infile); 38 if (!fin) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"File not found: %s",infile); 39 vector<PetscInt> myapp; 40 int tmp=-1; 41 while (!fin.eof()) { 42 tmp=-1; 43 fin >> tmp; 44 if (tmp==-1) break; 45 myapp.push_back(tmp); 46 } 47 ierr = PetscSynchronizedPrintf(PETSC_COMM_WORLD,"[%d] has %D indices.\n",rank,myapp.size());CHKERRQ(ierr); 48 ierr = PetscSynchronizedFlush(PETSC_COMM_WORLD,PETSC_STDOUT);CHKERRQ(ierr); 49 50 ierr = ISCreateGeneral(PETSC_COMM_WORLD, myapp.size(), &(myapp[0]), PETSC_USE_POINTER, &isapp);CHKERRQ(ierr); 51 52 ierr = AOCreate(PETSC_COMM_WORLD, &ao);CHKERRQ(ierr); 53 ierr = AOSetIS(ao, isapp, NULL);CHKERRQ(ierr); 54 ierr = AOSetType(ao, AOMEMORYSCALABLE);CHKERRQ(ierr); 55 ierr = AOSetFromOptions(ao);CHKERRQ(ierr); 56 57 if (rank==0) cout << "AO has been set up." << endl; 58 59 ierr = AODestroy(&ao);CHKERRQ(ierr); 60 ierr = ISDestroy(&isapp);CHKERRQ(ierr); 61 62 if (rank==0) cout << "AO is done." << endl; 63 64 ierr = PetscFinalize(); 65 return ierr; 66 } 67 68 69 /*TEST 70 71 build: 72 requires: !define(PETSC_USE_64BIT_INDICES) 73 74 test: 75 nsize: 12 76 requires: double !complex datafilespath 77 args: -datafiles ${DATAFILESPATH}/ao 78 output_file: output/ex3_1.out 79 80 test: 81 suffix: 2 82 nsize: 12 83 requires: double !complex datafilespath 84 args: -ao_type basic -datafiles ${DATAFILESPATH}/ao 85 output_file: output/ex3_1.out 86 87 test: 88 suffix: 3 89 nsize: 30 90 requires: double !complex datafilespath 91 args: -datafiles ${DATAFILESPATH}/ao 92 93 test: 94 suffix: 4 95 nsize: 30 96 requires: double !complex datafilespath 97 args: -ao_type basic -datafiles ${DATAFILESPATH}/ao 98 output_file: output/ex3_3.out 99 100 TEST*/ 101