182412ba7SHong Zhang 242c57489SHong Zhang /* 342c57489SHong Zhang Defines projective product routines where A is a MPIAIJ matrix 442c57489SHong Zhang C = P^T * A * P 542c57489SHong Zhang */ 642c57489SHong Zhang 7c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h> /*I "petscmat.h" I*/ 8c6db04a5SJed Brown #include <../src/mat/utils/freespace.h> 9c6db04a5SJed Brown #include <../src/mat/impls/aij/mpi/mpiaij.h> 10c6db04a5SJed Brown #include <petscbt.h> 1142c57489SHong Zhang 12fe615159SHong Zhang /* 13fe615159SHong Zhang #define DEBUG_MATPTAP 14fe615159SHong Zhang */ 15fe615159SHong Zhang 1609573ac7SBarry Smith extern PetscErrorCode MatDestroy_MPIAIJ(Mat); 17cc6cb787SHong Zhang #undef __FUNCT__ 18f8487c73SHong Zhang #define __FUNCT__ "MatDestroy_MPIAIJ_PtAP" 19f8487c73SHong Zhang PetscErrorCode MatDestroy_MPIAIJ_PtAP(Mat A) 20cc6cb787SHong Zhang { 21cc6cb787SHong Zhang PetscErrorCode ierr; 22f8487c73SHong Zhang Mat_MPIAIJ *a=(Mat_MPIAIJ*)A->data; 23f8487c73SHong Zhang Mat_PtAPMPI *ptap=a->ptap; 24cc6cb787SHong Zhang 25cc6cb787SHong Zhang PetscFunctionBegin; 26f8487c73SHong Zhang if (ptap){ 27*b7f45c76SHong Zhang ierr = PetscFree2(ptap->startsj_s,ptap->startsj_r);CHKERRQ(ierr); 28f8487c73SHong Zhang ierr = PetscFree(ptap->bufa);CHKERRQ(ierr); 29a1a4e84aSHong Zhang ierr = MatDestroy(&ptap->P_loc);CHKERRQ(ierr); 30a1a4e84aSHong Zhang ierr = MatDestroy(&ptap->P_oth);CHKERRQ(ierr); 31c5af039cSHong Zhang ierr = MatDestroy(&ptap->A_loc);CHKERRQ(ierr); /* used by MatTransposeMatMult() */ 32c5af039cSHong Zhang if (ptap->api){ierr = PetscFree(ptap->api);CHKERRQ(ierr);} 33c5af039cSHong Zhang if (ptap->apj){ierr = PetscFree(ptap->apj);CHKERRQ(ierr);} 34f8487c73SHong Zhang } 35f8487c73SHong Zhang if (ptap->merge) { 36f8487c73SHong Zhang Mat_Merge_SeqsToMPI *merge=ptap->merge; 37cc6cb787SHong Zhang ierr = PetscFree(merge->id_r);CHKERRQ(ierr); 38cc6cb787SHong Zhang ierr = PetscFree(merge->len_s);CHKERRQ(ierr); 39cc6cb787SHong Zhang ierr = PetscFree(merge->len_r);CHKERRQ(ierr); 40cc6cb787SHong Zhang ierr = PetscFree(merge->bi);CHKERRQ(ierr); 41cc6cb787SHong Zhang ierr = PetscFree(merge->bj);CHKERRQ(ierr); 42c05d87d6SBarry Smith ierr = PetscFree(merge->buf_ri[0]);CHKERRQ(ierr); 43cc6cb787SHong Zhang ierr = PetscFree(merge->buf_ri);CHKERRQ(ierr); 44c05d87d6SBarry Smith ierr = PetscFree(merge->buf_rj[0]);CHKERRQ(ierr); 45cc6cb787SHong Zhang ierr = PetscFree(merge->buf_rj);CHKERRQ(ierr); 4605b42c5fSBarry Smith ierr = PetscFree(merge->coi);CHKERRQ(ierr); 4705b42c5fSBarry Smith ierr = PetscFree(merge->coj);CHKERRQ(ierr); 4805b42c5fSBarry Smith ierr = PetscFree(merge->owners_co);CHKERRQ(ierr); 496bf464f9SBarry Smith ierr = PetscLayoutDestroy(&merge->rowmap);CHKERRQ(ierr); 50dce485f0SBarry Smith ierr = merge->destroy(A);CHKERRQ(ierr); 51f8487c73SHong Zhang ierr = PetscFree(ptap->merge);CHKERRQ(ierr); 52bf0cc555SLisandro Dalcin } 53f8487c73SHong Zhang ierr = PetscFree(ptap);CHKERRQ(ierr); 54cc6cb787SHong Zhang PetscFunctionReturn(0); 55cc6cb787SHong Zhang } 56cc6cb787SHong Zhang 57cc6cb787SHong Zhang #undef __FUNCT__ 58cc6cb787SHong Zhang #define __FUNCT__ "MatDuplicate_MPIAIJ_MatPtAP" 59f0c0a3a6SBarry Smith PetscErrorCode MatDuplicate_MPIAIJ_MatPtAP(Mat A, MatDuplicateOption op, Mat *M) 60f0c0a3a6SBarry Smith { 61cc6cb787SHong Zhang PetscErrorCode ierr; 62cc6cb787SHong Zhang Mat_Merge_SeqsToMPI *merge; 63776b82aeSLisandro Dalcin PetscContainer container; 64cc6cb787SHong Zhang 65cc6cb787SHong Zhang PetscFunctionBegin; 66cc6cb787SHong Zhang ierr = PetscObjectQuery((PetscObject)A,"MatMergeSeqsToMPI",(PetscObject *)&container);CHKERRQ(ierr); 67cc6cb787SHong Zhang if (container) { 68776b82aeSLisandro Dalcin ierr = PetscContainerGetPointer(container,(void **)&merge);CHKERRQ(ierr); 69e7e72b3dSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Container does not exit"); 70dce485f0SBarry Smith ierr = (*merge->duplicate)(A,op,M);CHKERRQ(ierr); 71dce485f0SBarry Smith (*M)->ops->destroy = merge->destroy; /* =MatDestroy_MPIAIJ, *M doesn't duplicate A's container! */ 72dce485f0SBarry Smith (*M)->ops->duplicate = merge->duplicate; /* =MatDuplicate_ MPIAIJ */ 73cc6cb787SHong Zhang PetscFunctionReturn(0); 74cc6cb787SHong Zhang } 75cc6cb787SHong Zhang 7642c57489SHong Zhang #undef __FUNCT__ 777a7894deSKris Buschelman #define __FUNCT__ "MatPtAPSymbolic_MPIAIJ" 787a7894deSKris Buschelman PetscErrorCode MatPtAPSymbolic_MPIAIJ(Mat A,Mat P,PetscReal fill,Mat *C) 7942c57489SHong Zhang { 8042c57489SHong Zhang PetscErrorCode ierr; 8142c57489SHong Zhang 8242c57489SHong Zhang PetscFunctionBegin; 8365e19b50SBarry Smith if (!P->ops->ptapsymbolic_mpiaij) SETERRQ2(((PetscObject)A)->comm,PETSC_ERR_SUP,"Not implemented for A=%s and P=%s",((PetscObject)A)->type_name,((PetscObject)P)->type_name); 847a7894deSKris Buschelman ierr = (*P->ops->ptapsymbolic_mpiaij)(A,P,fill,C);CHKERRQ(ierr); 857a7894deSKris Buschelman PetscFunctionReturn(0); 867a7894deSKris Buschelman } 877a7894deSKris Buschelman 887a7894deSKris Buschelman #undef __FUNCT__ 897a7894deSKris Buschelman #define __FUNCT__ "MatPtAPNumeric_MPIAIJ" 907a7894deSKris Buschelman PetscErrorCode MatPtAPNumeric_MPIAIJ(Mat A,Mat P,Mat C) 917a7894deSKris Buschelman { 927a7894deSKris Buschelman PetscErrorCode ierr; 937a7894deSKris Buschelman 947a7894deSKris Buschelman PetscFunctionBegin; 9565e19b50SBarry Smith if (!P->ops->ptapnumeric_mpiaij) SETERRQ2(((PetscObject)A)->comm,PETSC_ERR_SUP,"Not implemented for A=%s and P=%s",((PetscObject)A)->type_name,((PetscObject)P)->type_name); 967a7894deSKris Buschelman ierr = (*P->ops->ptapnumeric_mpiaij)(A,P,C);CHKERRQ(ierr); 9742c57489SHong Zhang PetscFunctionReturn(0); 9842c57489SHong Zhang } 9942c57489SHong Zhang 10042c57489SHong Zhang #undef __FUNCT__ 10142c57489SHong Zhang #define __FUNCT__ "MatPtAPSymbolic_MPIAIJ_MPIAIJ" 10242c57489SHong Zhang PetscErrorCode MatPtAPSymbolic_MPIAIJ_MPIAIJ(Mat A,Mat P,PetscReal fill,Mat *C) 10342c57489SHong Zhang { 10442c57489SHong Zhang PetscErrorCode ierr; 10577584fe6SHong Zhang Mat Cmpi; 106f8487c73SHong Zhang Mat_PtAPMPI *ptap; 107a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 108f8487c73SHong Zhang Mat_MPIAIJ *a=(Mat_MPIAIJ*)A->data,*p=(Mat_MPIAIJ*)P->data,*c; 10942c57489SHong Zhang Mat_SeqAIJ *ad=(Mat_SeqAIJ*)(a->A)->data,*ao=(Mat_SeqAIJ*)(a->B)->data; 11042c57489SHong Zhang Mat_SeqAIJ *p_loc,*p_oth; 111ded4f561SHong Zhang PetscInt *pi_loc,*pj_loc,*pi_oth,*pj_oth,*pdti,*pdtj,*poti,*potj,*ptJ; 11277584fe6SHong Zhang PetscInt *adi=ad->i,*aj,*aoi=ao->i,nnz; 11313f74950SBarry Smith PetscInt nlnk,*lnk,*owners_co,*coi,*coj,i,k,pnz,row; 114d16ca5f0SHong Zhang PetscInt am=A->rmap->n,pN=P->cmap->N,pm=P->rmap->n,pn=P->cmap->n; 11542c57489SHong Zhang PetscBT lnkbt; 1167adad957SLisandro Dalcin MPI_Comm comm=((PetscObject)A)->comm; 117529bc5dcSHong Zhang PetscMPIInt size,rank,tagi,tagj,*len_si,*len_s,*len_ri; 11842c57489SHong Zhang PetscInt **buf_rj,**buf_ri,**buf_ri_k; 11942c57489SHong Zhang PetscInt len,proc,*dnz,*onz,*owners; 120ded4f561SHong Zhang PetscInt nzi,*bi,*bj; 12142c57489SHong Zhang PetscInt nrows,*buf_s,*buf_si,*buf_si_i,**nextrow,**nextci; 122ded4f561SHong Zhang MPI_Request *swaits,*rwaits; 123ded4f561SHong Zhang MPI_Status *sstatus,rstatus; 12442c57489SHong Zhang Mat_Merge_SeqsToMPI *merge; 12577584fe6SHong Zhang PetscInt *api,*apj,*Jptr,apnz,*prmap=p->garray,pon,nspacedouble=0,j,ap_rmax=0; 126d16ca5f0SHong Zhang PetscReal afill=1.0,afill_tmp; 12708cb4532SHong Zhang PetscInt rstart = P->cmap->rstart,rmax; 12808cb4532SHong Zhang PetscScalar *vals; 12942c57489SHong Zhang 13042c57489SHong Zhang PetscFunctionBegin; 131c5af039cSHong Zhang /* check if matrix local sizes are compatible */ 132c5af039cSHong Zhang if (A->rmap->rstart != P->rmap->rstart || A->rmap->rend != P->rmap->rend){ 133c5af039cSHong Zhang SETERRQ4(comm,PETSC_ERR_ARG_SIZ,"Matrix local dimensions are incompatible, Arow (%D, %D) != Prow (%D,%D)",A->rmap->rstart,A->rmap->rend,P->rmap->rstart,P->rmap->rend); 134c5af039cSHong Zhang } 135c5af039cSHong Zhang if (A->cmap->rstart != P->rmap->rstart || A->cmap->rend != P->rmap->rend){ 136c5af039cSHong Zhang SETERRQ4(comm,PETSC_ERR_ARG_SIZ,"Matrix local dimensions are incompatible, Acol (%D, %D) != Prow (%D,%D)",A->cmap->rstart,A->cmap->rend,P->rmap->rstart,P->rmap->rend); 137c5af039cSHong Zhang } 138c5af039cSHong Zhang 13983445d95SHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 14083445d95SHong Zhang ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 14183445d95SHong Zhang 142f8487c73SHong Zhang /* create struct Mat_PtAPMPI and attached it to C later */ 143f8487c73SHong Zhang ierr = PetscNew(Mat_PtAPMPI,&ptap);CHKERRQ(ierr); 144f8487c73SHong Zhang ptap->reuse = MAT_INITIAL_MATRIX; 1456c6e00e0SHong Zhang 14677584fe6SHong Zhang 1476c6e00e0SHong Zhang /* get P_oth by taking rows of P (= non-zero cols of local A) from other processors */ 148*b7f45c76SHong Zhang ierr = MatGetBrowsOfAoCols_MPIAIJ(A,P,MAT_INITIAL_MATRIX,&ptap->startsj_s,&ptap->startsj_r,&ptap->bufa,&ptap->P_oth);CHKERRQ(ierr); 149fe615159SHong Zhang #if defined(DEBUG_MATPTAP) 150fe615159SHong Zhang if (!rank) ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] P_oth is done...\n",rank); 151fe615159SHong Zhang #endif 152fe615159SHong Zhang 1536c6e00e0SHong Zhang /* get P_loc by taking all local rows of P */ 154a1a4e84aSHong Zhang ierr = MatMPIAIJGetLocalMat(P,MAT_INITIAL_MATRIX,&ptap->P_loc);CHKERRQ(ierr); 155fe615159SHong Zhang #if defined(DEBUG_MATPTAP) 156fe615159SHong Zhang ierr = MPI_Barrier(comm);CHKERRQ(ierr); 157fe615159SHong Zhang if (!rank) ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] P_loc is done...\n",rank); 158fe615159SHong Zhang #endif 1596c6e00e0SHong Zhang 160a1a4e84aSHong Zhang p_loc = (Mat_SeqAIJ*)(ptap->P_loc)->data; 161a1a4e84aSHong Zhang p_oth = (Mat_SeqAIJ*)(ptap->P_oth)->data; 1626c6e00e0SHong Zhang pi_loc = p_loc->i; pj_loc = p_loc->j; 1636c6e00e0SHong Zhang pi_oth = p_oth->i; pj_oth = p_oth->j; 16442c57489SHong Zhang 165fd0ff01cSHong Zhang /* first, compute symbolic AP = A_loc*P = A_diag*P_loc + A_off*P_oth */ 166fd0ff01cSHong Zhang /*-------------------------------------------------------------------*/ 16777584fe6SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&api);CHKERRQ(ierr); 16882412ba7SHong Zhang api[0] = 0; 16983445d95SHong Zhang 17083445d95SHong Zhang /* create and initialize a linked list */ 17183445d95SHong Zhang nlnk = pN+1; 172fe615159SHong Zhang #if defined(DEBUG_MATPTAP) 173fe615159SHong Zhang ierr = MPI_Barrier(comm);CHKERRQ(ierr); 174fe615159SHong Zhang if (!rank) ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] call PetscLLCreate(), pN %D, P_rmax %D %D; Annz %D\n",rank,pN,p_loc->rmax,p_oth->rmax,adi[am]+aoi[am]); 175fe615159SHong Zhang #endif 17683445d95SHong Zhang ierr = PetscLLCreate(pN,pN,nlnk,lnk,lnkbt);CHKERRQ(ierr); 177fe615159SHong Zhang #if defined(DEBUG_MATPTAP) 178fe615159SHong Zhang ierr = MPI_Barrier(comm);CHKERRQ(ierr); 179fe615159SHong Zhang if (!rank) ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] call PetscLLCreate() is done\n",rank); 180fe615159SHong Zhang #endif 18183445d95SHong Zhang 182d16ca5f0SHong Zhang /* Initial FreeSpace size is fill*(nnz(A) + nnz(P)) */ 183d16ca5f0SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(adi[am]+aoi[am]+pi_loc[pm])),&free_space);CHKERRQ(ierr); 184f4a743e1SHong Zhang current_space = free_space; 185f4a743e1SHong Zhang 186f4a743e1SHong Zhang for (i=0; i<am; i++) { 18782412ba7SHong Zhang apnz = 0; 188f4a743e1SHong Zhang /* diagonal portion of A */ 189ded4f561SHong Zhang nzi = adi[i+1] - adi[i]; 19077584fe6SHong Zhang aj = ad->j + adi[i]; 191ded4f561SHong Zhang for (j=0; j<nzi; j++){ 19277584fe6SHong Zhang row = aj[j]; 19382412ba7SHong Zhang pnz = pi_loc[row+1] - pi_loc[row]; 194ded4f561SHong Zhang Jptr = pj_loc + pi_loc[row]; 19583445d95SHong Zhang /* add non-zero cols of P into the sorted linked list lnk */ 196dadf0e6bSHong Zhang ierr = PetscLLAddSorted(pnz,Jptr,pN,nlnk,lnk,lnkbt);CHKERRQ(ierr); 19782412ba7SHong Zhang apnz += nlnk; 198f4a743e1SHong Zhang } 199f4a743e1SHong Zhang /* off-diagonal portion of A */ 200ded4f561SHong Zhang nzi = aoi[i+1] - aoi[i]; 20177584fe6SHong Zhang aj = ao->j + aoi[i]; 202ded4f561SHong Zhang for (j=0; j<nzi; j++){ 20377584fe6SHong Zhang row = aj[j]; 20482412ba7SHong Zhang pnz = pi_oth[row+1] - pi_oth[row]; 205ded4f561SHong Zhang Jptr = pj_oth + pi_oth[row]; 206dadf0e6bSHong Zhang ierr = PetscLLAddSorted(pnz,Jptr,pN,nlnk,lnk,lnkbt);CHKERRQ(ierr); 20782412ba7SHong Zhang apnz += nlnk; 208f4a743e1SHong Zhang } 209f4a743e1SHong Zhang 21082412ba7SHong Zhang api[i+1] = api[i] + apnz; 21177584fe6SHong Zhang if (ap_rmax < apnz) ap_rmax = apnz; 212f4a743e1SHong Zhang 21383445d95SHong Zhang /* if free space is not available, double the total space in the list */ 21482412ba7SHong Zhang if (current_space->local_remaining<apnz) { 2152ba1acd5SHong Zhang ierr = PetscFreeSpaceGet(apnz+current_space->total_array_size,¤t_space);CHKERRQ(ierr); 216f2b054eeSHong Zhang nspacedouble++; 217f4a743e1SHong Zhang } 218f4a743e1SHong Zhang 219f4a743e1SHong Zhang /* Copy data into free space, then initialize lnk */ 22082412ba7SHong Zhang ierr = PetscLLClean(pN,pN,apnz,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 22182412ba7SHong Zhang current_space->array += apnz; 22282412ba7SHong Zhang current_space->local_used += apnz; 22382412ba7SHong Zhang current_space->local_remaining -= apnz; 224f4a743e1SHong Zhang } 22582412ba7SHong Zhang /* Allocate space for apj, initialize apj, and */ 226f4a743e1SHong Zhang /* destroy list of free space and other temporary array(s) */ 22777584fe6SHong Zhang ierr = PetscMalloc((api[am]+1)*sizeof(PetscInt),&apj);CHKERRQ(ierr); 22877584fe6SHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,apj);CHKERRQ(ierr); 229d16ca5f0SHong Zhang afill_tmp = (PetscReal)api[am]/(adi[am]+aoi[am]+pi_loc[pm]); 230d16ca5f0SHong Zhang if (afill_tmp > afill) afill = afill_tmp; 231f4a743e1SHong Zhang 232ded4f561SHong Zhang /* determine symbolic Co=(p->B)^T*AP - send to others */ 233ded4f561SHong Zhang /*----------------------------------------------------*/ 234de0260b3SHong Zhang ierr = MatGetSymbolicTranspose_SeqAIJ(p->B,&poti,&potj);CHKERRQ(ierr); 23542c57489SHong Zhang 236ded4f561SHong Zhang /* then, compute symbolic Co = (p->B)^T*AP */ 237d0f46423SBarry Smith pon = (p->B)->cmap->n; /* total num of rows to be sent to other processors 23883445d95SHong Zhang >= (num of nonzero rows of C_seq) - pn */ 239de0260b3SHong Zhang ierr = PetscMalloc((pon+1)*sizeof(PetscInt),&coi);CHKERRQ(ierr); 240de0260b3SHong Zhang coi[0] = 0; 24142c57489SHong Zhang 242d16ca5f0SHong Zhang /* set initial free space to be fill*(nnz(p->B) + nnz(AP)) */ 243d16ca5f0SHong Zhang nnz = fill*(poti[pon] + api[am]); 244a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,&free_space); 24542c57489SHong Zhang current_space = free_space; 24642c57489SHong Zhang 247de0260b3SHong Zhang for (i=0; i<pon; i++) { 248ded4f561SHong Zhang nnz = 0; 24982412ba7SHong Zhang pnz = poti[i+1] - poti[i]; 25077584fe6SHong Zhang ptJ = potj + poti[i]; 25177584fe6SHong Zhang for (j=0; j<pnz; j++){ 25277584fe6SHong Zhang row = ptJ[j]; /* row of AP == col of Pot */ 25382412ba7SHong Zhang apnz = api[row+1] - api[row]; 254ded4f561SHong Zhang Jptr = apj + api[row]; 25583445d95SHong Zhang /* add non-zero cols of AP into the sorted linked list lnk */ 256dadf0e6bSHong Zhang ierr = PetscLLAddSorted(apnz,Jptr,pN,nlnk,lnk,lnkbt);CHKERRQ(ierr); 257ded4f561SHong Zhang nnz += nlnk; 25842c57489SHong Zhang } 25942c57489SHong Zhang 26083445d95SHong Zhang /* If free space is not available, double the total space in the list */ 261ded4f561SHong Zhang if (current_space->local_remaining<nnz) { 2624238b7adSHong Zhang ierr = PetscFreeSpaceGet(nnz+current_space->total_array_size,¤t_space);CHKERRQ(ierr); 263d16ca5f0SHong Zhang nspacedouble++; 26442c57489SHong Zhang } 26542c57489SHong Zhang 26642c57489SHong Zhang /* Copy data into free space, and zero out denserows */ 267ded4f561SHong Zhang ierr = PetscLLClean(pN,pN,nnz,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 268ded4f561SHong Zhang current_space->array += nnz; 269ded4f561SHong Zhang current_space->local_used += nnz; 270ded4f561SHong Zhang current_space->local_remaining -= nnz; 271ded4f561SHong Zhang coi[i+1] = coi[i] + nnz; 27242c57489SHong Zhang } 273de0260b3SHong Zhang ierr = PetscMalloc((coi[pon]+1)*sizeof(PetscInt),&coj);CHKERRQ(ierr); 274a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,coj);CHKERRQ(ierr); 275d16ca5f0SHong Zhang afill_tmp = (PetscReal)coi[pon]/(poti[pon] + api[am]); 276d16ca5f0SHong Zhang if (afill_tmp > afill) afill = afill_tmp; 277de0260b3SHong Zhang ierr = MatRestoreSymbolicTranspose_SeqAIJ(p->B,&poti,&potj);CHKERRQ(ierr); 27842c57489SHong Zhang 279fe615159SHong Zhang #if defined(DEBUG_MATPTAP) 280fe615159SHong Zhang ierr = MPI_Barrier(comm);CHKERRQ(ierr); 281fe615159SHong Zhang if (!rank) ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] local Co is done\n",rank); 282fe615159SHong Zhang #endif 283fe615159SHong Zhang 284ded4f561SHong Zhang /* send j-array (coj) of Co to other processors */ 285ded4f561SHong Zhang /*----------------------------------------------*/ 28642c57489SHong Zhang /* determine row ownership */ 28783445d95SHong Zhang ierr = PetscNew(Mat_Merge_SeqsToMPI,&merge);CHKERRQ(ierr); 28826283091SBarry Smith ierr = PetscLayoutCreate(comm,&merge->rowmap);CHKERRQ(ierr); 2897a2fc3feSBarry Smith merge->rowmap->n = pn; 2907a2fc3feSBarry Smith merge->rowmap->bs = 1; 29126283091SBarry Smith ierr = PetscLayoutSetUp(merge->rowmap);CHKERRQ(ierr); 2927a2fc3feSBarry Smith owners = merge->rowmap->range; 29342c57489SHong Zhang 29442c57489SHong Zhang /* determine the number of messages to send, their lengths */ 29542c57489SHong Zhang ierr = PetscMalloc(size*sizeof(PetscMPIInt),&len_si);CHKERRQ(ierr); 29683445d95SHong Zhang ierr = PetscMemzero(len_si,size*sizeof(PetscMPIInt));CHKERRQ(ierr); 29742c57489SHong Zhang ierr = PetscMalloc(size*sizeof(PetscMPIInt),&merge->len_s);CHKERRQ(ierr); 29842c57489SHong Zhang len_s = merge->len_s; 29942c57489SHong Zhang merge->nsend = 0; 30083445d95SHong Zhang 301de0260b3SHong Zhang ierr = PetscMalloc((size+2)*sizeof(PetscInt),&owners_co);CHKERRQ(ierr); 302de0260b3SHong Zhang ierr = PetscMemzero(len_s,size*sizeof(PetscMPIInt));CHKERRQ(ierr); 303de0260b3SHong Zhang 30483445d95SHong Zhang proc = 0; 305de0260b3SHong Zhang for (i=0; i<pon; i++){ 306de0260b3SHong Zhang while (prmap[i] >= owners[proc+1]) proc++; 307de0260b3SHong Zhang len_si[proc]++; /* num of rows in Co to be sent to [proc] */ 308de0260b3SHong Zhang len_s[proc] += coi[i+1] - coi[i]; 30983445d95SHong Zhang } 310de0260b3SHong Zhang 311de0260b3SHong Zhang len = 0; /* max length of buf_si[] */ 312de0260b3SHong Zhang owners_co[0] = 0; 31342c57489SHong Zhang for (proc=0; proc<size; proc++){ 314de0260b3SHong Zhang owners_co[proc+1] = owners_co[proc] + len_si[proc]; 31583445d95SHong Zhang if (len_si[proc]){ 31642c57489SHong Zhang merge->nsend++; 31783445d95SHong Zhang len_si[proc] = 2*(len_si[proc] + 1); 31842c57489SHong Zhang len += len_si[proc]; 31942c57489SHong Zhang } 32042c57489SHong Zhang } 32142c57489SHong Zhang 322ded4f561SHong Zhang /* determine the number and length of messages to receive for coi and coj */ 32342c57489SHong Zhang ierr = PetscGatherNumberOfMessages(comm,PETSC_NULL,len_s,&merge->nrecv);CHKERRQ(ierr); 32442c57489SHong Zhang ierr = PetscGatherMessageLengths2(comm,merge->nsend,merge->nrecv,len_s,len_si,&merge->id_r,&merge->len_r,&len_ri);CHKERRQ(ierr); 32542c57489SHong Zhang 326ded4f561SHong Zhang /* post the Irecv and Isend of coj */ 327529bc5dcSHong Zhang ierr = PetscCommGetNewTag(comm,&tagj);CHKERRQ(ierr); 328529bc5dcSHong Zhang ierr = PetscPostIrecvInt(comm,tagj,merge->nrecv,merge->id_r,merge->len_r,&buf_rj,&rwaits);CHKERRQ(ierr); 329ded4f561SHong Zhang ierr = PetscMalloc((merge->nsend+1)*sizeof(MPI_Request),&swaits);CHKERRQ(ierr); 33042c57489SHong Zhang for (proc=0, k=0; proc<size; proc++){ 33142c57489SHong Zhang if (!len_s[proc]) continue; 332de0260b3SHong Zhang i = owners_co[proc]; 333529bc5dcSHong Zhang ierr = MPI_Isend(coj+coi[i],len_s[proc],MPIU_INT,proc,tagj,comm,swaits+k);CHKERRQ(ierr); 33442c57489SHong Zhang k++; 33542c57489SHong Zhang } 33642c57489SHong Zhang 337ded4f561SHong Zhang /* receives and sends of coj are complete */ 338ded4f561SHong Zhang ierr = PetscMalloc(size*sizeof(MPI_Status),&sstatus);CHKERRQ(ierr); 33977584fe6SHong Zhang for (i=0; i<merge->nrecv; i++){ 340ded4f561SHong Zhang ierr = MPI_Waitany(merge->nrecv,rwaits,&j,&rstatus);CHKERRQ(ierr); 341ded4f561SHong Zhang } 342ded4f561SHong Zhang ierr = PetscFree(rwaits);CHKERRQ(ierr); 3430c468ba9SBarry Smith if (merge->nsend) {ierr = MPI_Waitall(merge->nsend,swaits,sstatus);CHKERRQ(ierr);} 34482412ba7SHong Zhang 345ded4f561SHong Zhang /* send and recv coi */ 346ded4f561SHong Zhang /*-------------------*/ 347529bc5dcSHong Zhang ierr = PetscCommGetNewTag(comm,&tagi);CHKERRQ(ierr); 348529bc5dcSHong Zhang ierr = PetscPostIrecvInt(comm,tagi,merge->nrecv,merge->id_r,len_ri,&buf_ri,&rwaits);CHKERRQ(ierr); 34942c57489SHong Zhang ierr = PetscMalloc((len+1)*sizeof(PetscInt),&buf_s);CHKERRQ(ierr); 35042c57489SHong Zhang buf_si = buf_s; /* points to the beginning of k-th msg to be sent */ 35142c57489SHong Zhang for (proc=0,k=0; proc<size; proc++){ 35242c57489SHong Zhang if (!len_s[proc]) continue; 35342c57489SHong Zhang /* form outgoing message for i-structure: 35442c57489SHong Zhang buf_si[0]: nrows to be sent 35542c57489SHong Zhang [1:nrows]: row index (global) 35642c57489SHong Zhang [nrows+1:2*nrows+1]: i-structure index 35742c57489SHong Zhang */ 35842c57489SHong Zhang /*-------------------------------------------*/ 35942c57489SHong Zhang nrows = len_si[proc]/2 - 1; 36042c57489SHong Zhang buf_si_i = buf_si + nrows+1; 36142c57489SHong Zhang buf_si[0] = nrows; 36242c57489SHong Zhang buf_si_i[0] = 0; 36342c57489SHong Zhang nrows = 0; 364de0260b3SHong Zhang for (i=owners_co[proc]; i<owners_co[proc+1]; i++){ 365ded4f561SHong Zhang nzi = coi[i+1] - coi[i]; 366ded4f561SHong Zhang buf_si_i[nrows+1] = buf_si_i[nrows] + nzi; /* i-structure */ 367de0260b3SHong Zhang buf_si[nrows+1] =prmap[i] -owners[proc]; /* local row index */ 36842c57489SHong Zhang nrows++; 36942c57489SHong Zhang } 370529bc5dcSHong Zhang ierr = MPI_Isend(buf_si,len_si[proc],MPIU_INT,proc,tagi,comm,swaits+k);CHKERRQ(ierr); 37142c57489SHong Zhang k++; 37242c57489SHong Zhang buf_si += len_si[proc]; 37342c57489SHong Zhang } 374ded4f561SHong Zhang i = merge->nrecv; 375ded4f561SHong Zhang while (i--) { 376ded4f561SHong Zhang ierr = MPI_Waitany(merge->nrecv,rwaits,&j,&rstatus);CHKERRQ(ierr); 377ded4f561SHong Zhang } 378ded4f561SHong Zhang ierr = PetscFree(rwaits);CHKERRQ(ierr); 3790c468ba9SBarry Smith if (merge->nsend) {ierr = MPI_Waitall(merge->nsend,swaits,sstatus);CHKERRQ(ierr);} 380f2b054eeSHong Zhang /* 381ae15b995SBarry Smith ierr = PetscInfo2(A,"nsend: %d, nrecv: %d\n",merge->nsend,merge->nrecv);CHKERRQ(ierr); 38242c57489SHong Zhang for (i=0; i<merge->nrecv; i++){ 383ae15b995SBarry Smith ierr = PetscInfo3(A,"recv len_ri=%d, len_rj=%d from [%d]\n",len_ri[i],merge->len_r[i],merge->id_r[i]);CHKERRQ(ierr); 38442c57489SHong Zhang } 385f2b054eeSHong Zhang */ 38642c57489SHong Zhang ierr = PetscFree(len_si);CHKERRQ(ierr); 38742c57489SHong Zhang ierr = PetscFree(len_ri);CHKERRQ(ierr); 388ded4f561SHong Zhang ierr = PetscFree(swaits);CHKERRQ(ierr); 389ded4f561SHong Zhang ierr = PetscFree(sstatus);CHKERRQ(ierr); 39042c57489SHong Zhang ierr = PetscFree(buf_s);CHKERRQ(ierr); 39142c57489SHong Zhang 392fe615159SHong Zhang #if defined(DEBUG_MATPTAP) 393fe615159SHong Zhang ierr = MPI_Barrier(comm);CHKERRQ(ierr); 394fe615159SHong Zhang if (!rank) ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] Co is done\n",rank); 395fe615159SHong Zhang #endif 396fe615159SHong Zhang 397de0260b3SHong Zhang /* compute the local portion of C (mpi mat) */ 398de0260b3SHong Zhang /*------------------------------------------*/ 399ded4f561SHong Zhang ierr = MatGetSymbolicTranspose_SeqAIJ(p->A,&pdti,&pdtj);CHKERRQ(ierr); 400ded4f561SHong Zhang 40142c57489SHong Zhang /* allocate bi array and free space for accumulating nonzero column info */ 40242c57489SHong Zhang ierr = PetscMalloc((pn+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 40342c57489SHong Zhang bi[0] = 0; 40442c57489SHong Zhang 405d16ca5f0SHong Zhang /* set initial free space to be fill*(nnz(P) + nnz(AP)) */ 406d16ca5f0SHong Zhang nnz = fill*(pi_loc[pm] + api[am]); 407a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,&free_space); 40842c57489SHong Zhang current_space = free_space; 40942c57489SHong Zhang 410687f1162SBarry Smith ierr = PetscMalloc3(merge->nrecv,PetscInt**,&buf_ri_k,merge->nrecv,PetscInt*,&nextrow,merge->nrecv,PetscInt*,&nextci);CHKERRQ(ierr); 41142c57489SHong Zhang for (k=0; k<merge->nrecv; k++){ 41242c57489SHong Zhang buf_ri_k[k] = buf_ri[k]; /* beginning of k-th recved i-structure */ 41342c57489SHong Zhang nrows = *buf_ri_k[k]; 41442c57489SHong Zhang nextrow[k] = buf_ri_k[k] + 1; /* next row number of k-th recved i-structure */ 41542c57489SHong Zhang nextci[k] = buf_ri_k[k] + (nrows + 1);/* poins to the next i-structure of k-th recved i-structure */ 41642c57489SHong Zhang } 41742c57489SHong Zhang ierr = MatPreallocateInitialize(comm,pn,pn,dnz,onz);CHKERRQ(ierr); 41808cb4532SHong Zhang rmax = 0; 41942c57489SHong Zhang for (i=0; i<pn; i++) { 420ded4f561SHong Zhang /* add pdt[i,:]*AP into lnk */ 421ded4f561SHong Zhang nnz = 0; 422ded4f561SHong Zhang pnz = pdti[i+1] - pdti[i]; 42377584fe6SHong Zhang ptJ = pdtj + pdti[i]; 42477584fe6SHong Zhang for (j=0; j<pnz; j++){ 42577584fe6SHong Zhang row = ptJ[j]; /* row of AP == col of Pt */ 426ded4f561SHong Zhang apnz = api[row+1] - api[row]; 427ded4f561SHong Zhang Jptr = apj + api[row]; 428ded4f561SHong Zhang /* add non-zero cols of AP into the sorted linked list lnk */ 429dadf0e6bSHong Zhang ierr = PetscLLAddSorted(apnz,Jptr,pN,nlnk,lnk,lnkbt);CHKERRQ(ierr); 430ded4f561SHong Zhang nnz += nlnk; 431ded4f561SHong Zhang } 43242c57489SHong Zhang /* add received col data into lnk */ 43342c57489SHong Zhang for (k=0; k<merge->nrecv; k++){ /* k-th received message */ 43442c57489SHong Zhang if (i == *nextrow[k]) { /* i-th row */ 435ded4f561SHong Zhang nzi = *(nextci[k]+1) - *nextci[k]; 436ded4f561SHong Zhang Jptr = buf_rj[k] + *nextci[k]; 437dadf0e6bSHong Zhang ierr = PetscLLAddSorted(nzi,Jptr,pN,nlnk,lnk,lnkbt);CHKERRQ(ierr); 438ded4f561SHong Zhang nnz += nlnk; 43942c57489SHong Zhang nextrow[k]++; nextci[k]++; 44042c57489SHong Zhang } 44142c57489SHong Zhang } 44242c57489SHong Zhang 44342c57489SHong Zhang /* if free space is not available, make more free space */ 444ded4f561SHong Zhang if (current_space->local_remaining<nnz) { 4454238b7adSHong Zhang ierr = PetscFreeSpaceGet(nnz+current_space->total_array_size,¤t_space);CHKERRQ(ierr); 446d16ca5f0SHong Zhang nspacedouble++; 44742c57489SHong Zhang } 44842c57489SHong Zhang /* copy data into free space, then initialize lnk */ 449ded4f561SHong Zhang ierr = PetscLLClean(pN,pN,nnz,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 450ded4f561SHong Zhang ierr = MatPreallocateSet(i+owners[rank],nnz,current_space->array,dnz,onz);CHKERRQ(ierr); 451ded4f561SHong Zhang current_space->array += nnz; 452ded4f561SHong Zhang current_space->local_used += nnz; 453ded4f561SHong Zhang current_space->local_remaining -= nnz; 454ded4f561SHong Zhang bi[i+1] = bi[i] + nnz; 45508cb4532SHong Zhang if (nnz > rmax) rmax = nnz; 45642c57489SHong Zhang } 457ded4f561SHong Zhang ierr = MatRestoreSymbolicTranspose_SeqAIJ(p->A,&pdti,&pdtj);CHKERRQ(ierr); 4580572522cSBarry Smith ierr = PetscFree3(buf_ri_k,nextrow,nextci);CHKERRQ(ierr); 45942c57489SHong Zhang 46042c57489SHong Zhang ierr = PetscMalloc((bi[pn]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 461a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 462d16ca5f0SHong Zhang afill_tmp = (PetscReal)bi[pn]/(pi_loc[pm] + api[am]); 463d16ca5f0SHong Zhang if (afill_tmp > afill) afill = afill_tmp; 46442c57489SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 46542c57489SHong Zhang 46608cb4532SHong Zhang /* create symbolic parallel matrix Cmpi - why cannot be assembled in Numeric part - check runex55_SA ? */ 46708cb4532SHong Zhang /*------------------------------------------------------------------------------------------------------*/ 46808cb4532SHong Zhang ierr = PetscMalloc((rmax+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 46908cb4532SHong Zhang ierr = PetscMemzero(vals,rmax*sizeof(PetscScalar));CHKERRQ(ierr); 47008cb4532SHong Zhang 47177584fe6SHong Zhang ierr = MatCreate(comm,&Cmpi);CHKERRQ(ierr); 47277584fe6SHong Zhang ierr = MatSetSizes(Cmpi,pn,pn,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr); 47377584fe6SHong Zhang ierr = MatSetType(Cmpi,MATMPIAIJ);CHKERRQ(ierr); 47477584fe6SHong Zhang ierr = MatMPIAIJSetPreallocation(Cmpi,0,dnz,0,onz);CHKERRQ(ierr); 47542c57489SHong Zhang ierr = MatPreallocateFinalize(dnz,onz);CHKERRQ(ierr); 47608cb4532SHong Zhang ierr = MatSetBlockSize(Cmpi,1);CHKERRQ(ierr); 47708cb4532SHong Zhang for (i=0; i<pn; i++){ 47808cb4532SHong Zhang row = i + rstart; 47908cb4532SHong Zhang nnz = bi[i+1] - bi[i]; 48008cb4532SHong Zhang Jptr = bj + bi[i]; 48108cb4532SHong Zhang ierr = MatSetValues(Cmpi,1,&row,nnz,Jptr,vals,INSERT_VALUES);CHKERRQ(ierr); 48208cb4532SHong Zhang } 48308cb4532SHong Zhang ierr = MatAssemblyBegin(Cmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 48408cb4532SHong Zhang ierr = MatAssemblyEnd(Cmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 48508cb4532SHong Zhang ierr = PetscFree(vals);CHKERRQ(ierr); 48642c57489SHong Zhang 48742c57489SHong Zhang merge->bi = bi; 48842c57489SHong Zhang merge->bj = bj; 489de0260b3SHong Zhang merge->coi = coi; 490de0260b3SHong Zhang merge->coj = coj; 49142c57489SHong Zhang merge->buf_ri = buf_ri; 49242c57489SHong Zhang merge->buf_rj = buf_rj; 493de0260b3SHong Zhang merge->owners_co = owners_co; 49477584fe6SHong Zhang merge->destroy = Cmpi->ops->destroy; 49577584fe6SHong Zhang merge->duplicate = Cmpi->ops->duplicate; 496cc6cb787SHong Zhang 49777584fe6SHong Zhang /* Cmpi is not ready for use - assembly will be done by MatPtAPNumeric() */ 49808cb4532SHong Zhang /* Cmpi->assembled = PETSC_FALSE; */ 49977584fe6SHong Zhang Cmpi->ops->destroy = MatDestroy_MPIAIJ_PtAP; 50077584fe6SHong Zhang Cmpi->ops->duplicate = MatDuplicate_MPIAIJ_MatPtAP; 50142c57489SHong Zhang 50277584fe6SHong Zhang /* attach the supporting struct to Cmpi for reuse */ 50377584fe6SHong Zhang c = (Mat_MPIAIJ*)Cmpi->data; 504f8487c73SHong Zhang c->ptap = ptap; 50577584fe6SHong Zhang ptap->api = api; 50677584fe6SHong Zhang ptap->apj = apj; 507f8487c73SHong Zhang ptap->merge = merge; 508d6ab1dc0SHong Zhang ptap->rmax = ap_rmax; 509f8487c73SHong Zhang 51077584fe6SHong Zhang *C = Cmpi; 511f2b054eeSHong Zhang #if defined(PETSC_USE_INFO) 512f2b054eeSHong Zhang if (bi[pn] != 0) { 51377584fe6SHong Zhang ierr = PetscInfo3(Cmpi,"Reallocs %D; Fill ratio: given %G needed %G.\n",nspacedouble,fill,afill);CHKERRQ(ierr); 51477584fe6SHong Zhang ierr = PetscInfo1(Cmpi,"Use MatPtAP(A,P,MatReuse,%G,&C) for best performance.\n",afill);CHKERRQ(ierr); 515f2b054eeSHong Zhang } else { 51677584fe6SHong Zhang ierr = PetscInfo(Cmpi,"Empty matrix product\n");CHKERRQ(ierr); 517f2b054eeSHong Zhang } 518f2b054eeSHong Zhang #endif 51942c57489SHong Zhang PetscFunctionReturn(0); 52042c57489SHong Zhang } 52142c57489SHong Zhang 52242c57489SHong Zhang #undef __FUNCT__ 52342c57489SHong Zhang #define __FUNCT__ "MatPtAPNumeric_MPIAIJ_MPIAIJ" 52442c57489SHong Zhang PetscErrorCode MatPtAPNumeric_MPIAIJ_MPIAIJ(Mat A,Mat P,Mat C) 52542c57489SHong Zhang { 52642c57489SHong Zhang PetscErrorCode ierr; 52742c57489SHong Zhang Mat_Merge_SeqsToMPI *merge; 528f8487c73SHong Zhang Mat_MPIAIJ *a=(Mat_MPIAIJ*)A->data,*p=(Mat_MPIAIJ*)P->data,*c=(Mat_MPIAIJ*)C->data; 52942c57489SHong Zhang Mat_SeqAIJ *ad=(Mat_SeqAIJ*)(a->A)->data,*ao=(Mat_SeqAIJ*)(a->B)->data; 530de0260b3SHong Zhang Mat_SeqAIJ *pd=(Mat_SeqAIJ*)(p->A)->data,*po=(Mat_SeqAIJ*)(p->B)->data; 53142c57489SHong Zhang Mat_SeqAIJ *p_loc,*p_oth; 532f8487c73SHong Zhang Mat_PtAPMPI *ptap; 5331d633602SHong Zhang PetscInt *adi=ad->i,*aoi=ao->i,*adj,*aoj,*apJ,nextp; 53482412ba7SHong Zhang PetscInt *pi_loc,*pj_loc,*pi_oth,*pj_oth,*pJ,*pj; 53582412ba7SHong Zhang PetscInt i,j,k,anz,pnz,apnz,nextap,row,*cj; 536e900a757SHong Zhang MatScalar *ada,*aoa,*apa,*pa,*ca,*pa_loc,*pa_oth,valtmp; 537d0f46423SBarry Smith PetscInt am=A->rmap->n,cm=C->rmap->n,pon=(p->B)->cmap->n; 5387adad957SLisandro Dalcin MPI_Comm comm=((PetscObject)C)->comm; 53942c57489SHong Zhang PetscMPIInt size,rank,taga,*len_s; 54082412ba7SHong Zhang PetscInt *owners,proc,nrows,**buf_ri_k,**nextrow,**nextci; 54142c57489SHong Zhang PetscInt **buf_ri,**buf_rj; 54250f5bf86SHong Zhang PetscInt cnz=0,*bj_i,*bi,*bj,bnz,nextcj; /* bi,bj,ba: local array of C(mpi mat) */ 54342c57489SHong Zhang MPI_Request *s_waits,*r_waits; 54442c57489SHong Zhang MPI_Status *status; 54582412ba7SHong Zhang MatScalar **abuf_r,*ba_i,*pA,*coa,*ba; 54682412ba7SHong Zhang PetscInt *api,*apj,*coi,*coj; 547d0f46423SBarry Smith PetscInt *poJ=po->j,*pdJ=pd->j,pcstart=P->cmap->rstart,pcend=P->cmap->rend; 548a9d06231SHong Zhang PetscInt sparse_axpy; 54942c57489SHong Zhang 55042c57489SHong Zhang PetscFunctionBegin; 55142c57489SHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 55242c57489SHong Zhang ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 55342c57489SHong Zhang 554f8487c73SHong Zhang ptap = c->ptap; 555f8487c73SHong Zhang merge = ptap->merge; 5566c6e00e0SHong Zhang 557a1a4e84aSHong Zhang /* 1) get P_oth = ptap->P_oth and P_loc = ptap->P_loc */ 558f9473cf7SHong Zhang /*--------------------------------------------------*/ 559f8487c73SHong Zhang if (ptap->reuse == MAT_INITIAL_MATRIX){ 560f8487c73SHong Zhang ptap->reuse = MAT_REUSE_MATRIX; 5616c6e00e0SHong Zhang } else { /* update numerical values of P_oth and P_loc */ 562*b7f45c76SHong Zhang ierr = MatGetBrowsOfAoCols_MPIAIJ(A,P,MAT_REUSE_MATRIX,&ptap->startsj_s,&ptap->startsj_r,&ptap->bufa,&ptap->P_oth);CHKERRQ(ierr); 563a1a4e84aSHong Zhang ierr = MatMPIAIJGetLocalMat(P,MAT_REUSE_MATRIX,&ptap->P_loc);CHKERRQ(ierr); 5646c6e00e0SHong Zhang } 565f8487c73SHong Zhang 566f9473cf7SHong Zhang /* 2) compute numeric C_seq = P_loc^T*A_loc*P - dominating part */ 567f9473cf7SHong Zhang /*--------------------------------------------------------------*/ 56842c57489SHong Zhang /* get data from symbolic products */ 569a1a4e84aSHong Zhang p_loc = (Mat_SeqAIJ*)(ptap->P_loc)->data; 570a1a4e84aSHong Zhang p_oth = (Mat_SeqAIJ*)(ptap->P_oth)->data; 571a9d06231SHong Zhang pi_loc=p_loc->i; pj_loc=p_loc->j; pJ=pj_loc; pa_loc=p_loc->a; 57242c57489SHong Zhang pi_oth=p_oth->i; pj_oth=p_oth->j; pa_oth=p_oth->a; 57342c57489SHong Zhang 574de0260b3SHong Zhang coi = merge->coi; coj = merge->coj; 575de0260b3SHong Zhang ierr = PetscMalloc((coi[pon]+1)*sizeof(MatScalar),&coa);CHKERRQ(ierr); 576de0260b3SHong Zhang ierr = PetscMemzero(coa,coi[pon]*sizeof(MatScalar));CHKERRQ(ierr); 577de0260b3SHong Zhang 578de0260b3SHong Zhang bi = merge->bi; bj = merge->bj; 5797a2fc3feSBarry Smith owners = merge->rowmap->range; 580de0260b3SHong Zhang ierr = PetscMalloc((bi[cm]+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 581de0260b3SHong Zhang ierr = PetscMemzero(ba,bi[cm]*sizeof(MatScalar));CHKERRQ(ierr); 58242c57489SHong Zhang 583a1a4e84aSHong Zhang api = ptap->api; apj = ptap->apj; 5840496f32cSHong Zhang /* flag 'sparse_axpy' determines which implementations to be used: 5850496f32cSHong Zhang 0: do dense axpy in MatPtAPNumeric() - fastest, but requires storage of a dense array apa; (default) 5860496f32cSHong Zhang 1: do one sparse axpy - uses same memory as sparse_axpy=0 and might execute less flops 5870496f32cSHong Zhang (apnz vs. cnz in the outerproduct), slower than case '0' when cnz is not too large than apnz; 5880496f32cSHong Zhang 2: do two sparse axpy in MatPtAPNumeric() - slowest, uses a sparse array apa */ 589a9d06231SHong Zhang /* set default sparse_axpy */ 590e900a757SHong Zhang sparse_axpy = 0; 5910496f32cSHong Zhang ierr = PetscOptionsGetInt(PETSC_NULL,"-matptap_sparseaxpy",&sparse_axpy,PETSC_NULL);CHKERRQ(ierr); 592a9d06231SHong Zhang if (sparse_axpy == 0){ /* Do not perform sparse axpy */ 593a9d06231SHong Zhang /*--------------------------------------------------*/ 594a9d06231SHong Zhang /* malloc apa to store dense row A[i,:]*P */ 595a9d06231SHong Zhang ierr = PetscMalloc((P->cmap->N)*sizeof(PetscScalar),&apa);CHKERRQ(ierr); 596a9d06231SHong Zhang ierr = PetscMemzero(apa,P->cmap->N*sizeof(PetscScalar));CHKERRQ(ierr); 597a9d06231SHong Zhang 598a9d06231SHong Zhang for (i=0; i<am; i++) { 599a9d06231SHong Zhang /* 2-a) form i-th sparse row of A_loc*P = Ad*P_loc + Ao*P_oth */ 600a9d06231SHong Zhang /*------------------------------------------------------------*/ 601a9d06231SHong Zhang apJ = apj + api[i]; 602a9d06231SHong Zhang 603a9d06231SHong Zhang /* diagonal portion of A */ 604a9d06231SHong Zhang anz = adi[i+1] - adi[i]; 605a9d06231SHong Zhang adj = ad->j + adi[i]; 606a9d06231SHong Zhang ada = ad->a + adi[i]; 607a9d06231SHong Zhang for (j=0; j<anz; j++) { 608a9d06231SHong Zhang row = adj[j]; 609a9d06231SHong Zhang pnz = pi_loc[row+1] - pi_loc[row]; 610a9d06231SHong Zhang pj = pj_loc + pi_loc[row]; 611a9d06231SHong Zhang pa = pa_loc + pi_loc[row]; 612a9d06231SHong Zhang 613a9d06231SHong Zhang /* perform dense axpy */ 614e900a757SHong Zhang valtmp = ada[j]; 615a9d06231SHong Zhang for (k=0; k<pnz; k++){ 616e900a757SHong Zhang apa[pj[k]] += valtmp*pa[k]; 617a9d06231SHong Zhang } 618a9d06231SHong Zhang ierr = PetscLogFlops(2.0*pnz);CHKERRQ(ierr); 619a9d06231SHong Zhang } 620a9d06231SHong Zhang 621a9d06231SHong Zhang /* off-diagonal portion of A */ 622a9d06231SHong Zhang anz = aoi[i+1] - aoi[i]; 623a9d06231SHong Zhang aoj = ao->j + aoi[i]; 624a9d06231SHong Zhang aoa = ao->a + aoi[i]; 625a9d06231SHong Zhang for (j=0; j<anz; j++) { 626a9d06231SHong Zhang row = aoj[j]; 627a9d06231SHong Zhang pnz = pi_oth[row+1] - pi_oth[row]; 628a9d06231SHong Zhang pj = pj_oth + pi_oth[row]; 629a9d06231SHong Zhang pa = pa_oth + pi_oth[row]; 630a9d06231SHong Zhang 631a9d06231SHong Zhang /* perform dense axpy */ 632e900a757SHong Zhang valtmp = aoa[j]; 633a9d06231SHong Zhang for (k=0; k<pnz; k++){ 634e900a757SHong Zhang apa[pj[k]] += valtmp*pa[k]; 635a9d06231SHong Zhang } 636a9d06231SHong Zhang ierr = PetscLogFlops(2.0*pnz);CHKERRQ(ierr); 637a9d06231SHong Zhang } 638a9d06231SHong Zhang 639a9d06231SHong Zhang /* 2-b) Compute Cseq = P_loc[i,:]^T*AP[i,:] using outer product */ 640a9d06231SHong Zhang /*--------------------------------------------------------------*/ 641a9d06231SHong Zhang apnz = api[i+1] - api[i]; 642a9d06231SHong Zhang /* put the value into Co=(p->B)^T*AP (off-diagonal part, send to others) */ 643a9d06231SHong Zhang pnz = po->i[i+1] - po->i[i]; 644e900a757SHong Zhang poJ = po->j + po->i[i]; 645a9d06231SHong Zhang pA = po->a + po->i[i]; 646a9d06231SHong Zhang for (j=0; j<pnz; j++){ 647e900a757SHong Zhang row = poJ[j]; 648e900a757SHong Zhang cnz = coi[row+1] - coi[row]; 649e900a757SHong Zhang cj = coj + coi[row]; 650e900a757SHong Zhang ca = coa + coi[row]; 651a9d06231SHong Zhang /* perform dense axpy */ 652e900a757SHong Zhang valtmp = pA[j]; 653a9d06231SHong Zhang for (k=0; k<cnz; k++) { 654e900a757SHong Zhang ca[k] += valtmp*apa[cj[k]]; 655a9d06231SHong Zhang } 656a9d06231SHong Zhang ierr = PetscLogFlops(2.0*cnz);CHKERRQ(ierr); 657a9d06231SHong Zhang } 658a9d06231SHong Zhang 659a9d06231SHong Zhang /* put the value into Cd (diagonal part) */ 660a9d06231SHong Zhang pnz = pd->i[i+1] - pd->i[i]; 661e900a757SHong Zhang pdJ = pd->j + pd->i[i]; 662a9d06231SHong Zhang pA = pd->a + pd->i[i]; 663a9d06231SHong Zhang for (j=0; j<pnz; j++){ 664e900a757SHong Zhang row = pdJ[j]; 665e900a757SHong Zhang cnz = bi[row+1] - bi[row]; 666e900a757SHong Zhang cj = bj + bi[row]; 667e900a757SHong Zhang ca = ba + bi[row]; 668a9d06231SHong Zhang /* perform dense axpy */ 669e900a757SHong Zhang valtmp = pA[j]; 670a9d06231SHong Zhang for (k=0; k<cnz; k++) { 671e900a757SHong Zhang ca[k] += valtmp*apa[cj[k]]; 672a9d06231SHong Zhang } 673a9d06231SHong Zhang ierr = PetscLogFlops(2.0*cnz);CHKERRQ(ierr); 674a9d06231SHong Zhang } 675a9d06231SHong Zhang 676a9d06231SHong Zhang /* zero the current row of A*P */ 677a9d06231SHong Zhang for (k=0; k<apnz; k++) apa[apJ[k]] = 0.0; 678a9d06231SHong Zhang } 679a9d06231SHong Zhang } else if (sparse_axpy == 1){ /* Perform one sparse axpy */ 680a9d06231SHong Zhang /*------------------------------------------------------*/ 6811d633602SHong Zhang /* malloc apa to store dense row A[i,:]*P */ 6821d633602SHong Zhang ierr = PetscMalloc((P->cmap->N)*sizeof(PetscScalar),&apa);CHKERRQ(ierr); 6831d633602SHong Zhang ierr = PetscMemzero(apa,P->cmap->N*sizeof(PetscScalar));CHKERRQ(ierr); 6841d633602SHong Zhang 6851d633602SHong Zhang for (i=0; i<am; i++) { 686f9473cf7SHong Zhang /* 2-a) form i-th sparse row of A_loc*P = Ad*P_loc + Ao*P_oth */ 687f9473cf7SHong Zhang /*------------------------------------------------------------*/ 6881d633602SHong Zhang apJ = apj + api[i]; 6891d633602SHong Zhang 6901d633602SHong Zhang /* diagonal portion of A */ 6911d633602SHong Zhang anz = adi[i+1] - adi[i]; 6921d633602SHong Zhang adj = ad->j + adi[i]; 6931d633602SHong Zhang ada = ad->a + adi[i]; 6941d633602SHong Zhang for (j=0; j<anz; j++) { 6951d633602SHong Zhang row = adj[j]; 6961d633602SHong Zhang pnz = pi_loc[row+1] - pi_loc[row]; 6971d633602SHong Zhang pj = pj_loc + pi_loc[row]; 6981d633602SHong Zhang pa = pa_loc + pi_loc[row]; 6991d633602SHong Zhang 7001d633602SHong Zhang /* perform dense axpy */ 701e900a757SHong Zhang valtmp = ada[j]; 7021d633602SHong Zhang for (k=0; k<pnz; k++){ 703e900a757SHong Zhang apa[pj[k]] += valtmp*pa[k]; 7041d633602SHong Zhang } 7051d633602SHong Zhang ierr = PetscLogFlops(2.0*pnz);CHKERRQ(ierr); 7061d633602SHong Zhang } 7071d633602SHong Zhang 7081d633602SHong Zhang /* off-diagonal portion of A */ 7091d633602SHong Zhang anz = aoi[i+1] - aoi[i]; 7101d633602SHong Zhang aoj = ao->j + aoi[i]; 7111d633602SHong Zhang aoa = ao->a + aoi[i]; 7121d633602SHong Zhang for (j=0; j<anz; j++) { 7131d633602SHong Zhang row = aoj[j]; 7141d633602SHong Zhang pnz = pi_oth[row+1] - pi_oth[row]; 7151d633602SHong Zhang pj = pj_oth + pi_oth[row]; 7161d633602SHong Zhang pa = pa_oth + pi_oth[row]; 7171d633602SHong Zhang 7181d633602SHong Zhang /* perform dense axpy */ 719e900a757SHong Zhang valtmp = aoa[j]; 7201d633602SHong Zhang for (k=0; k<pnz; k++){ 721e900a757SHong Zhang apa[pj[k]] += valtmp*pa[k]; 7221d633602SHong Zhang } 7231d633602SHong Zhang ierr = PetscLogFlops(2.0*pnz);CHKERRQ(ierr); 7241d633602SHong Zhang } 7251d633602SHong Zhang 726f9473cf7SHong Zhang /* 2-b) Compute Cseq = P_loc[i,:]^T*AP[i,:] using outer product */ 727f9473cf7SHong Zhang /*--------------------------------------------------------------*/ 7281d633602SHong Zhang apnz = api[i+1] - api[i]; 729f9473cf7SHong Zhang /* put the value into Co=(p->B)^T*AP (off-diagonal part, send to others) */ 730f9473cf7SHong Zhang pnz = po->i[i+1] - po->i[i]; 731e900a757SHong Zhang poJ = po->j + po->i[i]; 732a9d06231SHong Zhang pA = po->a + po->i[i]; 7331d633602SHong Zhang for (j=0; j<pnz; j++){ 734e900a757SHong Zhang row = poJ[j]; 735e900a757SHong Zhang cj = coj + coi[row]; 736e900a757SHong Zhang ca = coa + coi[row]; 737e900a757SHong Zhang valtmp = pA[j]; 7381d633602SHong Zhang /* perform sparse axpy */ 7391d633602SHong Zhang nextap = 0; 7401d633602SHong Zhang for (k=0; nextap<apnz; k++) { 7411d633602SHong Zhang if (cj[k]==apJ[nextap]) { /* global column index */ 742e900a757SHong Zhang ca[k] += valtmp*apa[cj[k]]; nextap++; 7431d633602SHong Zhang } 7441d633602SHong Zhang } 7451d633602SHong Zhang ierr = PetscLogFlops(2.0*apnz);CHKERRQ(ierr); 7461d633602SHong Zhang } 747f9473cf7SHong Zhang 748f9473cf7SHong Zhang /* put the value into Cd (diagonal part) */ 749f9473cf7SHong Zhang pnz = pd->i[i+1] - pd->i[i]; 750e900a757SHong Zhang pdJ = pd->j + pd->i[i]; 751a9d06231SHong Zhang pA = pd->a + pd->i[i]; 752f9473cf7SHong Zhang for (j=0; j<pnz; j++){ 753e900a757SHong Zhang row = pdJ[j]; 754e900a757SHong Zhang cj = bj + bi[row]; 755e900a757SHong Zhang ca = ba + bi[row]; 756e900a757SHong Zhang valtmp = pA[j]; 757f9473cf7SHong Zhang /* perform sparse axpy */ 758f9473cf7SHong Zhang nextap = 0; 759f9473cf7SHong Zhang for (k=0; nextap<apnz; k++) { 760f9473cf7SHong Zhang if (cj[k]==apJ[nextap]) { /* global column index */ 761e900a757SHong Zhang ca[k] += valtmp*apa[cj[k]]; 762a9d06231SHong Zhang nextap++; 763f9473cf7SHong Zhang } 764f9473cf7SHong Zhang } 765f9473cf7SHong Zhang ierr = PetscLogFlops(2.0*apnz);CHKERRQ(ierr); 766f9473cf7SHong Zhang } 767f9473cf7SHong Zhang 768f9473cf7SHong Zhang /* zero the current row of A*P */ 7691d633602SHong Zhang for (k=0; k<apnz; k++) apa[apJ[k]] = 0.0; 7701d633602SHong Zhang } 7710496f32cSHong Zhang } else if (sparse_axpy == 2){/* Perform two sparse axpy */ 772a9d06231SHong Zhang /*----------------------------------------------------*/ 7731d633602SHong Zhang /* malloc apa to store sparse row A[i,:]*P */ 774d6ab1dc0SHong Zhang ierr = PetscMalloc((ptap->rmax+1)*sizeof(MatScalar),&apa);CHKERRQ(ierr); 775d6ab1dc0SHong Zhang ierr = PetscMemzero(apa,ptap->rmax*sizeof(MatScalar));CHKERRQ(ierr); 77642c57489SHong Zhang 777a9d06231SHong Zhang pA=pa_loc; 77842c57489SHong Zhang for (i=0; i<am; i++) { 779f4a743e1SHong Zhang /* form i-th sparse row of A*P */ 78082412ba7SHong Zhang apnz = api[i+1] - api[i]; 78182412ba7SHong Zhang apJ = apj + api[i]; 78242c57489SHong Zhang /* diagonal portion of A */ 78382412ba7SHong Zhang anz = adi[i+1] - adi[i]; 7841d633602SHong Zhang adj = ad->j + adi[i]; 7851d633602SHong Zhang ada = ad->a + adi[i]; 78682412ba7SHong Zhang for (j=0; j<anz; j++) { 7871d633602SHong Zhang row = adj[j]; 78882412ba7SHong Zhang pnz = pi_loc[row+1] - pi_loc[row]; 78982412ba7SHong Zhang pj = pj_loc + pi_loc[row]; 79082412ba7SHong Zhang pa = pa_loc + pi_loc[row]; 791e900a757SHong Zhang valtmp = ada[j]; 792f4a743e1SHong Zhang nextp = 0; 79382412ba7SHong Zhang for (k=0; nextp<pnz; k++) { 79482412ba7SHong Zhang if (apJ[k] == pj[nextp]) { /* col of AP == col of P */ 795e900a757SHong Zhang apa[k] += valtmp*pa[nextp++]; 79642c57489SHong Zhang } 79742c57489SHong Zhang } 798dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*pnz);CHKERRQ(ierr); 79942c57489SHong Zhang } 80042c57489SHong Zhang /* off-diagonal portion of A */ 80182412ba7SHong Zhang anz = aoi[i+1] - aoi[i]; 8021d633602SHong Zhang aoj = ao->j + aoi[i]; 8031d633602SHong Zhang aoa = ao->a + aoi[i]; 80482412ba7SHong Zhang for (j=0; j<anz; j++) { 8051d633602SHong Zhang row = aoj[j]; 80682412ba7SHong Zhang pnz = pi_oth[row+1] - pi_oth[row]; 80782412ba7SHong Zhang pj = pj_oth + pi_oth[row]; 80882412ba7SHong Zhang pa = pa_oth + pi_oth[row]; 809e900a757SHong Zhang valtmp = aoa[j]; 810f4a743e1SHong Zhang nextp = 0; 81182412ba7SHong Zhang for (k=0; nextp<pnz; k++) { 81282412ba7SHong Zhang if (apJ[k] == pj[nextp]) { /* col of AP == col of P */ 813e900a757SHong Zhang apa[k] += valtmp*pa[nextp++]; 81442c57489SHong Zhang } 81542c57489SHong Zhang } 816dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*pnz);CHKERRQ(ierr); 81742c57489SHong Zhang } 81842c57489SHong Zhang 819a9d06231SHong Zhang /* 2-b) Compute Cseq = P_loc[i,:]^T*AP[i,:] using outer product */ 820a9d06231SHong Zhang /*--------------------------------------------------------------*/ 82182412ba7SHong Zhang pnz = pi_loc[i+1] - pi_loc[i]; 822f9473cf7SHong Zhang pJ = pj_loc + pi_loc[i]; 82382412ba7SHong Zhang for (j=0; j<pnz; j++) { 82442c57489SHong Zhang nextap = 0; 825f9473cf7SHong Zhang row = pJ[j]; /* global index */ 82682412ba7SHong Zhang if (row < pcstart || row >=pcend) { /* put the value into Co */ 827e900a757SHong Zhang row = *poJ; 828e900a757SHong Zhang cj = coj + coi[row]; 829e900a757SHong Zhang ca = coa + coi[row]; poJ++; 83082412ba7SHong Zhang } else { /* put the value into Cd */ 831e900a757SHong Zhang row = *pdJ; 832e900a757SHong Zhang cj = bj + bi[row]; 833e900a757SHong Zhang ca = ba + bi[row]; pdJ++; 83442c57489SHong Zhang } 835e900a757SHong Zhang valtmp = pA[j]; 83682412ba7SHong Zhang for (k=0; nextap<apnz; k++) { 837e900a757SHong Zhang if (cj[k]==apJ[nextap]) ca[k] += valtmp*apa[nextap++]; 83842c57489SHong Zhang } 839dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*apnz);CHKERRQ(ierr); 840de0260b3SHong Zhang } 8410496f32cSHong Zhang pA += pnz; 84242c57489SHong Zhang /* zero the current row info for A*P */ 84382412ba7SHong Zhang ierr = PetscMemzero(apa,apnz*sizeof(MatScalar));CHKERRQ(ierr); 84442c57489SHong Zhang } 845a9d06231SHong Zhang } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"sparse_axpy only takes values 0, 1 and 2"); 84642c57489SHong Zhang ierr = PetscFree(apa);CHKERRQ(ierr); 84742c57489SHong Zhang 848a9d06231SHong Zhang /* 3) send and recv matrix values coa */ 849a9d06231SHong Zhang /*------------------------------------*/ 85042c57489SHong Zhang buf_ri = merge->buf_ri; 85142c57489SHong Zhang buf_rj = merge->buf_rj; 85242c57489SHong Zhang len_s = merge->len_s; 853fc42d0c8SSatish Balay ierr = PetscCommGetNewTag(comm,&taga);CHKERRQ(ierr); 85442c57489SHong Zhang ierr = PetscPostIrecvScalar(comm,taga,merge->nrecv,merge->id_r,merge->len_r,&abuf_r,&r_waits);CHKERRQ(ierr); 85542c57489SHong Zhang 856a9d06231SHong Zhang ierr = PetscMalloc2(merge->nsend+1,MPI_Request,&s_waits,size,MPI_Status,&status);CHKERRQ(ierr); 85742c57489SHong Zhang for (proc=0,k=0; proc<size; proc++){ 85842c57489SHong Zhang if (!len_s[proc]) continue; 859de0260b3SHong Zhang i = merge->owners_co[proc]; 860de0260b3SHong Zhang ierr = MPI_Isend(coa+coi[i],len_s[proc],MPIU_MATSCALAR,proc,taga,comm,s_waits+k);CHKERRQ(ierr); 86142c57489SHong Zhang k++; 86242c57489SHong Zhang } 8630c468ba9SBarry Smith if (merge->nrecv) {ierr = MPI_Waitall(merge->nrecv,r_waits,status);CHKERRQ(ierr);} 8640c468ba9SBarry Smith if (merge->nsend) {ierr = MPI_Waitall(merge->nsend,s_waits,status);CHKERRQ(ierr);} 86542c57489SHong Zhang 866a9d06231SHong Zhang ierr = PetscFree2(s_waits,status);CHKERRQ(ierr); 86742c57489SHong Zhang ierr = PetscFree(r_waits);CHKERRQ(ierr); 86882412ba7SHong Zhang ierr = PetscFree(coa);CHKERRQ(ierr); 86942c57489SHong Zhang 870a9d06231SHong Zhang /* 4) insert local Cseq and received values into Cmpi */ 871a9d06231SHong Zhang /*------------------------------------------------------*/ 872687f1162SBarry Smith ierr = PetscMalloc3(merge->nrecv,PetscInt**,&buf_ri_k,merge->nrecv,PetscInt*,&nextrow,merge->nrecv,PetscInt*,&nextci);CHKERRQ(ierr); 87342c57489SHong Zhang for (k=0; k<merge->nrecv; k++){ 87442c57489SHong Zhang buf_ri_k[k] = buf_ri[k]; /* beginning of k-th recved i-structure */ 87542c57489SHong Zhang nrows = *(buf_ri_k[k]); 87642c57489SHong Zhang nextrow[k] = buf_ri_k[k]+1; /* next row number of k-th recved i-structure */ 87782412ba7SHong Zhang nextci[k] = buf_ri_k[k] + (nrows + 1);/* poins to the next i-structure of k-th recved i-structure */ 87842c57489SHong Zhang } 87942c57489SHong Zhang 880de0260b3SHong Zhang for (i=0; i<cm; i++) { 88182412ba7SHong Zhang row = owners[rank] + i; /* global row index of C_seq */ 88242c57489SHong Zhang bj_i = bj + bi[i]; /* col indices of the i-th row of C */ 883de0260b3SHong Zhang ba_i = ba + bi[i]; 88482412ba7SHong Zhang bnz = bi[i+1] - bi[i]; 88542c57489SHong Zhang /* add received vals into ba */ 88642c57489SHong Zhang for (k=0; k<merge->nrecv; k++){ /* k-th received message */ 88742c57489SHong Zhang /* i-th row */ 88842c57489SHong Zhang if (i == *nextrow[k]) { 88982412ba7SHong Zhang cnz = *(nextci[k]+1) - *nextci[k]; 89082412ba7SHong Zhang cj = buf_rj[k] + *(nextci[k]); 89182412ba7SHong Zhang ca = abuf_r[k] + *(nextci[k]); 89282412ba7SHong Zhang nextcj = 0; 89382412ba7SHong Zhang for (j=0; nextcj<cnz; j++){ 89482412ba7SHong Zhang if (bj_i[j] == cj[nextcj]){ /* bcol == ccol */ 89582412ba7SHong Zhang ba_i[j] += ca[nextcj++]; 89642c57489SHong Zhang } 89742c57489SHong Zhang } 89882412ba7SHong Zhang nextrow[k]++; nextci[k]++; 899c5af039cSHong Zhang ierr = PetscLogFlops(2.0*cnz);CHKERRQ(ierr); 90042c57489SHong Zhang } 90142c57489SHong Zhang } 90282412ba7SHong Zhang ierr = MatSetValues(C,1,&row,bnz,bj_i,ba_i,INSERT_VALUES);CHKERRQ(ierr); 90342c57489SHong Zhang } 90442c57489SHong Zhang ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 90542c57489SHong Zhang ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 90642c57489SHong Zhang 907de0260b3SHong Zhang ierr = PetscFree(ba);CHKERRQ(ierr); 908c05d87d6SBarry Smith ierr = PetscFree(abuf_r[0]);CHKERRQ(ierr); 90942c57489SHong Zhang ierr = PetscFree(abuf_r);CHKERRQ(ierr); 9100572522cSBarry Smith ierr = PetscFree3(buf_ri_k,nextrow,nextci);CHKERRQ(ierr); 91142c57489SHong Zhang PetscFunctionReturn(0); 91242c57489SHong Zhang } 913