1 2 static char help[] = "Test MatAXPY and SUBSET_NONZERO_PATTERN [-different] [-skip]\n by default subset pattern is used \n\n"; 3 4 /* A test contributed by Jose E. Roman, Oct. 2014 */ 5 6 #include <petscmat.h> 7 8 int main(int argc,char **args) 9 { 10 PetscErrorCode ierr; 11 Mat A,B,C; 12 PetscBool different=PETSC_FALSE,skip=PETSC_FALSE; 13 PetscInt m0,m1,n=128,i; 14 15 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 16 ierr = PetscOptionsGetBool(NULL,NULL,"-different",&different,NULL);CHKERRQ(ierr); 17 ierr = PetscOptionsGetBool(NULL,NULL,"-skip",&skip,NULL);CHKERRQ(ierr); 18 /* 19 Create matrices 20 A = tridiag(1,-2,1) and B = diag(7); 21 */ 22 ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 23 ierr = MatCreate(PETSC_COMM_WORLD,&B);CHKERRQ(ierr); 24 ierr = MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr); 25 ierr = MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr); 26 ierr = MatSetFromOptions(A);CHKERRQ(ierr); 27 ierr = MatSetFromOptions(B);CHKERRQ(ierr); 28 ierr = MatSetUp(A);CHKERRQ(ierr); 29 ierr = MatSetUp(B);CHKERRQ(ierr); 30 ierr = MatGetOwnershipRange(A,&m0,&m1);CHKERRQ(ierr); 31 for (i=m0;i<m1;i++) { 32 if (i>0) { ierr = MatSetValue(A,i,i-1,-1.0,INSERT_VALUES);CHKERRQ(ierr); } 33 if (i<n-1) { ierr = MatSetValue(A,i,i+1,-1.0,INSERT_VALUES);CHKERRQ(ierr); } 34 ierr = MatSetValue(A,i,i,2.0,INSERT_VALUES);CHKERRQ(ierr); 35 ierr = MatSetValue(B,i,i,7.0,INSERT_VALUES);CHKERRQ(ierr); 36 } 37 ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 38 ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 39 ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 40 ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 41 42 ierr = MatDuplicate(A,MAT_COPY_VALUES,&C);CHKERRQ(ierr); 43 /* Add B */ 44 ierr = MatAXPY(C,1.0,B,(different)?DIFFERENT_NONZERO_PATTERN:SUBSET_NONZERO_PATTERN);CHKERRQ(ierr); 45 /* Add A */ 46 if (!skip) { ierr = MatAXPY(C,1.0,A,SUBSET_NONZERO_PATTERN);CHKERRQ(ierr); } 47 48 /* 49 Free memory 50 */ 51 ierr = MatDestroy(&A);CHKERRQ(ierr); 52 ierr = MatDestroy(&B);CHKERRQ(ierr); 53 ierr = MatDestroy(&C);CHKERRQ(ierr); 54 ierr = PetscFinalize(); 55 return ierr; 56 } 57 58 /*TEST 59 60 test: 61 nsize: 4 62 output_file: output/ex172.out 63 64 test: 65 suffix: 2 66 nsize: 4 67 args: -different 68 output_file: output/ex172.out 69 70 test: 71 suffix: 3 72 nsize: 4 73 args: -skip 74 output_file: output/ex172.out 75 76 test: 77 suffix: 4 78 nsize: 4 79 args: -different -skip 80 output_file: output/ex172.out 81 82 test: 83 suffix: baij 84 args: -mat_type baij> ex172.tmp 2>&1 85 output_file: output/ex172.out 86 87 test: 88 suffix: mpibaij 89 nsize: 4 90 args: -mat_type baij> ex172.tmp 2>&1 91 output_file: output/ex172.out 92 93 test: 94 suffix: mpisbaij 95 nsize: 4 96 args: -mat_type sbaij> ex172.tmp 2>&1 97 output_file: output/ex172.out 98 99 test: 100 suffix: sbaij 101 args: -mat_type sbaij> ex172.tmp 2>&1 102 output_file: output/ex172.out 103 104 TEST*/ 105