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