Actual source code: ex172.c


  2: static char help[] = "Test MatAXPY and SUBSET_NONZERO_PATTERN [-different] [-skip]\n by default subset pattern is used \n\n";

  4: /* A test contributed by Jose E. Roman, Oct. 2014 */

  6: #include <petscmat.h>

  8: int main(int argc, char **args)
  9: {
 10:   Mat       A, B, C;
 11:   PetscBool different = PETSC_FALSE, skip = PETSC_FALSE;
 12:   PetscInt  m0, m1, n = 128, i;

 14:   PetscFunctionBeginUser;
 15:   PetscCall(PetscInitialize(&argc, &args, (char *)0, help));
 16:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-different", &different, NULL));
 17:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-skip", &skip, NULL));
 18:   /*
 19:      Create matrices
 20:      A = tridiag(1,-2,1) and B = diag(7);
 21:   */
 22:   PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
 23:   PetscCall(MatCreate(PETSC_COMM_WORLD, &B));
 24:   PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, n, n));
 25:   PetscCall(MatSetSizes(B, PETSC_DECIDE, PETSC_DECIDE, n, n));
 26:   PetscCall(MatSetFromOptions(A));
 27:   PetscCall(MatSetFromOptions(B));
 28:   PetscCall(MatSetUp(A));
 29:   PetscCall(MatSetUp(B));
 30:   PetscCall(MatGetOwnershipRange(A, &m0, &m1));
 31:   for (i = m0; i < m1; i++) {
 32:     if (i > 0) PetscCall(MatSetValue(A, i, i - 1, -1.0, INSERT_VALUES));
 33:     if (i < n - 1) PetscCall(MatSetValue(A, i, i + 1, -1.0, INSERT_VALUES));
 34:     PetscCall(MatSetValue(A, i, i, 2.0, INSERT_VALUES));
 35:     PetscCall(MatSetValue(B, i, i, 7.0, INSERT_VALUES));
 36:   }
 37:   PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
 38:   PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
 39:   PetscCall(MatAssemblyBegin(B, MAT_FINAL_ASSEMBLY));
 40:   PetscCall(MatAssemblyEnd(B, MAT_FINAL_ASSEMBLY));

 42:   PetscCall(MatDuplicate(A, MAT_COPY_VALUES, &C));
 43:   /* Add B */
 44:   PetscCall(MatAXPY(C, 1.0, B, (different) ? DIFFERENT_NONZERO_PATTERN : SUBSET_NONZERO_PATTERN));
 45:   /* Add A */
 46:   if (!skip) PetscCall(MatAXPY(C, 1.0, A, SUBSET_NONZERO_PATTERN));

 48:   /*
 49:      Free memory
 50:   */
 51:   PetscCall(MatDestroy(&A));
 52:   PetscCall(MatDestroy(&B));
 53:   PetscCall(MatDestroy(&C));
 54:   PetscCall(PetscFinalize());
 55:   return 0;
 56: }

 58: /*TEST

 60:    test:
 61:       nsize: 4
 62:       output_file: output/ex172.out

 64:    test:
 65:       suffix: 2
 66:       nsize: 4
 67:       args: -different
 68:       output_file: output/ex172.out

 70:    test:
 71:       suffix: 3
 72:       nsize: 4
 73:       args: -skip
 74:       output_file: output/ex172.out

 76:    test:
 77:       suffix: 4
 78:       nsize: 4
 79:       args: -different -skip
 80:       output_file: output/ex172.out

 82:    test:
 83:       suffix: baij
 84:       args: -mat_type baij> ex172.tmp 2>&1
 85:       output_file: output/ex172.out

 87:    test:
 88:       suffix: mpibaij
 89:       nsize: 4
 90:       args: -mat_type baij> ex172.tmp 2>&1
 91:       output_file: output/ex172.out

 93:    test:
 94:       suffix: mpisbaij
 95:       nsize: 4
 96:       args: -mat_type sbaij> ex172.tmp 2>&1
 97:       output_file: output/ex172.out

 99:    test:
100:       suffix: sbaij
101:       args: -mat_type sbaij> ex172.tmp 2>&1
102:       output_file: output/ex172.out

104: TEST*/