1c4762a1bSJed Brown static char help[] = "Tests sequential and parallel MatMatMult() and MatAXPY(...,SUBSET_NONZERO_PATTERN) \n\
2c4762a1bSJed Brown Input arguments are:\n\
3c4762a1bSJed Brown -f <input_file> : file to load\n\n";
4c4762a1bSJed Brown /* e.g., mpiexec -n 3 ./ex113 -f <file> */
5c4762a1bSJed Brown
6c4762a1bSJed Brown #include <petscmat.h>
7c4762a1bSJed Brown
main(int argc,char ** args)8d71ae5a4SJacob Faibussowitsch int main(int argc, char **args)
9d71ae5a4SJacob Faibussowitsch {
10c4762a1bSJed Brown Mat A, A1, A2, Mtmp, dstMat;
11c4762a1bSJed Brown PetscViewer viewer;
12c4762a1bSJed Brown PetscReal fill = 4.0;
13c4762a1bSJed Brown char file[128];
14c4762a1bSJed Brown PetscBool flg;
15c4762a1bSJed Brown
16327415f7SBarry Smith PetscFunctionBeginUser;
17*c8025a54SPierre Jolivet PetscCall(PetscInitialize(&argc, &args, NULL, help));
18c4762a1bSJed Brown /* Load the matrix A */
199566063dSJacob Faibussowitsch PetscCall(PetscOptionsGetString(NULL, NULL, "-f", file, sizeof(file), &flg));
2028b400f6SJacob Faibussowitsch PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must indicate a file name for matrix A with the -f option.");
21c4762a1bSJed Brown
229566063dSJacob Faibussowitsch PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, file, FILE_MODE_READ, &viewer));
239566063dSJacob Faibussowitsch PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
249566063dSJacob Faibussowitsch PetscCall(MatLoad(A, viewer));
259566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&viewer));
26c4762a1bSJed Brown
279566063dSJacob Faibussowitsch PetscCall(MatDuplicate(A, MAT_COPY_VALUES, &A1));
289566063dSJacob Faibussowitsch PetscCall(MatDuplicate(A, MAT_COPY_VALUES, &A2));
29c4762a1bSJed Brown
30c4762a1bSJed Brown /* dstMat = A*A1*A2 */
319566063dSJacob Faibussowitsch PetscCall(MatMatMult(A1, A2, MAT_INITIAL_MATRIX, fill, &Mtmp));
329566063dSJacob Faibussowitsch PetscCall(MatMatMult(A, Mtmp, MAT_INITIAL_MATRIX, fill, &dstMat));
339566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Mtmp));
34c4762a1bSJed Brown
35c4762a1bSJed Brown /* dstMat += A1*A2 */
369566063dSJacob Faibussowitsch PetscCall(MatMatMult(A1, A2, MAT_INITIAL_MATRIX, fill, &Mtmp));
379566063dSJacob Faibussowitsch PetscCall(MatAXPY(dstMat, 1.0, Mtmp, SUBSET_NONZERO_PATTERN));
389566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Mtmp));
39c4762a1bSJed Brown
40c4762a1bSJed Brown /* dstMat += A*A1 */
419566063dSJacob Faibussowitsch PetscCall(MatMatMult(A, A1, MAT_INITIAL_MATRIX, fill, &Mtmp));
429566063dSJacob Faibussowitsch PetscCall(MatAXPY(dstMat, 1.0, Mtmp, SUBSET_NONZERO_PATTERN));
439566063dSJacob Faibussowitsch PetscCall(MatDestroy(&Mtmp));
44c4762a1bSJed Brown
45c4762a1bSJed Brown /* dstMat += A */
469566063dSJacob Faibussowitsch PetscCall(MatAXPY(dstMat, 1.0, A, SUBSET_NONZERO_PATTERN));
47c4762a1bSJed Brown
489566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A));
499566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A1));
509566063dSJacob Faibussowitsch PetscCall(MatDestroy(&A2));
519566063dSJacob Faibussowitsch PetscCall(MatDestroy(&dstMat));
529566063dSJacob Faibussowitsch PetscCall(PetscFinalize());
53b122ec5aSJacob Faibussowitsch return 0;
54c4762a1bSJed Brown }
55