Actual source code: ex122.c

  1: static char help[] = "Test MatMatMult() for AIJ and Dense matrices.\n\n";

  3: #include <petscmat.h>

  5: int main(int argc, char **argv)
  6: {
  7:   Mat         A, B, C;
  8:   PetscInt    M = 10, N = 5;
  9:   PetscRandom r;
 10:   PetscBool   equal = PETSC_FALSE;
 11:   PetscReal   fill  = 1.0;
 12:   PetscInt    nza, am, an;

 14:   PetscFunctionBeginUser;
 15:   PetscCall(PetscInitialize(&argc, &argv, NULL, help));
 16:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL));
 17:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL));
 18:   PetscCall(PetscOptionsGetReal(NULL, NULL, "-fill", &fill, NULL));

 20:   PetscCall(PetscRandomCreate(PETSC_COMM_WORLD, &r));
 21:   PetscCall(PetscRandomSetFromOptions(r));

 23:   /* create a aij matrix A */
 24:   PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
 25:   PetscCall(MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, N, M));
 26:   PetscCall(MatSetType(A, MATAIJ));
 27:   nza = (PetscInt)(.3 * M); /* num of nonzeros in each row of A */
 28:   PetscCall(MatSeqAIJSetPreallocation(A, nza, NULL));
 29:   PetscCall(MatMPIAIJSetPreallocation(A, nza, NULL, nza, NULL));
 30:   PetscCall(MatSetRandom(A, r));

 32:   /* create a dense matrix B */
 33:   PetscCall(MatGetLocalSize(A, &am, &an));
 34:   PetscCall(MatCreate(PETSC_COMM_WORLD, &B));
 35:   PetscCall(MatSetSizes(B, PETSC_DECIDE, am, N, PETSC_DECIDE));
 36:   PetscCall(MatSetType(B, MATDENSE));
 37:   PetscCall(MatSeqDenseSetPreallocation(B, NULL));
 38:   PetscCall(MatMPIDenseSetPreallocation(B, NULL));
 39:   PetscCall(MatSetRandom(B, r));
 40:   PetscCall(PetscRandomDestroy(&r));

 42:   /* Test MatMatMult() */
 43:   PetscCall(MatMatMult(B, A, MAT_INITIAL_MATRIX, fill, &C));
 44:   PetscCall(MatMatMult(B, A, MAT_REUSE_MATRIX, fill, &C));
 45:   PetscCall(MatMatMultEqual(B, A, C, 10, &equal));
 46:   PetscCheck(equal, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "C != B*A");

 48:   PetscCall(MatDestroy(&C));
 49:   PetscCall(MatDestroy(&B));
 50:   PetscCall(MatDestroy(&A));
 51:   PetscCall(PetscFinalize());
 52:   return 0;
 53: }

 55: /*TEST

 57:    test:
 58:       output_file: output/ex122.out

 60: TEST*/