1 static const char help[] = "Test ParMETIS handling of negative weights.\n\n"; 2 3 /* Test contributed by John Fettig */ 4 5 /* 6 * This file implements two tests for a bug reported in ParMETIS. These tests are not expected to pass without the 7 * patches in the PETSc distribution of ParMetis. See parmetis.py 8 * 9 * 10 * The bug was reported upstream, but has received no action so far. 11 * 12 * http://glaros.dtc.umn.edu/gkhome/node/837 13 * 14 */ 15 16 #include <petscsys.h> 17 #include <parmetis.h> 18 19 #define CHKERRQPARMETIS(n) \ 20 if (n == METIS_ERROR_INPUT) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"ParMETIS error due to wrong inputs and/or options"); \ 21 else if (n == METIS_ERROR_MEMORY) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"ParMETIS error due to insufficient memory"); \ 22 else if (n == METIS_ERROR) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"ParMETIS general error"); \ 23 24 int main(int argc, char *argv[]) 25 { 26 PetscErrorCode ierr; 27 PetscBool flg; 28 PetscMPIInt rank, size; 29 int i, status; 30 idx_t ni,isize,*vtxdist, *xadj, *adjncy, *vwgt, *part; 31 idx_t wgtflag=0, numflag=0, ncon=1, ndims=3, edgecut=0; 32 idx_t options[5]; 33 PetscReal *xyz; 34 real_t *sxyz, *tpwgts, ubvec[1]; 35 MPI_Comm comm; 36 FILE *fp; 37 char fname[PETSC_MAX_PATH_LEN],prefix[PETSC_MAX_PATH_LEN] = ""; 38 size_t red; 39 40 ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr; 41 #if defined(PETSC_USE_64BIT_INDICES) 42 ierr = PetscPrintf(PETSC_COMM_WORLD,"This example only works with 32 bit indices\n"); 43 ierr = PetscFinalize(); 44 return ierr; 45 #endif 46 MPI_Comm_rank(PETSC_COMM_WORLD,&rank); 47 MPI_Comm_size(PETSC_COMM_WORLD,&size); 48 49 ierr = PetscOptionsBegin(PETSC_COMM_WORLD,NULL,"Parmetis test options","");CHKERRQ(ierr); 50 ierr = PetscOptionsString("-prefix","Path and prefix of test file","",prefix,prefix,sizeof(prefix),&flg);CHKERRQ(ierr); 51 if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_USER,"Must specify -prefix"); 52 ierr = PetscOptionsEnd();CHKERRQ(ierr); 53 54 ierr = PetscMalloc1(size+1,&vtxdist);CHKERRQ(ierr); 55 56 ierr = PetscSNPrintf(fname,sizeof(fname),"%s.%d.graph",prefix,rank);CHKERRQ(ierr); 57 58 ierr = PetscFOpen(PETSC_COMM_SELF,fname,"r",&fp);CHKERRQ(ierr); 59 60 red = fread(vtxdist, sizeof(idx_t), size+1, fp);if (red != (size_t) (size+1)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"Unable to read from data file"); 61 62 ni = vtxdist[rank+1]-vtxdist[rank]; 63 64 ierr = PetscMalloc1(ni+1,&xadj);CHKERRQ(ierr); 65 66 red = fread(xadj, sizeof(idx_t), ni+1, fp);if (red != (size_t) (ni+1)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"Unable to read from data file"); 67 68 ierr = PetscMalloc1(xadj[ni],&adjncy);CHKERRQ(ierr); 69 70 for (i=0; i<ni; i++) { 71 red = fread(&adjncy[xadj[i]], sizeof(idx_t), xadj[i+1]-xadj[i], fp);if (red != (size_t) (xadj[i+1]-xadj[i])) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"Unable to read from data file"); 72 } 73 74 ierr = PetscFClose(PETSC_COMM_SELF,fp);CHKERRQ(ierr); 75 76 ierr = PetscSNPrintf(fname,sizeof(fname),"%s.%d.graph.xyz",prefix,rank);CHKERRQ(ierr); 77 ierr = PetscFOpen(PETSC_COMM_SELF,fname,"r",&fp);CHKERRQ(ierr); 78 79 ierr = PetscMalloc3(ni*ndims,&xyz,ni,&part,size,&tpwgts);CHKERRQ(ierr); 80 ierr = PetscMalloc1(ni*ndims,&sxyz);CHKERRQ(ierr); 81 82 red = fread(xyz, sizeof(PetscReal), ndims*ni, fp);if (red != (size_t) (ndims*ni)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"Unable to read from data file"); 83 for (i=0; i<ni*ndims; i++) sxyz[i] = (size_t) xyz[i]; 84 85 ierr = PetscFClose(PETSC_COMM_SELF,fp);CHKERRQ(ierr); 86 87 vwgt = NULL; 88 89 for (i = 0; i < size; i++) tpwgts[i] = 1. / size; 90 isize = size; 91 92 ubvec[0] = 1.05; 93 options[0] = 0; 94 options[1] = 2; 95 options[2] = 15; 96 options[3] = 0; 97 options[4] = 0; 98 99 ierr = MPI_Comm_dup(MPI_COMM_WORLD, &comm);CHKERRQ(ierr); 100 status = ParMETIS_V3_PartGeomKway(vtxdist, xadj, adjncy, vwgt, NULL, &wgtflag, &numflag, &ndims, sxyz, &ncon, &isize, tpwgts, ubvec,options, &edgecut, part, &comm);CHKERRQPARMETIS(status); 101 ierr = MPI_Comm_free(&comm);CHKERRQ(ierr); 102 103 ierr = PetscFree(vtxdist);CHKERRQ(ierr); 104 ierr = PetscFree(xadj);CHKERRQ(ierr); 105 ierr = PetscFree(adjncy);CHKERRQ(ierr); 106 ierr = PetscFree3(xyz,part,tpwgts);CHKERRQ(ierr); 107 ierr = PetscFree(sxyz);CHKERRQ(ierr); 108 ierr = PetscFinalize(); 109 return ierr; 110 } 111 112 113 /*TEST 114 115 build: 116 requires: parmetis 117 118 test: 119 nsize: 2 120 requires: parmetis datafilespath !complex double !define(PETSC_USE_64BIT_INDICES) 121 args: -prefix ${DATAFILESPATH}/parmetis-test/testnp2 122 123 test: 124 suffix: 2 125 nsize: 4 126 requires: parmetis datafilespath !complex double !define(PETSC_USE_64BIT_INDICES) 127 args: -prefix ${DATAFILESPATH}/parmetis-test/testnp4 128 129 TEST*/ 130