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 PetscCallPARMETIS(...) \ 20 do { \ 21 int metis_ierr = __VA_ARGS__; \ 22 PetscCheck(metis_ierr != METIS_ERROR_INPUT, PETSC_COMM_SELF, PETSC_ERR_LIB, "ParMETIS error due to wrong inputs and/or options"); \ 23 PetscCheck(metis_ierr != METIS_ERROR_MEMORY, PETSC_COMM_SELF, PETSC_ERR_LIB, "ParMETIS error due to insufficient memory"); \ 24 PetscCheck(metis_ierr != METIS_ERROR, PETSC_COMM_SELF, PETSC_ERR_LIB, "ParMETIS general error"); \ 25 } while (0) 26 27 int main(int argc, char *argv[]) 28 { 29 PetscBool flg; 30 PetscMPIInt rank, size; 31 idx_t ni, isize, *vtxdist, *xadj, *adjncy, *vwgt, *part; 32 idx_t wgtflag = 0, numflag = 0, ncon = 1, ndims = 3, edgecut = 0; 33 idx_t options[5]; 34 PetscReal *xyz; 35 real_t *sxyz, *tpwgts, ubvec[1]; 36 MPI_Comm comm; 37 FILE *fp; 38 char fname[PETSC_MAX_PATH_LEN], prefix[PETSC_MAX_PATH_LEN] = ""; 39 size_t red; 40 41 PetscFunctionBeginUser; 42 PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 43 #if defined(PETSC_USE_64BIT_INDICES) 44 PetscCall(PetscPrintf(PETSC_COMM_WORLD, "This example only works with 32-bit indices\n")); 45 PetscCall(PetscFinalize()); 46 return 0; 47 #endif 48 MPI_Comm_rank(PETSC_COMM_WORLD, &rank); 49 MPI_Comm_size(PETSC_COMM_WORLD, &size); 50 51 PetscOptionsBegin(PETSC_COMM_WORLD, NULL, "Parmetis test options", ""); 52 PetscCall(PetscOptionsString("-prefix", "Path and prefix of test file", "", prefix, prefix, sizeof(prefix), &flg)); 53 PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must specify -prefix"); 54 PetscOptionsEnd(); 55 56 PetscCall(PetscMalloc1(size + 1, &vtxdist)); 57 58 PetscCall(PetscSNPrintf(fname, sizeof(fname), "%s.%d.graph", prefix, rank)); 59 60 PetscCall(PetscFOpen(PETSC_COMM_SELF, fname, "r", &fp)); 61 62 red = fread(vtxdist, sizeof(idx_t), size + 1, fp); 63 PetscCheck(red == (size_t)(size + 1), PETSC_COMM_SELF, PETSC_ERR_SYS, "Unable to read from data file"); 64 65 ni = vtxdist[rank + 1] - vtxdist[rank]; 66 67 PetscCall(PetscMalloc1(ni + 1, &xadj)); 68 69 red = fread(xadj, sizeof(idx_t), ni + 1, fp); 70 PetscCheck(red == (size_t)(ni + 1), PETSC_COMM_SELF, PETSC_ERR_SYS, "Unable to read from data file"); 71 72 PetscCall(PetscMalloc1(xadj[ni], &adjncy)); 73 74 for (PetscInt i = 0; i < ni; i++) { 75 red = fread(&adjncy[xadj[i]], sizeof(idx_t), xadj[i + 1] - xadj[i], fp); 76 PetscCheck(red == (size_t)(xadj[i + 1] - xadj[i]), PETSC_COMM_SELF, PETSC_ERR_SYS, "Unable to read from data file"); 77 } 78 79 PetscCall(PetscFClose(PETSC_COMM_SELF, fp)); 80 81 PetscCall(PetscSNPrintf(fname, sizeof(fname), "%s.%d.graph.xyz", prefix, rank)); 82 PetscCall(PetscFOpen(PETSC_COMM_SELF, fname, "r", &fp)); 83 84 PetscCall(PetscMalloc3(ni * ndims, &xyz, ni, &part, size, &tpwgts)); 85 PetscCall(PetscMalloc1(ni * ndims, &sxyz)); 86 87 red = fread(xyz, sizeof(PetscReal), ndims * ni, fp); 88 PetscCheck(red == (size_t)(ndims * ni), PETSC_COMM_SELF, PETSC_ERR_SYS, "Unable to read from data file"); 89 for (PetscInt i = 0; i < ni * ndims; i++) sxyz[i] = (size_t)xyz[i]; 90 91 PetscCall(PetscFClose(PETSC_COMM_SELF, fp)); 92 93 vwgt = NULL; 94 95 for (PetscInt i = 0; i < size; i++) tpwgts[i] = 1. / size; 96 isize = size; 97 98 ubvec[0] = 1.05; 99 options[0] = 0; 100 options[1] = 2; 101 options[2] = 15; 102 options[3] = 0; 103 options[4] = 0; 104 105 PetscCallMPI(MPI_Comm_dup(MPI_COMM_WORLD, &comm)); 106 PetscCallPARMETIS(ParMETIS_V3_PartGeomKway(vtxdist, xadj, adjncy, vwgt, NULL, &wgtflag, &numflag, &ndims, sxyz, &ncon, &isize, tpwgts, ubvec, options, &edgecut, part, &comm)); 107 PetscCallMPI(MPI_Comm_free(&comm)); 108 109 PetscCall(PetscFree(vtxdist)); 110 PetscCall(PetscFree(xadj)); 111 PetscCall(PetscFree(adjncy)); 112 PetscCall(PetscFree3(xyz, part, tpwgts)); 113 PetscCall(PetscFree(sxyz)); 114 PetscCall(PetscFinalize()); 115 return 0; 116 } 117 118 /*TEST 119 120 build: 121 requires: parmetis 122 123 test: 124 nsize: 2 125 requires: parmetis datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES) 126 args: -prefix ${DATAFILESPATH}/parmetis-test/testnp2 127 128 test: 129 suffix: 2 130 nsize: 4 131 requires: parmetis datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES) 132 args: -prefix ${DATAFILESPATH}/parmetis-test/testnp4 133 134 TEST*/ 135