Actual source code: ex232.c

  1: /* tests MatSeqSBAIJSetPreallocationCSR() and MatMPISBAIJSetPreallocationCSR() */

  3: #include <petsc.h>

  5: int main(int argc, char **args)
  6: {
  7:   PetscInt    ia[3] = {0, 2, 4};
  8:   PetscInt    ja[4] = {0, 1, 0, 1};
  9:   PetscScalar c[16] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
 10:   Mat         ssbaij, msbaij;
 11:   PetscBool   v2 = PETSC_FALSE;

 13:   PetscFunctionBeginUser;
 14:   PetscCall(PetscInitialize(&argc, &args, (char *)0, (char *)0));
 15:   PetscCall(PetscOptionsGetBool(NULL, NULL, "-v2", &v2, NULL));

 17:   PetscCall(MatCreate(PETSC_COMM_SELF, &ssbaij));
 18:   PetscCall(MatCreate(PETSC_COMM_SELF, &msbaij));
 19:   if (!v2) {
 20:     PetscCall(MatSetType(ssbaij, MATSEQSBAIJ));
 21:     PetscCall(MatSetType(msbaij, MATMPISBAIJ));
 22:   }
 23:   PetscCall(MatSetBlockSize(ssbaij, 2));
 24:   PetscCall(MatSetSizes(ssbaij, 4, 4, 4, 4));
 25:   PetscCall(MatSetBlockSize(msbaij, 2));
 26:   PetscCall(MatSetSizes(msbaij, 4, 4, 4, 4));
 27:   if (v2) {
 28:     PetscCall(MatSetUp(ssbaij));
 29:     PetscCall(MatSetUp(msbaij));
 30:     PetscCall(MatSetType(ssbaij, MATSEQSBAIJ));
 31:     PetscCall(MatSetType(msbaij, MATMPISBAIJ));
 32:   }
 33:   PetscCall(MatSeqSBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c));
 34:   PetscCall(MatMPISBAIJSetPreallocationCSR(msbaij, 2, ia, ja, c));
 35:   PetscCall(MatView(ssbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF)));
 36:   PetscCall(MatView(msbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF)));
 37:   PetscCall(MatDestroy(&ssbaij));
 38:   PetscCall(MatDestroy(&msbaij));
 39:   PetscCall(PetscFinalize());
 40:   return 0;
 41: }

 43: /*TEST

 45:    test:
 46:      filter: sed "s?\.??g"

 48:    test:
 49:      suffix: 2
 50:      filter: sed "s?\.??g"
 51:      args : -v2
 52:      output_file: output/ex232_1.out

 54: TEST*/