int main(int argc,char **args)
{
Mat A,B,MA;
PetscViewer fd;
char file[PETSC_MAX_PATH_LEN];
PetscInt m,n,M,N,dof=1;
PetscMPIInt rank,size;
PetscErrorCode ierr;
PetscBool flg;
PetscInitialize(&argc,&args,(char*)0,help);
ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
#if defined(PETSC_USE_COMPLEX)
SETERRQ(PETSC_COMM_WORLD,1,"This example does not work with complex numbers");
#else
/* Load aij matrix A */
ierr = PetscOptionsGetString(NULL,"-f",file,PETSC_MAX_PATH_LEN,&flg);CHKERRQ(ierr);
if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_USER,"Must indicate binary file with the -f option");
ierr = PetscViewerBinaryOpen(PETSC_COMM_WORLD,file,FILE_MODE_READ,&fd);CHKERRQ(ierr);
ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr);
ierr = MatLoad(A,fd);CHKERRQ(ierr);
ierr = PetscViewerDestroy(&fd);CHKERRQ(ierr);
/* Get dof, then create maij matrix MA */
ierr = PetscOptionsGetInt(NULL,"-dof",&dof,NULL);CHKERRQ(ierr);
ierr = MatCreateMAIJ(A,dof,&MA);CHKERRQ(ierr);
ierr = MatGetLocalSize(MA,&m,&n);CHKERRQ(ierr);
ierr = MatGetSize(MA,&M,&N);CHKERRQ(ierr);
if (size == 1) {
ierr = MatConvert(MA,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
} else {
ierr = MatConvert(MA,MATMPIAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
}
/* Test MatMult() */
ierr = MatMultEqual(MA,B,10,&flg);CHKERRQ(ierr);
if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_CONV_FAILED,"Error: MatMul() for MAIJ matrix");
/* Test MatMultAdd() */
ierr = MatMultAddEqual(MA,B,10,&flg);CHKERRQ(ierr);
if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_CONV_FAILED,"Error: MatMulAdd() for MAIJ matrix");
/* Test MatMultTranspose() */
ierr = MatMultTransposeEqual(MA,B,10,&flg);CHKERRQ(ierr);
if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_CONV_FAILED,"Error: MatMulAdd() for MAIJ matrix");
/* Test MatMultTransposeAdd() */
ierr = MatMultTransposeAddEqual(MA,B,10,&flg);CHKERRQ(ierr);
if (!flg) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_CONV_FAILED,"Error: MatMulTransposeAdd() for MAIJ matrix");
ierr = MatDestroy(&MA);CHKERRQ(ierr);
ierr = MatDestroy(&A);CHKERRQ(ierr);
ierr = MatDestroy(&B);CHKERRQ(ierr);
ierr = PetscFinalize();
#endif
return 0;
}
static PetscErrorCode MatHermitianTransposeMatMult_MPIAIJ_MPIDense0(Mat A,PetscScalar *Barray, PetscInt BN, PRIMME_INT ldb,PetscScalar *Carray, PetscInt ldc)
{
PetscErrorCode ierr;
PetscInt i,j,m=A->rmap->n,n=A->cmap->n;
PetscScalar *btarray,*ctarray;
Vec bt,ct;
Mat mA;
PetscFunctionBegin;
/* create MAIJ matrix mA from A -- should be done in symbolic phase */
ierr = MatCreateMAIJ(A,BN,&mA);CHKERRQ(ierr);
/* create vectors bt and ct to hold locally transposed arrays of B and C */
ierr = VecCreate(PetscObjectComm((PetscObject)A),&bt);CHKERRQ(ierr);
ierr = VecSetSizes(bt,m*BN,PETSC_DECIDE);CHKERRQ(ierr);
ierr = VecSetType(bt,VECSTANDARD);CHKERRQ(ierr);
ierr = VecCreate(PetscObjectComm((PetscObject)A),&ct);CHKERRQ(ierr);
ierr = VecSetSizes(ct,n*BN,PETSC_DECIDE);CHKERRQ(ierr);
ierr = VecSetType(ct,VECSTANDARD);CHKERRQ(ierr);
/* transpose local arry of B, then copy it to vector bt */
ierr = VecGetArray(bt,&btarray);CHKERRQ(ierr);
for (j=0; j<BN; j++) {
for (i=0; i<m; i++) btarray[i*BN + j] = PetscConj(Barray[ldb*j+i]);
}
ierr = VecRestoreArray(bt,&btarray);CHKERRQ(ierr);
/* compute ct = mA^T * cb */
ierr = MatMultTranspose(mA,bt,ct);CHKERRQ(ierr);
/* transpose local arry of ct to matrix C */
ierr = VecGetArray(ct,&ctarray);CHKERRQ(ierr);
for (j=0; j<BN; j++) {
for (i=0; i<n; i++) Carray[j*ldc+i] = PetscConj(ctarray[i*BN + j]);
}
ierr = VecRestoreArray(ct,&ctarray);CHKERRQ(ierr);
ierr = VecDestroy(&bt);CHKERRQ(ierr);
ierr = VecDestroy(&ct);CHKERRQ(ierr);
ierr = MatDestroy(&mA);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode MatTransposeMatMultNumeric_MPIAIJ_MPIDense(Mat A,Mat B,Mat C)
{
PetscErrorCode ierr;
PetscInt i,j,k,m=A->rmap->n,n=A->cmap->n,BN=B->cmap->N;
PetscScalar *Barray,*Carray,*btarray,*ctarray;
Mat_MPIDense *c=(Mat_MPIDense*)C->data;
Mat_MatTransMatMult *atb=c->atb;
Vec bt=atb->bt,ct=atb->ct;
PetscFunctionBegin;
/* create MAIJ matrix mA from A -- should be done in symbolic phase */
ierr = MatDestroy(&atb->mA);CHKERRQ(ierr);
ierr = MatCreateMAIJ(A,BN,&atb->mA);CHKERRQ(ierr);
/* transpose local arry of B, then copy it to vector bt */
ierr = MatDenseGetArray(B,&Barray);CHKERRQ(ierr);
ierr = VecGetArray(bt,&btarray);CHKERRQ(ierr);
k=0;
for (j=0; j<BN; j++) {
for (i=0; i<m; i++) btarray[i*BN + j] = Barray[k++];
}
ierr = VecRestoreArray(bt,&btarray);CHKERRQ(ierr);
ierr = MatDenseRestoreArray(B,&Barray);CHKERRQ(ierr);
/* compute ct = mA^T * cb */
ierr = MatMultTranspose(atb->mA,bt,ct);CHKERRQ(ierr);
/* transpose local arry of ct to matrix C */
ierr = MatDenseGetArray(C,&Carray);CHKERRQ(ierr);
ierr = VecGetArray(ct,&ctarray);CHKERRQ(ierr);
k = 0;
for (j=0; j<BN; j++) {
for (i=0; i<n; i++) Carray[k++] = ctarray[i*BN + j];
}
ierr = VecRestoreArray(ct,&ctarray);CHKERRQ(ierr);
ierr = MatDenseRestoreArray(C,&Carray);CHKERRQ(ierr);
ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int main(int argc,char **argv)
{
Mat pA,P,aijP;
PetscScalar pa[]={1.,-1.,0.,0.,1.,-1.,0.,0.,1.};
PetscInt i,pij[]={0,1,2};
PetscInt aij[3][3]={{0,1,2},{3,4,5},{6,7,8}};
Mat A,mC,C;
PetscScalar one=1.;
PetscErrorCode ierr;
PetscMPIInt size;
ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
if (size != 1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"This is a uniprocessor example only!");
/* Create MAIJ matrix, P */
ierr = MatCreate(PETSC_COMM_SELF,&pA);CHKERRQ(ierr);
ierr = MatSetSizes(pA,3,3,3,3);CHKERRQ(ierr);
ierr = MatSetType(pA,MATSEQAIJ);CHKERRQ(ierr);
ierr = MatSetUp(pA);CHKERRQ(ierr);
ierr = MatSetOption(pA,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE);CHKERRQ(ierr);
ierr = MatSetValues(pA,3,pij,3,pij,pa,ADD_VALUES);CHKERRQ(ierr);
ierr = MatAssemblyBegin(pA,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(pA,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatCreateMAIJ(pA,3,&P);CHKERRQ(ierr);
ierr = MatDestroy(&pA);CHKERRQ(ierr);
/* Create AIJ equivalent matrix, aijP, for comparison testing */
ierr = MatConvert(P,MATSEQAIJ,MAT_INITIAL_MATRIX,&aijP);CHKERRQ(ierr);
/* Create AIJ matrix, A */
ierr = MatCreate(PETSC_COMM_SELF,&A);CHKERRQ(ierr);
ierr = MatSetSizes(A,9,9,9,9);CHKERRQ(ierr);
ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr);
ierr = MatSetUp(A);CHKERRQ(ierr);
ierr = MatSetOption(A,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE);CHKERRQ(ierr);
ierr = MatSetValues(A,3,aij[0],3,aij[0],pa,ADD_VALUES);CHKERRQ(ierr);
ierr = MatSetValues(A,3,aij[1],3,aij[1],pa,ADD_VALUES);CHKERRQ(ierr);
ierr = MatSetValues(A,3,aij[2],3,aij[2],pa,ADD_VALUES);CHKERRQ(ierr);
for (i=0; i<9; i++) {
ierr = MatSetValue(A,i,i,one,ADD_VALUES);CHKERRQ(ierr);
}
ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* Perform PtAP_SeqAIJ_SeqMAIJ */
ierr = MatPtAP(A,P,MAT_INITIAL_MATRIX,1.,&mC);CHKERRQ(ierr);
ierr = MatPtAP(A,P,MAT_REUSE_MATRIX,1.,&mC);CHKERRQ(ierr);
ierr = MatView(mC,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
/* Perform PtAP_SeqAIJ_SeqAIJ for comparison testing */
ierr = MatPtAP(A,aijP,MAT_INITIAL_MATRIX,1.,&C);CHKERRQ(ierr);
ierr = MatView(C,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
/* Check mC = C */
ierr = MatAXPY(C,-1.0,mC,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
/* Note: We should be able to use SAME_NONZERO_PATTERN on the line above, */
/* but don't because this flag doesn't assist testing. */
ierr = MatView(C,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
/* Cleanup */
ierr = MatDestroy(&P);CHKERRQ(ierr);
ierr = MatDestroy(&aijP);CHKERRQ(ierr);
ierr = MatDestroy(&A);CHKERRQ(ierr);
ierr = MatDestroy(&C);CHKERRQ(ierr);
ierr = MatDestroy(&mC);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
int main(int argc,char **argv) {
Mat pA,P,aijP;
PetscScalar pa[]={1.,-1.,0.,0.,1.,-1.,0.,0.,1.};
PetscInt pij[]={0,1,2};
PetscInt aij[3][3]={{0,1,2},{3,4,5},{6,7,8}};
Mat A,mC,C;
PetscScalar one=1.;
PetscErrorCode ierr;
PetscInitialize(&argc,&argv,(char *)0,help);
/* Create MAIJ matrix, P */
ierr = MatCreate(PETSC_COMM_SELF,&pA);CHKERRQ(ierr);
ierr = MatSetSizes(pA,3,3,3,3);CHKERRQ(ierr);
ierr = MatSetType(pA,MATSEQAIJ);CHKERRQ(ierr);
ierr = MatSetOption(pA,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE);CHKERRQ(ierr);
ierr = MatSetValues(pA,3,pij,3,pij,pa,ADD_VALUES);CHKERRQ(ierr);
ierr = MatAssemblyBegin(pA,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(pA,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatCreateMAIJ(pA,3,&P);CHKERRQ(ierr);
ierr = MatDestroy(pA);
/* Create AIJ equivalent matrix, aijP, for comparison testing */
ierr = MatConvert(P,MATSEQAIJ,MAT_INITIAL_MATRIX,&aijP);
/* Create AIJ matrix, A */
ierr = MatCreate(PETSC_COMM_SELF,&A);CHKERRQ(ierr);
ierr = MatSetSizes(A,9,9,9,9);CHKERRQ(ierr);
ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr);
ierr = MatSetOption(A,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE);CHKERRQ(ierr);
ierr = MatSetValues(A,3,aij[0],3,aij[0],pa,ADD_VALUES);CHKERRQ(ierr);
ierr = MatSetValues(A,3,aij[1],3,aij[1],pa,ADD_VALUES);CHKERRQ(ierr);
ierr = MatSetValues(A,3,aij[2],3,aij[2],pa,ADD_VALUES);CHKERRQ(ierr);
{int i;
for (i=0;i<9;i++) {
ierr = MatSetValue(A,i,i,one,ADD_VALUES);CHKERRQ(ierr);
}
}
ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
/* Perform SeqAIJ_SeqMAIJ PtAP */
ierr = MatPtAP(A,P,MAT_INITIAL_MATRIX,1.,&mC);CHKERRQ(ierr);
ierr = MatView(mC,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
/* Perform SeqAIJ_SeqAIJ PtAP for comparison testing */
ierr = MatPtAP(A,aijP,MAT_INITIAL_MATRIX,1.,&C);CHKERRQ(ierr);
ierr = MatView(C,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
/* Perform diff of two matrices */
ierr = MatAXPY(C,-1.0,mC,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr);
/* Note: We should be able to use SAME_NONZERO_PATTERN on the line above, */
/* but don't because this flag doesn't assist testing. */
ierr = MatView(C,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr);
/* Cleanup */
ierr = MatDestroy(P);
ierr = MatDestroy(aijP);
ierr = MatDestroy(A);
ierr = MatDestroy(C);
ierr = MatDestroy(mC);
PetscFinalize();
return(0);
}
