1c4762a1bSJed Brown static char help[] = "Tests various routines in MatMAIJ format.\n";
2c4762a1bSJed Brown
3c4762a1bSJed Brown #include <petscmat.h>
4c4762a1bSJed Brown #define IMAX 15
main(int argc,char ** args)5d71ae5a4SJacob Faibussowitsch int main(int argc, char **args)
6d71ae5a4SJacob Faibussowitsch {
7c4762a1bSJed Brown Mat A, B, MA;
8c4762a1bSJed Brown PetscViewer fd;
9c4762a1bSJed Brown char file[PETSC_MAX_PATH_LEN];
10c4762a1bSJed Brown PetscInt m, n, M, N, dof = 1;
11c4762a1bSJed Brown PetscMPIInt rank, size;
12c4762a1bSJed Brown PetscBool flg;
13c4762a1bSJed Brown
14327415f7SBarry Smith PetscFunctionBeginUser;
15c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &args, NULL, help));
169566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
179566063dSJacob Faibussowitsch PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
18c4762a1bSJed Brown
19c4762a1bSJed Brown /* Load aij matrix A */
209566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetString(NULL, NULL, "-f", file, sizeof(file), &flg));
2128b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must indicate binary file with the -f option");
229566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, file, FILE_MODE_READ, &fd));
239566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
249566063dSJacob Faibussowitsch PetscCall(MatLoad(A, fd));
259566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&fd));
26c4762a1bSJed Brown
27c4762a1bSJed Brown /* Get dof, then create maij matrix MA */
289566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL));
299566063dSJacob Faibussowitsch PetscCall(MatCreateMAIJ(A, dof, &MA));
309566063dSJacob Faibussowitsch PetscCall(MatGetLocalSize(MA, &m, &n));
319566063dSJacob Faibussowitsch PetscCall(MatGetSize(MA, &M, &N));
32c4762a1bSJed Brown
33c4762a1bSJed Brown if (size == 1) {
349566063dSJacob Faibussowitsch PetscCall(MatConvert(MA, MATSEQAIJ, MAT_INITIAL_MATRIX, &B));
35c4762a1bSJed Brown } else {
369566063dSJacob Faibussowitsch PetscCall(MatConvert(MA, MATMPIAIJ, MAT_INITIAL_MATRIX, &B));
37c4762a1bSJed Brown }
38c4762a1bSJed Brown
39c4762a1bSJed Brown /* Test MatMult() */
409566063dSJacob Faibussowitsch PetscCall(MatMultEqual(MA, B, 10, &flg));
4128b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_CONV_FAILED, "Error: MatMul() for MAIJ matrix");
42c4762a1bSJed Brown /* Test MatMultAdd() */
439566063dSJacob Faibussowitsch PetscCall(MatMultAddEqual(MA, B, 10, &flg));
4428b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_CONV_FAILED, "Error: MatMulAdd() for MAIJ matrix");
45c4762a1bSJed Brown
46c4762a1bSJed Brown /* Test MatMultTranspose() */
479566063dSJacob Faibussowitsch PetscCall(MatMultTransposeEqual(MA, B, 10, &flg));
4828b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_CONV_FAILED, "Error: MatMulAdd() for MAIJ matrix");
49c4762a1bSJed Brown
50c4762a1bSJed Brown /* Test MatMultTransposeAdd() */
519566063dSJacob Faibussowitsch PetscCall(MatMultTransposeAddEqual(MA, B, 10, &flg));
5228b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_CONV_FAILED, "Error: MatMulTransposeAdd() for MAIJ matrix");
53c4762a1bSJed Brown
549566063dSJacob Faibussowitsch PetscCall(MatDestroy(&MA));
559566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A));
569566063dSJacob Faibussowitsch PetscCall(MatDestroy(&B));
579566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
58b122ec5aSJacob Faibussowitsch return 0;
59c4762a1bSJed Brown }
60c4762a1bSJed Brown
61c4762a1bSJed Brown /*TEST
62c4762a1bSJed Brown
63c4762a1bSJed Brown build:
64c4762a1bSJed Brown requires: !complex
65c4762a1bSJed Brown
66c4762a1bSJed Brown test:
67c4762a1bSJed Brown nsize: {{1 3}}
68dfd57a17SPierre Jolivet requires: datafilespath !complex double !defined(PETSC_USE_64BIT_INDICES)
69c4762a1bSJed Brown args: -f ${DATAFILESPATH}/matrices/arco1 -dof {{1 2 3 4 5 6 8 9 16}} -viewer_binary_skip_info
70*3886731fSPierre Jolivet output_file: output/empty.out
71c4762a1bSJed Brown
72c4762a1bSJed Brown TEST*/
73