1 static char help[] = "Test MatMultHermitianTranspose() and MatMultHermitianTransposeAdd().\n\n"; 2 3 #include <petscmat.h> 4 5 int main(int argc,char **args) 6 { 7 Mat A; 8 Vec x,y,ys; 9 PetscErrorCode ierr; 10 PetscInt i,j; 11 PetscScalar v; 12 13 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 14 ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 15 ierr = MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,2,2);CHKERRQ(ierr); 16 ierr = MatSetType(A,MATAIJ);CHKERRQ(ierr); 17 ierr = MatSetFromOptions(A);CHKERRQ(ierr); 18 ierr = MatSetUp(A);CHKERRQ(ierr); 19 20 i = 0; j = 0; v = 2.0; 21 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 22 i = 0; j = 1; v = 3.0 + 4.0*PETSC_i; 23 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 24 i = 1; j = 0; v = 5.0 + 6.0*PETSC_i; 25 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 26 i = 1; j = 1; v = 7.0 + 8.0*PETSC_i; 27 ierr = MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr); 28 ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 29 ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 30 31 /* Create vectors */ 32 ierr = VecCreate(PETSC_COMM_WORLD,&y);CHKERRQ(ierr); 33 ierr = VecSetSizes(y,PETSC_DECIDE,2);CHKERRQ(ierr); 34 ierr = VecSetFromOptions(y);CHKERRQ(ierr); 35 ierr = VecDuplicate(y,&ys);CHKERRQ(ierr); 36 ierr = VecDuplicate(y,&x);CHKERRQ(ierr); 37 38 i = 0; v = 10.0 + 11.0*PETSC_i; 39 ierr = VecSetValues(x,1,&i,&v,INSERT_VALUES);CHKERRQ(ierr); 40 i = 1; v = 100.0 + 120.0*PETSC_i; 41 ierr = VecSetValues(x,1,&i,&v,INSERT_VALUES);CHKERRQ(ierr); 42 ierr = VecAssemblyBegin(x);CHKERRQ(ierr); 43 ierr = VecAssemblyEnd(x);CHKERRQ(ierr); 44 45 ierr = MatMultHermitianTranspose(A,x,y);CHKERRQ(ierr); 46 ierr = VecView(y,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 47 ierr = MatMultHermitianTransposeAdd(A,x,y,ys);CHKERRQ(ierr); 48 ierr = VecView(ys,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 49 50 ierr = MatDestroy(&A);CHKERRQ(ierr); 51 52 ierr = VecDestroy(&x);CHKERRQ(ierr); 53 ierr = VecDestroy(&y);CHKERRQ(ierr); 54 ierr = VecDestroy(&ys);CHKERRQ(ierr); 55 ierr = PetscFinalize(); 56 return ierr; 57 } 58 59 60 /*TEST 61 62 build: 63 requires: complex 64 test: 65 66 test: 67 suffix: 2 68 nsize: 2 69 70 TEST*/ 71