1da81f932SPierre Jolivet static char help[] = "Tests the different MatColoring implementations and ISColoringTestValid() \n\ 2da81f932SPierre Jolivet Modified from the code contributed by Ali Berk Kahraman. \n\n"; 3c4762a1bSJed Brown #include <petscmat.h> 4c4762a1bSJed Brown 5d71ae5a4SJacob Faibussowitsch PetscErrorCode FormJacobian(Mat A) 6d71ae5a4SJacob Faibussowitsch { 7c4762a1bSJed Brown PetscInt M, ownbegin, ownend, i, j; 8c4762a1bSJed Brown PetscScalar dummy = 0.0; 9c4762a1bSJed Brown 10c4762a1bSJed Brown PetscFunctionBeginUser; 119566063dSJacob Faibussowitsch PetscCall(MatGetSize(A, &M, NULL)); 129566063dSJacob Faibussowitsch PetscCall(MatGetOwnershipRange(A, &ownbegin, &ownend)); 13c4762a1bSJed Brown 14c4762a1bSJed Brown for (i = ownbegin; i < ownend; i++) { 15c4762a1bSJed Brown for (j = i - 3; j < i + 3; j++) { 1648a46eb9SPierre Jolivet if (j >= 0 && j < M) PetscCall(MatSetValues(A, 1, &i, 1, &j, &dummy, INSERT_VALUES)); 17c4762a1bSJed Brown } 18c4762a1bSJed Brown } 199566063dSJacob Faibussowitsch PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY)); 209566063dSJacob Faibussowitsch PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY)); 213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 22c4762a1bSJed Brown } 23c4762a1bSJed Brown 24d71ae5a4SJacob Faibussowitsch int main(int argc, char *argv[]) 25d71ae5a4SJacob Faibussowitsch { 26c4762a1bSJed Brown Mat J; 27c4762a1bSJed Brown PetscMPIInt size; 28c4762a1bSJed Brown PetscInt M = 8; 29c4762a1bSJed Brown ISColoring iscoloring; 30c4762a1bSJed Brown MatColoring coloring; 31c4762a1bSJed Brown 32327415f7SBarry Smith PetscFunctionBeginUser; 33*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &argv, NULL, help)); 349566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); 35c4762a1bSJed Brown 369566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &J)); 379566063dSJacob Faibussowitsch PetscCall(MatSetSizes(J, PETSC_DECIDE, PETSC_DECIDE, M, M)); 389566063dSJacob Faibussowitsch PetscCall(MatSetFromOptions(J)); 399566063dSJacob Faibussowitsch PetscCall(MatSetUp(J)); 40c4762a1bSJed Brown 419566063dSJacob Faibussowitsch PetscCall(FormJacobian(J)); 429566063dSJacob Faibussowitsch PetscCall(MatView(J, PETSC_VIEWER_STDOUT_WORLD)); 43c4762a1bSJed Brown 44c4762a1bSJed Brown /* 45c4762a1bSJed Brown Color the matrix, i.e. determine groups of columns that share no common 46c4762a1bSJed Brown rows. These columns in the Jacobian can all be computed simultaneously. 47c4762a1bSJed Brown */ 489566063dSJacob Faibussowitsch PetscCall(MatColoringCreate(J, &coloring)); 499566063dSJacob Faibussowitsch PetscCall(MatColoringSetType(coloring, MATCOLORINGGREEDY)); 509566063dSJacob Faibussowitsch PetscCall(MatColoringSetFromOptions(coloring)); 519566063dSJacob Faibussowitsch PetscCall(MatColoringApply(coloring, &iscoloring)); 52c4762a1bSJed Brown 5348a46eb9SPierre Jolivet if (size == 1) PetscCall(MatISColoringTest(J, iscoloring)); 54c4762a1bSJed Brown 559566063dSJacob Faibussowitsch PetscCall(ISColoringDestroy(&iscoloring)); 569566063dSJacob Faibussowitsch PetscCall(MatColoringDestroy(&coloring)); 579566063dSJacob Faibussowitsch PetscCall(MatDestroy(&J)); 589566063dSJacob Faibussowitsch PetscCall(PetscFinalize()); 59b122ec5aSJacob Faibussowitsch return 0; 60c4762a1bSJed Brown } 61c4762a1bSJed Brown 62c4762a1bSJed Brown /*TEST 63c4762a1bSJed Brown 64c4762a1bSJed Brown test: 65c4762a1bSJed Brown suffix: sl 66dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 67c4762a1bSJed Brown args: -mat_coloring_type sl 68c4762a1bSJed Brown output_file: output/ex24_1.out 69c4762a1bSJed Brown 70c4762a1bSJed Brown test: 71c4762a1bSJed Brown suffix: lf 72dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 73c4762a1bSJed Brown args: -mat_coloring_type lf 74c4762a1bSJed Brown output_file: output/ex24_1.out 75c4762a1bSJed Brown 76c4762a1bSJed Brown test: 77c4762a1bSJed Brown suffix: id 78dfd57a17SPierre Jolivet requires: !complex double !defined(PETSC_USE_64BIT_INDICES) 79c4762a1bSJed Brown args: -mat_coloring_type id 80c4762a1bSJed Brown output_file: output/ex24_1.out 81c4762a1bSJed Brown 82c4762a1bSJed Brown TEST*/ 83