xref: /petsc/src/mat/impls/aij/mpi/mpiptap.c (revision b7f45c76db6feae4085b9eefce882ba3e3973571)
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,&current_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,&current_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,&current_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