1 2 static char help[] = "Tests MatSeqAIJKron.\n\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **argv) 7 { 8 Mat A,B,C,K,Ad,Bd; 9 const PetscScalar *Bv; 10 PetscInt n = 10, m = 20, p = 7, q = 17; 11 PetscBool flg; 12 PetscErrorCode ierr; 13 14 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 15 ierr = MatCreateDense(PETSC_COMM_SELF,m,n,m,n,NULL,&Ad);CHKERRQ(ierr); 16 ierr = MatCreateDense(PETSC_COMM_SELF,p,q,p,q,NULL,&Bd);CHKERRQ(ierr); 17 ierr = MatSetRandom(Ad,NULL);CHKERRQ(ierr); 18 ierr = MatSetRandom(Bd,NULL);CHKERRQ(ierr); 19 ierr = MatChop(Ad,0.2);CHKERRQ(ierr); 20 ierr = MatChop(Bd,0.2);CHKERRQ(ierr); 21 ierr = MatConvert(Ad,MATAIJ,MAT_INITIAL_MATRIX,&A);CHKERRQ(ierr); 22 ierr = MatConvert(Bd,MATAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 23 ierr = MatSeqAIJKron(A,B,MAT_INITIAL_MATRIX,&C);CHKERRQ(ierr); 24 ierr = MatViewFromOptions(A,NULL,"-A_view");CHKERRQ(ierr); 25 ierr = MatViewFromOptions(B,NULL,"-B_view");CHKERRQ(ierr); 26 ierr = MatViewFromOptions(C,NULL,"-C_view");CHKERRQ(ierr); 27 ierr = MatDenseGetArrayRead(Bd,&Bv);CHKERRQ(ierr); 28 ierr = MatCreateKAIJ(A,p,q,NULL,Bv,&K);CHKERRQ(ierr); 29 ierr = MatDenseRestoreArrayRead(Bd,&Bv);CHKERRQ(ierr); 30 ierr = MatMultEqual(C,K,10,&flg);CHKERRQ(ierr); 31 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"K*x != C*x"); 32 ierr = MatScale(A,1.3);CHKERRQ(ierr); 33 ierr = MatScale(B,0.3);CHKERRQ(ierr); 34 ierr = MatScale(Bd,0.3);CHKERRQ(ierr); 35 ierr = MatSeqAIJKron(A,B,MAT_REUSE_MATRIX,&C);CHKERRQ(ierr); 36 ierr = MatDenseGetArrayRead(Bd,&Bv);CHKERRQ(ierr); 37 ierr = MatKAIJSetT(K,p,q,Bv);CHKERRQ(ierr); 38 ierr = MatDenseRestoreArrayRead(Bd,&Bv);CHKERRQ(ierr); 39 ierr = MatMultEqual(C,K,10,&flg);CHKERRQ(ierr); 40 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"K*x != C*x"); 41 ierr = MatDestroy(&K);CHKERRQ(ierr); 42 ierr = MatDestroy(&C);CHKERRQ(ierr); 43 ierr = MatDestroy(&B);CHKERRQ(ierr); 44 ierr = MatDestroy(&A);CHKERRQ(ierr); 45 ierr = MatDestroy(&Bd);CHKERRQ(ierr); 46 ierr = MatDestroy(&Ad);CHKERRQ(ierr); 47 ierr = PetscFinalize(); 48 return ierr; 49 } 50 51 /*TEST 52 53 test: 54 suffix: 1 55 nsize: 1 56 output_file: output/ex101.out 57 58 TEST*/ 59