static char help[] = "Test scalability of PetscHSetIJ hash set.\n\n"; #include #include #include int main(int argc, char **argv) { PetscHSetIJ table; PetscInt N = 0, i, j, n; PetscHashIJKey key; PetscBool flag; PetscLogDouble t_add = 0; PetscLogDouble t_has = 0; PetscLogDouble t_del = 0; PetscErrorCode ierr; ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr; ierr = PetscOptionsGetInt(NULL,NULL,"-N",&N,NULL);CHKERRQ(ierr); ierr = PetscHSetIJCreate(&table);CHKERRQ(ierr); /* The following line silences warnings from Clang Static Analyzer */ ierr = PetscHSetIJResize(table,0);CHKERRQ(ierr); ierr = PetscTimeSubtract(&t_add);CHKERRQ(ierr); for (i = 0; i < N; ++i) { for (j = 0; j < N; ++j) { key.i = PetscMin(i, j); key.j = PetscMax(i, j); ierr = PetscHSetIJQueryAdd(table, key, &flag);CHKERRQ(ierr); } } ierr = PetscTimeAdd(&t_add);CHKERRQ(ierr); ierr = PetscHSetIJGetSize(table,&n);CHKERRQ(ierr); ierr = PetscTimeSubtract(&t_has);CHKERRQ(ierr); for (i = 0; i < N; ++i) { for (j = 0; j < N; ++j) { key.i = i; key.j = j; ierr = PetscHSetIJHas(table, key, &flag);CHKERRQ(ierr); } } ierr = PetscTimeAdd(&t_has);CHKERRQ(ierr); ierr = PetscTimeSubtract(&t_del);CHKERRQ(ierr); for (i = 0; i < N; ++i) { for (j = 0; j < N; ++j) { key.i = i; key.j = j; ierr = PetscHSetIJQueryDel(table, key, &flag);CHKERRQ(ierr); } } ierr = PetscTimeAdd(&t_del);CHKERRQ(ierr); ierr = PetscPrintf(PETSC_COMM_WORLD,"N = %D - table size: %D, add: %g, has: %g, del: %g\n",N,n,t_add,t_has,t_del);CHKERRQ(ierr); ierr = PetscHSetIJDestroy(&table);CHKERRQ(ierr); ierr = PetscFinalize(); return ierr; } /*TEST test: args: -N 32 TEST*/