1 2 static char help[] = "Tests the different MatColoring implementatons.\n\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **args) 7 { 8 Mat C; 9 PetscViewer viewer; 10 char file[128]; 11 PetscBool flg; 12 MatColoring ctx; 13 ISColoring coloring; 14 PetscMPIInt size; 15 16 CHKERRQ(PetscInitialize(&argc,&args,(char*)0,help)); 17 CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 18 19 CHKERRQ(PetscOptionsGetString(NULL,NULL,"-f",file,sizeof(file),&flg)); 20 PetscCheck(flg,PETSC_COMM_WORLD,PETSC_ERR_USER,"Must use -f filename to load sparse matrix"); 21 CHKERRQ(PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&viewer)); 22 CHKERRQ(MatCreate(PETSC_COMM_WORLD,&C)); 23 CHKERRQ(MatLoad(C,viewer)); 24 CHKERRQ(PetscViewerDestroy(&viewer)); 25 26 CHKERRQ(MatColoringCreate(C,&ctx)); 27 CHKERRQ(MatColoringSetFromOptions(ctx)); 28 CHKERRQ(MatColoringApply(ctx,&coloring)); 29 CHKERRQ(MatColoringTest(ctx,coloring)); 30 if (size == 1) { 31 /* jp, power and greedy have bug -- need to be fixed */ 32 CHKERRQ(MatISColoringTest(C,coloring)); 33 } 34 35 /* Free data structures */ 36 CHKERRQ(ISColoringDestroy(&coloring)); 37 CHKERRQ(MatColoringDestroy(&ctx)); 38 CHKERRQ(MatDestroy(&C)); 39 CHKERRQ(PetscFinalize()); 40 return 0; 41 } 42 43 /*TEST 44 45 test: 46 nsize: {{3}} 47 requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES) 48 args: -f ${DATAFILESPATH}/matrices/arco1 -mat_coloring_type {{ jp power natural greedy}} -mat_coloring_distance {{ 1 2}} 49 50 test: 51 suffix: 2 52 nsize: {{1 2}} 53 requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES) 54 args: -f ${DATAFILESPATH}/matrices/arco1 -mat_coloring_type {{ sl lf id }} -mat_coloring_distance 2 55 output_file: output/ex199_1.out 56 57 TEST*/ 58