Actual source code: ex198.c

petsc-3.10.5 2019-03-28
Report Typos and Errors
  1: static char help[] = "Test MatMatMatMult\n\
  2: Reads PETSc matrix A B and C, then comput D=A*B*C \n\
  3: Input parameters include\n\
  4:   -fA <input_file> -fB <input_file> -fC <input_file> \n\n";

  6:  #include <petscmat.h>

  8: int main(int argc,char **args)
  9: {
 10:   Mat            A,B,C,D,BC,ABC;
 11:   PetscViewer    fd;
 12:   char           file[3][PETSC_MAX_PATH_LEN];
 13:   PetscBool      flg;

 16:   PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr;
 17:   /* read matrices A, B and C */
 18:   PetscOptionsGetString(NULL,NULL,"-fA",file[0],PETSC_MAX_PATH_LEN,&flg);
 19:   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_USER,"Must indicate binary file with the -fA options");

 21:   PetscOptionsGetString(NULL,NULL,"-fB",file[1],PETSC_MAX_PATH_LEN,&flg);
 22:   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_USER,"Must indicate binary file with the -fB options");

 24:   PetscOptionsGetString(NULL,NULL,"-fC",file[2],PETSC_MAX_PATH_LEN,&flg);
 25:   if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_USER,"Must indicate binary file with the -fC options");

 27:   /* Load matrices */
 28:   PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[0],FILE_MODE_READ,&fd);
 29:   MatCreate(PETSC_COMM_WORLD,&A);
 30:   MatLoad(A,fd);
 31:   PetscViewerDestroy(&fd);

 33:   PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[1],FILE_MODE_READ,&fd);
 34:   MatCreate(PETSC_COMM_WORLD,&B);
 35:   MatLoad(B,fd);
 36:   PetscViewerDestroy(&fd);

 38:   PetscViewerBinaryOpen(PETSC_COMM_WORLD,file[2],FILE_MODE_READ,&fd);
 39:   MatCreate(PETSC_COMM_WORLD,&C);
 40:   MatLoad(C,fd);
 41:   PetscViewerDestroy(&fd);

 43:   /* Test MatMatMult() */
 44:   MatMatMult(B,C,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&BC);
 45:   MatMatMult(A,BC,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&ABC);

 47:   MatMatMatMult(A,B,C,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&D);
 48:   MatMatMatMult(A,B,C,MAT_REUSE_MATRIX,PETSC_DEFAULT,&D);
 49:   /* MatView(D,PETSC_VIEWER_STDOUT_WORLD); */

 51:   MatEqual(ABC,D,&flg);
 52:   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"ABC != D");
 53: 
 54:   MatDestroy(&ABC);
 55:   MatDestroy(&BC);
 56:   MatDestroy(&D);
 57:   MatDestroy(&C);
 58:   MatDestroy(&B);
 59:   MatDestroy(&A);
 60:   PetscFinalize();
 61:   return ierr;
 62: }

 64: /*TEST

 66:    test:
 67:       requires: datafilespath !complex double !define(PETSC_USE_64BIT_INDICES)
 68:       args: -fA ${DATAFILESPATH}/matrices/matmatmatmult/A.bin -fB ${DATAFILESPATH}/matrices/matmatmatmult/B.bin -fC ${DATAFILESPATH}/matrices/matmatmatmult/C.bin
 69:       output_file: output/ex198.out

 71:    test:
 72:       suffix: 2
 73:       nsize: 3
 74:       requires: datafilespah !complex double !define(PETSC_USE_64BIT_INDICES)
 75:       args: -fA ${DATAFILESPATH}/matrices/matmatmatmult/A.bin -fB ${DATAFILESPATH}/matrices/matmatmatmult/B.bin -fC ${DATAFILESPATH}/matrices/matmatmatmult/C.bin
 76:       output_file: output/ex198.out

 78: TEST*/