1b4319ba4SBarry Smith /* 2b4319ba4SBarry Smith Creates a matrix class for using the Neumann-Neumann type preconditioners. 3b4319ba4SBarry Smith This stores the matrices in globally unassembled form. Each processor 4b4319ba4SBarry Smith assembles only its local Neumann problem and the parallel matrix vector 5b4319ba4SBarry Smith product is handled "implicitly". 6b4319ba4SBarry Smith 7b4319ba4SBarry Smith Currently this allows for only one subdomain per processor. 8b4319ba4SBarry Smith */ 9b4319ba4SBarry Smith 10c6db04a5SJed Brown #include <../src/mat/impls/is/matis.h> /*I "petscmat.h" I*/ 114f2d7cafSStefano Zampini #include <petsc/private/sfimpl.h> 12a72d46e8SStefano Zampini #include <petsc/private/vecimpl.h> 1328f4e0baSStefano Zampini 14f26d0771SStefano Zampini #define MATIS_MAX_ENTRIES_INSERTION 2048 15b4f971dfSStefano Zampini static PetscErrorCode MatSetValuesLocal_IS(Mat,PetscInt,const PetscInt*,PetscInt,const PetscInt*,const PetscScalar*,InsertMode); 16b4f971dfSStefano Zampini static PetscErrorCode MatSetValuesBlockedLocal_IS(Mat,PetscInt,const PetscInt*,PetscInt,const PetscInt*,const PetscScalar*,InsertMode); 178546b261SStefano Zampini static PetscErrorCode MatISSetUpScatters_Private(Mat); 18f26d0771SStefano Zampini 1975d48cdbSStefano Zampini static PetscErrorCode MatISContainerDestroyPtAP_Private(void *ptr) 2075d48cdbSStefano Zampini { 2175d48cdbSStefano Zampini MatISPtAP ptap = (MatISPtAP)ptr; 2275d48cdbSStefano Zampini PetscErrorCode ierr; 2375d48cdbSStefano Zampini 2475d48cdbSStefano Zampini PetscFunctionBegin; 2575d48cdbSStefano Zampini ierr = MatDestroySubMatrices(ptap->ris1 ? 2 : 1,&ptap->lP);CHKERRQ(ierr); 2675d48cdbSStefano Zampini ierr = ISDestroy(&ptap->cis0);CHKERRQ(ierr); 2775d48cdbSStefano Zampini ierr = ISDestroy(&ptap->cis1);CHKERRQ(ierr); 2875d48cdbSStefano Zampini ierr = ISDestroy(&ptap->ris0);CHKERRQ(ierr); 2975d48cdbSStefano Zampini ierr = ISDestroy(&ptap->ris1);CHKERRQ(ierr); 3075d48cdbSStefano Zampini ierr = PetscFree(ptap);CHKERRQ(ierr); 3175d48cdbSStefano Zampini PetscFunctionReturn(0); 3275d48cdbSStefano Zampini } 3375d48cdbSStefano Zampini 3475d48cdbSStefano Zampini static PetscErrorCode MatPtAPNumeric_IS_XAIJ(Mat A, Mat P, Mat C) 3575d48cdbSStefano Zampini { 3675d48cdbSStefano Zampini MatISPtAP ptap; 3775d48cdbSStefano Zampini Mat_IS *matis = (Mat_IS*)A->data; 3875d48cdbSStefano Zampini Mat lA,lC; 3975d48cdbSStefano Zampini MatReuse reuse; 4075d48cdbSStefano Zampini IS ris[2],cis[2]; 4175d48cdbSStefano Zampini PetscContainer c; 4275d48cdbSStefano Zampini PetscInt n; 4375d48cdbSStefano Zampini PetscErrorCode ierr; 4475d48cdbSStefano Zampini 4575d48cdbSStefano Zampini PetscFunctionBegin; 466afe12f5SStefano Zampini ierr = PetscObjectQuery((PetscObject)C,"_MatIS_PtAP",(PetscObject*)&c);CHKERRQ(ierr); 4775d48cdbSStefano Zampini if (!c) SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_PLIB,"Missing PtAP information"); 4875d48cdbSStefano Zampini ierr = PetscContainerGetPointer(c,(void**)&ptap);CHKERRQ(ierr); 4975d48cdbSStefano Zampini ris[0] = ptap->ris0; 5075d48cdbSStefano Zampini ris[1] = ptap->ris1; 5175d48cdbSStefano Zampini cis[0] = ptap->cis0; 5275d48cdbSStefano Zampini cis[1] = ptap->cis1; 5375d48cdbSStefano Zampini n = ptap->ris1 ? 2 : 1; 5475d48cdbSStefano Zampini reuse = ptap->lP ? MAT_REUSE_MATRIX : MAT_INITIAL_MATRIX; 5575d48cdbSStefano Zampini ierr = MatCreateSubMatrices(P,n,ris,cis,reuse,&ptap->lP);CHKERRQ(ierr); 5675d48cdbSStefano Zampini 5775d48cdbSStefano Zampini ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr); 5875d48cdbSStefano Zampini ierr = MatISGetLocalMat(C,&lC);CHKERRQ(ierr); 5975d48cdbSStefano Zampini if (ptap->ris1) { /* unsymmetric A mapping */ 6075d48cdbSStefano Zampini Mat lPt; 6175d48cdbSStefano Zampini 6275d48cdbSStefano Zampini ierr = MatTranspose(ptap->lP[1],MAT_INITIAL_MATRIX,&lPt);CHKERRQ(ierr); 6375d48cdbSStefano Zampini ierr = MatMatMatMult(lPt,lA,ptap->lP[0],reuse,ptap->fill,&lC);CHKERRQ(ierr); 6475d48cdbSStefano Zampini if (matis->storel2l) { 6575d48cdbSStefano Zampini ierr = PetscObjectCompose((PetscObject)(A),"_MatIS_PtAP_l2l",(PetscObject)lPt);CHKERRQ(ierr); 6675d48cdbSStefano Zampini } 6775d48cdbSStefano Zampini ierr = MatDestroy(&lPt);CHKERRQ(ierr); 6875d48cdbSStefano Zampini } else { 6975d48cdbSStefano Zampini ierr = MatPtAP(lA,ptap->lP[0],reuse,ptap->fill,&lC);CHKERRQ(ierr); 7075d48cdbSStefano Zampini if (matis->storel2l) { 714222ddf1SHong Zhang ierr = PetscObjectCompose((PetscObject)C,"_MatIS_PtAP_l2l",(PetscObject)ptap->lP[0]);CHKERRQ(ierr); 7275d48cdbSStefano Zampini } 7375d48cdbSStefano Zampini } 7475d48cdbSStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 7575d48cdbSStefano Zampini ierr = MatISSetLocalMat(C,lC);CHKERRQ(ierr); 7675d48cdbSStefano Zampini ierr = MatDestroy(&lC);CHKERRQ(ierr); 7775d48cdbSStefano Zampini } 7875d48cdbSStefano Zampini ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 7975d48cdbSStefano Zampini ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 8075d48cdbSStefano Zampini PetscFunctionReturn(0); 8175d48cdbSStefano Zampini } 8275d48cdbSStefano Zampini 8375d48cdbSStefano Zampini static PetscErrorCode MatGetNonzeroColumnsLocal_Private(Mat PT,IS *cis) 8475d48cdbSStefano Zampini { 8575d48cdbSStefano Zampini Mat Po,Pd; 8675d48cdbSStefano Zampini IS zd,zo; 8775d48cdbSStefano Zampini const PetscInt *garray; 8875d48cdbSStefano Zampini PetscInt *aux,i,bs; 8975d48cdbSStefano Zampini PetscInt dc,stc,oc,ctd,cto; 9075d48cdbSStefano Zampini PetscBool ismpiaij,ismpibaij,isseqaij,isseqbaij; 9175d48cdbSStefano Zampini MPI_Comm comm; 9275d48cdbSStefano Zampini PetscErrorCode ierr; 9375d48cdbSStefano Zampini 9475d48cdbSStefano Zampini PetscFunctionBegin; 9575d48cdbSStefano Zampini PetscValidHeaderSpecific(PT,MAT_CLASSID,1); 9675d48cdbSStefano Zampini PetscValidPointer(cis,2); 9775d48cdbSStefano Zampini ierr = PetscObjectGetComm((PetscObject)PT,&comm);CHKERRQ(ierr); 9875d48cdbSStefano Zampini bs = 1; 99b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)PT,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 100b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)PT,MATMPIBAIJ,&ismpibaij);CHKERRQ(ierr); 10104637862SRichard Tran Mills ierr = PetscObjectBaseTypeCompare((PetscObject)PT,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 10275d48cdbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)PT,MATSEQBAIJ,&isseqbaij);CHKERRQ(ierr); 10375d48cdbSStefano Zampini if (isseqaij || isseqbaij) { 10475d48cdbSStefano Zampini Pd = PT; 10575d48cdbSStefano Zampini Po = NULL; 10675d48cdbSStefano Zampini garray = NULL; 10775d48cdbSStefano Zampini } else if (ismpiaij) { 10875d48cdbSStefano Zampini ierr = MatMPIAIJGetSeqAIJ(PT,&Pd,&Po,&garray);CHKERRQ(ierr); 10975d48cdbSStefano Zampini } else if (ismpibaij) { 11075d48cdbSStefano Zampini ierr = MatMPIBAIJGetSeqBAIJ(PT,&Pd,&Po,&garray);CHKERRQ(ierr); 11175d48cdbSStefano Zampini ierr = MatGetBlockSize(PT,&bs);CHKERRQ(ierr); 11275d48cdbSStefano Zampini } else SETERRQ1(comm,PETSC_ERR_SUP,"Not for matrix type %s",((PetscObject)(PT))->type_name); 11375d48cdbSStefano Zampini 11475d48cdbSStefano Zampini /* identify any null columns in Pd or Po */ 11522f7620eSStefano Zampini /* We use a tolerance comparison since it may happen that, with geometric multigrid, 11622f7620eSStefano Zampini some of the columns are not really zero, but very close to */ 11775d48cdbSStefano Zampini zo = zd = NULL; 11875d48cdbSStefano Zampini if (Po) { 11922f7620eSStefano Zampini ierr = MatFindNonzeroRowsOrCols_Basic(Po,PETSC_TRUE,PETSC_SMALL,&zo);CHKERRQ(ierr); 12075d48cdbSStefano Zampini } 12122f7620eSStefano Zampini ierr = MatFindNonzeroRowsOrCols_Basic(Pd,PETSC_TRUE,PETSC_SMALL,&zd);CHKERRQ(ierr); 12275d48cdbSStefano Zampini 12375d48cdbSStefano Zampini ierr = MatGetLocalSize(PT,NULL,&dc);CHKERRQ(ierr); 12475d48cdbSStefano Zampini ierr = MatGetOwnershipRangeColumn(PT,&stc,NULL);CHKERRQ(ierr); 12575d48cdbSStefano Zampini if (Po) { ierr = MatGetLocalSize(Po,NULL,&oc);CHKERRQ(ierr); } 12675d48cdbSStefano Zampini else oc = 0; 12775d48cdbSStefano Zampini ierr = PetscMalloc1((dc+oc)/bs,&aux);CHKERRQ(ierr); 12875d48cdbSStefano Zampini if (zd) { 12975d48cdbSStefano Zampini const PetscInt *idxs; 13075d48cdbSStefano Zampini PetscInt nz; 13175d48cdbSStefano Zampini 13275d48cdbSStefano Zampini /* this will throw an error if bs is not valid */ 13375d48cdbSStefano Zampini ierr = ISSetBlockSize(zd,bs);CHKERRQ(ierr); 13475d48cdbSStefano Zampini ierr = ISGetLocalSize(zd,&nz);CHKERRQ(ierr); 13575d48cdbSStefano Zampini ierr = ISGetIndices(zd,&idxs);CHKERRQ(ierr); 13675d48cdbSStefano Zampini ctd = nz/bs; 13775d48cdbSStefano Zampini for (i=0; i<ctd; i++) aux[i] = (idxs[bs*i]+stc)/bs; 13875d48cdbSStefano Zampini ierr = ISRestoreIndices(zd,&idxs);CHKERRQ(ierr); 13975d48cdbSStefano Zampini } else { 14075d48cdbSStefano Zampini ctd = dc/bs; 14175d48cdbSStefano Zampini for (i=0; i<ctd; i++) aux[i] = i+stc/bs; 14275d48cdbSStefano Zampini } 14375d48cdbSStefano Zampini if (zo) { 14475d48cdbSStefano Zampini const PetscInt *idxs; 14575d48cdbSStefano Zampini PetscInt nz; 14675d48cdbSStefano Zampini 14775d48cdbSStefano Zampini /* this will throw an error if bs is not valid */ 14875d48cdbSStefano Zampini ierr = ISSetBlockSize(zo,bs);CHKERRQ(ierr); 14975d48cdbSStefano Zampini ierr = ISGetLocalSize(zo,&nz);CHKERRQ(ierr); 15075d48cdbSStefano Zampini ierr = ISGetIndices(zo,&idxs);CHKERRQ(ierr); 15175d48cdbSStefano Zampini cto = nz/bs; 15275d48cdbSStefano Zampini for (i=0; i<cto; i++) aux[i+ctd] = garray[idxs[bs*i]/bs]; 15375d48cdbSStefano Zampini ierr = ISRestoreIndices(zo,&idxs);CHKERRQ(ierr); 15475d48cdbSStefano Zampini } else { 15575d48cdbSStefano Zampini cto = oc/bs; 15675d48cdbSStefano Zampini for (i=0; i<cto; i++) aux[i+ctd] = garray[i]; 15775d48cdbSStefano Zampini } 15875d48cdbSStefano Zampini ierr = ISCreateBlock(comm,bs,ctd+cto,aux,PETSC_OWN_POINTER,cis);CHKERRQ(ierr); 15975d48cdbSStefano Zampini ierr = ISDestroy(&zd);CHKERRQ(ierr); 16075d48cdbSStefano Zampini ierr = ISDestroy(&zo);CHKERRQ(ierr); 16175d48cdbSStefano Zampini PetscFunctionReturn(0); 16275d48cdbSStefano Zampini } 16375d48cdbSStefano Zampini 1644222ddf1SHong Zhang static PetscErrorCode MatPtAPSymbolic_IS_XAIJ(Mat A,Mat P,PetscReal fill,Mat C) 16575d48cdbSStefano Zampini { 1668546b261SStefano Zampini Mat PT,lA; 16775d48cdbSStefano Zampini MatISPtAP ptap; 16875d48cdbSStefano Zampini ISLocalToGlobalMapping Crl2g,Ccl2g,rl2g,cl2g; 16975d48cdbSStefano Zampini PetscContainer c; 1708546b261SStefano Zampini MatType lmtype; 17175d48cdbSStefano Zampini const PetscInt *garray; 17275d48cdbSStefano Zampini PetscInt ibs,N,dc; 17375d48cdbSStefano Zampini MPI_Comm comm; 17475d48cdbSStefano Zampini PetscErrorCode ierr; 17575d48cdbSStefano Zampini 17675d48cdbSStefano Zampini PetscFunctionBegin; 17775d48cdbSStefano Zampini ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 1784222ddf1SHong Zhang ierr = MatSetType(C,MATIS);CHKERRQ(ierr); 1798546b261SStefano Zampini ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr); 1808546b261SStefano Zampini ierr = MatGetType(lA,&lmtype);CHKERRQ(ierr); 1814222ddf1SHong Zhang ierr = MatISSetLocalMatType(C,lmtype);CHKERRQ(ierr); 18275d48cdbSStefano Zampini ierr = MatGetSize(P,NULL,&N);CHKERRQ(ierr); 18375d48cdbSStefano Zampini ierr = MatGetLocalSize(P,NULL,&dc);CHKERRQ(ierr); 1844222ddf1SHong Zhang ierr = MatSetSizes(C,dc,dc,N,N);CHKERRQ(ierr); 18575d48cdbSStefano Zampini /* Not sure about this 18675d48cdbSStefano Zampini ierr = MatGetBlockSizes(P,NULL,&ibs);CHKERRQ(ierr); 18775d48cdbSStefano Zampini ierr = MatSetBlockSize(*C,ibs);CHKERRQ(ierr); 18875d48cdbSStefano Zampini */ 18975d48cdbSStefano Zampini 19075d48cdbSStefano Zampini ierr = PetscNew(&ptap);CHKERRQ(ierr); 19175d48cdbSStefano Zampini ierr = PetscContainerCreate(PETSC_COMM_SELF,&c);CHKERRQ(ierr); 19275d48cdbSStefano Zampini ierr = PetscContainerSetPointer(c,ptap);CHKERRQ(ierr); 19375d48cdbSStefano Zampini ierr = PetscContainerSetUserDestroy(c,MatISContainerDestroyPtAP_Private);CHKERRQ(ierr); 1944222ddf1SHong Zhang ierr = PetscObjectCompose((PetscObject)C,"_MatIS_PtAP",(PetscObject)c);CHKERRQ(ierr); 19575d48cdbSStefano Zampini ierr = PetscContainerDestroy(&c);CHKERRQ(ierr); 19675d48cdbSStefano Zampini ptap->fill = fill; 19775d48cdbSStefano Zampini 19875d48cdbSStefano Zampini ierr = MatGetLocalToGlobalMapping(A,&rl2g,&cl2g);CHKERRQ(ierr); 19975d48cdbSStefano Zampini 20075d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(cl2g,&ibs);CHKERRQ(ierr); 20175d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingGetSize(cl2g,&N);CHKERRQ(ierr); 20275d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(cl2g,&garray);CHKERRQ(ierr); 20375d48cdbSStefano Zampini ierr = ISCreateBlock(comm,ibs,N/ibs,garray,PETSC_COPY_VALUES,&ptap->ris0);CHKERRQ(ierr); 20475d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(cl2g,&garray);CHKERRQ(ierr); 20575d48cdbSStefano Zampini 20675d48cdbSStefano Zampini ierr = MatCreateSubMatrix(P,ptap->ris0,NULL,MAT_INITIAL_MATRIX,&PT);CHKERRQ(ierr); 20775d48cdbSStefano Zampini ierr = MatGetNonzeroColumnsLocal_Private(PT,&ptap->cis0);CHKERRQ(ierr); 20875d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(ptap->cis0,&Ccl2g);CHKERRQ(ierr); 20975d48cdbSStefano Zampini ierr = MatDestroy(&PT);CHKERRQ(ierr); 21075d48cdbSStefano Zampini 21175d48cdbSStefano Zampini Crl2g = NULL; 21275d48cdbSStefano Zampini if (rl2g != cl2g) { /* unsymmetric A mapping */ 21375d48cdbSStefano Zampini PetscBool same,lsame = PETSC_FALSE; 21475d48cdbSStefano Zampini PetscInt N1,ibs1; 21575d48cdbSStefano Zampini 21675d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingGetSize(rl2g,&N1);CHKERRQ(ierr); 21775d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(rl2g,&ibs1);CHKERRQ(ierr); 21875d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(rl2g,&garray);CHKERRQ(ierr); 21975d48cdbSStefano Zampini ierr = ISCreateBlock(comm,ibs,N/ibs,garray,PETSC_COPY_VALUES,&ptap->ris1);CHKERRQ(ierr); 22075d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(rl2g,&garray);CHKERRQ(ierr); 22175d48cdbSStefano Zampini if (ibs1 == ibs && N1 == N) { /* check if the l2gmaps are the same */ 22275d48cdbSStefano Zampini const PetscInt *i1,*i2; 22375d48cdbSStefano Zampini 22475d48cdbSStefano Zampini ierr = ISBlockGetIndices(ptap->ris0,&i1);CHKERRQ(ierr); 22575d48cdbSStefano Zampini ierr = ISBlockGetIndices(ptap->ris1,&i2);CHKERRQ(ierr); 226580bdb30SBarry Smith ierr = PetscArraycmp(i1,i2,N,&lsame);CHKERRQ(ierr); 22775d48cdbSStefano Zampini } 228820f2d46SBarry Smith ierr = MPIU_Allreduce(&lsame,&same,1,MPIU_BOOL,MPI_LAND,comm);CHKERRMPI(ierr); 22975d48cdbSStefano Zampini if (same) { 23075d48cdbSStefano Zampini ierr = ISDestroy(&ptap->ris1);CHKERRQ(ierr); 23175d48cdbSStefano Zampini } else { 23275d48cdbSStefano Zampini ierr = MatCreateSubMatrix(P,ptap->ris1,NULL,MAT_INITIAL_MATRIX,&PT);CHKERRQ(ierr); 23375d48cdbSStefano Zampini ierr = MatGetNonzeroColumnsLocal_Private(PT,&ptap->cis1);CHKERRQ(ierr); 23475d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(ptap->cis1,&Crl2g);CHKERRQ(ierr); 23575d48cdbSStefano Zampini ierr = MatDestroy(&PT);CHKERRQ(ierr); 23675d48cdbSStefano Zampini } 23775d48cdbSStefano Zampini } 23875d48cdbSStefano Zampini /* Not sure about this 23975d48cdbSStefano Zampini if (!Crl2g) { 2404222ddf1SHong Zhang ierr = MatGetBlockSize(C,&ibs);CHKERRQ(ierr); 24175d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingSetBlockSize(Ccl2g,ibs);CHKERRQ(ierr); 24275d48cdbSStefano Zampini } 24375d48cdbSStefano Zampini */ 2444222ddf1SHong Zhang ierr = MatSetLocalToGlobalMapping(C,Crl2g ? Crl2g : Ccl2g,Ccl2g);CHKERRQ(ierr); 24575d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&Crl2g);CHKERRQ(ierr); 24675d48cdbSStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&Ccl2g);CHKERRQ(ierr); 24775d48cdbSStefano Zampini 2484222ddf1SHong Zhang C->ops->ptapnumeric = MatPtAPNumeric_IS_XAIJ; 24975d48cdbSStefano Zampini PetscFunctionReturn(0); 25075d48cdbSStefano Zampini } 25175d48cdbSStefano Zampini 2524222ddf1SHong Zhang /* ----------------------------------------- */ 2534222ddf1SHong Zhang static PetscErrorCode MatProductSymbolic_PtAP_IS_XAIJ(Mat C) 25475d48cdbSStefano Zampini { 25575d48cdbSStefano Zampini PetscErrorCode ierr; 2564222ddf1SHong Zhang Mat_Product *product = C->product; 2574222ddf1SHong Zhang Mat A=product->A,P=product->B; 2584222ddf1SHong Zhang PetscReal fill=product->fill; 25975d48cdbSStefano Zampini 26075d48cdbSStefano Zampini PetscFunctionBegin; 26175d48cdbSStefano Zampini ierr = MatPtAPSymbolic_IS_XAIJ(A,P,fill,C);CHKERRQ(ierr); 2624222ddf1SHong Zhang C->ops->productnumeric = MatProductNumeric_PtAP; 26375d48cdbSStefano Zampini PetscFunctionReturn(0); 26475d48cdbSStefano Zampini } 26575d48cdbSStefano Zampini 2664222ddf1SHong Zhang static PetscErrorCode MatProductSetFromOptions_IS_XAIJ_PtAP(Mat C) 2674222ddf1SHong Zhang { 2684222ddf1SHong Zhang PetscFunctionBegin; 2694222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_PtAP_IS_XAIJ; 2704222ddf1SHong Zhang PetscFunctionReturn(0); 2714222ddf1SHong Zhang } 2724222ddf1SHong Zhang 2734222ddf1SHong Zhang PETSC_INTERN PetscErrorCode MatProductSetFromOptions_IS_XAIJ(Mat C) 2744222ddf1SHong Zhang { 2754222ddf1SHong Zhang PetscErrorCode ierr; 2764222ddf1SHong Zhang Mat_Product *product = C->product; 2774222ddf1SHong Zhang 2784222ddf1SHong Zhang PetscFunctionBegin; 2794222ddf1SHong Zhang if (product->type == MATPRODUCT_PtAP) { 2804222ddf1SHong Zhang ierr = MatProductSetFromOptions_IS_XAIJ_PtAP(C);CHKERRQ(ierr); 2816718818eSStefano Zampini } 2824222ddf1SHong Zhang PetscFunctionReturn(0); 2834222ddf1SHong Zhang } 2844222ddf1SHong Zhang 2854222ddf1SHong Zhang /* ----------------------------------------- */ 2865b003df0Sstefano_zampini static PetscErrorCode MatISContainerDestroyFields_Private(void *ptr) 2875b003df0Sstefano_zampini { 2885b003df0Sstefano_zampini MatISLocalFields lf = (MatISLocalFields)ptr; 2895b003df0Sstefano_zampini PetscInt i; 2905b003df0Sstefano_zampini PetscErrorCode ierr; 2915b003df0Sstefano_zampini 292ab4d48faSStefano Zampini PetscFunctionBegin; 2935b003df0Sstefano_zampini for (i=0;i<lf->nr;i++) { 2945b003df0Sstefano_zampini ierr = ISDestroy(&lf->rf[i]);CHKERRQ(ierr); 2955b003df0Sstefano_zampini } 2965b003df0Sstefano_zampini for (i=0;i<lf->nc;i++) { 2975b003df0Sstefano_zampini ierr = ISDestroy(&lf->cf[i]);CHKERRQ(ierr); 2985b003df0Sstefano_zampini } 2995b003df0Sstefano_zampini ierr = PetscFree2(lf->rf,lf->cf);CHKERRQ(ierr); 3005b003df0Sstefano_zampini ierr = PetscFree(lf);CHKERRQ(ierr); 3015b003df0Sstefano_zampini PetscFunctionReturn(0); 3025b003df0Sstefano_zampini } 303a72627d2SStefano Zampini 304c9225affSStefano Zampini static PetscErrorCode MatConvert_SeqXAIJ_IS(Mat A,MatType type,MatReuse reuse,Mat *newmat) 3056989cf23SStefano Zampini { 306c9225affSStefano Zampini Mat B,lB; 307c9225affSStefano Zampini PetscErrorCode ierr; 308c9225affSStefano Zampini 309c9225affSStefano Zampini PetscFunctionBegin; 310c9225affSStefano Zampini if (reuse != MAT_REUSE_MATRIX) { 311c9225affSStefano Zampini ISLocalToGlobalMapping rl2g,cl2g; 312c9225affSStefano Zampini PetscInt bs; 313c9225affSStefano Zampini IS is; 314c9225affSStefano Zampini 315c9225affSStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 316c9225affSStefano Zampini ierr = ISCreateStride(PetscObjectComm((PetscObject)A),A->rmap->n/bs,0,1,&is);CHKERRQ(ierr); 317c9225affSStefano Zampini if (bs > 1) { 318c9225affSStefano Zampini IS is2; 319c9225affSStefano Zampini PetscInt i,*aux; 320c9225affSStefano Zampini 321c9225affSStefano Zampini ierr = ISGetLocalSize(is,&i);CHKERRQ(ierr); 322c9225affSStefano Zampini ierr = ISGetIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr); 323c9225affSStefano Zampini ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),bs,i,aux,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr); 324c9225affSStefano Zampini ierr = ISRestoreIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr); 325c9225affSStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 326c9225affSStefano Zampini is = is2; 327c9225affSStefano Zampini } 328c9225affSStefano Zampini ierr = ISSetIdentity(is);CHKERRQ(ierr); 329c9225affSStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr); 330c9225affSStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 331c9225affSStefano Zampini ierr = ISCreateStride(PetscObjectComm((PetscObject)A),A->cmap->n/bs,0,1,&is);CHKERRQ(ierr); 332c9225affSStefano Zampini if (bs > 1) { 333c9225affSStefano Zampini IS is2; 334c9225affSStefano Zampini PetscInt i,*aux; 335c9225affSStefano Zampini 336c9225affSStefano Zampini ierr = ISGetLocalSize(is,&i);CHKERRQ(ierr); 337c9225affSStefano Zampini ierr = ISGetIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr); 338c9225affSStefano Zampini ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),bs,i,aux,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr); 339c9225affSStefano Zampini ierr = ISRestoreIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr); 340c9225affSStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 341c9225affSStefano Zampini is = is2; 342c9225affSStefano Zampini } 343c9225affSStefano Zampini ierr = ISSetIdentity(is);CHKERRQ(ierr); 344c9225affSStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr); 345c9225affSStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 346c9225affSStefano Zampini ierr = MatCreateIS(PetscObjectComm((PetscObject)A),bs,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N,rl2g,cl2g,&B);CHKERRQ(ierr); 347c9225affSStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 348c9225affSStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 349c9225affSStefano Zampini ierr = MatDuplicate(A,MAT_COPY_VALUES,&lB);CHKERRQ(ierr); 350c9225affSStefano Zampini if (reuse == MAT_INITIAL_MATRIX) *newmat = B; 351c9225affSStefano Zampini } else { 352c9225affSStefano Zampini B = *newmat; 353c9225affSStefano Zampini ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 354c9225affSStefano Zampini lB = A; 355c9225affSStefano Zampini } 356c9225affSStefano Zampini ierr = MatISSetLocalMat(B,lB);CHKERRQ(ierr); 357c9225affSStefano Zampini ierr = MatDestroy(&lB);CHKERRQ(ierr); 358c9225affSStefano Zampini ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 359c9225affSStefano Zampini ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 360c9225affSStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 361c9225affSStefano Zampini ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 362c9225affSStefano Zampini } 363c9225affSStefano Zampini PetscFunctionReturn(0); 364c9225affSStefano Zampini } 365c9225affSStefano Zampini 366c9225affSStefano Zampini static PetscErrorCode MatISScaleDisassembling_Private(Mat A) 367c9225affSStefano Zampini { 368c9225affSStefano Zampini Mat_IS *matis = (Mat_IS*)(A->data); 369c9225affSStefano Zampini PetscScalar *aa; 370c9225affSStefano Zampini const PetscInt *ii,*jj; 371c9225affSStefano Zampini PetscInt i,n,m; 372fabe8965SStefano Zampini PetscInt *ecount,**eneighs; 373c9225affSStefano Zampini PetscBool flg; 374c9225affSStefano Zampini PetscErrorCode ierr; 375c9225affSStefano Zampini 376c9225affSStefano Zampini PetscFunctionBegin; 377c9225affSStefano Zampini ierr = MatGetRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&m,&ii,&jj,&flg);CHKERRQ(ierr); 378c9225affSStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure"); 379fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingGetNodeInfo(A->rmap->mapping,&n,&ecount,&eneighs);CHKERRQ(ierr); 380c9225affSStefano Zampini if (m != n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected %D != %D",m,n); 381c9225affSStefano Zampini ierr = MatSeqAIJGetArray(matis->A,&aa);CHKERRQ(ierr); 382c9225affSStefano Zampini for (i=0;i<n;i++) { 383fabe8965SStefano Zampini if (ecount[i] > 1) { 384c9225affSStefano Zampini PetscInt j; 385c9225affSStefano Zampini 386c9225affSStefano Zampini for (j=ii[i];j<ii[i+1];j++) { 387c9225affSStefano Zampini PetscInt i2 = jj[j],p,p2; 388fabe8965SStefano Zampini PetscReal scal = 0.0; 389c9225affSStefano Zampini 390c9225affSStefano Zampini for (p=0;p<ecount[i];p++) { 391c9225affSStefano Zampini for (p2=0;p2<ecount[i2];p2++) { 392c9225affSStefano Zampini if (eneighs[i][p] == eneighs[i2][p2]) { scal += 1.0; break; } 393c9225affSStefano Zampini } 394c9225affSStefano Zampini } 395fabe8965SStefano Zampini if (scal) aa[j] /= scal; 396c9225affSStefano Zampini } 397c9225affSStefano Zampini } 398c9225affSStefano Zampini } 399fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingRestoreNodeInfo(A->rmap->mapping,&n,&ecount,&eneighs);CHKERRQ(ierr); 400c9225affSStefano Zampini ierr = MatSeqAIJRestoreArray(matis->A,&aa);CHKERRQ(ierr); 401c9225affSStefano Zampini ierr = MatRestoreRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&m,&ii,&jj,&flg);CHKERRQ(ierr); 402c9225affSStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot restore IJ structure"); 403c9225affSStefano Zampini PetscFunctionReturn(0); 404c9225affSStefano Zampini } 405c9225affSStefano Zampini 406fabe8965SStefano Zampini typedef enum {MAT_IS_DISASSEMBLE_L2G_NATURAL,MAT_IS_DISASSEMBLE_L2G_MAT, MAT_IS_DISASSEMBLE_L2G_ND} MatISDisassemblel2gType; 407fabe8965SStefano Zampini 408c9225affSStefano Zampini static PetscErrorCode MatMPIXAIJComputeLocalToGlobalMapping_Private(Mat A, ISLocalToGlobalMapping *l2g) 409c9225affSStefano Zampini { 410fabe8965SStefano Zampini Mat Ad,Ao; 411fabe8965SStefano Zampini IS is,ndmap,ndsub; 412c9225affSStefano Zampini MPI_Comm comm; 413fabe8965SStefano Zampini const PetscInt *garray,*ndmapi; 414fabe8965SStefano Zampini PetscInt bs,i,cnt,nl,*ncount,*ndmapc; 415fabe8965SStefano Zampini PetscBool ismpiaij,ismpibaij; 416f4259b30SLisandro Dalcin const char *const MatISDisassemblel2gTypes[] = {"NATURAL","MAT","ND","MatISDisassemblel2gType","MAT_IS_DISASSEMBLE_L2G_",NULL}; 417fabe8965SStefano Zampini MatISDisassemblel2gType mode = MAT_IS_DISASSEMBLE_L2G_NATURAL; 418fabe8965SStefano Zampini MatPartitioning part; 419fabe8965SStefano Zampini PetscSF sf; 420c9225affSStefano Zampini PetscErrorCode ierr; 421c9225affSStefano Zampini 422c9225affSStefano Zampini PetscFunctionBegin; 423fabe8965SStefano Zampini ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"MatIS l2g disassembling options","Mat");CHKERRQ(ierr); 424fabe8965SStefano Zampini ierr = PetscOptionsEnum("-mat_is_disassemble_l2g_type","Type of local-to-global mapping to be used for disassembling","MatISDisassemblel2gType",MatISDisassemblel2gTypes,(PetscEnum)mode,(PetscEnum*)&mode,NULL);CHKERRQ(ierr); 425fabe8965SStefano Zampini ierr = PetscOptionsEnd();CHKERRQ(ierr); 426fabe8965SStefano Zampini if (mode == MAT_IS_DISASSEMBLE_L2G_MAT) { 427c9225affSStefano Zampini ierr = MatGetLocalToGlobalMapping(A,l2g,NULL);CHKERRQ(ierr); 428c9225affSStefano Zampini PetscFunctionReturn(0); 429c9225affSStefano Zampini } 430c9225affSStefano Zampini ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 431b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ ,&ismpiaij);CHKERRQ(ierr); 432b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIBAIJ,&ismpibaij);CHKERRQ(ierr); 433c9225affSStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 434fabe8965SStefano Zampini switch (mode) { 435fabe8965SStefano Zampini case MAT_IS_DISASSEMBLE_L2G_ND: 436fabe8965SStefano Zampini ierr = MatPartitioningCreate(comm,&part);CHKERRQ(ierr); 437fabe8965SStefano Zampini ierr = MatPartitioningSetAdjacency(part,A);CHKERRQ(ierr); 438fabe8965SStefano Zampini ierr = PetscObjectSetOptionsPrefix((PetscObject)part,((PetscObject)A)->prefix);CHKERRQ(ierr); 439fabe8965SStefano Zampini ierr = MatPartitioningSetFromOptions(part);CHKERRQ(ierr); 440fabe8965SStefano Zampini ierr = MatPartitioningApplyND(part,&ndmap);CHKERRQ(ierr); 441fabe8965SStefano Zampini ierr = MatPartitioningDestroy(&part);CHKERRQ(ierr); 442fabe8965SStefano Zampini ierr = ISBuildTwoSided(ndmap,NULL,&ndsub);CHKERRQ(ierr); 443fabe8965SStefano Zampini ierr = MatMPIAIJSetUseScalableIncreaseOverlap(A,PETSC_TRUE);CHKERRQ(ierr); 444fabe8965SStefano Zampini ierr = MatIncreaseOverlap(A,1,&ndsub,1);CHKERRQ(ierr); 445fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(ndsub,l2g);CHKERRQ(ierr); 446fabe8965SStefano Zampini 447fabe8965SStefano Zampini /* it may happen that a separator node is not properly shared */ 448fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingGetNodeInfo(*l2g,&nl,&ncount,NULL);CHKERRQ(ierr); 449fabe8965SStefano Zampini ierr = PetscSFCreate(comm,&sf);CHKERRQ(ierr); 450fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(*l2g,&garray);CHKERRQ(ierr); 451fabe8965SStefano Zampini ierr = PetscSFSetGraphLayout(sf,A->rmap,nl,NULL,PETSC_OWN_POINTER,garray);CHKERRQ(ierr); 452fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(*l2g,&garray);CHKERRQ(ierr); 453fabe8965SStefano Zampini ierr = PetscCalloc1(A->rmap->n,&ndmapc);CHKERRQ(ierr); 45483df288dSJunchao Zhang ierr = PetscSFReduceBegin(sf,MPIU_INT,ncount,ndmapc,MPI_REPLACE);CHKERRQ(ierr); 45583df288dSJunchao Zhang ierr = PetscSFReduceEnd(sf,MPIU_INT,ncount,ndmapc,MPI_REPLACE);CHKERRQ(ierr); 456fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingRestoreNodeInfo(*l2g,NULL,&ncount,NULL);CHKERRQ(ierr); 457fabe8965SStefano Zampini ierr = ISGetIndices(ndmap,&ndmapi);CHKERRQ(ierr); 458fabe8965SStefano Zampini for (i = 0, cnt = 0; i < A->rmap->n; i++) 459fabe8965SStefano Zampini if (ndmapi[i] < 0 && ndmapc[i] < 2) 460fabe8965SStefano Zampini cnt++; 461fabe8965SStefano Zampini 462820f2d46SBarry Smith ierr = MPIU_Allreduce(&cnt,&i,1,MPIU_INT,MPI_MAX,comm);CHKERRMPI(ierr); 463fabe8965SStefano Zampini if (i) { /* we detected isolated separator nodes */ 464fabe8965SStefano Zampini Mat A2,A3; 465fabe8965SStefano Zampini IS *workis,is2; 466fabe8965SStefano Zampini PetscScalar *vals; 467fabe8965SStefano Zampini PetscInt gcnt = i,*dnz,*onz,j,*lndmapi; 468fabe8965SStefano Zampini ISLocalToGlobalMapping ll2g; 469fabe8965SStefano Zampini PetscBool flg; 470fabe8965SStefano Zampini const PetscInt *ii,*jj; 471fabe8965SStefano Zampini 472fabe8965SStefano Zampini /* communicate global id of separators */ 473fabe8965SStefano Zampini ierr = MatPreallocateInitialize(comm,A->rmap->n,A->cmap->n,dnz,onz);CHKERRQ(ierr); 474fabe8965SStefano Zampini for (i = 0, cnt = 0; i < A->rmap->n; i++) 475fabe8965SStefano Zampini dnz[i] = ndmapi[i] < 0 ? i + A->rmap->rstart : -1; 476fabe8965SStefano Zampini 477fabe8965SStefano Zampini ierr = PetscMalloc1(nl,&lndmapi);CHKERRQ(ierr); 478ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(sf,MPIU_INT,dnz,lndmapi,MPI_REPLACE);CHKERRQ(ierr); 479fabe8965SStefano Zampini 480fabe8965SStefano Zampini /* compute adjacency of isolated separators node */ 481fabe8965SStefano Zampini ierr = PetscMalloc1(gcnt,&workis);CHKERRQ(ierr); 482fabe8965SStefano Zampini for (i = 0, cnt = 0; i < A->rmap->n; i++) { 483fabe8965SStefano Zampini if (ndmapi[i] < 0 && ndmapc[i] < 2) { 484fabe8965SStefano Zampini ierr = ISCreateStride(comm,1,i+A->rmap->rstart,1,&workis[cnt++]);CHKERRQ(ierr); 485fabe8965SStefano Zampini } 486fabe8965SStefano Zampini } 487fabe8965SStefano Zampini for (i = cnt; i < gcnt; i++) { 488fabe8965SStefano Zampini ierr = ISCreateStride(comm,0,0,1,&workis[i]);CHKERRQ(ierr); 489fabe8965SStefano Zampini } 490fabe8965SStefano Zampini for (i = 0; i < gcnt; i++) { 491fabe8965SStefano Zampini ierr = PetscObjectSetName((PetscObject)workis[i],"ISOLATED");CHKERRQ(ierr); 492fabe8965SStefano Zampini ierr = ISViewFromOptions(workis[i],NULL,"-view_isolated_separators");CHKERRQ(ierr); 493fabe8965SStefano Zampini } 494fabe8965SStefano Zampini 495fabe8965SStefano Zampini /* no communications since all the ISes correspond to locally owned rows */ 496fabe8965SStefano Zampini ierr = MatIncreaseOverlap(A,gcnt,workis,1);CHKERRQ(ierr); 497fabe8965SStefano Zampini 498fabe8965SStefano Zampini /* end communicate global id of separators */ 499ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(sf,MPIU_INT,dnz,lndmapi,MPI_REPLACE);CHKERRQ(ierr); 500fabe8965SStefano Zampini 501fabe8965SStefano Zampini /* communicate new layers : create a matrix and transpose it */ 502580bdb30SBarry Smith ierr = PetscArrayzero(dnz,A->rmap->n);CHKERRQ(ierr); 503580bdb30SBarry Smith ierr = PetscArrayzero(onz,A->rmap->n);CHKERRQ(ierr); 504fabe8965SStefano Zampini for (i = 0, j = 0; i < A->rmap->n; i++) { 505fabe8965SStefano Zampini if (ndmapi[i] < 0 && ndmapc[i] < 2) { 506fabe8965SStefano Zampini const PetscInt* idxs; 507fabe8965SStefano Zampini PetscInt s; 508fabe8965SStefano Zampini 509fabe8965SStefano Zampini ierr = ISGetLocalSize(workis[j],&s);CHKERRQ(ierr); 510fabe8965SStefano Zampini ierr = ISGetIndices(workis[j],&idxs);CHKERRQ(ierr); 511fabe8965SStefano Zampini ierr = MatPreallocateSet(i+A->rmap->rstart,s,idxs,dnz,onz);CHKERRQ(ierr); 512fabe8965SStefano Zampini j++; 513fabe8965SStefano Zampini } 514fabe8965SStefano Zampini } 515fabe8965SStefano Zampini if (j != cnt) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected local count %D != %D",j,cnt); 516fabe8965SStefano Zampini 517fabe8965SStefano Zampini for (i = 0; i < gcnt; i++) { 518fabe8965SStefano Zampini ierr = PetscObjectSetName((PetscObject)workis[i],"EXTENDED");CHKERRQ(ierr); 519fabe8965SStefano Zampini ierr = ISViewFromOptions(workis[i],NULL,"-view_isolated_separators");CHKERRQ(ierr); 520fabe8965SStefano Zampini } 521fabe8965SStefano Zampini 522fabe8965SStefano Zampini for (i = 0, j = 0; i < A->rmap->n; i++) j = PetscMax(j,dnz[i]+onz[i]); 523fabe8965SStefano Zampini ierr = PetscMalloc1(j,&vals);CHKERRQ(ierr); 524fabe8965SStefano Zampini for (i = 0; i < j; i++) vals[i] = 1.0; 525fabe8965SStefano Zampini 526fabe8965SStefano Zampini ierr = MatCreate(comm,&A2);CHKERRQ(ierr); 527fabe8965SStefano Zampini ierr = MatSetType(A2,MATMPIAIJ);CHKERRQ(ierr); 528fabe8965SStefano Zampini ierr = MatSetSizes(A2,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 529fabe8965SStefano Zampini ierr = MatMPIAIJSetPreallocation(A2,0,dnz,0,onz);CHKERRQ(ierr); 530fabe8965SStefano Zampini ierr = MatSetOption(A2,MAT_NO_OFF_PROC_ENTRIES,PETSC_TRUE);CHKERRQ(ierr); 531fabe8965SStefano Zampini for (i = 0, j = 0; i < A2->rmap->n; i++) { 532fabe8965SStefano Zampini PetscInt row = i+A2->rmap->rstart,s = dnz[i] + onz[i]; 533fabe8965SStefano Zampini const PetscInt* idxs; 534fabe8965SStefano Zampini 535fabe8965SStefano Zampini if (s) { 536fabe8965SStefano Zampini ierr = ISGetIndices(workis[j],&idxs);CHKERRQ(ierr); 537fabe8965SStefano Zampini ierr = MatSetValues(A2,1,&row,s,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 538fabe8965SStefano Zampini ierr = ISRestoreIndices(workis[j],&idxs);CHKERRQ(ierr); 539fabe8965SStefano Zampini j++; 540fabe8965SStefano Zampini } 541fabe8965SStefano Zampini } 542fabe8965SStefano Zampini if (j != cnt) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected local count %D != %D",j,cnt); 543fabe8965SStefano Zampini ierr = PetscFree(vals);CHKERRQ(ierr); 544fabe8965SStefano Zampini ierr = MatAssemblyBegin(A2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 545fabe8965SStefano Zampini ierr = MatAssemblyEnd(A2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 546fabe8965SStefano Zampini ierr = MatTranspose(A2,MAT_INPLACE_MATRIX,&A2);CHKERRQ(ierr); 547fabe8965SStefano Zampini 548fabe8965SStefano Zampini /* extract submatrix corresponding to the coupling "owned separators" x "isolated separators" */ 549fabe8965SStefano Zampini for (i = 0, j = 0; i < nl; i++) 550fabe8965SStefano Zampini if (lndmapi[i] >= 0) lndmapi[j++] = lndmapi[i]; 551fabe8965SStefano Zampini ierr = ISCreateGeneral(comm,j,lndmapi,PETSC_USE_POINTER,&is);CHKERRQ(ierr); 552fabe8965SStefano Zampini ierr = MatMPIAIJGetLocalMatCondensed(A2,MAT_INITIAL_MATRIX,&is,NULL,&A3);CHKERRQ(ierr); 553fabe8965SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 554fabe8965SStefano Zampini ierr = MatDestroy(&A2);CHKERRQ(ierr); 555fabe8965SStefano Zampini 556fabe8965SStefano Zampini /* extend local to global map to include connected isolated separators */ 557fabe8965SStefano Zampini ierr = PetscObjectQuery((PetscObject)A3,"_petsc_GetLocalMatCondensed_iscol",(PetscObject*)&is);CHKERRQ(ierr); 558fabe8965SStefano Zampini if (!is) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Missing column map"); 559fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&ll2g);CHKERRQ(ierr); 560fabe8965SStefano Zampini ierr = MatGetRowIJ(A3,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&flg);CHKERRQ(ierr); 561fabe8965SStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure"); 562fabe8965SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,ii[i],jj,PETSC_COPY_VALUES,&is);CHKERRQ(ierr); 563fabe8965SStefano Zampini ierr = MatRestoreRowIJ(A3,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&flg);CHKERRQ(ierr); 564fabe8965SStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure"); 565fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingApplyIS(ll2g,is,&is2);CHKERRQ(ierr); 566fabe8965SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 567fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&ll2g);CHKERRQ(ierr); 568fabe8965SStefano Zampini 569fabe8965SStefano Zampini /* add new nodes to the local-to-global map */ 570fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(l2g);CHKERRQ(ierr); 571fabe8965SStefano Zampini ierr = ISExpand(ndsub,is2,&is);CHKERRQ(ierr); 572fabe8965SStefano Zampini ierr = ISDestroy(&is2);CHKERRQ(ierr); 573fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,l2g);CHKERRQ(ierr); 574fabe8965SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 575fabe8965SStefano Zampini 576fabe8965SStefano Zampini ierr = MatDestroy(&A3);CHKERRQ(ierr); 577fabe8965SStefano Zampini ierr = PetscFree(lndmapi);CHKERRQ(ierr); 578fabe8965SStefano Zampini ierr = MatPreallocateFinalize(dnz,onz);CHKERRQ(ierr); 579fabe8965SStefano Zampini for (i = 0; i < gcnt; i++) { 580fabe8965SStefano Zampini ierr = ISDestroy(&workis[i]);CHKERRQ(ierr); 581fabe8965SStefano Zampini } 582fabe8965SStefano Zampini ierr = PetscFree(workis);CHKERRQ(ierr); 583fabe8965SStefano Zampini } 584fabe8965SStefano Zampini ierr = ISRestoreIndices(ndmap,&ndmapi);CHKERRQ(ierr); 585fabe8965SStefano Zampini ierr = PetscSFDestroy(&sf);CHKERRQ(ierr); 586fabe8965SStefano Zampini ierr = PetscFree(ndmapc);CHKERRQ(ierr); 587fabe8965SStefano Zampini ierr = ISDestroy(&ndmap);CHKERRQ(ierr); 588fabe8965SStefano Zampini ierr = ISDestroy(&ndsub);CHKERRQ(ierr); 589fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingSetBlockSize(*l2g,bs);CHKERRQ(ierr); 590fabe8965SStefano Zampini ierr = ISLocalToGlobalMappingViewFromOptions(*l2g,NULL,"-matis_nd_l2g_view");CHKERRQ(ierr); 591fabe8965SStefano Zampini break; 592fabe8965SStefano Zampini case MAT_IS_DISASSEMBLE_L2G_NATURAL: 593fabe8965SStefano Zampini if (ismpiaij) { 594fabe8965SStefano Zampini ierr = MatMPIAIJGetSeqAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr); 595fabe8965SStefano Zampini } else if (ismpibaij) { 596fabe8965SStefano Zampini ierr = MatMPIBAIJGetSeqBAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr); 597c9225affSStefano Zampini } else SETERRQ1(comm,PETSC_ERR_SUP,"Type %s",((PetscObject)A)->type_name); 598c9225affSStefano Zampini if (!garray) SETERRQ(comm,PETSC_ERR_ARG_WRONGSTATE,"garray not present"); 599c9225affSStefano Zampini if (A->rmap->n) { 600fabe8965SStefano Zampini PetscInt dc,oc,stc,*aux; 601c9225affSStefano Zampini 602c9225affSStefano Zampini ierr = MatGetLocalSize(A,NULL,&dc);CHKERRQ(ierr); 603c9225affSStefano Zampini ierr = MatGetLocalSize(Ao,NULL,&oc);CHKERRQ(ierr); 604c9225affSStefano Zampini ierr = MatGetOwnershipRangeColumn(A,&stc,NULL);CHKERRQ(ierr); 605c9225affSStefano Zampini ierr = PetscMalloc1((dc+oc)/bs,&aux);CHKERRQ(ierr); 606c9225affSStefano Zampini for (i=0; i<dc/bs; i++) aux[i] = i+stc/bs; 607c9225affSStefano Zampini for (i=0; i<oc/bs; i++) aux[i+dc/bs] = garray[i]; 608c9225affSStefano Zampini ierr = ISCreateBlock(comm,bs,(dc+oc)/bs,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 609c9225affSStefano Zampini } else { 610c9225affSStefano Zampini ierr = ISCreateBlock(comm,1,0,NULL,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 611c9225affSStefano Zampini } 612c9225affSStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,l2g);CHKERRQ(ierr); 613c9225affSStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 614fabe8965SStefano Zampini break; 615fabe8965SStefano Zampini default: 616fabe8965SStefano Zampini SETERRQ1(comm,PETSC_ERR_ARG_WRONG,"Unsupported l2g disassembling type %D",mode); 617c9225affSStefano Zampini } 618c9225affSStefano Zampini PetscFunctionReturn(0); 619c9225affSStefano Zampini } 620c9225affSStefano Zampini 621c9225affSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_XAIJ_IS(Mat A,MatType type,MatReuse reuse,Mat *newmat) 622c9225affSStefano Zampini { 623c9225affSStefano Zampini Mat lA,Ad,Ao,B = NULL; 6246989cf23SStefano Zampini ISLocalToGlobalMapping rl2g,cl2g; 6256989cf23SStefano Zampini IS is; 6266989cf23SStefano Zampini MPI_Comm comm; 6276989cf23SStefano Zampini void *ptrs[2]; 6286989cf23SStefano Zampini const char *names[2] = {"_convert_csr_aux","_convert_csr_data"}; 629c9225affSStefano Zampini const PetscInt *garray; 6306989cf23SStefano Zampini PetscScalar *dd,*od,*aa,*data; 631c9225affSStefano Zampini const PetscInt *di,*dj,*oi,*oj; 632c9225affSStefano Zampini const PetscInt *odi,*odj,*ooi,*ooj; 6336989cf23SStefano Zampini PetscInt *aux,*ii,*jj; 634c9225affSStefano Zampini PetscInt bs,lc,dr,dc,oc,str,stc,nnz,i,jd,jo,cum; 635c9225affSStefano Zampini PetscBool flg,ismpiaij,ismpibaij,was_inplace = PETSC_FALSE; 636c9225affSStefano Zampini PetscMPIInt size; 6376989cf23SStefano Zampini PetscErrorCode ierr; 6386989cf23SStefano Zampini 639ab4d48faSStefano Zampini PetscFunctionBegin; 6406989cf23SStefano Zampini ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 641ffc4695bSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr); 642c9225affSStefano Zampini if (size == 1) { 643c9225affSStefano Zampini ierr = MatConvert_SeqXAIJ_IS(A,type,reuse,newmat);CHKERRQ(ierr); 644c9225affSStefano Zampini PetscFunctionReturn(0); 645c9225affSStefano Zampini } 646c9225affSStefano Zampini if (reuse != MAT_REUSE_MATRIX && A->cmap->N == A->rmap->N) { 647c9225affSStefano Zampini ierr = MatMPIXAIJComputeLocalToGlobalMapping_Private(A,&rl2g);CHKERRQ(ierr); 648c9225affSStefano Zampini ierr = MatCreate(comm,&B);CHKERRQ(ierr); 649c9225affSStefano Zampini ierr = MatSetType(B,MATIS);CHKERRQ(ierr); 650c9225affSStefano Zampini ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 651c9225affSStefano Zampini ierr = MatSetLocalToGlobalMapping(B,rl2g,rl2g);CHKERRQ(ierr); 652c9225affSStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 653c9225affSStefano Zampini ierr = MatSetBlockSize(B,bs);CHKERRQ(ierr); 654c9225affSStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 655c9225affSStefano Zampini if (reuse == MAT_INPLACE_MATRIX) was_inplace = PETSC_TRUE; 656c9225affSStefano Zampini reuse = MAT_REUSE_MATRIX; 657c9225affSStefano Zampini } 658c9225affSStefano Zampini if (reuse == MAT_REUSE_MATRIX) { 659c9225affSStefano Zampini Mat *newlA, lA; 660c9225affSStefano Zampini IS rows, cols; 661c9225affSStefano Zampini const PetscInt *ridx, *cidx; 662c9225affSStefano Zampini PetscInt rbs, cbs, nr, nc; 663c9225affSStefano Zampini 664c9225affSStefano Zampini if (!B) B = *newmat; 665c9225affSStefano Zampini ierr = MatGetLocalToGlobalMapping(B,&rl2g,&cl2g);CHKERRQ(ierr); 666c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(rl2g,&ridx);CHKERRQ(ierr); 667c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(cl2g,&cidx);CHKERRQ(ierr); 668c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetSize(rl2g,&nr);CHKERRQ(ierr); 669c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetSize(cl2g,&nc);CHKERRQ(ierr); 670c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(rl2g,&rbs);CHKERRQ(ierr); 671c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(cl2g,&cbs);CHKERRQ(ierr); 672c9225affSStefano Zampini ierr = ISCreateBlock(comm,rbs,nr/rbs,ridx,PETSC_USE_POINTER,&rows);CHKERRQ(ierr); 673c9225affSStefano Zampini if (rl2g != cl2g) { 674c9225affSStefano Zampini ierr = ISCreateBlock(comm,cbs,nc/cbs,cidx,PETSC_USE_POINTER,&cols);CHKERRQ(ierr); 675c9225affSStefano Zampini } else { 676c9225affSStefano Zampini ierr = PetscObjectReference((PetscObject)rows);CHKERRQ(ierr); 677c9225affSStefano Zampini cols = rows; 678c9225affSStefano Zampini } 679c9225affSStefano Zampini ierr = MatISGetLocalMat(B,&lA);CHKERRQ(ierr); 680c9225affSStefano Zampini ierr = MatCreateSubMatrices(A,1,&rows,&cols,MAT_INITIAL_MATRIX,&newlA);CHKERRQ(ierr); 681c9225affSStefano Zampini ierr = MatConvert(newlA[0],MATSEQAIJ,MAT_INPLACE_MATRIX,&newlA[0]);CHKERRQ(ierr); 682c9225affSStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(rl2g,&ridx);CHKERRQ(ierr); 683c9225affSStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(cl2g,&cidx);CHKERRQ(ierr); 684c9225affSStefano Zampini ierr = ISDestroy(&rows);CHKERRQ(ierr); 685c9225affSStefano Zampini ierr = ISDestroy(&cols);CHKERRQ(ierr); 686c9225affSStefano Zampini if (!lA->preallocated) { /* first time */ 687c9225affSStefano Zampini ierr = MatDuplicate(newlA[0],MAT_COPY_VALUES,&lA);CHKERRQ(ierr); 688c9225affSStefano Zampini ierr = MatISSetLocalMat(B,lA);CHKERRQ(ierr); 689c9225affSStefano Zampini ierr = PetscObjectDereference((PetscObject)lA);CHKERRQ(ierr); 690c9225affSStefano Zampini } 691c9225affSStefano Zampini ierr = MatCopy(newlA[0],lA,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 692c9225affSStefano Zampini ierr = MatDestroySubMatrices(1,&newlA);CHKERRQ(ierr); 693c9225affSStefano Zampini ierr = MatISScaleDisassembling_Private(B);CHKERRQ(ierr); 694c9225affSStefano Zampini ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 695c9225affSStefano Zampini ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 696c9225affSStefano Zampini if (was_inplace) { ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); } 697c9225affSStefano Zampini else *newmat = B; 698c9225affSStefano Zampini PetscFunctionReturn(0); 699c9225affSStefano Zampini } 700c9225affSStefano Zampini /* rectangular case, just compress out the column space */ 701b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ ,&ismpiaij);CHKERRQ(ierr); 702b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIBAIJ,&ismpibaij);CHKERRQ(ierr); 703c9225affSStefano Zampini if (ismpiaij) { 704c9225affSStefano Zampini bs = 1; 705c9225affSStefano Zampini ierr = MatMPIAIJGetSeqAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr); 706c9225affSStefano Zampini } else if (ismpibaij) { 707c9225affSStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 708c9225affSStefano Zampini ierr = MatMPIBAIJGetSeqBAIJ(A,&Ad,&Ao,&garray);CHKERRQ(ierr); 709c9225affSStefano Zampini ierr = MatConvert(Ad,MATSEQAIJ,MAT_INITIAL_MATRIX,&Ad);CHKERRQ(ierr); 710c9225affSStefano Zampini ierr = MatConvert(Ao,MATSEQAIJ,MAT_INITIAL_MATRIX,&Ao);CHKERRQ(ierr); 711c9225affSStefano Zampini } else SETERRQ1(comm,PETSC_ERR_SUP,"Type %s",((PetscObject)A)->type_name); 712c9225affSStefano Zampini ierr = MatSeqAIJGetArray(Ad,&dd);CHKERRQ(ierr); 713c9225affSStefano Zampini ierr = MatSeqAIJGetArray(Ao,&od);CHKERRQ(ierr); 714c9225affSStefano Zampini if (!garray) SETERRQ(comm,PETSC_ERR_ARG_WRONGSTATE,"garray not present"); 7156989cf23SStefano Zampini 7166989cf23SStefano Zampini /* access relevant information from MPIAIJ */ 7176989cf23SStefano Zampini ierr = MatGetOwnershipRange(A,&str,NULL);CHKERRQ(ierr); 7186989cf23SStefano Zampini ierr = MatGetOwnershipRangeColumn(A,&stc,NULL);CHKERRQ(ierr); 7196989cf23SStefano Zampini ierr = MatGetLocalSize(A,&dr,&dc);CHKERRQ(ierr); 720c9225affSStefano Zampini ierr = MatGetLocalSize(Ao,NULL,&oc);CHKERRQ(ierr); 721c9225affSStefano Zampini ierr = MatGetRowIJ(Ad,0,PETSC_FALSE,PETSC_FALSE,&i,&di,&dj,&flg);CHKERRQ(ierr); 722c9225affSStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure"); 723c9225affSStefano Zampini ierr = MatGetRowIJ(Ao,0,PETSC_FALSE,PETSC_FALSE,&i,&oi,&oj,&flg);CHKERRQ(ierr); 724c9225affSStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get IJ structure"); 725c9225affSStefano Zampini nnz = di[dr] + oi[dr]; 726c9225affSStefano Zampini /* store original pointers to be restored later */ 727c9225affSStefano Zampini odi = di; odj = dj; ooi = oi; ooj = oj; 7286989cf23SStefano Zampini 7296989cf23SStefano Zampini /* generate l2g maps for rows and cols */ 730c9225affSStefano Zampini ierr = ISCreateStride(comm,dr/bs,str/bs,1,&is);CHKERRQ(ierr); 731c9225affSStefano Zampini if (bs > 1) { 732c9225affSStefano Zampini IS is2; 733c9225affSStefano Zampini 734c9225affSStefano Zampini ierr = ISGetLocalSize(is,&i);CHKERRQ(ierr); 735c9225affSStefano Zampini ierr = ISGetIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr); 736c9225affSStefano Zampini ierr = ISCreateBlock(comm,bs,i,aux,PETSC_COPY_VALUES,&is2);CHKERRQ(ierr); 737c9225affSStefano Zampini ierr = ISRestoreIndices(is,(const PetscInt**)&aux);CHKERRQ(ierr); 738c9225affSStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 739c9225affSStefano Zampini is = is2; 740c9225affSStefano Zampini } 7416989cf23SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr); 7426989cf23SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 743e363d98aSStefano Zampini if (dr) { 744c9225affSStefano Zampini ierr = PetscMalloc1((dc+oc)/bs,&aux);CHKERRQ(ierr); 745c9225affSStefano Zampini for (i=0; i<dc/bs; i++) aux[i] = i+stc/bs; 746c9225affSStefano Zampini for (i=0; i<oc/bs; i++) aux[i+dc/bs] = garray[i]; 747c9225affSStefano Zampini ierr = ISCreateBlock(comm,bs,(dc+oc)/bs,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 748e363d98aSStefano Zampini lc = dc+oc; 749e363d98aSStefano Zampini } else { 750c9225affSStefano Zampini ierr = ISCreateBlock(comm,bs,0,NULL,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 751e363d98aSStefano Zampini lc = 0; 752e363d98aSStefano Zampini } 7536989cf23SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr); 7546989cf23SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 7556989cf23SStefano Zampini 7566989cf23SStefano Zampini /* create MATIS object */ 757c9225affSStefano Zampini ierr = MatCreate(comm,&B);CHKERRQ(ierr); 758c9225affSStefano Zampini ierr = MatSetSizes(B,dr,dc,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 759c9225affSStefano Zampini ierr = MatSetType(B,MATIS);CHKERRQ(ierr); 760c9225affSStefano Zampini ierr = MatSetBlockSize(B,bs);CHKERRQ(ierr); 761c9225affSStefano Zampini ierr = MatSetLocalToGlobalMapping(B,rl2g,cl2g);CHKERRQ(ierr); 7626989cf23SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 7636989cf23SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 7646989cf23SStefano Zampini 7656989cf23SStefano Zampini /* merge local matrices */ 7666989cf23SStefano Zampini ierr = PetscMalloc1(nnz+dr+1,&aux);CHKERRQ(ierr); 7676989cf23SStefano Zampini ierr = PetscMalloc1(nnz,&data);CHKERRQ(ierr); 7686989cf23SStefano Zampini ii = aux; 7696989cf23SStefano Zampini jj = aux+dr+1; 7706989cf23SStefano Zampini aa = data; 7716989cf23SStefano Zampini *ii = *(di++) + *(oi++); 7726989cf23SStefano Zampini for (jd=0,jo=0,cum=0;*ii<nnz;cum++) 7736989cf23SStefano Zampini { 7746989cf23SStefano Zampini for (;jd<*di;jd++) { *jj++ = *dj++; *aa++ = *dd++; } 7756989cf23SStefano Zampini for (;jo<*oi;jo++) { *jj++ = *oj++ + dc; *aa++ = *od++; } 7766989cf23SStefano Zampini *(++ii) = *(di++) + *(oi++); 7776989cf23SStefano Zampini } 7786989cf23SStefano Zampini for (;cum<dr;cum++) *(++ii) = nnz; 779c9225affSStefano Zampini 780c9225affSStefano Zampini ierr = MatRestoreRowIJ(Ad,0,PETSC_FALSE,PETSC_FALSE,&i,&odi,&odj,&flg);CHKERRQ(ierr); 781c9225affSStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot restore IJ structure"); 782c9225affSStefano Zampini ierr = MatRestoreRowIJ(Ao,0,PETSC_FALSE,PETSC_FALSE,&i,&ooi,&ooj,&flg);CHKERRQ(ierr); 783c9225affSStefano Zampini if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot restore IJ structure"); 784c9225affSStefano Zampini ierr = MatSeqAIJRestoreArray(Ad,&dd);CHKERRQ(ierr); 785c9225affSStefano Zampini ierr = MatSeqAIJRestoreArray(Ao,&od);CHKERRQ(ierr); 786c9225affSStefano Zampini 7876989cf23SStefano Zampini ii = aux; 7886989cf23SStefano Zampini jj = aux+dr+1; 7896989cf23SStefano Zampini aa = data; 790e363d98aSStefano Zampini ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,dr,lc,ii,jj,aa,&lA);CHKERRQ(ierr); 7916989cf23SStefano Zampini 7926989cf23SStefano Zampini /* create containers to destroy the data */ 7936989cf23SStefano Zampini ptrs[0] = aux; 7946989cf23SStefano Zampini ptrs[1] = data; 7956989cf23SStefano Zampini for (i=0; i<2; i++) { 7966989cf23SStefano Zampini PetscContainer c; 7976989cf23SStefano Zampini 7986989cf23SStefano Zampini ierr = PetscContainerCreate(PETSC_COMM_SELF,&c);CHKERRQ(ierr); 7996989cf23SStefano Zampini ierr = PetscContainerSetPointer(c,ptrs[i]);CHKERRQ(ierr); 800b81c21eeSStefano Zampini ierr = PetscContainerSetUserDestroy(c,PetscContainerUserDestroyDefault);CHKERRQ(ierr); 8016989cf23SStefano Zampini ierr = PetscObjectCompose((PetscObject)lA,names[i],(PetscObject)c);CHKERRQ(ierr); 8026989cf23SStefano Zampini ierr = PetscContainerDestroy(&c);CHKERRQ(ierr); 8036989cf23SStefano Zampini } 804c9225affSStefano Zampini if (ismpibaij) { /* destroy converted local matrices */ 805c9225affSStefano Zampini ierr = MatDestroy(&Ad);CHKERRQ(ierr); 806c9225affSStefano Zampini ierr = MatDestroy(&Ao);CHKERRQ(ierr); 807c9225affSStefano Zampini } 8086989cf23SStefano Zampini 8096989cf23SStefano Zampini /* finalize matrix */ 810c9225affSStefano Zampini ierr = MatISSetLocalMat(B,lA);CHKERRQ(ierr); 8116989cf23SStefano Zampini ierr = MatDestroy(&lA);CHKERRQ(ierr); 812c9225affSStefano Zampini ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 813c9225affSStefano Zampini ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 814c9225affSStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 815c9225affSStefano Zampini ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 816c9225affSStefano Zampini } else *newmat = B; 8176989cf23SStefano Zampini PetscFunctionReturn(0); 8186989cf23SStefano Zampini } 8196989cf23SStefano Zampini 8205e3038f0Sstefano_zampini PETSC_INTERN PetscErrorCode MatConvert_Nest_IS(Mat A,MatType type,MatReuse reuse,Mat *newmat) 8215e3038f0Sstefano_zampini { 8225e3038f0Sstefano_zampini Mat **nest,*snest,**rnest,lA,B; 8235e3038f0Sstefano_zampini IS *iscol,*isrow,*islrow,*islcol; 8245e3038f0Sstefano_zampini ISLocalToGlobalMapping rl2g,cl2g; 8255e3038f0Sstefano_zampini MPI_Comm comm; 8265b003df0Sstefano_zampini PetscInt *lr,*lc,*l2gidxs; 8275b003df0Sstefano_zampini PetscInt i,j,nr,nc,rbs,cbs; 8289e7b2b25Sstefano_zampini PetscBool convert,lreuse,*istrans; 8295e3038f0Sstefano_zampini PetscErrorCode ierr; 8305e3038f0Sstefano_zampini 831ab4d48faSStefano Zampini PetscFunctionBegin; 8325e3038f0Sstefano_zampini ierr = MatNestGetSubMats(A,&nr,&nc,&nest);CHKERRQ(ierr); 8335e3038f0Sstefano_zampini lreuse = PETSC_FALSE; 8345e3038f0Sstefano_zampini rnest = NULL; 8355e3038f0Sstefano_zampini if (reuse == MAT_REUSE_MATRIX) { 8365e3038f0Sstefano_zampini PetscBool ismatis,isnest; 8375e3038f0Sstefano_zampini 8385e3038f0Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)*newmat,MATIS,&ismatis);CHKERRQ(ierr); 839a32993e3SJed Brown if (!ismatis) SETERRQ1(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_USER,"Cannot reuse matrix of type %s",((PetscObject)(*newmat))->type_name); 8405e3038f0Sstefano_zampini ierr = MatISGetLocalMat(*newmat,&lA);CHKERRQ(ierr); 8415e3038f0Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)lA,MATNEST,&isnest);CHKERRQ(ierr); 8425e3038f0Sstefano_zampini if (isnest) { 8435e3038f0Sstefano_zampini ierr = MatNestGetSubMats(lA,&i,&j,&rnest);CHKERRQ(ierr); 8445e3038f0Sstefano_zampini lreuse = (PetscBool)(i == nr && j == nc); 8455e3038f0Sstefano_zampini if (!lreuse) rnest = NULL; 8465e3038f0Sstefano_zampini } 8475e3038f0Sstefano_zampini } 8485e3038f0Sstefano_zampini ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 8495b003df0Sstefano_zampini ierr = PetscCalloc2(nr,&lr,nc,&lc);CHKERRQ(ierr); 850071fcb05SBarry Smith ierr = PetscCalloc6(nr,&isrow,nc,&iscol,nr,&islrow,nc,&islcol,nr*nc,&snest,nr*nc,&istrans);CHKERRQ(ierr); 8515e3038f0Sstefano_zampini ierr = MatNestGetISs(A,isrow,iscol);CHKERRQ(ierr); 8525e3038f0Sstefano_zampini for (i=0;i<nr;i++) { 8535e3038f0Sstefano_zampini for (j=0;j<nc;j++) { 8545e3038f0Sstefano_zampini PetscBool ismatis; 8559e7b2b25Sstefano_zampini PetscInt l1,l2,lb1,lb2,ij=i*nc+j; 8565e3038f0Sstefano_zampini 8575e3038f0Sstefano_zampini /* Null matrix pointers are allowed in MATNEST */ 8585e3038f0Sstefano_zampini if (!nest[i][j]) continue; 8595e3038f0Sstefano_zampini 8605e3038f0Sstefano_zampini /* Nested matrices should be of type MATIS */ 8619e7b2b25Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)nest[i][j],MATTRANSPOSEMAT,&istrans[ij]);CHKERRQ(ierr); 8629e7b2b25Sstefano_zampini if (istrans[ij]) { 8639e7b2b25Sstefano_zampini Mat T,lT; 8649e7b2b25Sstefano_zampini ierr = MatTransposeGetMat(nest[i][j],&T);CHKERRQ(ierr); 8659e7b2b25Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)T,MATIS,&ismatis);CHKERRQ(ierr); 8669e7b2b25Sstefano_zampini if (!ismatis) SETERRQ2(comm,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) (transposed) is not of type MATIS",i,j); 8679e7b2b25Sstefano_zampini ierr = MatISGetLocalMat(T,&lT);CHKERRQ(ierr); 8689e7b2b25Sstefano_zampini ierr = MatCreateTranspose(lT,&snest[ij]);CHKERRQ(ierr); 8699e7b2b25Sstefano_zampini } else { 8705e3038f0Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)nest[i][j],MATIS,&ismatis);CHKERRQ(ierr); 8715e3038f0Sstefano_zampini if (!ismatis) SETERRQ2(comm,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) is not of type MATIS",i,j); 8729e7b2b25Sstefano_zampini ierr = MatISGetLocalMat(nest[i][j],&snest[ij]);CHKERRQ(ierr); 8739e7b2b25Sstefano_zampini } 8745e3038f0Sstefano_zampini 8755e3038f0Sstefano_zampini /* Check compatibility of local sizes */ 8765e3038f0Sstefano_zampini ierr = MatGetSize(snest[ij],&l1,&l2);CHKERRQ(ierr); 8779e7b2b25Sstefano_zampini ierr = MatGetBlockSizes(snest[ij],&lb1,&lb2);CHKERRQ(ierr); 8785e3038f0Sstefano_zampini if (!l1 || !l2) continue; 8795e3038f0Sstefano_zampini if (lr[i] && l1 != lr[i]) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid local size %D != %D",i,j,lr[i],l1); 8805e3038f0Sstefano_zampini if (lc[j] && l2 != lc[j]) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid local size %D != %D",i,j,lc[j],l2); 8815e3038f0Sstefano_zampini lr[i] = l1; 8825e3038f0Sstefano_zampini lc[j] = l2; 8835e3038f0Sstefano_zampini 8845e3038f0Sstefano_zampini /* check compatibilty for local matrix reusage */ 8855e3038f0Sstefano_zampini if (rnest && !rnest[i][j] != !snest[ij]) lreuse = PETSC_FALSE; 8865e3038f0Sstefano_zampini } 8875e3038f0Sstefano_zampini } 8885e3038f0Sstefano_zampini 88976bd3646SJed Brown if (PetscDefined (USE_DEBUG)) { 8905e3038f0Sstefano_zampini /* Check compatibility of l2g maps for rows */ 8915e3038f0Sstefano_zampini for (i=0;i<nr;i++) { 8925e3038f0Sstefano_zampini rl2g = NULL; 8935e3038f0Sstefano_zampini for (j=0;j<nc;j++) { 8945e3038f0Sstefano_zampini PetscInt n1,n2; 8955e3038f0Sstefano_zampini 8965e3038f0Sstefano_zampini if (!nest[i][j]) continue; 8979e7b2b25Sstefano_zampini if (istrans[i*nc+j]) { 8989e7b2b25Sstefano_zampini Mat T; 8999e7b2b25Sstefano_zampini 9009e7b2b25Sstefano_zampini ierr = MatTransposeGetMat(nest[i][j],&T);CHKERRQ(ierr); 9019e7b2b25Sstefano_zampini ierr = MatGetLocalToGlobalMapping(T,NULL,&cl2g);CHKERRQ(ierr); 9029e7b2b25Sstefano_zampini } else { 9035e3038f0Sstefano_zampini ierr = MatGetLocalToGlobalMapping(nest[i][j],&cl2g,NULL);CHKERRQ(ierr); 9049e7b2b25Sstefano_zampini } 9055e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetSize(cl2g,&n1);CHKERRQ(ierr); 9065e3038f0Sstefano_zampini if (!n1) continue; 9075e3038f0Sstefano_zampini if (!rl2g) { 9085e3038f0Sstefano_zampini rl2g = cl2g; 9095e3038f0Sstefano_zampini } else { 9105e3038f0Sstefano_zampini const PetscInt *idxs1,*idxs2; 9115e3038f0Sstefano_zampini PetscBool same; 9125e3038f0Sstefano_zampini 9135e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetSize(rl2g,&n2);CHKERRQ(ierr); 9145e3038f0Sstefano_zampini if (n1 != n2) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid row l2gmap size %D != %D",i,j,n1,n2); 9155e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetIndices(cl2g,&idxs1);CHKERRQ(ierr); 9165e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetIndices(rl2g,&idxs2);CHKERRQ(ierr); 917580bdb30SBarry Smith ierr = PetscArraycmp(idxs1,idxs2,n1,&same);CHKERRQ(ierr); 9185e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingRestoreIndices(cl2g,&idxs1);CHKERRQ(ierr); 9195e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingRestoreIndices(rl2g,&idxs2);CHKERRQ(ierr); 9205e3038f0Sstefano_zampini if (!same) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid row l2gmap",i,j); 9215e3038f0Sstefano_zampini } 9225e3038f0Sstefano_zampini } 9235e3038f0Sstefano_zampini } 9245e3038f0Sstefano_zampini /* Check compatibility of l2g maps for columns */ 9255e3038f0Sstefano_zampini for (i=0;i<nc;i++) { 9265e3038f0Sstefano_zampini rl2g = NULL; 9275e3038f0Sstefano_zampini for (j=0;j<nr;j++) { 9285e3038f0Sstefano_zampini PetscInt n1,n2; 9295e3038f0Sstefano_zampini 9305e3038f0Sstefano_zampini if (!nest[j][i]) continue; 9319e7b2b25Sstefano_zampini if (istrans[j*nc+i]) { 9329e7b2b25Sstefano_zampini Mat T; 9339e7b2b25Sstefano_zampini 9349e7b2b25Sstefano_zampini ierr = MatTransposeGetMat(nest[j][i],&T);CHKERRQ(ierr); 9359e7b2b25Sstefano_zampini ierr = MatGetLocalToGlobalMapping(T,&cl2g,NULL);CHKERRQ(ierr); 9369e7b2b25Sstefano_zampini } else { 9375e3038f0Sstefano_zampini ierr = MatGetLocalToGlobalMapping(nest[j][i],NULL,&cl2g);CHKERRQ(ierr); 9389e7b2b25Sstefano_zampini } 9395e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetSize(cl2g,&n1);CHKERRQ(ierr); 9405e3038f0Sstefano_zampini if (!n1) continue; 9415e3038f0Sstefano_zampini if (!rl2g) { 9425e3038f0Sstefano_zampini rl2g = cl2g; 9435e3038f0Sstefano_zampini } else { 9445e3038f0Sstefano_zampini const PetscInt *idxs1,*idxs2; 9455e3038f0Sstefano_zampini PetscBool same; 9465e3038f0Sstefano_zampini 9475e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetSize(rl2g,&n2);CHKERRQ(ierr); 9485e3038f0Sstefano_zampini if (n1 != n2) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid column l2gmap size %D != %D",j,i,n1,n2); 9495e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetIndices(cl2g,&idxs1);CHKERRQ(ierr); 9505e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingGetIndices(rl2g,&idxs2);CHKERRQ(ierr); 951580bdb30SBarry Smith ierr = PetscArraycmp(idxs1,idxs2,n1,&same);CHKERRQ(ierr); 9525e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingRestoreIndices(cl2g,&idxs1);CHKERRQ(ierr); 9535e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingRestoreIndices(rl2g,&idxs2);CHKERRQ(ierr); 9545e3038f0Sstefano_zampini if (!same) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot convert from MATNEST to MATIS! Matrix block (%D,%D) has invalid column l2gmap",j,i); 9555e3038f0Sstefano_zampini } 9565e3038f0Sstefano_zampini } 9575e3038f0Sstefano_zampini } 95876bd3646SJed Brown } 9595e3038f0Sstefano_zampini 9605e3038f0Sstefano_zampini B = NULL; 9615e3038f0Sstefano_zampini if (reuse != MAT_REUSE_MATRIX) { 9625b003df0Sstefano_zampini PetscInt stl; 9635b003df0Sstefano_zampini 9645e3038f0Sstefano_zampini /* Create l2g map for the rows of the new matrix and index sets for the local MATNEST */ 9655e3038f0Sstefano_zampini for (i=0,stl=0;i<nr;i++) stl += lr[i]; 9665e3038f0Sstefano_zampini ierr = PetscMalloc1(stl,&l2gidxs);CHKERRQ(ierr); 9675b003df0Sstefano_zampini for (i=0,stl=0;i<nr;i++) { 9685e3038f0Sstefano_zampini Mat usedmat; 9695e3038f0Sstefano_zampini Mat_IS *matis; 9705e3038f0Sstefano_zampini const PetscInt *idxs; 9715e3038f0Sstefano_zampini 9725e3038f0Sstefano_zampini /* local IS for local NEST */ 9735b003df0Sstefano_zampini ierr = ISCreateStride(PETSC_COMM_SELF,lr[i],stl,1,&islrow[i]);CHKERRQ(ierr); 9745e3038f0Sstefano_zampini 9755e3038f0Sstefano_zampini /* l2gmap */ 9765e3038f0Sstefano_zampini j = 0; 9775e3038f0Sstefano_zampini usedmat = nest[i][j]; 9789e7b2b25Sstefano_zampini while (!usedmat && j < nc-1) usedmat = nest[i][++j]; 9799e7b2b25Sstefano_zampini if (!usedmat) SETERRQ(comm,PETSC_ERR_SUP,"Cannot find valid row mat"); 9809e7b2b25Sstefano_zampini 9819e7b2b25Sstefano_zampini if (istrans[i*nc+j]) { 9829e7b2b25Sstefano_zampini Mat T; 9839e7b2b25Sstefano_zampini ierr = MatTransposeGetMat(usedmat,&T);CHKERRQ(ierr); 9849e7b2b25Sstefano_zampini usedmat = T; 9859e7b2b25Sstefano_zampini } 9865e3038f0Sstefano_zampini matis = (Mat_IS*)(usedmat->data); 9875e3038f0Sstefano_zampini ierr = ISGetIndices(isrow[i],&idxs);CHKERRQ(ierr); 9889e7b2b25Sstefano_zampini if (istrans[i*nc+j]) { 989ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 990ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 9919e7b2b25Sstefano_zampini } else { 992ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 993ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 9949e7b2b25Sstefano_zampini } 9955e3038f0Sstefano_zampini ierr = ISRestoreIndices(isrow[i],&idxs);CHKERRQ(ierr); 9965e3038f0Sstefano_zampini stl += lr[i]; 9975e3038f0Sstefano_zampini } 9985e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingCreate(comm,1,stl,l2gidxs,PETSC_OWN_POINTER,&rl2g);CHKERRQ(ierr); 9995e3038f0Sstefano_zampini 10005e3038f0Sstefano_zampini /* Create l2g map for columns of the new matrix and index sets for the local MATNEST */ 10015e3038f0Sstefano_zampini for (i=0,stl=0;i<nc;i++) stl += lc[i]; 10025e3038f0Sstefano_zampini ierr = PetscMalloc1(stl,&l2gidxs);CHKERRQ(ierr); 10035b003df0Sstefano_zampini for (i=0,stl=0;i<nc;i++) { 10045e3038f0Sstefano_zampini Mat usedmat; 10055e3038f0Sstefano_zampini Mat_IS *matis; 10065e3038f0Sstefano_zampini const PetscInt *idxs; 10075e3038f0Sstefano_zampini 10085e3038f0Sstefano_zampini /* local IS for local NEST */ 10095b003df0Sstefano_zampini ierr = ISCreateStride(PETSC_COMM_SELF,lc[i],stl,1,&islcol[i]);CHKERRQ(ierr); 10105e3038f0Sstefano_zampini 10115e3038f0Sstefano_zampini /* l2gmap */ 10125e3038f0Sstefano_zampini j = 0; 10135e3038f0Sstefano_zampini usedmat = nest[j][i]; 10149e7b2b25Sstefano_zampini while (!usedmat && j < nr-1) usedmat = nest[++j][i]; 10159e7b2b25Sstefano_zampini if (!usedmat) SETERRQ(comm,PETSC_ERR_SUP,"Cannot find valid column mat"); 10169e7b2b25Sstefano_zampini if (istrans[j*nc+i]) { 10179e7b2b25Sstefano_zampini Mat T; 10189e7b2b25Sstefano_zampini ierr = MatTransposeGetMat(usedmat,&T);CHKERRQ(ierr); 10199e7b2b25Sstefano_zampini usedmat = T; 10209e7b2b25Sstefano_zampini } 10215e3038f0Sstefano_zampini matis = (Mat_IS*)(usedmat->data); 10225e3038f0Sstefano_zampini ierr = ISGetIndices(iscol[i],&idxs);CHKERRQ(ierr); 10239e7b2b25Sstefano_zampini if (istrans[j*nc+i]) { 1024ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 1025ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 10269e7b2b25Sstefano_zampini } else { 1027ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 1028ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->csf,MPIU_INT,idxs,l2gidxs+stl,MPI_REPLACE);CHKERRQ(ierr); 10299e7b2b25Sstefano_zampini } 10305e3038f0Sstefano_zampini ierr = ISRestoreIndices(iscol[i],&idxs);CHKERRQ(ierr); 10315e3038f0Sstefano_zampini stl += lc[i]; 10325e3038f0Sstefano_zampini } 10335e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingCreate(comm,1,stl,l2gidxs,PETSC_OWN_POINTER,&cl2g);CHKERRQ(ierr); 10345e3038f0Sstefano_zampini 10355e3038f0Sstefano_zampini /* Create MATIS */ 10365e3038f0Sstefano_zampini ierr = MatCreate(comm,&B);CHKERRQ(ierr); 10375e3038f0Sstefano_zampini ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 10385e3038f0Sstefano_zampini ierr = MatGetBlockSizes(A,&rbs,&cbs);CHKERRQ(ierr); 10395e3038f0Sstefano_zampini ierr = MatSetBlockSizes(B,rbs,cbs);CHKERRQ(ierr); 10405e3038f0Sstefano_zampini ierr = MatSetType(B,MATIS);CHKERRQ(ierr); 10418546b261SStefano Zampini ierr = MatISSetLocalMatType(B,MATNEST);CHKERRQ(ierr); 10428546b261SStefano Zampini { /* hack : avoid setup of scatters */ 10438546b261SStefano Zampini Mat_IS *matis = (Mat_IS*)(B->data); 10448546b261SStefano Zampini matis->islocalref = PETSC_TRUE; 10458546b261SStefano Zampini } 10465e3038f0Sstefano_zampini ierr = MatSetLocalToGlobalMapping(B,rl2g,cl2g);CHKERRQ(ierr); 10475e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 10485e3038f0Sstefano_zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 10495e3038f0Sstefano_zampini ierr = MatCreateNest(PETSC_COMM_SELF,nr,islrow,nc,islcol,snest,&lA);CHKERRQ(ierr); 10508546b261SStefano Zampini ierr = MatNestSetVecType(lA,VECNEST);CHKERRQ(ierr); 10519e7b2b25Sstefano_zampini for (i=0;i<nr*nc;i++) { 10529e7b2b25Sstefano_zampini if (istrans[i]) { 10539e7b2b25Sstefano_zampini ierr = MatDestroy(&snest[i]);CHKERRQ(ierr); 10549e7b2b25Sstefano_zampini } 10559e7b2b25Sstefano_zampini } 10565e3038f0Sstefano_zampini ierr = MatISSetLocalMat(B,lA);CHKERRQ(ierr); 10575e3038f0Sstefano_zampini ierr = MatDestroy(&lA);CHKERRQ(ierr); 10588546b261SStefano Zampini { /* hack : setup of scatters done here */ 10598546b261SStefano Zampini Mat_IS *matis = (Mat_IS*)(B->data); 10608546b261SStefano Zampini 10618546b261SStefano Zampini matis->islocalref = PETSC_FALSE; 10628546b261SStefano Zampini ierr = MatISSetUpScatters_Private(B);CHKERRQ(ierr); 10638546b261SStefano Zampini } 10645e3038f0Sstefano_zampini ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 10655e3038f0Sstefano_zampini ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 10665e3038f0Sstefano_zampini if (reuse == MAT_INPLACE_MATRIX) { 10675e3038f0Sstefano_zampini ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 10685e3038f0Sstefano_zampini } else { 10695e3038f0Sstefano_zampini *newmat = B; 10705e3038f0Sstefano_zampini } 10715e3038f0Sstefano_zampini } else { 10725e3038f0Sstefano_zampini if (lreuse) { 10735e3038f0Sstefano_zampini ierr = MatISGetLocalMat(*newmat,&lA);CHKERRQ(ierr); 10745e3038f0Sstefano_zampini for (i=0;i<nr;i++) { 10755e3038f0Sstefano_zampini for (j=0;j<nc;j++) { 10765e3038f0Sstefano_zampini if (snest[i*nc+j]) { 10775e3038f0Sstefano_zampini ierr = MatNestSetSubMat(lA,i,j,snest[i*nc+j]);CHKERRQ(ierr); 10789e7b2b25Sstefano_zampini if (istrans[i*nc+j]) { 10799e7b2b25Sstefano_zampini ierr = MatDestroy(&snest[i*nc+j]);CHKERRQ(ierr); 10809e7b2b25Sstefano_zampini } 10815e3038f0Sstefano_zampini } 10825e3038f0Sstefano_zampini } 10835e3038f0Sstefano_zampini } 10845e3038f0Sstefano_zampini } else { 10855b003df0Sstefano_zampini PetscInt stl; 10865b003df0Sstefano_zampini for (i=0,stl=0;i<nr;i++) { 10875b003df0Sstefano_zampini ierr = ISCreateStride(PETSC_COMM_SELF,lr[i],stl,1,&islrow[i]);CHKERRQ(ierr); 10885b003df0Sstefano_zampini stl += lr[i]; 10895e3038f0Sstefano_zampini } 10905b003df0Sstefano_zampini for (i=0,stl=0;i<nc;i++) { 10915b003df0Sstefano_zampini ierr = ISCreateStride(PETSC_COMM_SELF,lc[i],stl,1,&islcol[i]);CHKERRQ(ierr); 10925b003df0Sstefano_zampini stl += lc[i]; 10935e3038f0Sstefano_zampini } 10945e3038f0Sstefano_zampini ierr = MatCreateNest(PETSC_COMM_SELF,nr,islrow,nc,islcol,snest,&lA);CHKERRQ(ierr); 1095ab4d48faSStefano Zampini for (i=0;i<nr*nc;i++) { 10969e7b2b25Sstefano_zampini if (istrans[i]) { 10979e7b2b25Sstefano_zampini ierr = MatDestroy(&snest[i]);CHKERRQ(ierr); 10989e7b2b25Sstefano_zampini } 1099ab4d48faSStefano Zampini } 11005e3038f0Sstefano_zampini ierr = MatISSetLocalMat(*newmat,lA);CHKERRQ(ierr); 11015e3038f0Sstefano_zampini ierr = MatDestroy(&lA);CHKERRQ(ierr); 11025e3038f0Sstefano_zampini } 11035e3038f0Sstefano_zampini ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11045e3038f0Sstefano_zampini ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11055e3038f0Sstefano_zampini } 11065e3038f0Sstefano_zampini 11075b003df0Sstefano_zampini /* Create local matrix in MATNEST format */ 11085b003df0Sstefano_zampini convert = PETSC_FALSE; 11095b003df0Sstefano_zampini ierr = PetscOptionsGetBool(NULL,((PetscObject)A)->prefix,"-matis_convert_local_nest",&convert,NULL);CHKERRQ(ierr); 11105b003df0Sstefano_zampini if (convert) { 11115b003df0Sstefano_zampini Mat M; 11125b003df0Sstefano_zampini MatISLocalFields lf; 11135b003df0Sstefano_zampini PetscContainer c; 11145b003df0Sstefano_zampini 11155b003df0Sstefano_zampini ierr = MatISGetLocalMat(*newmat,&lA);CHKERRQ(ierr); 11165b003df0Sstefano_zampini ierr = MatConvert(lA,MATAIJ,MAT_INITIAL_MATRIX,&M);CHKERRQ(ierr); 11175b003df0Sstefano_zampini ierr = MatISSetLocalMat(*newmat,M);CHKERRQ(ierr); 11185b003df0Sstefano_zampini ierr = MatDestroy(&M);CHKERRQ(ierr); 11195b003df0Sstefano_zampini 11205b003df0Sstefano_zampini /* attach local fields to the matrix */ 11215b003df0Sstefano_zampini ierr = PetscNew(&lf);CHKERRQ(ierr); 1122071fcb05SBarry Smith ierr = PetscMalloc2(nr,&lf->rf,nc,&lf->cf);CHKERRQ(ierr); 11235b003df0Sstefano_zampini for (i=0;i<nr;i++) { 11245b003df0Sstefano_zampini PetscInt n,st; 11255b003df0Sstefano_zampini 11265b003df0Sstefano_zampini ierr = ISGetLocalSize(islrow[i],&n);CHKERRQ(ierr); 11275b003df0Sstefano_zampini ierr = ISStrideGetInfo(islrow[i],&st,NULL);CHKERRQ(ierr); 11285b003df0Sstefano_zampini ierr = ISCreateStride(comm,n,st,1,&lf->rf[i]);CHKERRQ(ierr); 11295b003df0Sstefano_zampini } 11305b003df0Sstefano_zampini for (i=0;i<nc;i++) { 11315b003df0Sstefano_zampini PetscInt n,st; 11325b003df0Sstefano_zampini 11335b003df0Sstefano_zampini ierr = ISGetLocalSize(islcol[i],&n);CHKERRQ(ierr); 11345b003df0Sstefano_zampini ierr = ISStrideGetInfo(islcol[i],&st,NULL);CHKERRQ(ierr); 11355b003df0Sstefano_zampini ierr = ISCreateStride(comm,n,st,1,&lf->cf[i]);CHKERRQ(ierr); 11365b003df0Sstefano_zampini } 11375b003df0Sstefano_zampini lf->nr = nr; 11385b003df0Sstefano_zampini lf->nc = nc; 11395b003df0Sstefano_zampini ierr = PetscContainerCreate(PetscObjectComm((PetscObject)(*newmat)),&c);CHKERRQ(ierr); 11405b003df0Sstefano_zampini ierr = PetscContainerSetPointer(c,lf);CHKERRQ(ierr); 11415b003df0Sstefano_zampini ierr = PetscContainerSetUserDestroy(c,MatISContainerDestroyFields_Private);CHKERRQ(ierr); 11425b003df0Sstefano_zampini ierr = PetscObjectCompose((PetscObject)(*newmat),"_convert_nest_lfields",(PetscObject)c);CHKERRQ(ierr); 11435b003df0Sstefano_zampini ierr = PetscContainerDestroy(&c);CHKERRQ(ierr); 11445b003df0Sstefano_zampini } 11455b003df0Sstefano_zampini 11465e3038f0Sstefano_zampini /* Free workspace */ 11475e3038f0Sstefano_zampini for (i=0;i<nr;i++) { 11485e3038f0Sstefano_zampini ierr = ISDestroy(&islrow[i]);CHKERRQ(ierr); 11495e3038f0Sstefano_zampini } 11505e3038f0Sstefano_zampini for (i=0;i<nc;i++) { 11515e3038f0Sstefano_zampini ierr = ISDestroy(&islcol[i]);CHKERRQ(ierr); 11525e3038f0Sstefano_zampini } 11539e7b2b25Sstefano_zampini ierr = PetscFree6(isrow,iscol,islrow,islcol,snest,istrans);CHKERRQ(ierr); 11545b003df0Sstefano_zampini ierr = PetscFree2(lr,lc);CHKERRQ(ierr); 11555e3038f0Sstefano_zampini PetscFunctionReturn(0); 11565e3038f0Sstefano_zampini } 11575e3038f0Sstefano_zampini 1158ad219c80Sstefano_zampini static PetscErrorCode MatDiagonalScale_IS(Mat A, Vec l, Vec r) 1159ad219c80Sstefano_zampini { 1160ad219c80Sstefano_zampini Mat_IS *matis = (Mat_IS*)A->data; 1161ad219c80Sstefano_zampini Vec ll,rr; 1162ad219c80Sstefano_zampini const PetscScalar *Y,*X; 1163ad219c80Sstefano_zampini PetscScalar *x,*y; 1164ad219c80Sstefano_zampini PetscErrorCode ierr; 1165ad219c80Sstefano_zampini 1166ad219c80Sstefano_zampini PetscFunctionBegin; 1167ad219c80Sstefano_zampini if (l) { 1168ad219c80Sstefano_zampini ll = matis->y; 1169ad219c80Sstefano_zampini ierr = VecGetArrayRead(l,&Y);CHKERRQ(ierr); 1170ad219c80Sstefano_zampini ierr = VecGetArray(ll,&y);CHKERRQ(ierr); 1171ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->sf,MPIU_SCALAR,Y,y,MPI_REPLACE);CHKERRQ(ierr); 1172ad219c80Sstefano_zampini } else { 1173ad219c80Sstefano_zampini ll = NULL; 1174ad219c80Sstefano_zampini } 1175ad219c80Sstefano_zampini if (r) { 1176ad219c80Sstefano_zampini rr = matis->x; 1177ad219c80Sstefano_zampini ierr = VecGetArrayRead(r,&X);CHKERRQ(ierr); 1178ad219c80Sstefano_zampini ierr = VecGetArray(rr,&x);CHKERRQ(ierr); 1179ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->csf,MPIU_SCALAR,X,x,MPI_REPLACE);CHKERRQ(ierr); 1180ad219c80Sstefano_zampini } else { 1181ad219c80Sstefano_zampini rr = NULL; 1182ad219c80Sstefano_zampini } 1183ad219c80Sstefano_zampini if (ll) { 1184ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->sf,MPIU_SCALAR,Y,y,MPI_REPLACE);CHKERRQ(ierr); 1185ad219c80Sstefano_zampini ierr = VecRestoreArrayRead(l,&Y);CHKERRQ(ierr); 1186ad219c80Sstefano_zampini ierr = VecRestoreArray(ll,&y);CHKERRQ(ierr); 1187ad219c80Sstefano_zampini } 1188ad219c80Sstefano_zampini if (rr) { 1189ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->csf,MPIU_SCALAR,X,x,MPI_REPLACE);CHKERRQ(ierr); 1190ad219c80Sstefano_zampini ierr = VecRestoreArrayRead(r,&X);CHKERRQ(ierr); 1191ad219c80Sstefano_zampini ierr = VecRestoreArray(rr,&x);CHKERRQ(ierr); 1192ad219c80Sstefano_zampini } 1193ad219c80Sstefano_zampini ierr = MatDiagonalScale(matis->A,ll,rr);CHKERRQ(ierr); 1194ad219c80Sstefano_zampini PetscFunctionReturn(0); 1195ad219c80Sstefano_zampini } 1196ad219c80Sstefano_zampini 11977fa8f2d3SStefano Zampini static PetscErrorCode MatGetInfo_IS(Mat A,MatInfoType flag,MatInfo *ginfo) 11987fa8f2d3SStefano Zampini { 11997fa8f2d3SStefano Zampini Mat_IS *matis = (Mat_IS*)A->data; 12007fa8f2d3SStefano Zampini MatInfo info; 12013966268fSBarry Smith PetscLogDouble isend[6],irecv[6]; 12027fa8f2d3SStefano Zampini PetscInt bs; 12037fa8f2d3SStefano Zampini PetscErrorCode ierr; 12047fa8f2d3SStefano Zampini 12057fa8f2d3SStefano Zampini PetscFunctionBegin; 12067fa8f2d3SStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 1207a2ccb5f9Sstefano_zampini if (matis->A->ops->getinfo) { 12087fa8f2d3SStefano Zampini ierr = MatGetInfo(matis->A,MAT_LOCAL,&info);CHKERRQ(ierr); 12097fa8f2d3SStefano Zampini isend[0] = info.nz_used; 12107fa8f2d3SStefano Zampini isend[1] = info.nz_allocated; 12117fa8f2d3SStefano Zampini isend[2] = info.nz_unneeded; 12127fa8f2d3SStefano Zampini isend[3] = info.memory; 12137fa8f2d3SStefano Zampini isend[4] = info.mallocs; 1214a2ccb5f9Sstefano_zampini } else { 1215a2ccb5f9Sstefano_zampini isend[0] = 0.; 1216a2ccb5f9Sstefano_zampini isend[1] = 0.; 1217a2ccb5f9Sstefano_zampini isend[2] = 0.; 1218a2ccb5f9Sstefano_zampini isend[3] = 0.; 1219a2ccb5f9Sstefano_zampini isend[4] = 0.; 1220a2ccb5f9Sstefano_zampini } 1221314ce898Sstefano_zampini isend[5] = matis->A->num_ass; 12227fa8f2d3SStefano Zampini if (flag == MAT_LOCAL) { 12237fa8f2d3SStefano Zampini ginfo->nz_used = isend[0]; 12247fa8f2d3SStefano Zampini ginfo->nz_allocated = isend[1]; 12257fa8f2d3SStefano Zampini ginfo->nz_unneeded = isend[2]; 12267fa8f2d3SStefano Zampini ginfo->memory = isend[3]; 12277fa8f2d3SStefano Zampini ginfo->mallocs = isend[4]; 1228314ce898Sstefano_zampini ginfo->assemblies = isend[5]; 12297fa8f2d3SStefano Zampini } else if (flag == MAT_GLOBAL_MAX) { 1230820f2d46SBarry Smith ierr = MPIU_Allreduce(isend,irecv,6,MPIU_PETSCLOGDOUBLE,MPI_MAX,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 12317fa8f2d3SStefano Zampini 12327fa8f2d3SStefano Zampini ginfo->nz_used = irecv[0]; 12337fa8f2d3SStefano Zampini ginfo->nz_allocated = irecv[1]; 12347fa8f2d3SStefano Zampini ginfo->nz_unneeded = irecv[2]; 12357fa8f2d3SStefano Zampini ginfo->memory = irecv[3]; 12367fa8f2d3SStefano Zampini ginfo->mallocs = irecv[4]; 1237314ce898Sstefano_zampini ginfo->assemblies = irecv[5]; 12387fa8f2d3SStefano Zampini } else if (flag == MAT_GLOBAL_SUM) { 1239820f2d46SBarry Smith ierr = MPIU_Allreduce(isend,irecv,5,MPIU_PETSCLOGDOUBLE,MPI_SUM,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 12407fa8f2d3SStefano Zampini 12417fa8f2d3SStefano Zampini ginfo->nz_used = irecv[0]; 12427fa8f2d3SStefano Zampini ginfo->nz_allocated = irecv[1]; 12437fa8f2d3SStefano Zampini ginfo->nz_unneeded = irecv[2]; 12447fa8f2d3SStefano Zampini ginfo->memory = irecv[3]; 12457fa8f2d3SStefano Zampini ginfo->mallocs = irecv[4]; 12467fa8f2d3SStefano Zampini ginfo->assemblies = A->num_ass; 12477fa8f2d3SStefano Zampini } 12487fa8f2d3SStefano Zampini ginfo->block_size = bs; 12497fa8f2d3SStefano Zampini ginfo->fill_ratio_given = 0; 12507fa8f2d3SStefano Zampini ginfo->fill_ratio_needed = 0; 12517fa8f2d3SStefano Zampini ginfo->factor_mallocs = 0; 12525e3038f0Sstefano_zampini PetscFunctionReturn(0); 12535e3038f0Sstefano_zampini } 12545e3038f0Sstefano_zampini 1255d7f69cd0SStefano Zampini PetscErrorCode MatTranspose_IS(Mat A,MatReuse reuse,Mat *B) 1256d7f69cd0SStefano Zampini { 1257d7f69cd0SStefano Zampini Mat C,lC,lA; 1258d7f69cd0SStefano Zampini PetscErrorCode ierr; 1259d7f69cd0SStefano Zampini 1260d7f69cd0SStefano Zampini PetscFunctionBegin; 1261cf37664fSBarry Smith if (reuse == MAT_INITIAL_MATRIX || reuse == MAT_INPLACE_MATRIX) { 1262cf37664fSBarry Smith ISLocalToGlobalMapping rl2g,cl2g; 1263d7f69cd0SStefano Zampini ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 1264d7f69cd0SStefano Zampini ierr = MatSetSizes(C,A->cmap->n,A->rmap->n,A->cmap->N,A->rmap->N);CHKERRQ(ierr); 1265d7f69cd0SStefano Zampini ierr = MatSetBlockSizes(C,PetscAbs(A->cmap->bs),PetscAbs(A->rmap->bs));CHKERRQ(ierr); 1266d7f69cd0SStefano Zampini ierr = MatSetType(C,MATIS);CHKERRQ(ierr); 1267d7f69cd0SStefano Zampini ierr = MatGetLocalToGlobalMapping(A,&rl2g,&cl2g);CHKERRQ(ierr); 1268d7f69cd0SStefano Zampini ierr = MatSetLocalToGlobalMapping(C,cl2g,rl2g);CHKERRQ(ierr); 1269cf37664fSBarry Smith } else { 1270cf37664fSBarry Smith C = *B; 1271d7f69cd0SStefano Zampini } 1272d7f69cd0SStefano Zampini 1273d7f69cd0SStefano Zampini /* perform local transposition */ 1274d7f69cd0SStefano Zampini ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr); 1275d7f69cd0SStefano Zampini ierr = MatTranspose(lA,MAT_INITIAL_MATRIX,&lC);CHKERRQ(ierr); 1276d7f69cd0SStefano Zampini ierr = MatISSetLocalMat(C,lC);CHKERRQ(ierr); 1277d7f69cd0SStefano Zampini ierr = MatDestroy(&lC);CHKERRQ(ierr); 1278d7f69cd0SStefano Zampini 1279cf37664fSBarry Smith if (reuse == MAT_INITIAL_MATRIX || reuse == MAT_REUSE_MATRIX) { 1280d7f69cd0SStefano Zampini *B = C; 1281d7f69cd0SStefano Zampini } else { 1282d7f69cd0SStefano Zampini ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr); 1283d7f69cd0SStefano Zampini } 12847aa7aec5Sstefano_zampini ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 12857aa7aec5Sstefano_zampini ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1286d7f69cd0SStefano Zampini PetscFunctionReturn(0); 1287d7f69cd0SStefano Zampini } 1288d7f69cd0SStefano Zampini 12893fd1c9e7SStefano Zampini PetscErrorCode MatDiagonalSet_IS(Mat A,Vec D,InsertMode insmode) 12903fd1c9e7SStefano Zampini { 12913fd1c9e7SStefano Zampini Mat_IS *is = (Mat_IS*)A->data; 12923fd1c9e7SStefano Zampini PetscErrorCode ierr; 12933fd1c9e7SStefano Zampini 12943fd1c9e7SStefano Zampini PetscFunctionBegin; 12954b89b9cdSStefano Zampini if (D) { /* MatShift_IS pass D = NULL */ 12963fd1c9e7SStefano Zampini ierr = VecScatterBegin(is->rctx,D,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12973fd1c9e7SStefano Zampini ierr = VecScatterEnd(is->rctx,D,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12983fd1c9e7SStefano Zampini } 12993fd1c9e7SStefano Zampini ierr = VecPointwiseDivide(is->y,is->y,is->counter);CHKERRQ(ierr); 13003fd1c9e7SStefano Zampini ierr = MatDiagonalSet(is->A,is->y,insmode);CHKERRQ(ierr); 13013fd1c9e7SStefano Zampini PetscFunctionReturn(0); 13023fd1c9e7SStefano Zampini } 13033fd1c9e7SStefano Zampini 13043fd1c9e7SStefano Zampini PetscErrorCode MatShift_IS(Mat A,PetscScalar a) 13053fd1c9e7SStefano Zampini { 13064b89b9cdSStefano Zampini Mat_IS *is = (Mat_IS*)A->data; 13073fd1c9e7SStefano Zampini PetscErrorCode ierr; 13083fd1c9e7SStefano Zampini 13093fd1c9e7SStefano Zampini PetscFunctionBegin; 13104b89b9cdSStefano Zampini ierr = VecSet(is->y,a);CHKERRQ(ierr); 13113fd1c9e7SStefano Zampini ierr = MatDiagonalSet_IS(A,NULL,ADD_VALUES);CHKERRQ(ierr); 13123fd1c9e7SStefano Zampini PetscFunctionReturn(0); 13133fd1c9e7SStefano Zampini } 13143fd1c9e7SStefano Zampini 1315f26d0771SStefano Zampini static PetscErrorCode MatSetValuesLocal_SubMat_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv) 1316f26d0771SStefano Zampini { 1317f26d0771SStefano Zampini PetscErrorCode ierr; 1318f26d0771SStefano Zampini PetscInt rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION]; 1319f26d0771SStefano Zampini 1320f26d0771SStefano Zampini PetscFunctionBegin; 132176bd3646SJed Brown if (PetscUnlikely(m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION)) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n); 1322f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingApply(A->rmap->mapping,m,rows,rows_l);CHKERRQ(ierr); 1323f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingApply(A->cmap->mapping,n,cols,cols_l);CHKERRQ(ierr); 1324b4f971dfSStefano Zampini ierr = MatSetValuesLocal_IS(A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr); 1325f26d0771SStefano Zampini PetscFunctionReturn(0); 1326f26d0771SStefano Zampini } 1327f26d0771SStefano Zampini 1328f26d0771SStefano Zampini static PetscErrorCode MatSetValuesBlockedLocal_SubMat_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv) 1329f26d0771SStefano Zampini { 1330f26d0771SStefano Zampini PetscErrorCode ierr; 1331f26d0771SStefano Zampini PetscInt rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION]; 1332f26d0771SStefano Zampini 1333f26d0771SStefano Zampini PetscFunctionBegin; 133476bd3646SJed Brown if (PetscUnlikely(m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION)) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column block indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n); 1335f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingApplyBlock(A->rmap->mapping,m,rows,rows_l);CHKERRQ(ierr); 1336f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingApplyBlock(A->cmap->mapping,n,cols,cols_l);CHKERRQ(ierr); 1337b4f971dfSStefano Zampini ierr = MatSetValuesBlockedLocal_IS(A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr); 1338f26d0771SStefano Zampini PetscFunctionReturn(0); 1339f26d0771SStefano Zampini } 1340f26d0771SStefano Zampini 13417dae84e0SHong Zhang static PetscErrorCode MatCreateSubMatrix_IS(Mat mat,IS irow,IS icol,MatReuse scall,Mat *newmat) 1342a8116848SStefano Zampini { 1343a8116848SStefano Zampini Mat locmat,newlocmat; 1344a8116848SStefano Zampini Mat_IS *newmatis; 1345a8116848SStefano Zampini const PetscInt *idxs; 1346a8116848SStefano Zampini PetscInt i,m,n; 1347a8116848SStefano Zampini PetscErrorCode ierr; 1348a8116848SStefano Zampini 1349a8116848SStefano Zampini PetscFunctionBegin; 1350a8116848SStefano Zampini if (scall == MAT_REUSE_MATRIX) { 1351a8116848SStefano Zampini PetscBool ismatis; 1352a8116848SStefano Zampini 1353a8116848SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)*newmat,MATIS,&ismatis);CHKERRQ(ierr); 1354a8116848SStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_ARG_WRONG,"Cannot reuse matrix! Not of MATIS type"); 1355a8116848SStefano Zampini newmatis = (Mat_IS*)(*newmat)->data; 1356a8116848SStefano Zampini if (!newmatis->getsub_ris) SETERRQ(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_ARG_WRONG,"Cannot reuse matrix! Misses local row IS"); 1357a8116848SStefano Zampini if (!newmatis->getsub_cis) SETERRQ(PetscObjectComm((PetscObject)*newmat),PETSC_ERR_ARG_WRONG,"Cannot reuse matrix! Misses local col IS"); 1358a8116848SStefano Zampini } 1359a8116848SStefano Zampini /* irow and icol may not have duplicate entries */ 136076bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 136176bd3646SJed Brown Vec rtest,ltest; 136276bd3646SJed Brown const PetscScalar *array; 136376bd3646SJed Brown 1364a8116848SStefano Zampini ierr = MatCreateVecs(mat,<est,&rtest);CHKERRQ(ierr); 1365a8116848SStefano Zampini ierr = ISGetLocalSize(irow,&n);CHKERRQ(ierr); 1366a8116848SStefano Zampini ierr = ISGetIndices(irow,&idxs);CHKERRQ(ierr); 1367a8116848SStefano Zampini for (i=0;i<n;i++) { 1368a8116848SStefano Zampini ierr = VecSetValue(rtest,idxs[i],1.0,ADD_VALUES);CHKERRQ(ierr); 1369a8116848SStefano Zampini } 1370a8116848SStefano Zampini ierr = VecAssemblyBegin(rtest);CHKERRQ(ierr); 1371a8116848SStefano Zampini ierr = VecAssemblyEnd(rtest);CHKERRQ(ierr); 1372a8116848SStefano Zampini ierr = VecGetLocalSize(rtest,&n);CHKERRQ(ierr); 1373a8116848SStefano Zampini ierr = VecGetOwnershipRange(rtest,&m,NULL);CHKERRQ(ierr); 1374a8116848SStefano Zampini ierr = VecGetArrayRead(rtest,&array);CHKERRQ(ierr); 1375fd479f66SMatthew G. Knepley for (i=0;i<n;i++) if (array[i] != 0. && array[i] != 1.) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Index %D counted %D times! Irow may not have duplicate entries",i+m,(PetscInt)PetscRealPart(array[i])); 1376a8116848SStefano Zampini ierr = VecRestoreArrayRead(rtest,&array);CHKERRQ(ierr); 1377a8116848SStefano Zampini ierr = ISRestoreIndices(irow,&idxs);CHKERRQ(ierr); 1378a8116848SStefano Zampini ierr = ISGetLocalSize(icol,&n);CHKERRQ(ierr); 1379a8116848SStefano Zampini ierr = ISGetIndices(icol,&idxs);CHKERRQ(ierr); 1380a8116848SStefano Zampini for (i=0;i<n;i++) { 1381a8116848SStefano Zampini ierr = VecSetValue(ltest,idxs[i],1.0,ADD_VALUES);CHKERRQ(ierr); 1382a8116848SStefano Zampini } 1383a8116848SStefano Zampini ierr = VecAssemblyBegin(ltest);CHKERRQ(ierr); 1384a8116848SStefano Zampini ierr = VecAssemblyEnd(ltest);CHKERRQ(ierr); 1385a8116848SStefano Zampini ierr = VecGetLocalSize(ltest,&n);CHKERRQ(ierr); 1386a8116848SStefano Zampini ierr = VecGetOwnershipRange(ltest,&m,NULL);CHKERRQ(ierr); 1387a8116848SStefano Zampini ierr = VecGetArrayRead(ltest,&array);CHKERRQ(ierr); 1388fd479f66SMatthew G. Knepley for (i=0;i<n;i++) if (array[i] != 0. && array[i] != 1.) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Index %D counted %D times! Icol may not have duplicate entries",i+m,(PetscInt)PetscRealPart(array[i])); 1389a8116848SStefano Zampini ierr = VecRestoreArrayRead(ltest,&array);CHKERRQ(ierr); 1390a8116848SStefano Zampini ierr = ISRestoreIndices(icol,&idxs);CHKERRQ(ierr); 1391a8116848SStefano Zampini ierr = VecDestroy(&rtest);CHKERRQ(ierr); 1392a8116848SStefano Zampini ierr = VecDestroy(<est);CHKERRQ(ierr); 139376bd3646SJed Brown } 1394a8116848SStefano Zampini if (scall == MAT_INITIAL_MATRIX) { 1395a8116848SStefano Zampini Mat_IS *matis = (Mat_IS*)mat->data; 1396a8116848SStefano Zampini ISLocalToGlobalMapping rl2g; 1397a8116848SStefano Zampini IS is; 1398a8116848SStefano Zampini PetscInt *lidxs,*lgidxs,*newgidxs; 1399306cf5c7SStefano Zampini PetscInt ll,newloc,irbs,icbs,arbs,acbs,rbs,cbs; 140094342113SStefano Zampini PetscBool cong; 1401a8116848SStefano Zampini MPI_Comm comm; 1402a8116848SStefano Zampini 1403a8116848SStefano Zampini ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr); 1404306cf5c7SStefano Zampini ierr = MatGetBlockSizes(mat,&arbs,&acbs);CHKERRQ(ierr); 1405306cf5c7SStefano Zampini ierr = ISGetBlockSize(irow,&irbs);CHKERRQ(ierr); 1406306cf5c7SStefano Zampini ierr = ISGetBlockSize(icol,&icbs);CHKERRQ(ierr); 1407306cf5c7SStefano Zampini rbs = arbs == irbs ? irbs : 1; 1408306cf5c7SStefano Zampini cbs = acbs == icbs ? icbs : 1; 1409a8116848SStefano Zampini ierr = ISGetLocalSize(irow,&m);CHKERRQ(ierr); 1410a8116848SStefano Zampini ierr = ISGetLocalSize(icol,&n);CHKERRQ(ierr); 1411a8116848SStefano Zampini ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 1412a8116848SStefano Zampini ierr = MatSetType(*newmat,MATIS);CHKERRQ(ierr); 1413a8116848SStefano Zampini ierr = MatSetSizes(*newmat,m,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 1414306cf5c7SStefano Zampini ierr = MatSetBlockSizes(*newmat,rbs,cbs);CHKERRQ(ierr); 1415a8116848SStefano Zampini /* communicate irow to their owners in the layout */ 1416a8116848SStefano Zampini ierr = ISGetIndices(irow,&idxs);CHKERRQ(ierr); 1417a72d46e8SStefano Zampini ierr = PetscLayoutMapLocal(mat->rmap,m,idxs,&ll,&lidxs,&lgidxs);CHKERRQ(ierr); 1418a8116848SStefano Zampini ierr = ISRestoreIndices(irow,&idxs);CHKERRQ(ierr); 1419580bdb30SBarry Smith ierr = PetscArrayzero(matis->sf_rootdata,matis->sf->nroots);CHKERRQ(ierr); 1420a8116848SStefano Zampini for (i=0;i<ll;i++) matis->sf_rootdata[lidxs[i]] = lgidxs[i]+1; 1421a8116848SStefano Zampini ierr = PetscFree(lidxs);CHKERRQ(ierr); 1422a8116848SStefano Zampini ierr = PetscFree(lgidxs);CHKERRQ(ierr); 1423ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 1424ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 14253d996552SStefano Zampini for (i=0,newloc=0;i<matis->sf->nleaves;i++) if (matis->sf_leafdata[i]) newloc++; 1426a8116848SStefano Zampini ierr = PetscMalloc1(newloc,&newgidxs);CHKERRQ(ierr); 1427a8116848SStefano Zampini ierr = PetscMalloc1(newloc,&lidxs);CHKERRQ(ierr); 14283d996552SStefano Zampini for (i=0,newloc=0;i<matis->sf->nleaves;i++) 1429a8116848SStefano Zampini if (matis->sf_leafdata[i]) { 1430a8116848SStefano Zampini lidxs[newloc] = i; 1431a8116848SStefano Zampini newgidxs[newloc++] = matis->sf_leafdata[i]-1; 1432a8116848SStefano Zampini } 1433a8116848SStefano Zampini ierr = ISCreateGeneral(comm,newloc,newgidxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 1434a8116848SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr); 1435306cf5c7SStefano Zampini ierr = ISLocalToGlobalMappingSetBlockSize(rl2g,rbs);CHKERRQ(ierr); 1436a8116848SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 1437a8116848SStefano Zampini /* local is to extract local submatrix */ 1438a8116848SStefano Zampini newmatis = (Mat_IS*)(*newmat)->data; 1439a8116848SStefano Zampini ierr = ISCreateGeneral(comm,newloc,lidxs,PETSC_OWN_POINTER,&newmatis->getsub_ris);CHKERRQ(ierr); 144094342113SStefano Zampini ierr = MatHasCongruentLayouts(mat,&cong);CHKERRQ(ierr); 144194342113SStefano Zampini if (cong && irow == icol && matis->csf == matis->sf) { 1442a8116848SStefano Zampini ierr = MatSetLocalToGlobalMapping(*newmat,rl2g,rl2g);CHKERRQ(ierr); 1443a8116848SStefano Zampini ierr = PetscObjectReference((PetscObject)newmatis->getsub_ris);CHKERRQ(ierr); 1444a8116848SStefano Zampini newmatis->getsub_cis = newmatis->getsub_ris; 1445a8116848SStefano Zampini } else { 1446a8116848SStefano Zampini ISLocalToGlobalMapping cl2g; 1447a8116848SStefano Zampini 1448a8116848SStefano Zampini /* communicate icol to their owners in the layout */ 1449a8116848SStefano Zampini ierr = ISGetIndices(icol,&idxs);CHKERRQ(ierr); 1450a72d46e8SStefano Zampini ierr = PetscLayoutMapLocal(mat->cmap,n,idxs,&ll,&lidxs,&lgidxs);CHKERRQ(ierr); 1451a8116848SStefano Zampini ierr = ISRestoreIndices(icol,&idxs);CHKERRQ(ierr); 1452580bdb30SBarry Smith ierr = PetscArrayzero(matis->csf_rootdata,matis->csf->nroots);CHKERRQ(ierr); 1453a8116848SStefano Zampini for (i=0;i<ll;i++) matis->csf_rootdata[lidxs[i]] = lgidxs[i]+1; 1454a8116848SStefano Zampini ierr = PetscFree(lidxs);CHKERRQ(ierr); 1455a8116848SStefano Zampini ierr = PetscFree(lgidxs);CHKERRQ(ierr); 1456ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->csf,MPIU_INT,matis->csf_rootdata,matis->csf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 1457ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->csf,MPIU_INT,matis->csf_rootdata,matis->csf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 14583d996552SStefano Zampini for (i=0,newloc=0;i<matis->csf->nleaves;i++) if (matis->csf_leafdata[i]) newloc++; 1459a8116848SStefano Zampini ierr = PetscMalloc1(newloc,&newgidxs);CHKERRQ(ierr); 1460a8116848SStefano Zampini ierr = PetscMalloc1(newloc,&lidxs);CHKERRQ(ierr); 14613d996552SStefano Zampini for (i=0,newloc=0;i<matis->csf->nleaves;i++) 1462a8116848SStefano Zampini if (matis->csf_leafdata[i]) { 1463a8116848SStefano Zampini lidxs[newloc] = i; 1464a8116848SStefano Zampini newgidxs[newloc++] = matis->csf_leafdata[i]-1; 1465a8116848SStefano Zampini } 1466a8116848SStefano Zampini ierr = ISCreateGeneral(comm,newloc,newgidxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 1467a8116848SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr); 1468306cf5c7SStefano Zampini ierr = ISLocalToGlobalMappingSetBlockSize(cl2g,cbs);CHKERRQ(ierr); 1469a8116848SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 1470a8116848SStefano Zampini /* local is to extract local submatrix */ 1471a8116848SStefano Zampini ierr = ISCreateGeneral(comm,newloc,lidxs,PETSC_OWN_POINTER,&newmatis->getsub_cis);CHKERRQ(ierr); 1472a8116848SStefano Zampini ierr = MatSetLocalToGlobalMapping(*newmat,rl2g,cl2g);CHKERRQ(ierr); 1473a8116848SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 1474a8116848SStefano Zampini } 1475a8116848SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 1476a8116848SStefano Zampini } else { 1477a8116848SStefano Zampini ierr = MatISGetLocalMat(*newmat,&newlocmat);CHKERRQ(ierr); 1478a8116848SStefano Zampini } 1479a8116848SStefano Zampini ierr = MatISGetLocalMat(mat,&locmat);CHKERRQ(ierr); 1480a8116848SStefano Zampini newmatis = (Mat_IS*)(*newmat)->data; 14817dae84e0SHong Zhang ierr = MatCreateSubMatrix(locmat,newmatis->getsub_ris,newmatis->getsub_cis,scall,&newlocmat);CHKERRQ(ierr); 1482a8116848SStefano Zampini if (scall == MAT_INITIAL_MATRIX) { 1483a8116848SStefano Zampini ierr = MatISSetLocalMat(*newmat,newlocmat);CHKERRQ(ierr); 1484a8116848SStefano Zampini ierr = MatDestroy(&newlocmat);CHKERRQ(ierr); 1485a8116848SStefano Zampini } 1486a8116848SStefano Zampini ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1487a8116848SStefano Zampini ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1488a8116848SStefano Zampini PetscFunctionReturn(0); 1489a8116848SStefano Zampini } 1490a8116848SStefano Zampini 1491a8116848SStefano Zampini static PetscErrorCode MatCopy_IS(Mat A,Mat B,MatStructure str) 14922b404112SStefano Zampini { 14932b404112SStefano Zampini Mat_IS *a = (Mat_IS*)A->data,*b; 14942b404112SStefano Zampini PetscBool ismatis; 14952b404112SStefano Zampini PetscErrorCode ierr; 14962b404112SStefano Zampini 14972b404112SStefano Zampini PetscFunctionBegin; 14982b404112SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)B,MATIS,&ismatis);CHKERRQ(ierr); 14992b404112SStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"Need to be implemented"); 15002b404112SStefano Zampini b = (Mat_IS*)B->data; 15012b404112SStefano Zampini ierr = MatCopy(a->A,b->A,str);CHKERRQ(ierr); 1502cdc753b6SBarry Smith ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr); 15032b404112SStefano Zampini PetscFunctionReturn(0); 15042b404112SStefano Zampini } 15052b404112SStefano Zampini 1506a8116848SStefano Zampini static PetscErrorCode MatMissingDiagonal_IS(Mat A,PetscBool *missing,PetscInt *d) 15076bd84002SStefano Zampini { 1508527b2640SStefano Zampini Vec v; 1509527b2640SStefano Zampini const PetscScalar *array; 1510527b2640SStefano Zampini PetscInt i,n; 15116bd84002SStefano Zampini PetscErrorCode ierr; 15126bd84002SStefano Zampini 15136bd84002SStefano Zampini PetscFunctionBegin; 1514527b2640SStefano Zampini *missing = PETSC_FALSE; 1515527b2640SStefano Zampini ierr = MatCreateVecs(A,NULL,&v);CHKERRQ(ierr); 1516527b2640SStefano Zampini ierr = MatGetDiagonal(A,v);CHKERRQ(ierr); 1517527b2640SStefano Zampini ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 1518527b2640SStefano Zampini ierr = VecGetArrayRead(v,&array);CHKERRQ(ierr); 1519527b2640SStefano Zampini for (i=0;i<n;i++) if (array[i] == 0.) break; 1520527b2640SStefano Zampini ierr = VecRestoreArrayRead(v,&array);CHKERRQ(ierr); 1521527b2640SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 1522527b2640SStefano Zampini if (i != n) *missing = PETSC_TRUE; 1523527b2640SStefano Zampini if (d) { 1524527b2640SStefano Zampini *d = -1; 1525527b2640SStefano Zampini if (*missing) { 1526527b2640SStefano Zampini PetscInt rstart; 1527527b2640SStefano Zampini ierr = MatGetOwnershipRange(A,&rstart,NULL);CHKERRQ(ierr); 1528527b2640SStefano Zampini *d = i+rstart; 1529527b2640SStefano Zampini } 1530527b2640SStefano Zampini } 15316bd84002SStefano Zampini PetscFunctionReturn(0); 15326bd84002SStefano Zampini } 15336bd84002SStefano Zampini 1534cf0a3239SStefano Zampini static PetscErrorCode MatISSetUpSF_IS(Mat B) 153528f4e0baSStefano Zampini { 153628f4e0baSStefano Zampini Mat_IS *matis = (Mat_IS*)(B->data); 153728f4e0baSStefano Zampini const PetscInt *gidxs; 15384f2d7cafSStefano Zampini PetscInt nleaves; 153928f4e0baSStefano Zampini PetscErrorCode ierr; 154028f4e0baSStefano Zampini 154128f4e0baSStefano Zampini PetscFunctionBegin; 15424f2d7cafSStefano Zampini if (matis->sf) PetscFunctionReturn(0); 154328f4e0baSStefano Zampini ierr = PetscSFCreate(PetscObjectComm((PetscObject)B),&matis->sf);CHKERRQ(ierr); 15443bbff08aSStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(B->rmap->mapping,&gidxs);CHKERRQ(ierr); 15454f2d7cafSStefano Zampini ierr = ISLocalToGlobalMappingGetSize(B->rmap->mapping,&nleaves);CHKERRQ(ierr); 15464f2d7cafSStefano Zampini ierr = PetscSFSetGraphLayout(matis->sf,B->rmap,nleaves,NULL,PETSC_OWN_POINTER,gidxs);CHKERRQ(ierr); 15473bbff08aSStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(B->rmap->mapping,&gidxs);CHKERRQ(ierr); 15484f2d7cafSStefano Zampini ierr = PetscMalloc2(matis->sf->nroots,&matis->sf_rootdata,matis->sf->nleaves,&matis->sf_leafdata);CHKERRQ(ierr); 1549a8116848SStefano Zampini if (B->rmap->mapping != B->cmap->mapping) { /* setup SF for columns */ 15503d996552SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(B->cmap->mapping,&nleaves);CHKERRQ(ierr); 1551a8116848SStefano Zampini ierr = PetscSFCreate(PetscObjectComm((PetscObject)B),&matis->csf);CHKERRQ(ierr); 1552a8116848SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(B->cmap->mapping,&gidxs);CHKERRQ(ierr); 15533d996552SStefano Zampini ierr = PetscSFSetGraphLayout(matis->csf,B->cmap,nleaves,NULL,PETSC_OWN_POINTER,gidxs);CHKERRQ(ierr); 1554a8116848SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(B->cmap->mapping,&gidxs);CHKERRQ(ierr); 15553d996552SStefano Zampini ierr = PetscMalloc2(matis->csf->nroots,&matis->csf_rootdata,matis->csf->nleaves,&matis->csf_leafdata);CHKERRQ(ierr); 1556a8116848SStefano Zampini } else { 1557a8116848SStefano Zampini matis->csf = matis->sf; 1558a8116848SStefano Zampini matis->csf_leafdata = matis->sf_leafdata; 1559a8116848SStefano Zampini matis->csf_rootdata = matis->sf_rootdata; 1560a8116848SStefano Zampini } 156128f4e0baSStefano Zampini PetscFunctionReturn(0); 156228f4e0baSStefano Zampini } 15632e1947a5SStefano Zampini 1564eb82efa4SStefano Zampini /*@ 156575d48cdbSStefano Zampini MatISStoreL2L - Store local-to-local operators during the Galerkin process of MatPtAP. 156675d48cdbSStefano Zampini 1567d083f849SBarry Smith Collective 156875d48cdbSStefano Zampini 156975d48cdbSStefano Zampini Input Parameters: 157075d48cdbSStefano Zampini + A - the matrix 157175d48cdbSStefano Zampini - store - the boolean flag 157275d48cdbSStefano Zampini 157375d48cdbSStefano Zampini Level: advanced 157475d48cdbSStefano Zampini 157575d48cdbSStefano Zampini Notes: 157675d48cdbSStefano Zampini 157775d48cdbSStefano Zampini .seealso: MatCreate(), MatCreateIS(), MatISSetPreallocation(), MatPtAP() 157875d48cdbSStefano Zampini @*/ 157975d48cdbSStefano Zampini PetscErrorCode MatISStoreL2L(Mat A, PetscBool store) 158075d48cdbSStefano Zampini { 158175d48cdbSStefano Zampini PetscErrorCode ierr; 158275d48cdbSStefano Zampini 158375d48cdbSStefano Zampini PetscFunctionBegin; 158475d48cdbSStefano Zampini PetscValidHeaderSpecific(A,MAT_CLASSID,1); 158575d48cdbSStefano Zampini PetscValidType(A,1); 158675d48cdbSStefano Zampini PetscValidLogicalCollectiveBool(A,store,2); 158775d48cdbSStefano Zampini ierr = PetscTryMethod(A,"MatISStoreL2L_C",(Mat,PetscBool),(A,store));CHKERRQ(ierr); 158875d48cdbSStefano Zampini PetscFunctionReturn(0); 158975d48cdbSStefano Zampini } 159075d48cdbSStefano Zampini 159175d48cdbSStefano Zampini static PetscErrorCode MatISStoreL2L_IS(Mat A, PetscBool store) 159275d48cdbSStefano Zampini { 159375d48cdbSStefano Zampini Mat_IS *matis = (Mat_IS*)(A->data); 159475d48cdbSStefano Zampini PetscErrorCode ierr; 159575d48cdbSStefano Zampini 159675d48cdbSStefano Zampini PetscFunctionBegin; 159775d48cdbSStefano Zampini matis->storel2l = store; 159875d48cdbSStefano Zampini if (!store) { 159975d48cdbSStefano Zampini ierr = PetscObjectCompose((PetscObject)(A),"_MatIS_PtAP_l2l",NULL);CHKERRQ(ierr); 160075d48cdbSStefano Zampini } 160175d48cdbSStefano Zampini PetscFunctionReturn(0); 160275d48cdbSStefano Zampini } 160375d48cdbSStefano Zampini 160475d48cdbSStefano Zampini /*@ 1605f03112d0SStefano Zampini MatISFixLocalEmpty - Compress out zero local rows from the local matrices 1606f03112d0SStefano Zampini 1607d083f849SBarry Smith Collective 1608f03112d0SStefano Zampini 1609f03112d0SStefano Zampini Input Parameters: 1610f03112d0SStefano Zampini + A - the matrix 1611f03112d0SStefano Zampini - fix - the boolean flag 1612f03112d0SStefano Zampini 1613f03112d0SStefano Zampini Level: advanced 1614f03112d0SStefano Zampini 1615f03112d0SStefano Zampini Notes: When fix is true, new local matrices and l2g maps are generated during the final assembly process. 1616f03112d0SStefano Zampini 1617f03112d0SStefano Zampini .seealso: MatCreate(), MatCreateIS(), MatISSetPreallocation(), MatAssemblyEnd(), MAT_FINAL_ASSEMBLY 1618f03112d0SStefano Zampini @*/ 1619f03112d0SStefano Zampini PetscErrorCode MatISFixLocalEmpty(Mat A, PetscBool fix) 1620f03112d0SStefano Zampini { 1621f03112d0SStefano Zampini PetscErrorCode ierr; 1622f03112d0SStefano Zampini 1623f03112d0SStefano Zampini PetscFunctionBegin; 1624f03112d0SStefano Zampini PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1625f03112d0SStefano Zampini PetscValidType(A,1); 1626f03112d0SStefano Zampini PetscValidLogicalCollectiveBool(A,fix,2); 1627f03112d0SStefano Zampini ierr = PetscTryMethod(A,"MatISFixLocalEmpty_C",(Mat,PetscBool),(A,fix));CHKERRQ(ierr); 1628f03112d0SStefano Zampini PetscFunctionReturn(0); 1629f03112d0SStefano Zampini } 1630f03112d0SStefano Zampini 1631f03112d0SStefano Zampini static PetscErrorCode MatISFixLocalEmpty_IS(Mat A, PetscBool fix) 1632f03112d0SStefano Zampini { 1633f03112d0SStefano Zampini Mat_IS *matis = (Mat_IS*)(A->data); 1634f03112d0SStefano Zampini 1635f03112d0SStefano Zampini PetscFunctionBegin; 1636f03112d0SStefano Zampini matis->locempty = fix; 1637f03112d0SStefano Zampini PetscFunctionReturn(0); 1638f03112d0SStefano Zampini } 1639f03112d0SStefano Zampini 1640f03112d0SStefano Zampini /*@ 1641a88811baSStefano Zampini MatISSetPreallocation - Preallocates memory for a MATIS parallel matrix. 1642a88811baSStefano Zampini 1643d083f849SBarry Smith Collective 1644a88811baSStefano Zampini 1645a88811baSStefano Zampini Input Parameters: 1646a88811baSStefano Zampini + B - the matrix 1647a88811baSStefano Zampini . d_nz - number of nonzeros per row in DIAGONAL portion of local submatrix 1648a88811baSStefano Zampini (same value is used for all local rows) 1649a88811baSStefano Zampini . d_nnz - array containing the number of nonzeros in the various rows of the 1650a88811baSStefano Zampini DIAGONAL portion of the local submatrix (possibly different for each row) 1651a88811baSStefano Zampini or NULL, if d_nz is used to specify the nonzero structure. 1652a88811baSStefano Zampini The size of this array is equal to the number of local rows, i.e 'm'. 1653a88811baSStefano Zampini For matrices that will be factored, you must leave room for (and set) 1654a88811baSStefano Zampini the diagonal entry even if it is zero. 1655a88811baSStefano Zampini . o_nz - number of nonzeros per row in the OFF-DIAGONAL portion of local 1656a88811baSStefano Zampini submatrix (same value is used for all local rows). 1657a88811baSStefano Zampini - o_nnz - array containing the number of nonzeros in the various rows of the 1658a88811baSStefano Zampini OFF-DIAGONAL portion of the local submatrix (possibly different for 1659a88811baSStefano Zampini each row) or NULL, if o_nz is used to specify the nonzero 1660a88811baSStefano Zampini structure. The size of this array is equal to the number 1661a88811baSStefano Zampini of local rows, i.e 'm'. 1662a88811baSStefano Zampini 1663a88811baSStefano Zampini If the *_nnz parameter is given then the *_nz parameter is ignored 1664a88811baSStefano Zampini 1665a88811baSStefano Zampini Level: intermediate 1666a88811baSStefano Zampini 166795452b02SPatrick Sanan Notes: 166895452b02SPatrick Sanan This function has the same interface as the MPIAIJ preallocation routine in order to simplify the transition 1669a88811baSStefano Zampini from the asssembled format to the unassembled one. It overestimates the preallocation of MATIS local 1670a88811baSStefano Zampini matrices; for exact preallocation, the user should set the preallocation directly on local matrix objects. 1671a88811baSStefano Zampini 16723c212e90SHong Zhang .seealso: MatCreate(), MatCreateIS(), MatMPIAIJSetPreallocation(), MatISGetLocalMat(), MATIS 1673a88811baSStefano Zampini @*/ 16742e1947a5SStefano Zampini PetscErrorCode MatISSetPreallocation(Mat B,PetscInt d_nz,const PetscInt d_nnz[],PetscInt o_nz,const PetscInt o_nnz[]) 16752e1947a5SStefano Zampini { 16762e1947a5SStefano Zampini PetscErrorCode ierr; 16772e1947a5SStefano Zampini 16782e1947a5SStefano Zampini PetscFunctionBegin; 16792e1947a5SStefano Zampini PetscValidHeaderSpecific(B,MAT_CLASSID,1); 16802e1947a5SStefano Zampini PetscValidType(B,1); 16812e1947a5SStefano Zampini ierr = PetscTryMethod(B,"MatISSetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(B,d_nz,d_nnz,o_nz,o_nnz));CHKERRQ(ierr); 16822e1947a5SStefano Zampini PetscFunctionReturn(0); 16832e1947a5SStefano Zampini } 16842e1947a5SStefano Zampini 1685844bd0d7SStefano Zampini /* this is used by DMDA */ 1686844bd0d7SStefano Zampini PETSC_EXTERN PetscErrorCode MatISSetPreallocation_IS(Mat B,PetscInt d_nz,const PetscInt d_nnz[],PetscInt o_nz,const PetscInt o_nnz[]) 16872e1947a5SStefano Zampini { 16882e1947a5SStefano Zampini Mat_IS *matis = (Mat_IS*)(B->data); 168928f4e0baSStefano Zampini PetscInt bs,i,nlocalcols; 16902e1947a5SStefano Zampini PetscErrorCode ierr; 16912e1947a5SStefano Zampini 16922e1947a5SStefano Zampini PetscFunctionBegin; 16936c4ed002SBarry Smith if (!matis->A) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"You should first call MatSetLocalToGlobalMapping"); 16944f2d7cafSStefano Zampini 16954f2d7cafSStefano Zampini if (!d_nnz) for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] = d_nz; 16964f2d7cafSStefano Zampini else for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] = d_nnz[i]; 16974f2d7cafSStefano Zampini 16984f2d7cafSStefano Zampini if (!o_nnz) for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] += o_nz; 16994f2d7cafSStefano Zampini else for (i=0;i<matis->sf->nroots;i++) matis->sf_rootdata[i] += o_nnz[i]; 17004f2d7cafSStefano Zampini 1701ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 170228f4e0baSStefano Zampini ierr = MatGetSize(matis->A,NULL,&nlocalcols);CHKERRQ(ierr); 170328f4e0baSStefano Zampini ierr = MatGetBlockSize(matis->A,&bs);CHKERRQ(ierr); 1704ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 17054f2d7cafSStefano Zampini 17064f2d7cafSStefano Zampini for (i=0;i<matis->sf->nleaves;i++) matis->sf_leafdata[i] = PetscMin(matis->sf_leafdata[i],nlocalcols); 170728f4e0baSStefano Zampini ierr = MatSeqAIJSetPreallocation(matis->A,0,matis->sf_leafdata);CHKERRQ(ierr); 17080f2f62c7SStefano Zampini #if defined(PETSC_HAVE_HYPRE) 17090f2f62c7SStefano Zampini ierr = MatHYPRESetPreallocation(matis->A,0,matis->sf_leafdata,0,NULL);CHKERRQ(ierr); 17100f2f62c7SStefano Zampini #endif 17114f2d7cafSStefano Zampini 17124f2d7cafSStefano Zampini for (i=0;i<matis->sf->nleaves/bs;i++) matis->sf_leafdata[i] = matis->sf_leafdata[i*bs]/bs; 171328f4e0baSStefano Zampini ierr = MatSeqBAIJSetPreallocation(matis->A,bs,0,matis->sf_leafdata);CHKERRQ(ierr); 17144f2d7cafSStefano Zampini 171500a59248SStefano Zampini nlocalcols /= bs; 171600a59248SStefano Zampini for (i=0;i<matis->sf->nleaves/bs;i++) matis->sf_leafdata[i] = PetscMin(matis->sf_leafdata[i],nlocalcols - i); 171728f4e0baSStefano Zampini ierr = MatSeqSBAIJSetPreallocation(matis->A,bs,0,matis->sf_leafdata);CHKERRQ(ierr); 17180f2f62c7SStefano Zampini 17190f2f62c7SStefano Zampini /* for other matrix types */ 17200f2f62c7SStefano Zampini ierr = MatSetUp(matis->A);CHKERRQ(ierr); 17212e1947a5SStefano Zampini PetscFunctionReturn(0); 17222e1947a5SStefano Zampini } 1723b4319ba4SBarry Smith 17243927de2eSStefano Zampini PETSC_EXTERN PetscErrorCode MatISSetMPIXAIJPreallocation_Private(Mat A, Mat B, PetscBool maxreduce) 17253927de2eSStefano Zampini { 17263927de2eSStefano Zampini Mat_IS *matis = (Mat_IS*)(A->data); 17273927de2eSStefano Zampini PetscInt *my_dnz,*my_onz,*dnz,*onz,*mat_ranges,*row_ownership; 1728ecf5a873SStefano Zampini const PetscInt *global_indices_r,*global_indices_c; 17293927de2eSStefano Zampini PetscInt i,j,bs,rows,cols; 17303927de2eSStefano Zampini PetscInt lrows,lcols; 17313927de2eSStefano Zampini PetscInt local_rows,local_cols; 1732f03112d0SStefano Zampini PetscMPIInt size; 17333927de2eSStefano Zampini PetscBool isdense,issbaij; 17343927de2eSStefano Zampini PetscErrorCode ierr; 17353927de2eSStefano Zampini 17363927de2eSStefano Zampini PetscFunctionBegin; 1737ffc4695bSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRMPI(ierr); 17383927de2eSStefano Zampini ierr = MatGetSize(A,&rows,&cols);CHKERRQ(ierr); 17393927de2eSStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 17403927de2eSStefano Zampini ierr = MatGetSize(matis->A,&local_rows,&local_cols);CHKERRQ(ierr); 1741b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQDENSE,&isdense);CHKERRQ(ierr); 1742b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr); 1743ecf5a873SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(A->rmap->mapping,&global_indices_r);CHKERRQ(ierr); 1744ecf5a873SStefano Zampini if (A->rmap->mapping != A->cmap->mapping) { 17457230de76SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(A->cmap->mapping,&global_indices_c);CHKERRQ(ierr); 1746ecf5a873SStefano Zampini } else { 1747ecf5a873SStefano Zampini global_indices_c = global_indices_r; 1748ecf5a873SStefano Zampini } 1749ecf5a873SStefano Zampini 17503927de2eSStefano Zampini if (issbaij) { 17513927de2eSStefano Zampini ierr = MatGetRowUpperTriangular(matis->A);CHKERRQ(ierr); 17523927de2eSStefano Zampini } 17533927de2eSStefano Zampini /* 1754ecf5a873SStefano Zampini An SF reduce is needed to sum up properly on shared rows. 17553927de2eSStefano Zampini Note that generally preallocation is not exact, since it overestimates nonzeros 17563927de2eSStefano Zampini */ 17573927de2eSStefano Zampini ierr = MatGetLocalSize(A,&lrows,&lcols);CHKERRQ(ierr); 17583927de2eSStefano Zampini ierr = MatPreallocateInitialize(PetscObjectComm((PetscObject)A),lrows,lcols,dnz,onz);CHKERRQ(ierr); 17593927de2eSStefano Zampini /* All processes need to compute entire row ownership */ 17603927de2eSStefano Zampini ierr = PetscMalloc1(rows,&row_ownership);CHKERRQ(ierr); 17613927de2eSStefano Zampini ierr = MatGetOwnershipRanges(A,(const PetscInt**)&mat_ranges);CHKERRQ(ierr); 1762f03112d0SStefano Zampini for (i=0;i<size;i++) { 17633927de2eSStefano Zampini for (j=mat_ranges[i];j<mat_ranges[i+1];j++) { 17643927de2eSStefano Zampini row_ownership[j] = i; 17653927de2eSStefano Zampini } 17663927de2eSStefano Zampini } 17677230de76SStefano Zampini ierr = MatGetOwnershipRangesColumn(A,(const PetscInt**)&mat_ranges);CHKERRQ(ierr); 17683927de2eSStefano Zampini 17693927de2eSStefano Zampini /* 17703927de2eSStefano Zampini my_dnz and my_onz contains exact contribution to preallocation from each local mat 17713927de2eSStefano Zampini then, they will be summed up properly. This way, preallocation is always sufficient 17723927de2eSStefano Zampini */ 17733927de2eSStefano Zampini ierr = PetscCalloc2(local_rows,&my_dnz,local_rows,&my_onz);CHKERRQ(ierr); 17743927de2eSStefano Zampini /* preallocation as a MATAIJ */ 17753927de2eSStefano Zampini if (isdense) { /* special case for dense local matrices */ 17763927de2eSStefano Zampini for (i=0;i<local_rows;i++) { 177712dfadf8SStefano Zampini PetscInt owner = row_ownership[global_indices_r[i]]; 177812dfadf8SStefano Zampini for (j=0;j<local_cols;j++) { 1779ecf5a873SStefano Zampini PetscInt index_col = global_indices_c[j]; 17803927de2eSStefano Zampini if (index_col > mat_ranges[owner]-1 && index_col < mat_ranges[owner+1]) { /* diag block */ 17813927de2eSStefano Zampini my_dnz[i] += 1; 17823927de2eSStefano Zampini } else { /* offdiag block */ 17833927de2eSStefano Zampini my_onz[i] += 1; 17843927de2eSStefano Zampini } 17853927de2eSStefano Zampini } 17863927de2eSStefano Zampini } 1787bb1015c3SStefano Zampini } else if (matis->A->ops->getrowij) { 1788bb1015c3SStefano Zampini const PetscInt *ii,*jj,*jptr; 1789bb1015c3SStefano Zampini PetscBool done; 1790bb1015c3SStefano Zampini ierr = MatGetRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&local_rows,&ii,&jj,&done);CHKERRQ(ierr); 1791938e1ff8SStefano Zampini if (!done) SETERRQ(PetscObjectComm((PetscObject)(matis->A)),PETSC_ERR_PLIB,"Error in MatGetRowIJ"); 1792bb1015c3SStefano Zampini jptr = jj; 1793bb1015c3SStefano Zampini for (i=0;i<local_rows;i++) { 1794bb1015c3SStefano Zampini PetscInt index_row = global_indices_r[i]; 1795bb1015c3SStefano Zampini for (j=0;j<ii[i+1]-ii[i];j++,jptr++) { 1796bb1015c3SStefano Zampini PetscInt owner = row_ownership[index_row]; 1797bb1015c3SStefano Zampini PetscInt index_col = global_indices_c[*jptr]; 1798bb1015c3SStefano Zampini if (index_col > mat_ranges[owner]-1 && index_col < mat_ranges[owner+1]) { /* diag block */ 1799bb1015c3SStefano Zampini my_dnz[i] += 1; 1800bb1015c3SStefano Zampini } else { /* offdiag block */ 1801bb1015c3SStefano Zampini my_onz[i] += 1; 1802bb1015c3SStefano Zampini } 1803bb1015c3SStefano Zampini /* same as before, interchanging rows and cols */ 1804bb1015c3SStefano Zampini if (issbaij && index_col != index_row) { 1805bb1015c3SStefano Zampini owner = row_ownership[index_col]; 1806bb1015c3SStefano Zampini if (index_row > mat_ranges[owner]-1 && index_row < mat_ranges[owner+1]) { 1807bb1015c3SStefano Zampini my_dnz[*jptr] += 1; 1808bb1015c3SStefano Zampini } else { 1809bb1015c3SStefano Zampini my_onz[*jptr] += 1; 1810bb1015c3SStefano Zampini } 1811bb1015c3SStefano Zampini } 1812bb1015c3SStefano Zampini } 1813bb1015c3SStefano Zampini } 1814bb1015c3SStefano Zampini ierr = MatRestoreRowIJ(matis->A,0,PETSC_FALSE,PETSC_FALSE,&local_rows,&ii,&jj,&done);CHKERRQ(ierr); 1815938e1ff8SStefano Zampini if (!done) SETERRQ(PetscObjectComm((PetscObject)(matis->A)),PETSC_ERR_PLIB,"Error in MatRestoreRowIJ"); 1816bb1015c3SStefano Zampini } else { /* loop over rows and use MatGetRow */ 18173927de2eSStefano Zampini for (i=0;i<local_rows;i++) { 18183927de2eSStefano Zampini const PetscInt *cols; 1819ecf5a873SStefano Zampini PetscInt ncols,index_row = global_indices_r[i]; 18203927de2eSStefano Zampini ierr = MatGetRow(matis->A,i,&ncols,&cols,NULL);CHKERRQ(ierr); 18213927de2eSStefano Zampini for (j=0;j<ncols;j++) { 18223927de2eSStefano Zampini PetscInt owner = row_ownership[index_row]; 1823ecf5a873SStefano Zampini PetscInt index_col = global_indices_c[cols[j]]; 18243927de2eSStefano Zampini if (index_col > mat_ranges[owner]-1 && index_col < mat_ranges[owner+1]) { /* diag block */ 18253927de2eSStefano Zampini my_dnz[i] += 1; 18263927de2eSStefano Zampini } else { /* offdiag block */ 18273927de2eSStefano Zampini my_onz[i] += 1; 18283927de2eSStefano Zampini } 18293927de2eSStefano Zampini /* same as before, interchanging rows and cols */ 1830d9a9e74cSStefano Zampini if (issbaij && index_col != index_row) { 18313927de2eSStefano Zampini owner = row_ownership[index_col]; 18323927de2eSStefano Zampini if (index_row > mat_ranges[owner]-1 && index_row < mat_ranges[owner+1]) { 1833d9a9e74cSStefano Zampini my_dnz[cols[j]] += 1; 18343927de2eSStefano Zampini } else { 1835d9a9e74cSStefano Zampini my_onz[cols[j]] += 1; 18363927de2eSStefano Zampini } 18373927de2eSStefano Zampini } 18383927de2eSStefano Zampini } 18393927de2eSStefano Zampini ierr = MatRestoreRow(matis->A,i,&ncols,&cols,NULL);CHKERRQ(ierr); 18403927de2eSStefano Zampini } 18413927de2eSStefano Zampini } 1842ecf5a873SStefano Zampini if (global_indices_c != global_indices_r) { 18437230de76SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(A->cmap->mapping,&global_indices_c);CHKERRQ(ierr); 1844ecf5a873SStefano Zampini } 18454f619741Sstefano_zampini ierr = ISLocalToGlobalMappingRestoreIndices(A->rmap->mapping,&global_indices_r);CHKERRQ(ierr); 18463927de2eSStefano Zampini ierr = PetscFree(row_ownership);CHKERRQ(ierr); 1847ecf5a873SStefano Zampini 1848ecf5a873SStefano Zampini /* Reduce my_dnz and my_onz */ 18493927de2eSStefano Zampini if (maxreduce) { 18503927de2eSStefano Zampini ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_dnz,dnz,MPI_MAX);CHKERRQ(ierr); 18513927de2eSStefano Zampini ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_onz,onz,MPI_MAX);CHKERRQ(ierr); 1852bb1015c3SStefano Zampini ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_dnz,dnz,MPI_MAX);CHKERRQ(ierr); 18533927de2eSStefano Zampini ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_onz,onz,MPI_MAX);CHKERRQ(ierr); 18543927de2eSStefano Zampini } else { 18553927de2eSStefano Zampini ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_dnz,dnz,MPI_SUM);CHKERRQ(ierr); 18563927de2eSStefano Zampini ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,my_onz,onz,MPI_SUM);CHKERRQ(ierr); 1857bb1015c3SStefano Zampini ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_dnz,dnz,MPI_SUM);CHKERRQ(ierr); 18583927de2eSStefano Zampini ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,my_onz,onz,MPI_SUM);CHKERRQ(ierr); 18593927de2eSStefano Zampini } 18603927de2eSStefano Zampini ierr = PetscFree2(my_dnz,my_onz);CHKERRQ(ierr); 18613927de2eSStefano Zampini 18623927de2eSStefano Zampini /* Resize preallocation if overestimated */ 18633927de2eSStefano Zampini for (i=0;i<lrows;i++) { 18643927de2eSStefano Zampini dnz[i] = PetscMin(dnz[i],lcols); 18653927de2eSStefano Zampini onz[i] = PetscMin(onz[i],cols-lcols); 18663927de2eSStefano Zampini } 18671670daf9Sstefano_zampini 18681670daf9Sstefano_zampini /* Set preallocation */ 1869268753edSStefano Zampini ierr = MatSeqAIJSetPreallocation(B,0,dnz);CHKERRQ(ierr); 18703927de2eSStefano Zampini ierr = MatMPIAIJSetPreallocation(B,0,dnz,0,onz);CHKERRQ(ierr); 187153b44cf5SStefano Zampini for (i=0;i<lrows;i+=bs) { 187253b44cf5SStefano Zampini PetscInt b, d = dnz[i],o = onz[i]; 187353b44cf5SStefano Zampini 187453b44cf5SStefano Zampini for (b=1;b<bs;b++) { 187553b44cf5SStefano Zampini d = PetscMax(d,dnz[i+b]); 187653b44cf5SStefano Zampini o = PetscMax(o,onz[i+b]); 187753b44cf5SStefano Zampini } 187853b44cf5SStefano Zampini dnz[i/bs] = PetscMin(d/bs + d%bs,lcols/bs); 187953b44cf5SStefano Zampini onz[i/bs] = PetscMin(o/bs + o%bs,(cols-lcols)/bs); 18803927de2eSStefano Zampini } 1881268753edSStefano Zampini ierr = MatSeqBAIJSetPreallocation(B,bs,0,dnz);CHKERRQ(ierr); 18823927de2eSStefano Zampini ierr = MatMPIBAIJSetPreallocation(B,bs,0,dnz,0,onz);CHKERRQ(ierr); 18833927de2eSStefano Zampini ierr = MatMPISBAIJSetPreallocation(B,bs,0,dnz,0,onz);CHKERRQ(ierr); 18843927de2eSStefano Zampini ierr = MatPreallocateFinalize(dnz,onz);CHKERRQ(ierr); 18853927de2eSStefano Zampini if (issbaij) { 18863927de2eSStefano Zampini ierr = MatRestoreRowUpperTriangular(matis->A);CHKERRQ(ierr); 18873927de2eSStefano Zampini } 18889be90c3fSStefano Zampini ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 18893927de2eSStefano Zampini PetscFunctionReturn(0); 18903927de2eSStefano Zampini } 18913927de2eSStefano Zampini 1892487b449aSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_IS_XAIJ(Mat mat, MatType mtype, MatReuse reuse, Mat *M) 1893b7ce53b6SStefano Zampini { 1894b7ce53b6SStefano Zampini Mat_IS *matis = (Mat_IS*)(mat->data); 1895487b449aSStefano Zampini Mat local_mat,MT; 189653b44cf5SStefano Zampini PetscInt rbs,cbs,rows,cols,lrows,lcols; 1897b7ce53b6SStefano Zampini PetscInt local_rows,local_cols; 1898b9ed4604SStefano Zampini PetscBool isseqdense,isseqsbaij,isseqaij,isseqbaij; 1899f03112d0SStefano Zampini PetscMPIInt size; 19001683a169SBarry Smith const PetscScalar *array; 1901b7ce53b6SStefano Zampini PetscErrorCode ierr; 1902b7ce53b6SStefano Zampini 1903b7ce53b6SStefano Zampini PetscFunctionBegin; 1904ffc4695bSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)mat),&size);CHKERRMPI(ierr); 1905f03112d0SStefano Zampini if (size == 1 && mat->rmap->N == matis->A->rmap->N && mat->cmap->N == matis->A->cmap->N) { 19061670daf9Sstefano_zampini Mat B; 190753b44cf5SStefano Zampini IS irows = NULL,icols = NULL; 1908487b449aSStefano Zampini PetscInt rbs,cbs; 19091670daf9Sstefano_zampini 1910487b449aSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(mat->rmap->mapping,&rbs);CHKERRQ(ierr); 1911487b449aSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(mat->cmap->mapping,&cbs);CHKERRQ(ierr); 191253b44cf5SStefano Zampini if (reuse != MAT_REUSE_MATRIX) { /* check if l2g maps are one-to-one */ 191353b44cf5SStefano Zampini IS rows,cols; 191453b44cf5SStefano Zampini const PetscInt *ridxs,*cidxs; 191553b44cf5SStefano Zampini PetscInt i,nw,*work; 191653b44cf5SStefano Zampini 191753b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(mat->rmap->mapping,&ridxs);CHKERRQ(ierr); 191853b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(mat->rmap->mapping,&nw);CHKERRQ(ierr); 191953b44cf5SStefano Zampini nw = nw/rbs; 192053b44cf5SStefano Zampini ierr = PetscCalloc1(nw,&work);CHKERRQ(ierr); 192153b44cf5SStefano Zampini for (i=0;i<nw;i++) work[ridxs[i]] += 1; 192253b44cf5SStefano Zampini for (i=0;i<nw;i++) if (!work[i] || work[i] > 1) break; 192353b44cf5SStefano Zampini if (i == nw) { 192453b44cf5SStefano Zampini ierr = ISCreateBlock(PETSC_COMM_SELF,rbs,nw,ridxs,PETSC_USE_POINTER,&rows);CHKERRQ(ierr); 1925acdf38a7Sstefano_zampini ierr = ISSetPermutation(rows);CHKERRQ(ierr); 192653b44cf5SStefano Zampini ierr = ISInvertPermutation(rows,PETSC_DECIDE,&irows);CHKERRQ(ierr); 1927acdf38a7Sstefano_zampini ierr = ISDestroy(&rows);CHKERRQ(ierr); 192853b44cf5SStefano Zampini } 192953b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(mat->rmap->mapping,&ridxs);CHKERRQ(ierr); 193053b44cf5SStefano Zampini ierr = PetscFree(work);CHKERRQ(ierr); 193153b44cf5SStefano Zampini if (irows && mat->rmap->mapping != mat->cmap->mapping) { 193253b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(mat->cmap->mapping,&cidxs);CHKERRQ(ierr); 193353b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(mat->cmap->mapping,&nw);CHKERRQ(ierr); 193453b44cf5SStefano Zampini nw = nw/cbs; 193553b44cf5SStefano Zampini ierr = PetscCalloc1(nw,&work);CHKERRQ(ierr); 193653b44cf5SStefano Zampini for (i=0;i<nw;i++) work[cidxs[i]] += 1; 193753b44cf5SStefano Zampini for (i=0;i<nw;i++) if (!work[i] || work[i] > 1) break; 193853b44cf5SStefano Zampini if (i == nw) { 193953b44cf5SStefano Zampini ierr = ISCreateBlock(PETSC_COMM_SELF,cbs,nw,cidxs,PETSC_USE_POINTER,&cols);CHKERRQ(ierr); 194053b44cf5SStefano Zampini ierr = ISSetPermutation(cols);CHKERRQ(ierr); 194153b44cf5SStefano Zampini ierr = ISInvertPermutation(cols,PETSC_DECIDE,&icols);CHKERRQ(ierr); 194253b44cf5SStefano Zampini ierr = ISDestroy(&cols);CHKERRQ(ierr); 194353b44cf5SStefano Zampini } 194453b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(mat->cmap->mapping,&cidxs);CHKERRQ(ierr); 194553b44cf5SStefano Zampini ierr = PetscFree(work);CHKERRQ(ierr); 194653b44cf5SStefano Zampini } else if (irows) { 194753b44cf5SStefano Zampini ierr = PetscObjectReference((PetscObject)irows);CHKERRQ(ierr); 194853b44cf5SStefano Zampini icols = irows; 194953b44cf5SStefano Zampini } 195053b44cf5SStefano Zampini } else { 195153b44cf5SStefano Zampini ierr = PetscObjectQuery((PetscObject)(*M),"_MatIS_IS_XAIJ_irows",(PetscObject*)&irows);CHKERRQ(ierr); 195253b44cf5SStefano Zampini ierr = PetscObjectQuery((PetscObject)(*M),"_MatIS_IS_XAIJ_icols",(PetscObject*)&icols);CHKERRQ(ierr); 195353b44cf5SStefano Zampini if (irows) { ierr = PetscObjectReference((PetscObject)irows);CHKERRQ(ierr); } 195453b44cf5SStefano Zampini if (icols) { ierr = PetscObjectReference((PetscObject)icols);CHKERRQ(ierr); } 195553b44cf5SStefano Zampini } 195653b44cf5SStefano Zampini if (!irows || !icols) { 195753b44cf5SStefano Zampini ierr = ISDestroy(&icols);CHKERRQ(ierr); 195853b44cf5SStefano Zampini ierr = ISDestroy(&irows);CHKERRQ(ierr); 195953b44cf5SStefano Zampini goto general_assembly; 196053b44cf5SStefano Zampini } 1961d8e18881SStefano Zampini ierr = MatConvert(matis->A,mtype,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 1962487b449aSStefano Zampini if (reuse != MAT_INPLACE_MATRIX) { 19637dae84e0SHong Zhang ierr = MatCreateSubMatrix(B,irows,icols,reuse,M);CHKERRQ(ierr); 196453b44cf5SStefano Zampini ierr = PetscObjectCompose((PetscObject)(*M),"_MatIS_IS_XAIJ_irows",(PetscObject)irows);CHKERRQ(ierr); 196553b44cf5SStefano Zampini ierr = PetscObjectCompose((PetscObject)(*M),"_MatIS_IS_XAIJ_icols",(PetscObject)icols);CHKERRQ(ierr); 1966487b449aSStefano Zampini } else { 1967487b449aSStefano Zampini Mat C; 1968487b449aSStefano Zampini 1969487b449aSStefano Zampini ierr = MatCreateSubMatrix(B,irows,icols,MAT_INITIAL_MATRIX,&C);CHKERRQ(ierr); 1970487b449aSStefano Zampini ierr = MatHeaderReplace(mat,&C);CHKERRQ(ierr); 1971487b449aSStefano Zampini } 1972acdf38a7Sstefano_zampini ierr = MatDestroy(&B);CHKERRQ(ierr); 1973acdf38a7Sstefano_zampini ierr = ISDestroy(&icols);CHKERRQ(ierr); 1974acdf38a7Sstefano_zampini ierr = ISDestroy(&irows);CHKERRQ(ierr); 19757c03b4e8SStefano Zampini PetscFunctionReturn(0); 19767c03b4e8SStefano Zampini } 197753b44cf5SStefano Zampini general_assembly: 1978b7ce53b6SStefano Zampini ierr = MatGetSize(mat,&rows,&cols);CHKERRQ(ierr); 197953b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(mat->rmap->mapping,&rbs);CHKERRQ(ierr); 198053b44cf5SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(mat->cmap->mapping,&cbs);CHKERRQ(ierr); 19813cfa4ea4SStefano Zampini ierr = MatGetLocalSize(mat,&lrows,&lcols);CHKERRQ(ierr); 1982b7ce53b6SStefano Zampini ierr = MatGetSize(matis->A,&local_rows,&local_cols);CHKERRQ(ierr); 1983b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQDENSE,&isseqdense);CHKERRQ(ierr); 19844099cc6bSBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 1985b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQBAIJ,&isseqbaij);CHKERRQ(ierr); 1986b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)matis->A,MATSEQSBAIJ,&isseqsbaij);CHKERRQ(ierr); 1987b9ed4604SStefano Zampini if (!isseqdense && !isseqaij && !isseqbaij && !isseqsbaij) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not for matrix type %s",((PetscObject)(matis->A))->type_name); 198876bd3646SJed Brown if (PetscDefined (USE_DEBUG)) { 198976bd3646SJed Brown PetscBool lb[4],bb[4]; 199076bd3646SJed Brown 1991b9ed4604SStefano Zampini lb[0] = isseqdense; 1992b9ed4604SStefano Zampini lb[1] = isseqaij; 1993b9ed4604SStefano Zampini lb[2] = isseqbaij; 1994b9ed4604SStefano Zampini lb[3] = isseqsbaij; 1995820f2d46SBarry Smith ierr = MPIU_Allreduce(lb,bb,4,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)mat));CHKERRMPI(ierr); 1996b9ed4604SStefano Zampini if (!bb[0] && !bb[1] && !bb[2] && !bb[3]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Local matrices must have the same type"); 199776bd3646SJed Brown } 1998b7ce53b6SStefano Zampini 1999487b449aSStefano Zampini if (reuse != MAT_REUSE_MATRIX) { 2000487b449aSStefano Zampini ierr = MatCreate(PetscObjectComm((PetscObject)mat),&MT);CHKERRQ(ierr); 2001487b449aSStefano Zampini ierr = MatSetSizes(MT,lrows,lcols,rows,cols);CHKERRQ(ierr); 2002487b449aSStefano Zampini ierr = MatSetType(MT,mtype);CHKERRQ(ierr); 200353b44cf5SStefano Zampini ierr = MatSetBlockSizes(MT,rbs,cbs);CHKERRQ(ierr); 2004487b449aSStefano Zampini ierr = MatISSetMPIXAIJPreallocation_Private(mat,MT,PETSC_FALSE);CHKERRQ(ierr); 2005b7ce53b6SStefano Zampini } else { 200653b44cf5SStefano Zampini PetscInt mrbs,mcbs,mrows,mcols,mlrows,mlcols; 2007487b449aSStefano Zampini 2008b7ce53b6SStefano Zampini /* some checks */ 2009487b449aSStefano Zampini MT = *M; 201053b44cf5SStefano Zampini ierr = MatGetBlockSizes(MT,&mrbs,&mcbs);CHKERRQ(ierr); 2011487b449aSStefano Zampini ierr = MatGetSize(MT,&mrows,&mcols);CHKERRQ(ierr); 2012487b449aSStefano Zampini ierr = MatGetLocalSize(MT,&mlrows,&mlcols);CHKERRQ(ierr); 20136c4ed002SBarry Smith if (mrows != rows) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of rows (%d != %d)",rows,mrows); 20146c4ed002SBarry Smith if (mcols != cols) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of cols (%d != %d)",cols,mcols); 20156c4ed002SBarry Smith if (mlrows != lrows) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of local rows (%d != %d)",lrows,mlrows); 20166c4ed002SBarry Smith if (mlcols != lcols) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong number of local cols (%d != %d)",lcols,mlcols); 201753b44cf5SStefano Zampini if (mrbs != rbs) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong row block size (%d != %d)",rbs,mrbs); 201853b44cf5SStefano Zampini if (mcbs != cbs) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix. Wrong col block size (%d != %d)",cbs,mcbs); 2019487b449aSStefano Zampini ierr = MatZeroEntries(MT);CHKERRQ(ierr); 2020b7ce53b6SStefano Zampini } 2021d9a9e74cSStefano Zampini 20228546b261SStefano Zampini if (isseqsbaij || isseqbaij) { 20238546b261SStefano Zampini ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&local_mat);CHKERRQ(ierr); 20248546b261SStefano Zampini isseqaij = PETSC_TRUE; 2025d9a9e74cSStefano Zampini } else { 2026d9a9e74cSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 2027d9a9e74cSStefano Zampini local_mat = matis->A; 2028d9a9e74cSStefano Zampini } 2029686e3a49SStefano Zampini 2030b7ce53b6SStefano Zampini /* Set values */ 2031487b449aSStefano Zampini ierr = MatSetLocalToGlobalMapping(MT,mat->rmap->mapping,mat->cmap->mapping);CHKERRQ(ierr); 2032b9ed4604SStefano Zampini if (isseqdense) { /* special case for dense local matrices */ 203365066ba5SStefano Zampini PetscInt i,*dummy; 2034ecf5a873SStefano Zampini 203565066ba5SStefano Zampini ierr = PetscMalloc1(PetscMax(local_rows,local_cols),&dummy);CHKERRQ(ierr); 203665066ba5SStefano Zampini for (i=0;i<PetscMax(local_rows,local_cols);i++) dummy[i] = i; 2037487b449aSStefano Zampini ierr = MatSetOption(MT,MAT_ROW_ORIENTED,PETSC_FALSE);CHKERRQ(ierr); 20381683a169SBarry Smith ierr = MatDenseGetArrayRead(local_mat,&array);CHKERRQ(ierr); 2039487b449aSStefano Zampini ierr = MatSetValuesLocal(MT,local_rows,dummy,local_cols,dummy,array,ADD_VALUES);CHKERRQ(ierr); 20401683a169SBarry Smith ierr = MatDenseRestoreArrayRead(local_mat,&array);CHKERRQ(ierr); 204165066ba5SStefano Zampini ierr = PetscFree(dummy);CHKERRQ(ierr); 2042686e3a49SStefano Zampini } else if (isseqaij) { 20436afe12f5SStefano Zampini const PetscInt *blocks; 20446afe12f5SStefano Zampini PetscInt i,nvtxs,*xadj,*adjncy, nb; 2045686e3a49SStefano Zampini PetscBool done; 20461683a169SBarry Smith PetscScalar *sarray; 2047686e3a49SStefano Zampini 2048d9a9e74cSStefano Zampini ierr = MatGetRowIJ(local_mat,0,PETSC_FALSE,PETSC_FALSE,&nvtxs,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&done);CHKERRQ(ierr); 2049938e1ff8SStefano Zampini if (!done) SETERRQ(PetscObjectComm((PetscObject)local_mat),PETSC_ERR_PLIB,"Error in MatGetRowIJ"); 20501683a169SBarry Smith ierr = MatSeqAIJGetArray(local_mat,&sarray);CHKERRQ(ierr); 20516afe12f5SStefano Zampini ierr = MatGetVariableBlockSizes(local_mat,&nb,&blocks);CHKERRQ(ierr); 20526afe12f5SStefano Zampini if (nb) { /* speed up assembly for special blocked matrices (used by BDDC) */ 20536afe12f5SStefano Zampini PetscInt sum; 20546afe12f5SStefano Zampini 20556afe12f5SStefano Zampini for (i=0,sum=0;i<nb;i++) sum += blocks[i]; 20566afe12f5SStefano Zampini if (sum == nvtxs) { 20576afe12f5SStefano Zampini PetscInt r; 20586afe12f5SStefano Zampini 20596afe12f5SStefano Zampini for (i=0,r=0;i<nb;i++) { 2060cf9c20a2SJed Brown if (PetscUnlikelyDebug(blocks[i] != xadj[r+1] - xadj[r])) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Invalid block sizes prescribed for block %D: expected %D, got %D",i,blocks[i],xadj[r+1] - xadj[r]); 20611683a169SBarry Smith ierr = MatSetValuesLocal(MT,blocks[i],adjncy+xadj[r],blocks[i],adjncy+xadj[r],sarray+xadj[r],ADD_VALUES);CHKERRQ(ierr); 20626afe12f5SStefano Zampini r += blocks[i]; 20636afe12f5SStefano Zampini } 20646afe12f5SStefano Zampini } else { 2065686e3a49SStefano Zampini for (i=0;i<nvtxs;i++) { 20661683a169SBarry Smith ierr = MatSetValuesLocal(MT,1,&i,xadj[i+1]-xadj[i],adjncy+xadj[i],sarray+xadj[i],ADD_VALUES);CHKERRQ(ierr); 2067686e3a49SStefano Zampini } 20686afe12f5SStefano Zampini } 20696afe12f5SStefano Zampini } else { 20706afe12f5SStefano Zampini for (i=0;i<nvtxs;i++) { 20711683a169SBarry Smith ierr = MatSetValuesLocal(MT,1,&i,xadj[i+1]-xadj[i],adjncy+xadj[i],sarray+xadj[i],ADD_VALUES);CHKERRQ(ierr); 20726afe12f5SStefano Zampini } 20736afe12f5SStefano Zampini } 2074d9a9e74cSStefano Zampini ierr = MatRestoreRowIJ(local_mat,0,PETSC_FALSE,PETSC_FALSE,&nvtxs,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&done);CHKERRQ(ierr); 2075938e1ff8SStefano Zampini if (!done) SETERRQ(PetscObjectComm((PetscObject)local_mat),PETSC_ERR_PLIB,"Error in MatRestoreRowIJ"); 20761683a169SBarry Smith ierr = MatSeqAIJRestoreArray(local_mat,&sarray);CHKERRQ(ierr); 2077686e3a49SStefano Zampini } else { /* very basic values insertion for all other matrix types */ 2078ecf5a873SStefano Zampini PetscInt i; 2079c0962df8SStefano Zampini 2080686e3a49SStefano Zampini for (i=0;i<local_rows;i++) { 2081686e3a49SStefano Zampini PetscInt j; 2082ecf5a873SStefano Zampini const PetscInt *local_indices_cols; 2083686e3a49SStefano Zampini 20841683a169SBarry Smith ierr = MatGetRow(local_mat,i,&j,&local_indices_cols,&array);CHKERRQ(ierr); 2085487b449aSStefano Zampini ierr = MatSetValuesLocal(MT,1,&i,j,local_indices_cols,array,ADD_VALUES);CHKERRQ(ierr); 20861683a169SBarry Smith ierr = MatRestoreRow(local_mat,i,&j,&local_indices_cols,&array);CHKERRQ(ierr); 2087686e3a49SStefano Zampini } 2088b7ce53b6SStefano Zampini } 2089487b449aSStefano Zampini ierr = MatAssemblyBegin(MT,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2090d9a9e74cSStefano Zampini ierr = MatDestroy(&local_mat);CHKERRQ(ierr); 2091487b449aSStefano Zampini ierr = MatAssemblyEnd(MT,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2092b9ed4604SStefano Zampini if (isseqdense) { 2093487b449aSStefano Zampini ierr = MatSetOption(MT,MAT_ROW_ORIENTED,PETSC_TRUE);CHKERRQ(ierr); 2094487b449aSStefano Zampini } 2095487b449aSStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 2096487b449aSStefano Zampini ierr = MatHeaderReplace(mat,&MT);CHKERRQ(ierr); 2097487b449aSStefano Zampini } else if (reuse == MAT_INITIAL_MATRIX) { 2098487b449aSStefano Zampini *M = MT; 2099b7ce53b6SStefano Zampini } 2100b7ce53b6SStefano Zampini PetscFunctionReturn(0); 2101b7ce53b6SStefano Zampini } 2102b7ce53b6SStefano Zampini 2103b7ce53b6SStefano Zampini /*@ 2104b7ce53b6SStefano Zampini MatISGetMPIXAIJ - Converts MATIS matrix into a parallel AIJ format 2105b7ce53b6SStefano Zampini 2106*d8d19677SJose E. Roman Input Parameters: 2107a2b725a8SWilliam Gropp + mat - the matrix (should be of type MATIS) 2108a2b725a8SWilliam Gropp - reuse - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX 2109b7ce53b6SStefano Zampini 2110b7ce53b6SStefano Zampini Output Parameter: 2111b7ce53b6SStefano Zampini . newmat - the matrix in AIJ format 2112b7ce53b6SStefano Zampini 2113b7ce53b6SStefano Zampini Level: developer 2114b7ce53b6SStefano Zampini 211595452b02SPatrick Sanan Notes: 2116487b449aSStefano Zampini This function has been deprecated and it will be removed in future releases. Update your code to use the MatConvert() interface. 2117b7ce53b6SStefano Zampini 2118487b449aSStefano Zampini .seealso: MATIS, MatConvert() 2119b7ce53b6SStefano Zampini @*/ 2120b7ce53b6SStefano Zampini PetscErrorCode MatISGetMPIXAIJ(Mat mat, MatReuse reuse, Mat *newmat) 2121b7ce53b6SStefano Zampini { 2122b7ce53b6SStefano Zampini PetscErrorCode ierr; 2123b7ce53b6SStefano Zampini 2124b7ce53b6SStefano Zampini PetscFunctionBegin; 2125b7ce53b6SStefano Zampini PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 2126b7ce53b6SStefano Zampini PetscValidLogicalCollectiveEnum(mat,reuse,2); 2127b7ce53b6SStefano Zampini PetscValidPointer(newmat,3); 2128487b449aSStefano Zampini if (reuse == MAT_REUSE_MATRIX) { 2129b7ce53b6SStefano Zampini PetscValidHeaderSpecific(*newmat,MAT_CLASSID,3); 2130b7ce53b6SStefano Zampini PetscCheckSameComm(mat,1,*newmat,3); 21316c4ed002SBarry Smith if (mat == *newmat) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse the same matrix"); 2132b7ce53b6SStefano Zampini } 2133487b449aSStefano Zampini ierr = PetscUseMethod(mat,"MatISGetMPIXAIJ_C",(Mat,MatType,MatReuse,Mat*),(mat,MATAIJ,reuse,newmat));CHKERRQ(ierr); 2134b7ce53b6SStefano Zampini PetscFunctionReturn(0); 2135b7ce53b6SStefano Zampini } 2136b7ce53b6SStefano Zampini 2137ad6194a2SStefano Zampini PetscErrorCode MatDuplicate_IS(Mat mat,MatDuplicateOption op,Mat *newmat) 2138ad6194a2SStefano Zampini { 2139ad6194a2SStefano Zampini PetscErrorCode ierr; 2140ad6194a2SStefano Zampini Mat_IS *matis = (Mat_IS*)(mat->data); 2141c9225affSStefano Zampini PetscInt rbs,cbs,m,n,M,N; 2142ad6194a2SStefano Zampini Mat B,localmat; 2143ad6194a2SStefano Zampini 2144ad6194a2SStefano Zampini PetscFunctionBegin; 2145c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(mat->rmap->mapping,&rbs);CHKERRQ(ierr); 2146c9225affSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(mat->cmap->mapping,&cbs);CHKERRQ(ierr); 2147ad6194a2SStefano Zampini ierr = MatGetSize(mat,&M,&N);CHKERRQ(ierr); 2148ad6194a2SStefano Zampini ierr = MatGetLocalSize(mat,&m,&n);CHKERRQ(ierr); 21498546b261SStefano Zampini ierr = MatCreate(PetscObjectComm((PetscObject)mat),&B);CHKERRQ(ierr); 21508546b261SStefano Zampini ierr = MatSetSizes(B,m,n,M,N);CHKERRQ(ierr); 21518546b261SStefano Zampini ierr = MatSetBlockSize(B,rbs == cbs ? rbs : 1);CHKERRQ(ierr); 21528546b261SStefano Zampini ierr = MatSetType(B,MATIS);CHKERRQ(ierr); 21538546b261SStefano Zampini ierr = MatISSetLocalMatType(B,matis->lmattype);CHKERRQ(ierr); 21548546b261SStefano Zampini ierr = MatSetLocalToGlobalMapping(B,mat->rmap->mapping,mat->cmap->mapping);CHKERRQ(ierr); 2155ad6194a2SStefano Zampini ierr = MatDuplicate(matis->A,op,&localmat);CHKERRQ(ierr); 2156ad6194a2SStefano Zampini ierr = MatISSetLocalMat(B,localmat);CHKERRQ(ierr); 2157b3317aa8SStefano Zampini ierr = MatDestroy(&localmat);CHKERRQ(ierr); 2158ad6194a2SStefano Zampini ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2159ad6194a2SStefano Zampini ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2160ad6194a2SStefano Zampini *newmat = B; 2161ad6194a2SStefano Zampini PetscFunctionReturn(0); 2162ad6194a2SStefano Zampini } 2163ad6194a2SStefano Zampini 2164a8116848SStefano Zampini static PetscErrorCode MatIsHermitian_IS(Mat A,PetscReal tol,PetscBool *flg) 216569796d55SStefano Zampini { 216669796d55SStefano Zampini PetscErrorCode ierr; 216769796d55SStefano Zampini Mat_IS *matis = (Mat_IS*)A->data; 216869796d55SStefano Zampini PetscBool local_sym; 216969796d55SStefano Zampini 217069796d55SStefano Zampini PetscFunctionBegin; 217169796d55SStefano Zampini ierr = MatIsHermitian(matis->A,tol,&local_sym);CHKERRQ(ierr); 2172820f2d46SBarry Smith ierr = MPIU_Allreduce(&local_sym,flg,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 217369796d55SStefano Zampini PetscFunctionReturn(0); 217469796d55SStefano Zampini } 217569796d55SStefano Zampini 2176a8116848SStefano Zampini static PetscErrorCode MatIsSymmetric_IS(Mat A,PetscReal tol,PetscBool *flg) 217769796d55SStefano Zampini { 217869796d55SStefano Zampini PetscErrorCode ierr; 217969796d55SStefano Zampini Mat_IS *matis = (Mat_IS*)A->data; 218069796d55SStefano Zampini PetscBool local_sym; 218169796d55SStefano Zampini 218269796d55SStefano Zampini PetscFunctionBegin; 218369796d55SStefano Zampini ierr = MatIsSymmetric(matis->A,tol,&local_sym);CHKERRQ(ierr); 2184820f2d46SBarry Smith ierr = MPIU_Allreduce(&local_sym,flg,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 218569796d55SStefano Zampini PetscFunctionReturn(0); 218669796d55SStefano Zampini } 218769796d55SStefano Zampini 218845471136SStefano Zampini static PetscErrorCode MatIsStructurallySymmetric_IS(Mat A,PetscBool *flg) 218945471136SStefano Zampini { 219045471136SStefano Zampini PetscErrorCode ierr; 219145471136SStefano Zampini Mat_IS *matis = (Mat_IS*)A->data; 219245471136SStefano Zampini PetscBool local_sym; 219345471136SStefano Zampini 219445471136SStefano Zampini PetscFunctionBegin; 219545471136SStefano Zampini if (A->rmap->mapping != A->cmap->mapping) { 219645471136SStefano Zampini *flg = PETSC_FALSE; 219745471136SStefano Zampini PetscFunctionReturn(0); 219845471136SStefano Zampini } 219945471136SStefano Zampini ierr = MatIsStructurallySymmetric(matis->A,&local_sym);CHKERRQ(ierr); 2200820f2d46SBarry Smith ierr = MPIU_Allreduce(&local_sym,flg,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 220145471136SStefano Zampini PetscFunctionReturn(0); 220245471136SStefano Zampini } 220345471136SStefano Zampini 2204a8116848SStefano Zampini static PetscErrorCode MatDestroy_IS(Mat A) 2205b4319ba4SBarry Smith { 2206dfbe8321SBarry Smith PetscErrorCode ierr; 2207b4319ba4SBarry Smith Mat_IS *b = (Mat_IS*)A->data; 2208b4319ba4SBarry Smith 2209b4319ba4SBarry Smith PetscFunctionBegin; 2210b89f26deSStefano Zampini ierr = PetscFree(b->bdiag);CHKERRQ(ierr); 22118546b261SStefano Zampini ierr = PetscFree(b->lmattype);CHKERRQ(ierr); 22126bf464f9SBarry Smith ierr = MatDestroy(&b->A);CHKERRQ(ierr); 2213e176bc59SStefano Zampini ierr = VecScatterDestroy(&b->cctx);CHKERRQ(ierr); 2214e176bc59SStefano Zampini ierr = VecScatterDestroy(&b->rctx);CHKERRQ(ierr); 22156bf464f9SBarry Smith ierr = VecDestroy(&b->x);CHKERRQ(ierr); 22166bf464f9SBarry Smith ierr = VecDestroy(&b->y);CHKERRQ(ierr); 22173fd1c9e7SStefano Zampini ierr = VecDestroy(&b->counter);CHKERRQ(ierr); 2218a8116848SStefano Zampini ierr = ISDestroy(&b->getsub_ris);CHKERRQ(ierr); 2219a8116848SStefano Zampini ierr = ISDestroy(&b->getsub_cis);CHKERRQ(ierr); 2220a8116848SStefano Zampini if (b->sf != b->csf) { 2221a8116848SStefano Zampini ierr = PetscSFDestroy(&b->csf);CHKERRQ(ierr); 2222a8116848SStefano Zampini ierr = PetscFree2(b->csf_rootdata,b->csf_leafdata);CHKERRQ(ierr); 2223f03112d0SStefano Zampini } else b->csf = NULL; 222428f4e0baSStefano Zampini ierr = PetscSFDestroy(&b->sf);CHKERRQ(ierr); 222528f4e0baSStefano Zampini ierr = PetscFree2(b->sf_rootdata,b->sf_leafdata);CHKERRQ(ierr); 2226bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 2227f4259b30SLisandro Dalcin ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr); 22288546b261SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMatType_C",NULL);CHKERRQ(ierr); 2229bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetLocalMat_C",NULL);CHKERRQ(ierr); 2230b7ce53b6SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMat_C",NULL);CHKERRQ(ierr); 2231b7ce53b6SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetMPIXAIJ_C",NULL);CHKERRQ(ierr); 22322e1947a5SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetPreallocation_C",NULL);CHKERRQ(ierr); 223375d48cdbSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISStoreL2L_C",NULL);CHKERRQ(ierr); 2234f03112d0SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISFixLocalEmpty_C",NULL);CHKERRQ(ierr); 2235487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpiaij_C",NULL);CHKERRQ(ierr); 2236487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpibaij_C",NULL);CHKERRQ(ierr); 2237487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpisbaij_C",NULL);CHKERRQ(ierr); 2238487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqaij_C",NULL);CHKERRQ(ierr); 2239487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqbaij_C",NULL);CHKERRQ(ierr); 2240487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqsbaij_C",NULL);CHKERRQ(ierr); 2241487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_aij_C",NULL);CHKERRQ(ierr); 2242b4319ba4SBarry Smith PetscFunctionReturn(0); 2243b4319ba4SBarry Smith } 2244b4319ba4SBarry Smith 2245a8116848SStefano Zampini static PetscErrorCode MatMult_IS(Mat A,Vec x,Vec y) 2246b4319ba4SBarry Smith { 2247dfbe8321SBarry Smith PetscErrorCode ierr; 2248b4319ba4SBarry Smith Mat_IS *is = (Mat_IS*)A->data; 2249b4319ba4SBarry Smith PetscScalar zero = 0.0; 2250b4319ba4SBarry Smith 2251b4319ba4SBarry Smith PetscFunctionBegin; 2252b4319ba4SBarry Smith /* scatter the global vector x into the local work vector */ 2253e176bc59SStefano Zampini ierr = VecScatterBegin(is->cctx,x,is->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2254e176bc59SStefano Zampini ierr = VecScatterEnd(is->cctx,x,is->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2255b4319ba4SBarry Smith 2256b4319ba4SBarry Smith /* multiply the local matrix */ 2257b4319ba4SBarry Smith ierr = MatMult(is->A,is->x,is->y);CHKERRQ(ierr); 2258b4319ba4SBarry Smith 2259b4319ba4SBarry Smith /* scatter product back into global memory */ 22602dcb1b2aSMatthew Knepley ierr = VecSet(y,zero);CHKERRQ(ierr); 2261e176bc59SStefano Zampini ierr = VecScatterBegin(is->rctx,is->y,y,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2262e176bc59SStefano Zampini ierr = VecScatterEnd(is->rctx,is->y,y,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2263b4319ba4SBarry Smith PetscFunctionReturn(0); 2264b4319ba4SBarry Smith } 2265b4319ba4SBarry Smith 2266a8116848SStefano Zampini static PetscErrorCode MatMultAdd_IS(Mat A,Vec v1,Vec v2,Vec v3) 22672e74eeadSLisandro Dalcin { 2268650997f4SStefano Zampini Vec temp_vec; 22692e74eeadSLisandro Dalcin PetscErrorCode ierr; 22702e74eeadSLisandro Dalcin 22712e74eeadSLisandro Dalcin PetscFunctionBegin; /* v3 = v2 + A * v1.*/ 2272650997f4SStefano Zampini if (v3 != v2) { 2273650997f4SStefano Zampini ierr = MatMult(A,v1,v3);CHKERRQ(ierr); 2274650997f4SStefano Zampini ierr = VecAXPY(v3,1.0,v2);CHKERRQ(ierr); 2275650997f4SStefano Zampini } else { 2276650997f4SStefano Zampini ierr = VecDuplicate(v2,&temp_vec);CHKERRQ(ierr); 2277650997f4SStefano Zampini ierr = MatMult(A,v1,temp_vec);CHKERRQ(ierr); 2278650997f4SStefano Zampini ierr = VecAXPY(temp_vec,1.0,v2);CHKERRQ(ierr); 2279650997f4SStefano Zampini ierr = VecCopy(temp_vec,v3);CHKERRQ(ierr); 2280650997f4SStefano Zampini ierr = VecDestroy(&temp_vec);CHKERRQ(ierr); 2281650997f4SStefano Zampini } 22822e74eeadSLisandro Dalcin PetscFunctionReturn(0); 22832e74eeadSLisandro Dalcin } 22842e74eeadSLisandro Dalcin 2285a8116848SStefano Zampini static PetscErrorCode MatMultTranspose_IS(Mat A,Vec y,Vec x) 22862e74eeadSLisandro Dalcin { 22872e74eeadSLisandro Dalcin Mat_IS *is = (Mat_IS*)A->data; 22882e74eeadSLisandro Dalcin PetscErrorCode ierr; 22892e74eeadSLisandro Dalcin 2290e176bc59SStefano Zampini PetscFunctionBegin; 22912e74eeadSLisandro Dalcin /* scatter the global vector x into the local work vector */ 2292e176bc59SStefano Zampini ierr = VecScatterBegin(is->rctx,y,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2293e176bc59SStefano Zampini ierr = VecScatterEnd(is->rctx,y,is->y,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22942e74eeadSLisandro Dalcin 22952e74eeadSLisandro Dalcin /* multiply the local matrix */ 2296e176bc59SStefano Zampini ierr = MatMultTranspose(is->A,is->y,is->x);CHKERRQ(ierr); 22972e74eeadSLisandro Dalcin 22982e74eeadSLisandro Dalcin /* scatter product back into global vector */ 2299e176bc59SStefano Zampini ierr = VecSet(x,0);CHKERRQ(ierr); 2300e176bc59SStefano Zampini ierr = VecScatterBegin(is->cctx,is->x,x,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2301e176bc59SStefano Zampini ierr = VecScatterEnd(is->cctx,is->x,x,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23022e74eeadSLisandro Dalcin PetscFunctionReturn(0); 23032e74eeadSLisandro Dalcin } 23042e74eeadSLisandro Dalcin 2305a8116848SStefano Zampini static PetscErrorCode MatMultTransposeAdd_IS(Mat A,Vec v1,Vec v2,Vec v3) 23062e74eeadSLisandro Dalcin { 2307650997f4SStefano Zampini Vec temp_vec; 23082e74eeadSLisandro Dalcin PetscErrorCode ierr; 23092e74eeadSLisandro Dalcin 23102e74eeadSLisandro Dalcin PetscFunctionBegin; /* v3 = v2 + A' * v1.*/ 2311650997f4SStefano Zampini if (v3 != v2) { 2312650997f4SStefano Zampini ierr = MatMultTranspose(A,v1,v3);CHKERRQ(ierr); 2313650997f4SStefano Zampini ierr = VecAXPY(v3,1.0,v2);CHKERRQ(ierr); 2314650997f4SStefano Zampini } else { 2315650997f4SStefano Zampini ierr = VecDuplicate(v2,&temp_vec);CHKERRQ(ierr); 2316650997f4SStefano Zampini ierr = MatMultTranspose(A,v1,temp_vec);CHKERRQ(ierr); 2317650997f4SStefano Zampini ierr = VecAXPY(temp_vec,1.0,v2);CHKERRQ(ierr); 2318650997f4SStefano Zampini ierr = VecCopy(temp_vec,v3);CHKERRQ(ierr); 2319650997f4SStefano Zampini ierr = VecDestroy(&temp_vec);CHKERRQ(ierr); 2320650997f4SStefano Zampini } 23212e74eeadSLisandro Dalcin PetscFunctionReturn(0); 23222e74eeadSLisandro Dalcin } 23232e74eeadSLisandro Dalcin 2324a8116848SStefano Zampini static PetscErrorCode MatView_IS(Mat A,PetscViewer viewer) 2325b4319ba4SBarry Smith { 2326b4319ba4SBarry Smith Mat_IS *a = (Mat_IS*)A->data; 2327dfbe8321SBarry Smith PetscErrorCode ierr; 2328b4319ba4SBarry Smith PetscViewer sviewer; 2329ee2491ecSStefano Zampini PetscBool isascii,view = PETSC_TRUE; 2330b4319ba4SBarry Smith 2331b4319ba4SBarry Smith PetscFunctionBegin; 2332ee2491ecSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 2333ee2491ecSStefano Zampini if (isascii) { 2334ee2491ecSStefano Zampini PetscViewerFormat format; 2335ee2491ecSStefano Zampini 2336ee2491ecSStefano Zampini ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 2337ee2491ecSStefano Zampini if (format == PETSC_VIEWER_ASCII_INFO) view = PETSC_FALSE; 2338ee2491ecSStefano Zampini } 2339ee2491ecSStefano Zampini if (!view) PetscFunctionReturn(0); 23403f08860eSBarry Smith ierr = PetscViewerGetSubViewer(viewer,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 2341b4319ba4SBarry Smith ierr = MatView(a->A,sviewer);CHKERRQ(ierr); 23423f08860eSBarry Smith ierr = PetscViewerRestoreSubViewer(viewer,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 23436e520ac8SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 2344b4319ba4SBarry Smith PetscFunctionReturn(0); 2345b4319ba4SBarry Smith } 2346b4319ba4SBarry Smith 2347b89f26deSStefano Zampini static PetscErrorCode MatInvertBlockDiagonal_IS(Mat mat,const PetscScalar **values) 2348b89f26deSStefano Zampini { 2349b89f26deSStefano Zampini Mat_IS *is = (Mat_IS*)mat->data; 2350b89f26deSStefano Zampini MPI_Datatype nodeType; 2351b89f26deSStefano Zampini const PetscScalar *lv; 2352b89f26deSStefano Zampini PetscInt bs; 2353b89f26deSStefano Zampini PetscErrorCode ierr; 2354b89f26deSStefano Zampini 2355b89f26deSStefano Zampini PetscFunctionBegin; 2356b89f26deSStefano Zampini ierr = MatGetBlockSize(mat,&bs);CHKERRQ(ierr); 2357b89f26deSStefano Zampini ierr = MatSetBlockSize(is->A,bs);CHKERRQ(ierr); 2358b89f26deSStefano Zampini ierr = MatInvertBlockDiagonal(is->A,&lv);CHKERRQ(ierr); 2359b89f26deSStefano Zampini if (!is->bdiag) { 2360b89f26deSStefano Zampini ierr = PetscMalloc1(bs*mat->rmap->n,&is->bdiag);CHKERRQ(ierr); 2361b89f26deSStefano Zampini } 2362ffc4695bSBarry Smith ierr = MPI_Type_contiguous(bs,MPIU_SCALAR,&nodeType);CHKERRMPI(ierr); 2363ffc4695bSBarry Smith ierr = MPI_Type_commit(&nodeType);CHKERRMPI(ierr); 236483df288dSJunchao Zhang ierr = PetscSFReduceBegin(is->sf,nodeType,lv,is->bdiag,MPI_REPLACE);CHKERRQ(ierr); 236583df288dSJunchao Zhang ierr = PetscSFReduceEnd(is->sf,nodeType,lv,is->bdiag,MPI_REPLACE);CHKERRQ(ierr); 2366ffc4695bSBarry Smith ierr = MPI_Type_free(&nodeType);CHKERRMPI(ierr); 2367b89f26deSStefano Zampini if (values) *values = is->bdiag; 2368b89f26deSStefano Zampini PetscFunctionReturn(0); 2369b89f26deSStefano Zampini } 2370b89f26deSStefano Zampini 23718546b261SStefano Zampini static PetscErrorCode MatISSetUpScatters_Private(Mat A) 2372b4319ba4SBarry Smith { 2373e176bc59SStefano Zampini Vec cglobal,rglobal; 23748546b261SStefano Zampini IS from; 23758546b261SStefano Zampini Mat_IS *is = (Mat_IS*)A->data; 2376b89f26deSStefano Zampini PetscScalar sum; 23778546b261SStefano Zampini const PetscInt *garray; 23788546b261SStefano Zampini PetscInt nr,rbs,nc,cbs; 23798546b261SStefano Zampini PetscBool iscuda; 23808546b261SStefano Zampini PetscErrorCode ierr; 2381b4319ba4SBarry Smith 2382b4319ba4SBarry Smith PetscFunctionBegin; 23838546b261SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(A->rmap->mapping,&nr);CHKERRQ(ierr); 23848546b261SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(A->rmap->mapping,&rbs);CHKERRQ(ierr); 23858546b261SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(A->cmap->mapping,&nc);CHKERRQ(ierr); 23868546b261SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(A->cmap->mapping,&cbs);CHKERRQ(ierr); 238770cf5478SStefano Zampini ierr = VecDestroy(&is->x);CHKERRQ(ierr); 238870cf5478SStefano Zampini ierr = VecDestroy(&is->y);CHKERRQ(ierr); 23893fd1c9e7SStefano Zampini ierr = VecDestroy(&is->counter);CHKERRQ(ierr); 2390e176bc59SStefano Zampini ierr = VecScatterDestroy(&is->rctx);CHKERRQ(ierr); 2391e176bc59SStefano Zampini ierr = VecScatterDestroy(&is->cctx);CHKERRQ(ierr); 23928546b261SStefano Zampini ierr = MatCreateVecs(is->A,&is->x,&is->y);CHKERRQ(ierr); 2393b470e4b4SRichard Tran Mills ierr = VecBindToCPU(is->y,PETSC_TRUE);CHKERRQ(ierr); 23948546b261SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)is->y,VECSEQCUDA,&iscuda);CHKERRQ(ierr); 23958546b261SStefano Zampini if (iscuda) { 23968546b261SStefano Zampini ierr = PetscFree(A->defaultvectype);CHKERRQ(ierr); 23978546b261SStefano Zampini ierr = PetscStrallocpy(VECCUDA,&A->defaultvectype);CHKERRQ(ierr); 23988546b261SStefano Zampini } 23998546b261SStefano Zampini ierr = MatCreateVecs(A,&cglobal,&rglobal);CHKERRQ(ierr); 2400b470e4b4SRichard Tran Mills ierr = VecBindToCPU(rglobal,PETSC_TRUE);CHKERRQ(ierr); 24018546b261SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(A->rmap->mapping,&garray);CHKERRQ(ierr); 24028546b261SStefano Zampini ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),rbs,nr/rbs,garray,PETSC_USE_POINTER,&from);CHKERRQ(ierr); 24039448b7f1SJunchao Zhang ierr = VecScatterCreate(rglobal,from,is->y,NULL,&is->rctx);CHKERRQ(ierr); 24048546b261SStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(A->rmap->mapping,&garray);CHKERRQ(ierr); 24058546b261SStefano Zampini ierr = ISDestroy(&from);CHKERRQ(ierr); 24068546b261SStefano Zampini if (A->rmap->mapping != A->cmap->mapping) { 24078546b261SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(A->cmap->mapping,&garray);CHKERRQ(ierr); 24088546b261SStefano Zampini ierr = ISCreateBlock(PetscObjectComm((PetscObject)A),cbs,nc/cbs,garray,PETSC_USE_POINTER,&from);CHKERRQ(ierr); 24099448b7f1SJunchao Zhang ierr = VecScatterCreate(cglobal,from,is->x,NULL,&is->cctx);CHKERRQ(ierr); 24108546b261SStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(A->cmap->mapping,&garray);CHKERRQ(ierr); 24118546b261SStefano Zampini ierr = ISDestroy(&from);CHKERRQ(ierr); 24128546b261SStefano Zampini } else { 24138546b261SStefano Zampini ierr = PetscObjectReference((PetscObject)is->rctx);CHKERRQ(ierr); 24148546b261SStefano Zampini is->cctx = is->rctx; 24158546b261SStefano Zampini } 2416b89f26deSStefano Zampini ierr = VecDestroy(&cglobal);CHKERRQ(ierr); 2417b89f26deSStefano Zampini 24188546b261SStefano Zampini /* interface counter vector (local) */ 24198546b261SStefano Zampini ierr = VecDuplicate(is->y,&is->counter);CHKERRQ(ierr); 2420b470e4b4SRichard Tran Mills ierr = VecBindToCPU(is->counter,PETSC_TRUE);CHKERRQ(ierr); 24218546b261SStefano Zampini ierr = VecSet(is->y,1.);CHKERRQ(ierr); 24228546b261SStefano Zampini ierr = VecScatterBegin(is->rctx,is->y,rglobal,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 24238546b261SStefano Zampini ierr = VecScatterEnd(is->rctx,is->y,rglobal,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 24248546b261SStefano Zampini ierr = VecScatterBegin(is->rctx,rglobal,is->counter,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 24258546b261SStefano Zampini ierr = VecScatterEnd(is->rctx,rglobal,is->counter,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2426b470e4b4SRichard Tran Mills ierr = VecBindToCPU(is->y,PETSC_FALSE);CHKERRQ(ierr); 2427b470e4b4SRichard Tran Mills ierr = VecBindToCPU(is->counter,PETSC_FALSE);CHKERRQ(ierr); 2428b89f26deSStefano Zampini 2429b89f26deSStefano Zampini /* special functions for block-diagonal matrices */ 2430b89f26deSStefano Zampini ierr = VecSum(rglobal,&sum);CHKERRQ(ierr); 2431b89f26deSStefano Zampini if ((PetscInt)(PetscRealPart(sum)) == A->rmap->N && A->rmap->N == A->cmap->N && A->rmap->mapping == A->cmap->mapping) { 2432b89f26deSStefano Zampini A->ops->invertblockdiagonal = MatInvertBlockDiagonal_IS; 2433b89f26deSStefano Zampini } else { 2434b89f26deSStefano Zampini A->ops->invertblockdiagonal = NULL; 2435b89f26deSStefano Zampini } 24368546b261SStefano Zampini ierr = VecDestroy(&rglobal);CHKERRQ(ierr); 2437b0cc1f67SStefano Zampini 2438b0cc1f67SStefano Zampini /* setup SF for general purpose shared indices based communications */ 2439b0cc1f67SStefano Zampini ierr = MatISSetUpSF_IS(A);CHKERRQ(ierr); 24408546b261SStefano Zampini PetscFunctionReturn(0); 24418546b261SStefano Zampini } 24428546b261SStefano Zampini 24438546b261SStefano Zampini static PetscErrorCode MatSetLocalToGlobalMapping_IS(Mat A,ISLocalToGlobalMapping rmapping,ISLocalToGlobalMapping cmapping) 24448546b261SStefano Zampini { 24458546b261SStefano Zampini PetscErrorCode ierr; 24468546b261SStefano Zampini PetscInt nr,rbs,nc,cbs; 24478546b261SStefano Zampini Mat_IS *is = (Mat_IS*)A->data; 24488546b261SStefano Zampini 24498546b261SStefano Zampini PetscFunctionBegin; 24508546b261SStefano Zampini PetscCheckSameComm(A,1,rmapping,2); 24518546b261SStefano Zampini PetscCheckSameComm(A,1,cmapping,3); 24521c47cb0fSStefano Zampini ierr = MatDestroy(&is->A);CHKERRQ(ierr); 2453872cf891SStefano Zampini if (is->csf != is->sf) { 2454872cf891SStefano Zampini ierr = PetscSFDestroy(&is->csf);CHKERRQ(ierr); 2455872cf891SStefano Zampini ierr = PetscFree2(is->csf_rootdata,is->csf_leafdata);CHKERRQ(ierr); 2456f03112d0SStefano Zampini } else is->csf = NULL; 245728f4e0baSStefano Zampini ierr = PetscSFDestroy(&is->sf);CHKERRQ(ierr); 245828f4e0baSStefano Zampini ierr = PetscFree2(is->sf_rootdata,is->sf_leafdata);CHKERRQ(ierr); 2459b89f26deSStefano Zampini ierr = PetscFree(is->bdiag);CHKERRQ(ierr); 24603bbff08aSStefano Zampini 24613bbff08aSStefano Zampini /* Setup Layout and set local to global maps */ 2462fc27028aSStefano Zampini ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 2463fc27028aSStefano Zampini ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 2464e176bc59SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(rmapping,&nr);CHKERRQ(ierr); 2465e176bc59SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(rmapping,&rbs);CHKERRQ(ierr); 2466e176bc59SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(cmapping,&nc);CHKERRQ(ierr); 2467e176bc59SStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(cmapping,&cbs);CHKERRQ(ierr); 24686625354bSStefano Zampini /* check if the two mappings are actually the same for square matrices (DOLFIN passes 2 different objects) */ 24696625354bSStefano Zampini if (rmapping != cmapping && A->rmap->N == A->cmap->N) { 24706625354bSStefano Zampini PetscBool same,gsame; 24716625354bSStefano Zampini 24726625354bSStefano Zampini same = PETSC_FALSE; 24736625354bSStefano Zampini if (nr == nc && cbs == rbs) { 24746625354bSStefano Zampini const PetscInt *idxs1,*idxs2; 24756625354bSStefano Zampini 24766625354bSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(rmapping,&idxs1);CHKERRQ(ierr); 24776625354bSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockIndices(cmapping,&idxs2);CHKERRQ(ierr); 2478580bdb30SBarry Smith ierr = PetscArraycmp(idxs1,idxs2,nr/rbs,&same);CHKERRQ(ierr); 24796625354bSStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(rmapping,&idxs1);CHKERRQ(ierr); 24806625354bSStefano Zampini ierr = ISLocalToGlobalMappingRestoreBlockIndices(cmapping,&idxs2);CHKERRQ(ierr); 24816625354bSStefano Zampini } 2482820f2d46SBarry Smith ierr = MPIU_Allreduce(&same,&gsame,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 24836625354bSStefano Zampini if (gsame) cmapping = rmapping; 24846625354bSStefano Zampini } 248553b44cf5SStefano Zampini ierr = PetscLayoutSetBlockSize(A->rmap,rbs);CHKERRQ(ierr); 248653b44cf5SStefano Zampini ierr = PetscLayoutSetBlockSize(A->cmap,cbs);CHKERRQ(ierr); 24876625354bSStefano Zampini ierr = PetscLayoutSetISLocalToGlobalMapping(A->rmap,rmapping);CHKERRQ(ierr); 24886625354bSStefano Zampini ierr = PetscLayoutSetISLocalToGlobalMapping(A->cmap,cmapping);CHKERRQ(ierr); 24896625354bSStefano Zampini 24906625354bSStefano Zampini /* Create the local matrix A */ 2491f69a0ea3SMatthew Knepley ierr = MatCreate(PETSC_COMM_SELF,&is->A);CHKERRQ(ierr); 24928546b261SStefano Zampini ierr = MatSetType(is->A,is->lmattype);CHKERRQ(ierr); 2493e176bc59SStefano Zampini ierr = MatSetSizes(is->A,nr,nc,nr,nc);CHKERRQ(ierr); 2494e176bc59SStefano Zampini ierr = MatSetBlockSizes(is->A,rbs,cbs);CHKERRQ(ierr); 24958546b261SStefano Zampini ierr = MatSetOptionsPrefix(is->A,"is_");CHKERRQ(ierr); 24968546b261SStefano Zampini ierr = MatAppendOptionsPrefix(is->A,((PetscObject)A)->prefix);CHKERRQ(ierr); 2497c77832edSStefano Zampini ierr = PetscLayoutSetUp(is->A->rmap);CHKERRQ(ierr); 2498c77832edSStefano Zampini ierr = PetscLayoutSetUp(is->A->cmap);CHKERRQ(ierr); 2499b4319ba4SBarry Smith 2500f26d0771SStefano Zampini if (!is->islocalref) { /* setup scatters and local vectors for MatMult */ 25018546b261SStefano Zampini ierr = MatISSetUpScatters_Private(A);CHKERRQ(ierr); 2502f26d0771SStefano Zampini } 250348ff6bf3SStefano Zampini ierr = MatSetUp(A);CHKERRQ(ierr); 2504b4319ba4SBarry Smith PetscFunctionReturn(0); 2505b4319ba4SBarry Smith } 2506b4319ba4SBarry Smith 2507a8116848SStefano Zampini static PetscErrorCode MatSetValues_IS(Mat mat, PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols, const PetscScalar *values, InsertMode addv) 25082e74eeadSLisandro Dalcin { 25092e74eeadSLisandro Dalcin Mat_IS *is = (Mat_IS*)mat->data; 25102e74eeadSLisandro Dalcin PetscErrorCode ierr; 251197563a80SStefano Zampini PetscInt i,zm,zn; 2512f26d0771SStefano Zampini PetscInt rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION]; 25132e74eeadSLisandro Dalcin 25142e74eeadSLisandro Dalcin PetscFunctionBegin; 251576bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2516f26d0771SStefano Zampini if (m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n); 251797563a80SStefano Zampini /* count negative indices */ 251897563a80SStefano Zampini for (i=0,zm=0;i<m;i++) if (rows[i] < 0) zm++; 251997563a80SStefano Zampini for (i=0,zn=0;i<n;i++) if (cols[i] < 0) zn++; 252076bd3646SJed Brown } 252197563a80SStefano Zampini ierr = ISGlobalToLocalMappingApply(mat->rmap->mapping,IS_GTOLM_MASK,m,rows,&m,rows_l);CHKERRQ(ierr); 252297563a80SStefano Zampini ierr = ISGlobalToLocalMappingApply(mat->cmap->mapping,IS_GTOLM_MASK,n,cols,&n,cols_l);CHKERRQ(ierr); 252376bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 252497563a80SStefano Zampini /* count negative indices (should be the same as before) */ 252597563a80SStefano Zampini for (i=0;i<m;i++) if (rows_l[i] < 0) zm--; 252697563a80SStefano Zampini for (i=0;i<n;i++) if (cols_l[i] < 0) zn--; 2527b4f971dfSStefano Zampini if (!is->A->rmap->mapping && zm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the row indices can not be mapped! Maybe you should not use MATIS"); 2528b4f971dfSStefano Zampini if (!is->A->cmap->mapping && zn) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the column indices can not be mapped! Maybe you should not use MATIS"); 252976bd3646SJed Brown } 25302e74eeadSLisandro Dalcin ierr = MatSetValues(is->A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr); 25312e74eeadSLisandro Dalcin PetscFunctionReturn(0); 25322e74eeadSLisandro Dalcin } 25332e74eeadSLisandro Dalcin 2534a8116848SStefano Zampini static PetscErrorCode MatSetValuesBlocked_IS(Mat mat, PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols, const PetscScalar *values, InsertMode addv) 253597563a80SStefano Zampini { 253697563a80SStefano Zampini Mat_IS *is = (Mat_IS*)mat->data; 253797563a80SStefano Zampini PetscErrorCode ierr; 253897563a80SStefano Zampini PetscInt i,zm,zn; 2539f26d0771SStefano Zampini PetscInt rows_l[MATIS_MAX_ENTRIES_INSERTION],cols_l[MATIS_MAX_ENTRIES_INSERTION]; 254097563a80SStefano Zampini 254197563a80SStefano Zampini PetscFunctionBegin; 254276bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2543f26d0771SStefano Zampini if (m > MATIS_MAX_ENTRIES_INSERTION || n > MATIS_MAX_ENTRIES_INSERTION) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Number of row/column block indices must be <= %D: they are %D %D",MATIS_MAX_ENTRIES_INSERTION,m,n); 254497563a80SStefano Zampini /* count negative indices */ 254597563a80SStefano Zampini for (i=0,zm=0;i<m;i++) if (rows[i] < 0) zm++; 254697563a80SStefano Zampini for (i=0,zn=0;i<n;i++) if (cols[i] < 0) zn++; 254776bd3646SJed Brown } 254897563a80SStefano Zampini ierr = ISGlobalToLocalMappingApplyBlock(mat->rmap->mapping,IS_GTOLM_MASK,m,rows,&m,rows_l);CHKERRQ(ierr); 254997563a80SStefano Zampini ierr = ISGlobalToLocalMappingApplyBlock(mat->cmap->mapping,IS_GTOLM_MASK,n,cols,&n,cols_l);CHKERRQ(ierr); 255076bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 255197563a80SStefano Zampini /* count negative indices (should be the same as before) */ 255297563a80SStefano Zampini for (i=0;i<m;i++) if (rows_l[i] < 0) zm--; 255397563a80SStefano Zampini for (i=0;i<n;i++) if (cols_l[i] < 0) zn--; 2554b4f971dfSStefano Zampini if (!is->A->rmap->mapping && zm) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the row indices can not be mapped! Maybe you should not use MATIS"); 2555b4f971dfSStefano Zampini if (!is->A->cmap->mapping && zn) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Some of the column indices can not be mapped! Maybe you should not use MATIS"); 255676bd3646SJed Brown } 2557d59cf9ebSStefano Zampini ierr = MatSetValuesBlocked(is->A,m,rows_l,n,cols_l,values,addv);CHKERRQ(ierr); 255897563a80SStefano Zampini PetscFunctionReturn(0); 255997563a80SStefano Zampini } 256097563a80SStefano Zampini 2561a8116848SStefano Zampini static PetscErrorCode MatSetValuesLocal_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv) 2562b4319ba4SBarry Smith { 2563dfbe8321SBarry Smith PetscErrorCode ierr; 2564b4319ba4SBarry Smith Mat_IS *is = (Mat_IS*)A->data; 2565b4319ba4SBarry Smith 2566b4319ba4SBarry Smith PetscFunctionBegin; 2567b4f971dfSStefano Zampini if (is->A->rmap->mapping) { 2568872cf891SStefano Zampini ierr = MatSetValuesLocal(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr); 2569872cf891SStefano Zampini } else { 2570b4319ba4SBarry Smith ierr = MatSetValues(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr); 2571872cf891SStefano Zampini } 2572b4319ba4SBarry Smith PetscFunctionReturn(0); 2573b4319ba4SBarry Smith } 2574b4319ba4SBarry Smith 2575a8116848SStefano Zampini static PetscErrorCode MatSetValuesBlockedLocal_IS(Mat A,PetscInt m,const PetscInt *rows, PetscInt n,const PetscInt *cols,const PetscScalar *values,InsertMode addv) 2576f0006bf2SLisandro Dalcin { 2577f0006bf2SLisandro Dalcin PetscErrorCode ierr; 2578f0006bf2SLisandro Dalcin Mat_IS *is = (Mat_IS*)A->data; 2579f0006bf2SLisandro Dalcin 2580f0006bf2SLisandro Dalcin PetscFunctionBegin; 2581b4f971dfSStefano Zampini if (is->A->rmap->mapping) { 258276bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2583b4f971dfSStefano Zampini PetscInt ibs,bs; 2584b4f971dfSStefano Zampini 2585b4f971dfSStefano Zampini ierr = ISLocalToGlobalMappingGetBlockSize(is->A->rmap->mapping,&ibs);CHKERRQ(ierr); 2586b4f971dfSStefano Zampini ierr = MatGetBlockSize(is->A,&bs);CHKERRQ(ierr); 2587b4f971dfSStefano Zampini if (ibs != bs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Different block sizes! local mat %D, local l2g map %D",bs,ibs); 258876bd3646SJed Brown } 2589b4f971dfSStefano Zampini ierr = MatSetValuesBlockedLocal(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr); 2590b4f971dfSStefano Zampini } else { 2591f0006bf2SLisandro Dalcin ierr = MatSetValuesBlocked(is->A,m,rows,n,cols,values,addv);CHKERRQ(ierr); 2592b4f971dfSStefano Zampini } 2593f0006bf2SLisandro Dalcin PetscFunctionReturn(0); 2594f0006bf2SLisandro Dalcin } 2595f0006bf2SLisandro Dalcin 2596f0ae7da4SStefano Zampini static PetscErrorCode MatISZeroRowsColumnsLocal_Private(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,PetscBool columns) 2597f0ae7da4SStefano Zampini { 2598f0ae7da4SStefano Zampini Mat_IS *is = (Mat_IS*)A->data; 2599f0ae7da4SStefano Zampini PetscErrorCode ierr; 2600f0ae7da4SStefano Zampini 2601f0ae7da4SStefano Zampini PetscFunctionBegin; 2602f0ae7da4SStefano Zampini if (!n) { 2603f0ae7da4SStefano Zampini is->pure_neumann = PETSC_TRUE; 2604f0ae7da4SStefano Zampini } else { 2605f0ae7da4SStefano Zampini PetscInt i; 2606f0ae7da4SStefano Zampini is->pure_neumann = PETSC_FALSE; 2607f0ae7da4SStefano Zampini 2608f0ae7da4SStefano Zampini if (columns) { 2609f4259b30SLisandro Dalcin ierr = MatZeroRowsColumns(is->A,n,rows,diag,NULL,NULL);CHKERRQ(ierr); 2610f0ae7da4SStefano Zampini } else { 2611f4259b30SLisandro Dalcin ierr = MatZeroRows(is->A,n,rows,diag,NULL,NULL);CHKERRQ(ierr); 2612f0ae7da4SStefano Zampini } 2613f0ae7da4SStefano Zampini if (diag != 0.) { 2614f0ae7da4SStefano Zampini const PetscScalar *array; 2615f0ae7da4SStefano Zampini ierr = VecGetArrayRead(is->counter,&array);CHKERRQ(ierr); 2616f0ae7da4SStefano Zampini for (i=0; i<n; i++) { 2617f0ae7da4SStefano Zampini ierr = MatSetValue(is->A,rows[i],rows[i],diag/(array[rows[i]]),INSERT_VALUES);CHKERRQ(ierr); 2618f0ae7da4SStefano Zampini } 2619f0ae7da4SStefano Zampini ierr = VecRestoreArrayRead(is->counter,&array);CHKERRQ(ierr); 2620f0ae7da4SStefano Zampini } 2621f0ae7da4SStefano Zampini ierr = MatAssemblyBegin(is->A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2622f0ae7da4SStefano Zampini ierr = MatAssemblyEnd(is->A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2623f0ae7da4SStefano Zampini } 2624f0ae7da4SStefano Zampini PetscFunctionReturn(0); 2625f0ae7da4SStefano Zampini } 2626f0ae7da4SStefano Zampini 2627f0ae7da4SStefano Zampini static PetscErrorCode MatZeroRowsColumns_Private_IS(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,Vec x,Vec b,PetscBool columns) 26282e74eeadSLisandro Dalcin { 26296e520ac8SStefano Zampini Mat_IS *matis = (Mat_IS*)A->data; 26306e520ac8SStefano Zampini PetscInt nr,nl,len,i; 26316e520ac8SStefano Zampini PetscInt *lrows; 26322e74eeadSLisandro Dalcin PetscErrorCode ierr; 26332e74eeadSLisandro Dalcin 26342e74eeadSLisandro Dalcin PetscFunctionBegin; 2635cf9c20a2SJed Brown if (PetscUnlikelyDebug(columns || diag != 0. || (x && b))) { 2636f0ae7da4SStefano Zampini PetscBool cong; 263726b0207aSStefano Zampini 2638f0ae7da4SStefano Zampini ierr = PetscLayoutCompare(A->rmap,A->cmap,&cong);CHKERRQ(ierr); 263926b0207aSStefano Zampini cong = (PetscBool)(cong && matis->sf == matis->csf); 2640268753edSStefano Zampini if (!cong && columns) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Columns can be zeroed if and only if A->rmap and A->cmap are congruent and the l2g maps are the same for MATIS"); 2641268753edSStefano Zampini if (!cong && diag != 0.) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Nonzero diagonal value supported if and only if A->rmap and A->cmap are congruent and the l2g maps are the same for MATIS"); 2642268753edSStefano Zampini if (!cong && x && b) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"A->rmap and A->cmap need to be congruent, and the l2g maps be the same"); 2643f0ae7da4SStefano Zampini } 26446e520ac8SStefano Zampini /* get locally owned rows */ 2645a72d46e8SStefano Zampini ierr = PetscLayoutMapLocal(A->rmap,n,rows,&len,&lrows,NULL);CHKERRQ(ierr); 26466e520ac8SStefano Zampini /* fix right hand side if needed */ 26476e520ac8SStefano Zampini if (x && b) { 26486e520ac8SStefano Zampini const PetscScalar *xx; 26496e520ac8SStefano Zampini PetscScalar *bb; 26506e520ac8SStefano Zampini 26516e520ac8SStefano Zampini ierr = VecGetArrayRead(x, &xx);CHKERRQ(ierr); 26526e520ac8SStefano Zampini ierr = VecGetArray(b, &bb);CHKERRQ(ierr); 26536e520ac8SStefano Zampini for (i=0;i<len;++i) bb[lrows[i]] = diag*xx[lrows[i]]; 26546e520ac8SStefano Zampini ierr = VecRestoreArrayRead(x, &xx);CHKERRQ(ierr); 26556e520ac8SStefano Zampini ierr = VecRestoreArray(b, &bb);CHKERRQ(ierr); 26562e74eeadSLisandro Dalcin } 26576e520ac8SStefano Zampini /* get rows associated to the local matrices */ 26586e520ac8SStefano Zampini ierr = MatGetSize(matis->A,&nl,NULL);CHKERRQ(ierr); 2659580bdb30SBarry Smith ierr = PetscArrayzero(matis->sf_leafdata,nl);CHKERRQ(ierr); 2660580bdb30SBarry Smith ierr = PetscArrayzero(matis->sf_rootdata,A->rmap->n);CHKERRQ(ierr); 26616e520ac8SStefano Zampini for (i=0;i<len;i++) matis->sf_rootdata[lrows[i]] = 1; 26626e520ac8SStefano Zampini ierr = PetscFree(lrows);CHKERRQ(ierr); 2663ad227feaSJunchao Zhang ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 2664ad227feaSJunchao Zhang ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata,MPI_REPLACE);CHKERRQ(ierr); 26656e520ac8SStefano Zampini ierr = PetscMalloc1(nl,&lrows);CHKERRQ(ierr); 26666e520ac8SStefano Zampini for (i=0,nr=0;i<nl;i++) if (matis->sf_leafdata[i]) lrows[nr++] = i; 2667f0ae7da4SStefano Zampini ierr = MatISZeroRowsColumnsLocal_Private(A,nr,lrows,diag,columns);CHKERRQ(ierr); 26686e520ac8SStefano Zampini ierr = PetscFree(lrows);CHKERRQ(ierr); 26692e74eeadSLisandro Dalcin PetscFunctionReturn(0); 26702e74eeadSLisandro Dalcin } 26712e74eeadSLisandro Dalcin 2672f0ae7da4SStefano Zampini static PetscErrorCode MatZeroRows_IS(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,Vec x,Vec b) 2673b4319ba4SBarry Smith { 2674dfbe8321SBarry Smith PetscErrorCode ierr; 2675b4319ba4SBarry Smith 2676b4319ba4SBarry Smith PetscFunctionBegin; 2677f0ae7da4SStefano Zampini ierr = MatZeroRowsColumns_Private_IS(A,n,rows,diag,x,b,PETSC_FALSE);CHKERRQ(ierr); 2678f0ae7da4SStefano Zampini PetscFunctionReturn(0); 2679f0ae7da4SStefano Zampini } 26802205254eSKarl Rupp 2681f0ae7da4SStefano Zampini static PetscErrorCode MatZeroRowsColumns_IS(Mat A,PetscInt n,const PetscInt rows[],PetscScalar diag,Vec x,Vec b) 2682f0ae7da4SStefano Zampini { 2683f0ae7da4SStefano Zampini PetscErrorCode ierr; 2684f0ae7da4SStefano Zampini 2685f0ae7da4SStefano Zampini PetscFunctionBegin; 2686f0ae7da4SStefano Zampini ierr = MatZeroRowsColumns_Private_IS(A,n,rows,diag,x,b,PETSC_TRUE);CHKERRQ(ierr); 2687b4319ba4SBarry Smith PetscFunctionReturn(0); 2688b4319ba4SBarry Smith } 2689b4319ba4SBarry Smith 2690a8116848SStefano Zampini static PetscErrorCode MatAssemblyBegin_IS(Mat A,MatAssemblyType type) 2691b4319ba4SBarry Smith { 2692b4319ba4SBarry Smith Mat_IS *is = (Mat_IS*)A->data; 2693dfbe8321SBarry Smith PetscErrorCode ierr; 2694b4319ba4SBarry Smith 2695b4319ba4SBarry Smith PetscFunctionBegin; 2696b4319ba4SBarry Smith ierr = MatAssemblyBegin(is->A,type);CHKERRQ(ierr); 2697b4319ba4SBarry Smith PetscFunctionReturn(0); 2698b4319ba4SBarry Smith } 2699b4319ba4SBarry Smith 2700a8116848SStefano Zampini static PetscErrorCode MatAssemblyEnd_IS(Mat A,MatAssemblyType type) 2701b4319ba4SBarry Smith { 2702b4319ba4SBarry Smith Mat_IS *is = (Mat_IS*)A->data; 2703dfbe8321SBarry Smith PetscErrorCode ierr; 2704b4319ba4SBarry Smith 2705b4319ba4SBarry Smith PetscFunctionBegin; 2706b4319ba4SBarry Smith ierr = MatAssemblyEnd(is->A,type);CHKERRQ(ierr); 2707872cf891SStefano Zampini /* fix for local empty rows/cols */ 2708872cf891SStefano Zampini if (is->locempty && type == MAT_FINAL_ASSEMBLY) { 2709872cf891SStefano Zampini Mat newlA; 2710f03112d0SStefano Zampini ISLocalToGlobalMapping rl2g,cl2g; 2711f03112d0SStefano Zampini IS nzr,nzc; 2712f03112d0SStefano Zampini PetscInt nr,nc,nnzr,nnzc; 2713f03112d0SStefano Zampini PetscBool lnewl2g,newl2g; 2714872cf891SStefano Zampini 2715f03112d0SStefano Zampini ierr = MatGetSize(is->A,&nr,&nc);CHKERRQ(ierr); 2716f03112d0SStefano Zampini ierr = MatFindNonzeroRowsOrCols_Basic(is->A,PETSC_FALSE,PETSC_SMALL,&nzr);CHKERRQ(ierr); 2717f03112d0SStefano Zampini if (!nzr) { 2718f03112d0SStefano Zampini ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nr,0,1,&nzr);CHKERRQ(ierr); 2719872cf891SStefano Zampini } 2720f03112d0SStefano Zampini ierr = MatFindNonzeroRowsOrCols_Basic(is->A,PETSC_TRUE,PETSC_SMALL,&nzc);CHKERRQ(ierr); 2721f03112d0SStefano Zampini if (!nzc) { 2722f03112d0SStefano Zampini ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nc,0,1,&nzc);CHKERRQ(ierr); 2723872cf891SStefano Zampini } 2724f03112d0SStefano Zampini ierr = ISGetSize(nzr,&nnzr);CHKERRQ(ierr); 2725f03112d0SStefano Zampini ierr = ISGetSize(nzc,&nnzc);CHKERRQ(ierr); 2726f03112d0SStefano Zampini if (nnzr != nr || nnzc != nc) { 2727f03112d0SStefano Zampini ISLocalToGlobalMapping l2g; 2728f03112d0SStefano Zampini IS is1,is2; 2729f03112d0SStefano Zampini 2730f03112d0SStefano Zampini /* need new global l2g map */ 2731f03112d0SStefano Zampini lnewl2g = PETSC_TRUE; 273255b25c41SPierre Jolivet ierr = MPI_Allreduce(&lnewl2g,&newl2g,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 2733f03112d0SStefano Zampini 2734872cf891SStefano Zampini /* extract valid submatrix */ 2735f03112d0SStefano Zampini ierr = MatCreateSubMatrix(is->A,nzr,nzc,MAT_INITIAL_MATRIX,&newlA);CHKERRQ(ierr); 2736f03112d0SStefano Zampini 2737f03112d0SStefano Zampini /* attach local l2g maps for successive calls of MatSetValues on the local matrix */ 2738f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(nzr,&l2g);CHKERRQ(ierr); 2739f03112d0SStefano Zampini ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nr,0,1,&is1);CHKERRQ(ierr); 2740f03112d0SStefano Zampini ierr = ISGlobalToLocalMappingApplyIS(l2g,IS_GTOLM_MASK,is1,&is2);CHKERRQ(ierr); 2741872cf891SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&l2g);CHKERRQ(ierr); 274272ed36d8SStefano Zampini if (is->A->rmap->mapping) { /* the matrix has a local-to-local map already attached (probably DMDA generated) */ 274372ed36d8SStefano Zampini const PetscInt *idxs1,*idxs2; 274472ed36d8SStefano Zampini PetscInt j,i,nl,*tidxs; 274572ed36d8SStefano Zampini 274672ed36d8SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(is->A->rmap->mapping,&nl);CHKERRQ(ierr); 274772ed36d8SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(is->A->rmap->mapping,&idxs1);CHKERRQ(ierr); 274872ed36d8SStefano Zampini ierr = PetscMalloc1(nl,&tidxs);CHKERRQ(ierr); 274972ed36d8SStefano Zampini ierr = ISGetIndices(is2,&idxs2);CHKERRQ(ierr); 275072ed36d8SStefano Zampini for (i=0,j=0;i<nl;i++) tidxs[i] = idxs1[i] < 0 ? -1 : idxs2[j++]; 275172ed36d8SStefano Zampini if (j != nr) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected count %D != %D",j,nr); 275272ed36d8SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(is->A->rmap->mapping,&idxs1);CHKERRQ(ierr); 275372ed36d8SStefano Zampini ierr = ISRestoreIndices(is2,&idxs2);CHKERRQ(ierr); 275472ed36d8SStefano Zampini ierr = ISDestroy(&is2);CHKERRQ(ierr); 275572ed36d8SStefano Zampini ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is->A->rmap->mapping),nl,tidxs,PETSC_OWN_POINTER,&is2);CHKERRQ(ierr); 275672ed36d8SStefano Zampini } 2757f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is2,&rl2g);CHKERRQ(ierr); 2758f03112d0SStefano Zampini ierr = ISDestroy(&is1);CHKERRQ(ierr); 2759f03112d0SStefano Zampini ierr = ISDestroy(&is2);CHKERRQ(ierr); 276072ed36d8SStefano Zampini 2761f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(nzc,&l2g);CHKERRQ(ierr); 2762f03112d0SStefano Zampini ierr = ISCreateStride(PetscObjectComm((PetscObject)is->A),nc,0,1,&is1);CHKERRQ(ierr); 2763f03112d0SStefano Zampini ierr = ISGlobalToLocalMappingApplyIS(l2g,IS_GTOLM_MASK,is1,&is2);CHKERRQ(ierr); 2764f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&l2g);CHKERRQ(ierr); 276572ed36d8SStefano Zampini if (is->A->cmap->mapping) { /* the matrix has a local-to-local map already attached (probably DMDA generated) */ 276672ed36d8SStefano Zampini const PetscInt *idxs1,*idxs2; 276772ed36d8SStefano Zampini PetscInt j,i,nl,*tidxs; 276872ed36d8SStefano Zampini 276972ed36d8SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(is->A->cmap->mapping,&nl);CHKERRQ(ierr); 277072ed36d8SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(is->A->cmap->mapping,&idxs1);CHKERRQ(ierr); 277172ed36d8SStefano Zampini ierr = PetscMalloc1(nl,&tidxs);CHKERRQ(ierr); 277272ed36d8SStefano Zampini ierr = ISGetIndices(is2,&idxs2);CHKERRQ(ierr); 277372ed36d8SStefano Zampini for (i=0,j=0;i<nl;i++) tidxs[i] = idxs1[i] < 0 ? -1 : idxs2[j++]; 277472ed36d8SStefano Zampini if (j != nc) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected count %D != %D",j,nc); 277572ed36d8SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(is->A->cmap->mapping,&idxs1);CHKERRQ(ierr); 277672ed36d8SStefano Zampini ierr = ISRestoreIndices(is2,&idxs2);CHKERRQ(ierr); 277772ed36d8SStefano Zampini ierr = ISDestroy(&is2);CHKERRQ(ierr); 277872ed36d8SStefano Zampini ierr = ISCreateGeneral(PetscObjectComm((PetscObject)is->A->cmap->mapping),nl,tidxs,PETSC_OWN_POINTER,&is2);CHKERRQ(ierr); 277972ed36d8SStefano Zampini } 2780f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is2,&cl2g);CHKERRQ(ierr); 2781f03112d0SStefano Zampini ierr = ISDestroy(&is1);CHKERRQ(ierr); 2782f03112d0SStefano Zampini ierr = ISDestroy(&is2);CHKERRQ(ierr); 278372ed36d8SStefano Zampini 2784f03112d0SStefano Zampini ierr = MatSetLocalToGlobalMapping(newlA,rl2g,cl2g);CHKERRQ(ierr); 278572ed36d8SStefano Zampini 2786f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 2787f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 2788f03112d0SStefano Zampini } else { /* local matrix fully populated */ 2789f03112d0SStefano Zampini lnewl2g = PETSC_FALSE; 279055b25c41SPierre Jolivet ierr = MPI_Allreduce(&lnewl2g,&newl2g,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)A));CHKERRMPI(ierr); 2791f03112d0SStefano Zampini ierr = PetscObjectReference((PetscObject)is->A);CHKERRQ(ierr); 2792f03112d0SStefano Zampini newlA = is->A; 2793f03112d0SStefano Zampini } 2794f03112d0SStefano Zampini /* attach new global l2g map if needed */ 2795f03112d0SStefano Zampini if (newl2g) { 2796f03112d0SStefano Zampini IS gnzr,gnzc; 2797f03112d0SStefano Zampini const PetscInt *grid,*gcid; 2798f03112d0SStefano Zampini 2799f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingApplyIS(A->rmap->mapping,nzr,&gnzr);CHKERRQ(ierr); 2800f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingApplyIS(A->cmap->mapping,nzc,&gnzc);CHKERRQ(ierr); 2801f03112d0SStefano Zampini ierr = ISGetIndices(gnzr,&grid);CHKERRQ(ierr); 2802f03112d0SStefano Zampini ierr = ISGetIndices(gnzc,&gcid);CHKERRQ(ierr); 2803f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)A),1,nnzr,grid,PETSC_COPY_VALUES,&rl2g);CHKERRQ(ierr); 2804f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)A),1,nnzc,gcid,PETSC_COPY_VALUES,&cl2g);CHKERRQ(ierr); 2805f03112d0SStefano Zampini ierr = ISRestoreIndices(gnzr,&grid);CHKERRQ(ierr); 2806f03112d0SStefano Zampini ierr = ISRestoreIndices(gnzc,&gcid);CHKERRQ(ierr); 2807f03112d0SStefano Zampini ierr = ISDestroy(&gnzr);CHKERRQ(ierr); 2808f03112d0SStefano Zampini ierr = ISDestroy(&gnzc);CHKERRQ(ierr); 2809f03112d0SStefano Zampini ierr = MatSetLocalToGlobalMapping(A,rl2g,cl2g);CHKERRQ(ierr); 2810f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 2811f03112d0SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 2812f03112d0SStefano Zampini } 2813872cf891SStefano Zampini ierr = MatISSetLocalMat(A,newlA);CHKERRQ(ierr); 2814872cf891SStefano Zampini ierr = MatDestroy(&newlA);CHKERRQ(ierr); 2815f03112d0SStefano Zampini ierr = ISDestroy(&nzr);CHKERRQ(ierr); 2816f03112d0SStefano Zampini ierr = ISDestroy(&nzc);CHKERRQ(ierr); 2817872cf891SStefano Zampini is->locempty = PETSC_FALSE; 2818f03112d0SStefano Zampini } 2819b4319ba4SBarry Smith PetscFunctionReturn(0); 2820b4319ba4SBarry Smith } 2821b4319ba4SBarry Smith 2822a8116848SStefano Zampini static PetscErrorCode MatISGetLocalMat_IS(Mat mat,Mat *local) 2823b4319ba4SBarry Smith { 2824b4319ba4SBarry Smith Mat_IS *is = (Mat_IS*)mat->data; 2825b4319ba4SBarry Smith 2826b4319ba4SBarry Smith PetscFunctionBegin; 2827b4319ba4SBarry Smith *local = is->A; 2828b4319ba4SBarry Smith PetscFunctionReturn(0); 2829b4319ba4SBarry Smith } 2830b4319ba4SBarry Smith 28313b3b1effSJed Brown static PetscErrorCode MatISRestoreLocalMat_IS(Mat mat,Mat *local) 28323b3b1effSJed Brown { 28333b3b1effSJed Brown PetscFunctionBegin; 28343b3b1effSJed Brown *local = NULL; 28353b3b1effSJed Brown PetscFunctionReturn(0); 28363b3b1effSJed Brown } 28373b3b1effSJed Brown 2838b4319ba4SBarry Smith /*@ 2839b4319ba4SBarry Smith MatISGetLocalMat - Gets the local matrix stored inside a MATIS matrix. 2840b4319ba4SBarry Smith 2841b4319ba4SBarry Smith Input Parameter: 2842b4319ba4SBarry Smith . mat - the matrix 2843b4319ba4SBarry Smith 2844b4319ba4SBarry Smith Output Parameter: 2845eb82efa4SStefano Zampini . local - the local matrix 2846b4319ba4SBarry Smith 2847b4319ba4SBarry Smith Level: advanced 2848b4319ba4SBarry Smith 2849b4319ba4SBarry Smith Notes: 2850b4319ba4SBarry Smith This can be called if you have precomputed the nonzero structure of the 2851b4319ba4SBarry Smith matrix and want to provide it to the inner matrix object to improve the performance 2852b4319ba4SBarry Smith of the MatSetValues() operation. 2853b4319ba4SBarry Smith 28543b3b1effSJed Brown Call MatISRestoreLocalMat() when finished with the local matrix. 285596a6f129SJed Brown 2856b4319ba4SBarry Smith .seealso: MATIS 2857b4319ba4SBarry Smith @*/ 28587087cfbeSBarry Smith PetscErrorCode MatISGetLocalMat(Mat mat,Mat *local) 2859b4319ba4SBarry Smith { 28604ac538c5SBarry Smith PetscErrorCode ierr; 2861b4319ba4SBarry Smith 2862b4319ba4SBarry Smith PetscFunctionBegin; 28630700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 2864b4319ba4SBarry Smith PetscValidPointer(local,2); 28654ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatISGetLocalMat_C",(Mat,Mat*),(mat,local));CHKERRQ(ierr); 2866b4319ba4SBarry Smith PetscFunctionReturn(0); 2867b4319ba4SBarry Smith } 2868b4319ba4SBarry Smith 28693b3b1effSJed Brown /*@ 28703b3b1effSJed Brown MatISRestoreLocalMat - Restores the local matrix obtained with MatISGetLocalMat() 28713b3b1effSJed Brown 28723b3b1effSJed Brown Input Parameter: 28733b3b1effSJed Brown . mat - the matrix 28743b3b1effSJed Brown 28753b3b1effSJed Brown Output Parameter: 28763b3b1effSJed Brown . local - the local matrix 28773b3b1effSJed Brown 28783b3b1effSJed Brown Level: advanced 28793b3b1effSJed Brown 28803b3b1effSJed Brown .seealso: MATIS 28813b3b1effSJed Brown @*/ 28823b3b1effSJed Brown PetscErrorCode MatISRestoreLocalMat(Mat mat,Mat *local) 28833b3b1effSJed Brown { 28843b3b1effSJed Brown PetscErrorCode ierr; 28853b3b1effSJed Brown 28863b3b1effSJed Brown PetscFunctionBegin; 28873b3b1effSJed Brown PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 28883b3b1effSJed Brown PetscValidPointer(local,2); 28893b3b1effSJed Brown ierr = PetscUseMethod(mat,"MatISRestoreLocalMat_C",(Mat,Mat*),(mat,local));CHKERRQ(ierr); 28903b3b1effSJed Brown PetscFunctionReturn(0); 28913b3b1effSJed Brown } 28923b3b1effSJed Brown 28938546b261SStefano Zampini static PetscErrorCode MatISSetLocalMatType_IS(Mat mat,MatType mtype) 28948546b261SStefano Zampini { 28958546b261SStefano Zampini Mat_IS *is = (Mat_IS*)mat->data; 28968546b261SStefano Zampini PetscErrorCode ierr; 28978546b261SStefano Zampini 28988546b261SStefano Zampini PetscFunctionBegin; 28998546b261SStefano Zampini if (is->A) { 29008546b261SStefano Zampini ierr = MatSetType(is->A,mtype);CHKERRQ(ierr); 29018546b261SStefano Zampini } 29028546b261SStefano Zampini ierr = PetscFree(is->lmattype);CHKERRQ(ierr); 29038546b261SStefano Zampini ierr = PetscStrallocpy(mtype,&is->lmattype);CHKERRQ(ierr); 29048546b261SStefano Zampini PetscFunctionReturn(0); 29058546b261SStefano Zampini } 29068546b261SStefano Zampini 29078546b261SStefano Zampini /*@ 29088546b261SStefano Zampini MatISSetLocalMatType - Specifies the type of local matrix 29098546b261SStefano Zampini 2910*d8d19677SJose E. Roman Input Parameters: 2911a2b725a8SWilliam Gropp + mat - the matrix 2912a2b725a8SWilliam Gropp - mtype - the local matrix type 29138546b261SStefano Zampini 29148546b261SStefano Zampini Output Parameter: 29158546b261SStefano Zampini 29168546b261SStefano Zampini Level: advanced 29178546b261SStefano Zampini 29188546b261SStefano Zampini .seealso: MATIS, MatSetType(), MatType 29198546b261SStefano Zampini @*/ 29208546b261SStefano Zampini PetscErrorCode MatISSetLocalMatType(Mat mat,MatType mtype) 29218546b261SStefano Zampini { 29228546b261SStefano Zampini PetscErrorCode ierr; 29238546b261SStefano Zampini 29248546b261SStefano Zampini PetscFunctionBegin; 29258546b261SStefano Zampini PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 29268546b261SStefano Zampini ierr = PetscUseMethod(mat,"MatISSetLocalMatType_C",(Mat,MatType),(mat,mtype));CHKERRQ(ierr); 29278546b261SStefano Zampini PetscFunctionReturn(0); 29288546b261SStefano Zampini } 29298546b261SStefano Zampini 2930a8116848SStefano Zampini static PetscErrorCode MatISSetLocalMat_IS(Mat mat,Mat local) 29313b03a366Sstefano_zampini { 29323b03a366Sstefano_zampini Mat_IS *is = (Mat_IS*)mat->data; 29333b03a366Sstefano_zampini PetscInt nrows,ncols,orows,ocols; 29343b03a366Sstefano_zampini PetscErrorCode ierr; 29358546b261SStefano Zampini MatType mtype,otype; 29368546b261SStefano Zampini PetscBool sametype = PETSC_TRUE; 29373b03a366Sstefano_zampini 29383b03a366Sstefano_zampini PetscFunctionBegin; 29394e4c7dbeSStefano Zampini if (is->A) { 29403b03a366Sstefano_zampini ierr = MatGetSize(is->A,&orows,&ocols);CHKERRQ(ierr); 29413b03a366Sstefano_zampini ierr = MatGetSize(local,&nrows,&ncols);CHKERRQ(ierr); 2942f0ae7da4SStefano Zampini if (orows != nrows || ocols != ncols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Local MATIS matrix should be of size %Dx%D (you passed a %Dx%D matrix)",orows,ocols,nrows,ncols); 29438546b261SStefano Zampini ierr = MatGetType(local,&mtype);CHKERRQ(ierr); 29448546b261SStefano Zampini ierr = MatGetType(is->A,&otype);CHKERRQ(ierr); 29458546b261SStefano Zampini ierr = PetscStrcmp(mtype,otype,&sametype);CHKERRQ(ierr); 29464e4c7dbeSStefano Zampini } 29473b03a366Sstefano_zampini ierr = PetscObjectReference((PetscObject)local);CHKERRQ(ierr); 29483b03a366Sstefano_zampini ierr = MatDestroy(&is->A);CHKERRQ(ierr); 29493b03a366Sstefano_zampini is->A = local; 29508546b261SStefano Zampini ierr = MatGetType(is->A,&mtype);CHKERRQ(ierr); 29518546b261SStefano Zampini ierr = MatISSetLocalMatType(mat,mtype);CHKERRQ(ierr); 29528546b261SStefano Zampini if (!sametype && !is->islocalref) { 29538546b261SStefano Zampini ierr = MatISSetUpScatters_Private(mat);CHKERRQ(ierr); 29548546b261SStefano Zampini } 29553b03a366Sstefano_zampini PetscFunctionReturn(0); 29563b03a366Sstefano_zampini } 29573b03a366Sstefano_zampini 29583b03a366Sstefano_zampini /*@ 2959eb82efa4SStefano Zampini MatISSetLocalMat - Replace the local matrix stored inside a MATIS object. 29603b03a366Sstefano_zampini 29618546b261SStefano Zampini Collective on Mat 29628546b261SStefano Zampini 2963*d8d19677SJose E. Roman Input Parameters: 2964a2b725a8SWilliam Gropp + mat - the matrix 2965a2b725a8SWilliam Gropp - local - the local matrix 29663b03a366Sstefano_zampini 29673b03a366Sstefano_zampini Output Parameter: 29683b03a366Sstefano_zampini 29693b03a366Sstefano_zampini Level: advanced 29703b03a366Sstefano_zampini 29713b03a366Sstefano_zampini Notes: 29723b03a366Sstefano_zampini This can be called if you have precomputed the local matrix and 29733b03a366Sstefano_zampini want to provide it to the matrix object MATIS. 29743b03a366Sstefano_zampini 29753b03a366Sstefano_zampini .seealso: MATIS 29763b03a366Sstefano_zampini @*/ 29773b03a366Sstefano_zampini PetscErrorCode MatISSetLocalMat(Mat mat,Mat local) 29783b03a366Sstefano_zampini { 29793b03a366Sstefano_zampini PetscErrorCode ierr; 29803b03a366Sstefano_zampini 29813b03a366Sstefano_zampini PetscFunctionBegin; 29823b03a366Sstefano_zampini PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 2983b7ce53b6SStefano Zampini PetscValidHeaderSpecific(local,MAT_CLASSID,2); 29843b03a366Sstefano_zampini ierr = PetscUseMethod(mat,"MatISSetLocalMat_C",(Mat,Mat),(mat,local));CHKERRQ(ierr); 29853b03a366Sstefano_zampini PetscFunctionReturn(0); 29863b03a366Sstefano_zampini } 29873b03a366Sstefano_zampini 2988a8116848SStefano Zampini static PetscErrorCode MatZeroEntries_IS(Mat A) 29896726f965SBarry Smith { 29906726f965SBarry Smith Mat_IS *a = (Mat_IS*)A->data; 29916726f965SBarry Smith PetscErrorCode ierr; 29926726f965SBarry Smith 29936726f965SBarry Smith PetscFunctionBegin; 29946726f965SBarry Smith ierr = MatZeroEntries(a->A);CHKERRQ(ierr); 29956726f965SBarry Smith PetscFunctionReturn(0); 29966726f965SBarry Smith } 29976726f965SBarry Smith 2998a8116848SStefano Zampini static PetscErrorCode MatScale_IS(Mat A,PetscScalar a) 29992e74eeadSLisandro Dalcin { 30002e74eeadSLisandro Dalcin Mat_IS *is = (Mat_IS*)A->data; 30012e74eeadSLisandro Dalcin PetscErrorCode ierr; 30022e74eeadSLisandro Dalcin 30032e74eeadSLisandro Dalcin PetscFunctionBegin; 30042e74eeadSLisandro Dalcin ierr = MatScale(is->A,a);CHKERRQ(ierr); 30052e74eeadSLisandro Dalcin PetscFunctionReturn(0); 30062e74eeadSLisandro Dalcin } 30072e74eeadSLisandro Dalcin 3008a8116848SStefano Zampini static PetscErrorCode MatGetDiagonal_IS(Mat A, Vec v) 30092e74eeadSLisandro Dalcin { 30102e74eeadSLisandro Dalcin Mat_IS *is = (Mat_IS*)A->data; 30112e74eeadSLisandro Dalcin PetscErrorCode ierr; 30122e74eeadSLisandro Dalcin 30132e74eeadSLisandro Dalcin PetscFunctionBegin; 30142e74eeadSLisandro Dalcin /* get diagonal of the local matrix */ 3015e176bc59SStefano Zampini ierr = MatGetDiagonal(is->A,is->y);CHKERRQ(ierr); 30162e74eeadSLisandro Dalcin 30172e74eeadSLisandro Dalcin /* scatter diagonal back into global vector */ 30182e74eeadSLisandro Dalcin ierr = VecSet(v,0);CHKERRQ(ierr); 3019e176bc59SStefano Zampini ierr = VecScatterBegin(is->rctx,is->y,v,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 3020e176bc59SStefano Zampini ierr = VecScatterEnd(is->rctx,is->y,v,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 30212e74eeadSLisandro Dalcin PetscFunctionReturn(0); 30222e74eeadSLisandro Dalcin } 30232e74eeadSLisandro Dalcin 3024a8116848SStefano Zampini static PetscErrorCode MatSetOption_IS(Mat A,MatOption op,PetscBool flg) 30256726f965SBarry Smith { 30266726f965SBarry Smith Mat_IS *a = (Mat_IS*)A->data; 30276726f965SBarry Smith PetscErrorCode ierr; 30286726f965SBarry Smith 30296726f965SBarry Smith PetscFunctionBegin; 30304e0d8c25SBarry Smith ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 30316726f965SBarry Smith PetscFunctionReturn(0); 30326726f965SBarry Smith } 30336726f965SBarry Smith 3034f26d0771SStefano Zampini static PetscErrorCode MatAXPY_IS(Mat Y,PetscScalar a,Mat X,MatStructure str) 3035f26d0771SStefano Zampini { 3036f26d0771SStefano Zampini Mat_IS *y = (Mat_IS*)Y->data; 3037f26d0771SStefano Zampini Mat_IS *x; 3038f26d0771SStefano Zampini PetscErrorCode ierr; 3039f26d0771SStefano Zampini 3040f26d0771SStefano Zampini PetscFunctionBegin; 304176bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 304276bd3646SJed Brown PetscBool ismatis; 3043f26d0771SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)X,MATIS,&ismatis);CHKERRQ(ierr); 3044f26d0771SStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)Y),PETSC_ERR_SUP,"Cannot call MatAXPY(Y,a,X,str) with X not of type MATIS"); 304576bd3646SJed Brown } 3046f26d0771SStefano Zampini x = (Mat_IS*)X->data; 3047f26d0771SStefano Zampini ierr = MatAXPY(y->A,a,x->A,str);CHKERRQ(ierr); 3048f26d0771SStefano Zampini PetscFunctionReturn(0); 3049f26d0771SStefano Zampini } 3050f26d0771SStefano Zampini 3051f26d0771SStefano Zampini static PetscErrorCode MatGetLocalSubMatrix_IS(Mat A,IS row,IS col,Mat *submat) 3052f26d0771SStefano Zampini { 3053f26d0771SStefano Zampini Mat lA; 3054f26d0771SStefano Zampini Mat_IS *matis; 3055f26d0771SStefano Zampini ISLocalToGlobalMapping rl2g,cl2g; 3056f26d0771SStefano Zampini IS is; 3057f26d0771SStefano Zampini const PetscInt *rg,*rl; 3058f26d0771SStefano Zampini PetscInt nrg; 3059f26d0771SStefano Zampini PetscInt N,M,nrl,i,*idxs; 3060f26d0771SStefano Zampini PetscErrorCode ierr; 3061f26d0771SStefano Zampini 3062f26d0771SStefano Zampini PetscFunctionBegin; 3063f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(A->rmap->mapping,&rg);CHKERRQ(ierr); 3064f26d0771SStefano Zampini ierr = ISGetLocalSize(row,&nrl);CHKERRQ(ierr); 3065f26d0771SStefano Zampini ierr = ISGetIndices(row,&rl);CHKERRQ(ierr); 3066f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(A->rmap->mapping,&nrg);CHKERRQ(ierr); 306776bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 3068249c8a71SStefano Zampini for (i=0; i<nrl; i++) if (rl[i]>=nrg) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Local row index %D -> %D greater then maximum possible %D",i,rl[i],nrg); 306976bd3646SJed Brown } 3070f26d0771SStefano Zampini ierr = PetscMalloc1(nrg,&idxs);CHKERRQ(ierr); 3071f26d0771SStefano Zampini /* map from [0,nrl) to row */ 3072f26d0771SStefano Zampini for (i=0;i<nrl;i++) idxs[i] = rl[i]; 3073f26d0771SStefano Zampini for (i=nrl;i<nrg;i++) idxs[i] = -1; 3074f26d0771SStefano Zampini ierr = ISRestoreIndices(row,&rl);CHKERRQ(ierr); 3075f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(A->rmap->mapping,&rg);CHKERRQ(ierr); 3076f26d0771SStefano Zampini ierr = ISCreateGeneral(PetscObjectComm((PetscObject)A),nrg,idxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 3077f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr); 3078f26d0771SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 3079f26d0771SStefano Zampini /* compute new l2g map for columns */ 3080f26d0771SStefano Zampini if (col != row || A->rmap->mapping != A->cmap->mapping) { 3081f26d0771SStefano Zampini const PetscInt *cg,*cl; 3082f26d0771SStefano Zampini PetscInt ncg; 3083f26d0771SStefano Zampini PetscInt ncl; 3084f26d0771SStefano Zampini 3085f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(A->cmap->mapping,&cg);CHKERRQ(ierr); 3086f26d0771SStefano Zampini ierr = ISGetLocalSize(col,&ncl);CHKERRQ(ierr); 3087f26d0771SStefano Zampini ierr = ISGetIndices(col,&cl);CHKERRQ(ierr); 3088f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(A->cmap->mapping,&ncg);CHKERRQ(ierr); 308976bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 3090249c8a71SStefano Zampini for (i=0; i<ncl; i++) if (cl[i]>=ncg) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Local column index %D -> %D greater then maximum possible %D",i,cl[i],ncg); 309176bd3646SJed Brown } 3092f26d0771SStefano Zampini ierr = PetscMalloc1(ncg,&idxs);CHKERRQ(ierr); 3093f26d0771SStefano Zampini /* map from [0,ncl) to col */ 3094f26d0771SStefano Zampini for (i=0;i<ncl;i++) idxs[i] = cl[i]; 3095f26d0771SStefano Zampini for (i=ncl;i<ncg;i++) idxs[i] = -1; 3096f26d0771SStefano Zampini ierr = ISRestoreIndices(col,&cl);CHKERRQ(ierr); 3097f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingRestoreIndices(A->cmap->mapping,&cg);CHKERRQ(ierr); 3098f26d0771SStefano Zampini ierr = ISCreateGeneral(PetscObjectComm((PetscObject)A),ncg,idxs,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 3099f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr); 3100f26d0771SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 3101f26d0771SStefano Zampini } else { 3102f26d0771SStefano Zampini ierr = PetscObjectReference((PetscObject)rl2g);CHKERRQ(ierr); 3103f26d0771SStefano Zampini cl2g = rl2g; 3104f26d0771SStefano Zampini } 3105f26d0771SStefano Zampini /* create the MATIS submatrix */ 3106f26d0771SStefano Zampini ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr); 3107f26d0771SStefano Zampini ierr = MatCreate(PetscObjectComm((PetscObject)A),submat);CHKERRQ(ierr); 3108f26d0771SStefano Zampini ierr = MatSetSizes(*submat,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); 3109f26d0771SStefano Zampini ierr = MatSetType(*submat,MATIS);CHKERRQ(ierr); 3110b0aa3428SStefano Zampini matis = (Mat_IS*)((*submat)->data); 3111f26d0771SStefano Zampini matis->islocalref = PETSC_TRUE; 3112f26d0771SStefano Zampini ierr = MatSetLocalToGlobalMapping(*submat,rl2g,cl2g);CHKERRQ(ierr); 3113f26d0771SStefano Zampini ierr = MatISGetLocalMat(A,&lA);CHKERRQ(ierr); 3114f26d0771SStefano Zampini ierr = MatISSetLocalMat(*submat,lA);CHKERRQ(ierr); 3115f26d0771SStefano Zampini ierr = MatSetUp(*submat);CHKERRQ(ierr); 3116f26d0771SStefano Zampini ierr = MatAssemblyBegin(*submat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3117f26d0771SStefano Zampini ierr = MatAssemblyEnd(*submat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3118f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 3119f26d0771SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 3120f26d0771SStefano Zampini /* remove unsupported ops */ 3121f26d0771SStefano Zampini ierr = PetscMemzero((*submat)->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 3122f26d0771SStefano Zampini (*submat)->ops->destroy = MatDestroy_IS; 3123f26d0771SStefano Zampini (*submat)->ops->setvalueslocal = MatSetValuesLocal_SubMat_IS; 3124f26d0771SStefano Zampini (*submat)->ops->setvaluesblockedlocal = MatSetValuesBlockedLocal_SubMat_IS; 3125f26d0771SStefano Zampini (*submat)->ops->assemblybegin = MatAssemblyBegin_IS; 3126f26d0771SStefano Zampini (*submat)->ops->assemblyend = MatAssemblyEnd_IS; 3127f26d0771SStefano Zampini PetscFunctionReturn(0); 3128f26d0771SStefano Zampini } 3129f26d0771SStefano Zampini 3130872cf891SStefano Zampini static PetscErrorCode MatSetFromOptions_IS(PetscOptionItems *PetscOptionsObject, Mat A) 3131872cf891SStefano Zampini { 3132872cf891SStefano Zampini Mat_IS *a = (Mat_IS*)A->data; 31338546b261SStefano Zampini char type[256]; 31348546b261SStefano Zampini PetscBool flg; 3135872cf891SStefano Zampini PetscErrorCode ierr; 3136872cf891SStefano Zampini 3137872cf891SStefano Zampini PetscFunctionBegin; 3138872cf891SStefano Zampini ierr = PetscOptionsHead(PetscOptionsObject,"MATIS options");CHKERRQ(ierr); 3139f03112d0SStefano Zampini ierr = PetscOptionsBool("-matis_fixempty","Fix local matrices in case of empty local rows/columns","MatISFixLocalEmpty",a->locempty,&a->locempty,NULL);CHKERRQ(ierr); 314075d48cdbSStefano Zampini ierr = PetscOptionsBool("-matis_storel2l","Store local-to-local matrices generated from PtAP operations","MatISStoreL2L",a->storel2l,&a->storel2l,NULL);CHKERRQ(ierr); 31418546b261SStefano Zampini ierr = PetscOptionsFList("-matis_localmat_type","Matrix type","MatISSetLocalMatType",MatList,a->lmattype,type,256,&flg);CHKERRQ(ierr); 31428546b261SStefano Zampini if (flg) { 31438546b261SStefano Zampini ierr = MatISSetLocalMatType(A,type);CHKERRQ(ierr); 31448546b261SStefano Zampini } 31458546b261SStefano Zampini if (a->A) { 31468546b261SStefano Zampini ierr = MatSetFromOptions(a->A);CHKERRQ(ierr); 31478546b261SStefano Zampini } 31480af67c1bSStefano Zampini ierr = PetscOptionsTail();CHKERRQ(ierr); 3149872cf891SStefano Zampini PetscFunctionReturn(0); 3150872cf891SStefano Zampini } 3151872cf891SStefano Zampini 3152284134d9SBarry Smith /*@ 31533c212e90SHong Zhang MatCreateIS - Creates a "process" unassembled matrix, assembled on each 3154284134d9SBarry Smith process but not across processes. 3155284134d9SBarry Smith 3156284134d9SBarry Smith Input Parameters: 3157284134d9SBarry Smith + comm - MPI communicator that will share the matrix 3158e176bc59SStefano Zampini . bs - block size of the matrix 3159df3898eeSBarry Smith . m,n,M,N - local and/or global sizes of the left and right vector used in matrix vector products 3160e176bc59SStefano Zampini . rmap - local to global map for rows 3161e176bc59SStefano Zampini - cmap - local to global map for cols 3162284134d9SBarry Smith 3163284134d9SBarry Smith Output Parameter: 3164284134d9SBarry Smith . A - the resulting matrix 3165284134d9SBarry Smith 31668e6c10adSSatish Balay Level: advanced 31678e6c10adSSatish Balay 316895452b02SPatrick Sanan Notes: 316995452b02SPatrick Sanan See MATIS for more details. 31706fdf41d1SStefano Zampini m and n are NOT related to the size of the map; they represent the size of the local parts of the vectors 31716fdf41d1SStefano Zampini used in MatMult operations. The sizes of rmap and cmap define the size of the local matrices. 31723c212e90SHong Zhang If either rmap or cmap are NULL, then the matrix is assumed to be square. 3173284134d9SBarry Smith 3174284134d9SBarry Smith .seealso: MATIS, MatSetLocalToGlobalMapping() 3175284134d9SBarry Smith @*/ 3176e176bc59SStefano Zampini PetscErrorCode MatCreateIS(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt M,PetscInt N,ISLocalToGlobalMapping rmap,ISLocalToGlobalMapping cmap,Mat *A) 3177284134d9SBarry Smith { 3178284134d9SBarry Smith PetscErrorCode ierr; 3179284134d9SBarry Smith 3180284134d9SBarry Smith PetscFunctionBegin; 31816fdf41d1SStefano Zampini if (!rmap && !cmap) SETERRQ(comm,PETSC_ERR_USER,"You need to provide at least one of the mappings"); 3182284134d9SBarry Smith ierr = MatCreate(comm,A);CHKERRQ(ierr); 3183284134d9SBarry Smith ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr); 31846fdf41d1SStefano Zampini if (bs > 0) { 3185284134d9SBarry Smith ierr = MatSetBlockSize(*A,bs);CHKERRQ(ierr); 31866fdf41d1SStefano Zampini } 3187284134d9SBarry Smith ierr = MatSetType(*A,MATIS);CHKERRQ(ierr); 3188e176bc59SStefano Zampini if (rmap && cmap) { 3189e176bc59SStefano Zampini ierr = MatSetLocalToGlobalMapping(*A,rmap,cmap);CHKERRQ(ierr); 3190e176bc59SStefano Zampini } else if (!rmap) { 3191e176bc59SStefano Zampini ierr = MatSetLocalToGlobalMapping(*A,cmap,cmap);CHKERRQ(ierr); 3192e176bc59SStefano Zampini } else { 3193e176bc59SStefano Zampini ierr = MatSetLocalToGlobalMapping(*A,rmap,rmap);CHKERRQ(ierr); 3194e176bc59SStefano Zampini } 3195284134d9SBarry Smith PetscFunctionReturn(0); 3196284134d9SBarry Smith } 3197284134d9SBarry Smith 3198b4319ba4SBarry Smith /*MC 3199f26d0771SStefano Zampini MATIS - MATIS = "is" - A matrix type to be used for using the non-overlapping domain decomposition methods (e.g. PCBDDC or KSPFETIDP). 3200b89f26deSStefano Zampini This stores the matrices in globally unassembled form. Each processor assembles only its local Neumann problem and the parallel matrix vector 3201b4319ba4SBarry Smith product is handled "implicitly". 3202b4319ba4SBarry Smith 3203b4319ba4SBarry Smith Options Database Keys: 320475d48cdbSStefano Zampini + -mat_type is - sets the matrix type to "is" during a call to MatSetFromOptions() 320575d48cdbSStefano Zampini . -matis_fixempty - Fixes local matrices in case of empty local rows/columns. 320675d48cdbSStefano Zampini - -matis_storel2l - stores the local-to-local operators generated by the Galerkin process of MatPtAP(). 3207b4319ba4SBarry Smith 320895452b02SPatrick Sanan Notes: 320995452b02SPatrick Sanan Options prefix for the inner matrix are given by -is_mat_xxx 3210b4319ba4SBarry Smith 3211b4319ba4SBarry Smith You must call MatSetLocalToGlobalMapping() before using this matrix type. 3212b4319ba4SBarry Smith 3213b4319ba4SBarry Smith You can do matrix preallocation on the local matrix after you obtain it with 3214eb82efa4SStefano Zampini MatISGetLocalMat(); otherwise, you could use MatISSetPreallocation() 3215b4319ba4SBarry Smith 3216b4319ba4SBarry Smith Level: advanced 3217b4319ba4SBarry Smith 3218f26d0771SStefano Zampini .seealso: Mat, MatISGetLocalMat(), MatSetLocalToGlobalMapping(), MatISSetPreallocation(), MatCreateIS(), PCBDDC, KSPFETIDP 3219b4319ba4SBarry Smith 3220b4319ba4SBarry Smith M*/ 3221b4319ba4SBarry Smith 32228cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_IS(Mat A) 3223b4319ba4SBarry Smith { 3224dfbe8321SBarry Smith PetscErrorCode ierr; 3225b4319ba4SBarry Smith Mat_IS *b; 3226b4319ba4SBarry Smith 3227b4319ba4SBarry Smith PetscFunctionBegin; 3228b00a9115SJed Brown ierr = PetscNewLog(A,&b);CHKERRQ(ierr); 32298546b261SStefano Zampini ierr = PetscStrallocpy(MATAIJ,&b->lmattype);CHKERRQ(ierr); 3230b4319ba4SBarry Smith A->data = (void*)b; 3231b4319ba4SBarry Smith 3232e176bc59SStefano Zampini /* matrix ops */ 3233e176bc59SStefano Zampini ierr = PetscMemzero(A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 3234b4319ba4SBarry Smith A->ops->mult = MatMult_IS; 32352e74eeadSLisandro Dalcin A->ops->multadd = MatMultAdd_IS; 32362e74eeadSLisandro Dalcin A->ops->multtranspose = MatMultTranspose_IS; 32372e74eeadSLisandro Dalcin A->ops->multtransposeadd = MatMultTransposeAdd_IS; 3238b4319ba4SBarry Smith A->ops->destroy = MatDestroy_IS; 3239b4319ba4SBarry Smith A->ops->setlocaltoglobalmapping = MatSetLocalToGlobalMapping_IS; 32402e74eeadSLisandro Dalcin A->ops->setvalues = MatSetValues_IS; 324198921651SStefano Zampini A->ops->setvaluesblocked = MatSetValuesBlocked_IS; 3242b4319ba4SBarry Smith A->ops->setvalueslocal = MatSetValuesLocal_IS; 3243f0006bf2SLisandro Dalcin A->ops->setvaluesblockedlocal = MatSetValuesBlockedLocal_IS; 32442e74eeadSLisandro Dalcin A->ops->zerorows = MatZeroRows_IS; 3245f0ae7da4SStefano Zampini A->ops->zerorowscolumns = MatZeroRowsColumns_IS; 3246b4319ba4SBarry Smith A->ops->assemblybegin = MatAssemblyBegin_IS; 3247b4319ba4SBarry Smith A->ops->assemblyend = MatAssemblyEnd_IS; 3248b4319ba4SBarry Smith A->ops->view = MatView_IS; 32496726f965SBarry Smith A->ops->zeroentries = MatZeroEntries_IS; 32502e74eeadSLisandro Dalcin A->ops->scale = MatScale_IS; 32512e74eeadSLisandro Dalcin A->ops->getdiagonal = MatGetDiagonal_IS; 32526726f965SBarry Smith A->ops->setoption = MatSetOption_IS; 325369796d55SStefano Zampini A->ops->ishermitian = MatIsHermitian_IS; 325469796d55SStefano Zampini A->ops->issymmetric = MatIsSymmetric_IS; 325545471136SStefano Zampini A->ops->isstructurallysymmetric = MatIsStructurallySymmetric_IS; 3256ad6194a2SStefano Zampini A->ops->duplicate = MatDuplicate_IS; 32576bd84002SStefano Zampini A->ops->missingdiagonal = MatMissingDiagonal_IS; 32582b404112SStefano Zampini A->ops->copy = MatCopy_IS; 3259659959c5SStefano Zampini A->ops->getlocalsubmatrix = MatGetLocalSubMatrix_IS; 32607dae84e0SHong Zhang A->ops->createsubmatrix = MatCreateSubMatrix_IS; 3261f26d0771SStefano Zampini A->ops->axpy = MatAXPY_IS; 32623fd1c9e7SStefano Zampini A->ops->diagonalset = MatDiagonalSet_IS; 32633fd1c9e7SStefano Zampini A->ops->shift = MatShift_IS; 3264d7f69cd0SStefano Zampini A->ops->transpose = MatTranspose_IS; 32657fa8f2d3SStefano Zampini A->ops->getinfo = MatGetInfo_IS; 3266ad219c80Sstefano_zampini A->ops->diagonalscale = MatDiagonalScale_IS; 3267872cf891SStefano Zampini A->ops->setfromoptions = MatSetFromOptions_IS; 3268b4319ba4SBarry Smith 3269b7ce53b6SStefano Zampini /* special MATIS functions */ 32708546b261SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMatType_C",MatISSetLocalMatType_IS);CHKERRQ(ierr); 3271bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetLocalMat_C",MatISGetLocalMat_IS);CHKERRQ(ierr); 32723b3b1effSJed Brown ierr = PetscObjectComposeFunction((PetscObject)A,"MatISRestoreLocalMat_C",MatISRestoreLocalMat_IS);CHKERRQ(ierr); 3273bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetLocalMat_C",MatISSetLocalMat_IS);CHKERRQ(ierr); 3274487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISGetMPIXAIJ_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 32752e1947a5SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISSetPreallocation_C",MatISSetPreallocation_IS);CHKERRQ(ierr); 327675d48cdbSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISStoreL2L_C",MatISStoreL2L_IS);CHKERRQ(ierr); 3277f03112d0SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatISFixLocalEmpty_C",MatISFixLocalEmpty_IS);CHKERRQ(ierr); 3278487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpiaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 3279487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpibaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 3280487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_mpisbaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 3281487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 3282487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqbaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 3283487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_seqsbaij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 3284487b449aSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_is_aij_C",MatConvert_IS_XAIJ);CHKERRQ(ierr); 328517667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)A,MATIS);CHKERRQ(ierr); 3286b4319ba4SBarry Smith PetscFunctionReturn(0); 3287b4319ba4SBarry Smith } 3288