1 2 static char help[] = "Tests MatGetColumnNorms() for matrix read from file."; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **args) 7 { 8 Mat A; 9 PetscErrorCode ierr; 10 PetscReal *norms; 11 char file[PETSC_MAX_PATH_LEN]; 12 PetscBool flg; 13 PetscViewer fd; 14 PetscInt n; 15 PetscMPIInt rank; 16 17 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 18 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 19 ierr = PetscOptionsGetString(NULL,NULL,"-f",file,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr); 20 if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -f option"); 21 ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&fd);CHKERRQ(ierr); 22 ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 23 ierr = MatSetFromOptions(A);CHKERRQ(ierr); 24 ierr = MatLoad(A,fd);CHKERRQ(ierr); 25 ierr = PetscViewerDestroy(&fd);CHKERRQ(ierr); 26 27 ierr = MatGetSize(A,NULL,&n);CHKERRQ(ierr); 28 ierr = PetscMalloc1(n,&norms);CHKERRQ(ierr); 29 30 ierr = MatGetColumnNorms(A,NORM_2,norms);CHKERRQ(ierr); 31 if (!rank) { 32 ierr = PetscPrintf(PETSC_COMM_SELF,"NORM_2:\n");CHKERRQ(ierr); 33 ierr = PetscRealView(n,norms,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); 34 } 35 36 ierr = MatGetColumnNorms(A,NORM_1,norms);CHKERRQ(ierr); 37 if (!rank) { 38 ierr = PetscPrintf(PETSC_COMM_SELF,"NORM_1:\n");CHKERRQ(ierr); 39 ierr = PetscRealView(n,norms,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); 40 } 41 42 ierr = MatGetColumnNorms(A,NORM_INFINITY,norms);CHKERRQ(ierr); 43 if (!rank) { 44 ierr = PetscPrintf(PETSC_COMM_SELF,"NORM_INFINITY:\n");CHKERRQ(ierr); 45 ierr = PetscRealView(n,norms,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); 46 } 47 48 ierr = PetscFree(norms);CHKERRQ(ierr); 49 ierr = MatDestroy(&A);CHKERRQ(ierr); 50 ierr = PetscFinalize(); 51 return ierr; 52 } 53 54 /*TEST 55 56 test: 57 suffix: 1 58 nsize: 2 59 requires: datafilespath !complex double !define(PETSC_USE_64BIT_INDICES) 60 args: -f ${DATAFILESPATH}/matrices/small -mat_type aij 61 output_file: output/ex138.out 62 63 test: 64 suffix: 2 65 nsize: 2 66 requires: datafilespath !complex double !define(PETSC_USE_64BIT_INDICES) 67 args: -f ${DATAFILESPATH}/matrices/small -mat_type baij -matload_block_size {{2 3}} 68 output_file: output/ex138.out 69 70 TEST*/ 71