Actual source code: baijov.c
petsc-3.10.5 2019-03-28
2: /*
3: Routines to compute overlapping regions of a parallel MPI matrix
4: and to find submatrices that were shared across processors.
5: */
6: #include <../src/mat/impls/baij/mpi/mpibaij.h>
7: #include <petscbt.h>
9: static PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Local(Mat,PetscInt,char**,PetscInt*,PetscInt**);
10: static PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Receive(Mat,PetscInt,PetscInt**,PetscInt**,PetscInt*);
11: extern PetscErrorCode MatGetRow_MPIBAIJ(Mat,PetscInt,PetscInt*,PetscInt**,PetscScalar**);
12: extern PetscErrorCode MatRestoreRow_MPIBAIJ(Mat,PetscInt,PetscInt*,PetscInt**,PetscScalar**);
14: PetscErrorCode MatIncreaseOverlap_MPIBAIJ(Mat C,PetscInt imax,IS is[],PetscInt ov)
15: {
17: PetscInt i,N=C->cmap->N, bs=C->rmap->bs;
18: IS *is_new;
21: PetscMalloc1(imax,&is_new);
22: /* Convert the indices into block format */
23: ISCompressIndicesGeneral(N,C->rmap->n,bs,imax,is,is_new);
24: if (ov < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative overlap specified\n");
25: for (i=0; i<ov; ++i) {
26: MatIncreaseOverlap_MPIBAIJ_Once(C,imax,is_new);
27: }
28: for (i=0; i<imax; i++) {ISDestroy(&is[i]);}
29: ISExpandIndicesGeneral(N,N,bs,imax,is_new,is);
30: for (i=0; i<imax; i++) {ISDestroy(&is_new[i]);}
31: PetscFree(is_new);
32: return(0);
33: }
35: /*
36: Sample message format:
37: If a processor A wants processor B to process some elements corresponding
38: to index sets is[1], is[5]
39: mesg [0] = 2 (no of index sets in the mesg)
40: -----------
41: mesg [1] = 1 => is[1]
42: mesg [2] = sizeof(is[1]);
43: -----------
44: mesg [5] = 5 => is[5]
45: mesg [6] = sizeof(is[5]);
46: -----------
47: mesg [7]
48: mesg [n] data(is[1])
49: -----------
50: mesg[n+1]
51: mesg[m] data(is[5])
52: -----------
54: Notes:
55: nrqs - no of requests sent (or to be sent out)
56: nrqr - no of requests recieved (which have to be or which have been processed
57: */
58: PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Once(Mat C,PetscInt imax,IS is[])
59: {
60: Mat_MPIBAIJ *c = (Mat_MPIBAIJ*)C->data;
61: const PetscInt **idx,*idx_i;
62: PetscInt *n,*w3,*w4,**data,len;
64: PetscMPIInt size,rank,tag1,tag2,*w2,*w1,nrqr;
65: PetscInt Mbs,i,j,k,**rbuf,row,proc=-1,nrqs,msz,**outdat,**ptr;
66: PetscInt *ctr,*pa,*tmp,*isz,*isz1,**xdata,**rbuf2,*d_p;
67: PetscMPIInt *onodes1,*olengths1,*onodes2,*olengths2;
68: PetscBT *table;
69: MPI_Comm comm,*iscomms;
70: MPI_Request *s_waits1,*r_waits1,*s_waits2,*r_waits2;
71: MPI_Status *s_status,*recv_status;
72: char *t_p;
75: PetscObjectGetComm((PetscObject)C,&comm);
76: size = c->size;
77: rank = c->rank;
78: Mbs = c->Mbs;
80: PetscObjectGetNewTag((PetscObject)C,&tag1);
81: PetscObjectGetNewTag((PetscObject)C,&tag2);
83: PetscMalloc2(imax+1,(PetscInt***)&idx,imax,&n);
85: for (i=0; i<imax; i++) {
86: ISGetIndices(is[i],&idx[i]);
87: ISGetLocalSize(is[i],&n[i]);
88: }
90: /* evaluate communication - mesg to who,length of mesg, and buffer space
91: required. Based on this, buffers are allocated, and data copied into them*/
92: PetscCalloc4(size,&w1,size,&w2,size,&w3,size,&w4);
93: for (i=0; i<imax; i++) {
94: PetscMemzero(w4,size*sizeof(PetscInt)); /* initialise work vector*/
95: idx_i = idx[i];
96: len = n[i];
97: for (j=0; j<len; j++) {
98: row = idx_i[j];
99: if (row < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Index set cannot have negative entries");
100: PetscLayoutFindOwner(C->rmap,row*C->rmap->bs,&proc);
101: w4[proc]++;
102: }
103: for (j=0; j<size; j++) {
104: if (w4[j]) { w1[j] += w4[j]; w3[j]++;}
105: }
106: }
108: nrqs = 0; /* no of outgoing messages */
109: msz = 0; /* total mesg length (for all proc */
110: w1[rank] = 0; /* no mesg sent to itself */
111: w3[rank] = 0;
112: for (i=0; i<size; i++) {
113: if (w1[i]) {w2[i] = 1; nrqs++;} /* there exists a message to proc i */
114: }
115: /* pa - is list of processors to communicate with */
116: PetscMalloc1(nrqs+1,&pa);
117: for (i=0,j=0; i<size; i++) {
118: if (w1[i]) {pa[j] = i; j++;}
119: }
121: /* Each message would have a header = 1 + 2*(no of IS) + data */
122: for (i=0; i<nrqs; i++) {
123: j = pa[i];
124: w1[j] += w2[j] + 2*w3[j];
125: msz += w1[j];
126: }
128: /* Determine the number of messages to expect, their lengths, from from-ids */
129: PetscGatherNumberOfMessages(comm,w2,w1,&nrqr);
130: PetscGatherMessageLengths(comm,nrqs,nrqr,w1,&onodes1,&olengths1);
132: /* Now post the Irecvs corresponding to these messages */
133: PetscPostIrecvInt(comm,tag1,nrqr,onodes1,olengths1,&rbuf,&r_waits1);
135: /* Allocate Memory for outgoing messages */
136: PetscMalloc4(size,&outdat,size,&ptr,msz,&tmp,size,&ctr);
137: PetscMemzero(outdat,size*sizeof(PetscInt*));
138: PetscMemzero(ptr,size*sizeof(PetscInt*));
139: {
140: PetscInt *iptr = tmp,ict = 0;
141: for (i=0; i<nrqs; i++) {
142: j = pa[i];
143: iptr += ict;
144: outdat[j] = iptr;
145: ict = w1[j];
146: }
147: }
149: /* Form the outgoing messages */
150: /*plug in the headers*/
151: for (i=0; i<nrqs; i++) {
152: j = pa[i];
153: outdat[j][0] = 0;
154: PetscMemzero(outdat[j]+1,2*w3[j]*sizeof(PetscInt));
155: ptr[j] = outdat[j] + 2*w3[j] + 1;
156: }
158: /* Memory for doing local proc's work*/
159: {
160: PetscCalloc5(imax,&table, imax,&data, imax,&isz, Mbs*imax,&d_p, (Mbs/PETSC_BITS_PER_BYTE+1)*imax,&t_p);
162: for (i=0; i<imax; i++) {
163: table[i] = t_p + (Mbs/PETSC_BITS_PER_BYTE+1)*i;
164: data[i] = d_p + (Mbs)*i;
165: }
166: }
168: /* Parse the IS and update local tables and the outgoing buf with the data*/
169: {
170: PetscInt n_i,*data_i,isz_i,*outdat_j,ctr_j;
171: PetscBT table_i;
173: for (i=0; i<imax; i++) {
174: PetscMemzero(ctr,size*sizeof(PetscInt));
175: n_i = n[i];
176: table_i = table[i];
177: idx_i = idx[i];
178: data_i = data[i];
179: isz_i = isz[i];
180: for (j=0; j<n_i; j++) { /* parse the indices of each IS */
181: row = idx_i[j];
182: PetscLayoutFindOwner(C->rmap,row*C->rmap->bs,&proc);
183: if (proc != rank) { /* copy to the outgoing buffer */
184: ctr[proc]++;
185: *ptr[proc] = row;
186: ptr[proc]++;
187: } else { /* Update the local table */
188: if (!PetscBTLookupSet(table_i,row)) data_i[isz_i++] = row;
189: }
190: }
191: /* Update the headers for the current IS */
192: for (j=0; j<size; j++) { /* Can Optimise this loop by using pa[] */
193: if ((ctr_j = ctr[j])) {
194: outdat_j = outdat[j];
195: k = ++outdat_j[0];
196: outdat_j[2*k] = ctr_j;
197: outdat_j[2*k-1] = i;
198: }
199: }
200: isz[i] = isz_i;
201: }
202: }
204: /* Now post the sends */
205: PetscMalloc1(nrqs+1,&s_waits1);
206: for (i=0; i<nrqs; ++i) {
207: j = pa[i];
208: MPI_Isend(outdat[j],w1[j],MPIU_INT,j,tag1,comm,s_waits1+i);
209: }
211: /* No longer need the original indices*/
212: for (i=0; i<imax; ++i) {
213: ISRestoreIndices(is[i],idx+i);
214: }
215: PetscFree2(*(PetscInt***)&idx,n);
217: PetscMalloc1(imax,&iscomms);
218: for (i=0; i<imax; ++i) {
219: PetscCommDuplicate(PetscObjectComm((PetscObject)is[i]),&iscomms[i],NULL);
220: ISDestroy(&is[i]);
221: }
223: /* Do Local work*/
224: MatIncreaseOverlap_MPIBAIJ_Local(C,imax,table,isz,data);
226: /* Receive messages*/
227: PetscMalloc1(nrqr+1,&recv_status);
228: if (nrqr) {MPI_Waitall(nrqr,r_waits1,recv_status);}
230: PetscMalloc1(nrqs+1,&s_status);
231: if (nrqs) {MPI_Waitall(nrqs,s_waits1,s_status);}
233: /* Phase 1 sends are complete - deallocate buffers */
234: PetscFree4(outdat,ptr,tmp,ctr);
235: PetscFree4(w1,w2,w3,w4);
237: PetscMalloc1(nrqr+1,&xdata);
238: PetscMalloc1(nrqr+1,&isz1);
239: MatIncreaseOverlap_MPIBAIJ_Receive(C,nrqr,rbuf,xdata,isz1);
240: PetscFree(rbuf[0]);
241: PetscFree(rbuf);
243: /* Send the data back*/
244: /* Do a global reduction to know the buffer space req for incoming messages*/
245: {
246: PetscMPIInt *rw1;
248: PetscCalloc1(size,&rw1);
250: for (i=0; i<nrqr; ++i) {
251: proc = recv_status[i].MPI_SOURCE;
252: if (proc != onodes1[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MPI_SOURCE mismatch");
253: rw1[proc] = isz1[i];
254: }
256: PetscFree(onodes1);
257: PetscFree(olengths1);
259: /* Determine the number of messages to expect, their lengths, from from-ids */
260: PetscGatherMessageLengths(comm,nrqr,nrqs,rw1,&onodes2,&olengths2);
261: PetscFree(rw1);
262: }
263: /* Now post the Irecvs corresponding to these messages */
264: PetscPostIrecvInt(comm,tag2,nrqs,onodes2,olengths2,&rbuf2,&r_waits2);
266: /* Now post the sends */
267: PetscMalloc1(nrqr+1,&s_waits2);
268: for (i=0; i<nrqr; ++i) {
269: j = recv_status[i].MPI_SOURCE;
270: MPI_Isend(xdata[i],isz1[i],MPIU_INT,j,tag2,comm,s_waits2+i);
271: }
273: /* receive work done on other processors*/
274: {
275: PetscMPIInt idex;
276: PetscInt is_no,ct1,max,*rbuf2_i,isz_i,*data_i,jmax;
277: PetscBT table_i;
278: MPI_Status *status2;
280: PetscMalloc1(PetscMax(nrqr,nrqs)+1,&status2);
281: for (i=0; i<nrqs; ++i) {
282: MPI_Waitany(nrqs,r_waits2,&idex,status2+i);
283: /* Process the message*/
284: rbuf2_i = rbuf2[idex];
285: ct1 = 2*rbuf2_i[0]+1;
286: jmax = rbuf2[idex][0];
287: for (j=1; j<=jmax; j++) {
288: max = rbuf2_i[2*j];
289: is_no = rbuf2_i[2*j-1];
290: isz_i = isz[is_no];
291: data_i = data[is_no];
292: table_i = table[is_no];
293: for (k=0; k<max; k++,ct1++) {
294: row = rbuf2_i[ct1];
295: if (!PetscBTLookupSet(table_i,row)) data_i[isz_i++] = row;
296: }
297: isz[is_no] = isz_i;
298: }
299: }
300: if (nrqr) {MPI_Waitall(nrqr,s_waits2,status2);}
301: PetscFree(status2);
302: }
304: for (i=0; i<imax; ++i) {
305: ISCreateGeneral(iscomms[i],isz[i],data[i],PETSC_COPY_VALUES,is+i);
306: PetscCommDestroy(&iscomms[i]);
307: }
309: PetscFree(iscomms);
310: PetscFree(onodes2);
311: PetscFree(olengths2);
313: PetscFree(pa);
314: PetscFree(rbuf2[0]);
315: PetscFree(rbuf2);
316: PetscFree(s_waits1);
317: PetscFree(r_waits1);
318: PetscFree(s_waits2);
319: PetscFree(r_waits2);
320: PetscFree5(table,data,isz,d_p,t_p);
321: PetscFree(s_status);
322: PetscFree(recv_status);
323: PetscFree(xdata[0]);
324: PetscFree(xdata);
325: PetscFree(isz1);
326: return(0);
327: }
329: /*
330: MatIncreaseOverlap_MPIBAIJ_Local - Called by MatincreaseOverlap, to do
331: the work on the local processor.
333: Inputs:
334: C - MAT_MPIBAIJ;
335: imax - total no of index sets processed at a time;
336: table - an array of char - size = Mbs bits.
338: Output:
339: isz - array containing the count of the solution elements corresponding
340: to each index set;
341: data - pointer to the solutions
342: */
343: static PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Local(Mat C,PetscInt imax,PetscBT *table,PetscInt *isz,PetscInt **data)
344: {
345: Mat_MPIBAIJ *c = (Mat_MPIBAIJ*)C->data;
346: Mat A = c->A,B = c->B;
347: Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data,*b = (Mat_SeqBAIJ*)B->data;
348: PetscInt start,end,val,max,rstart,cstart,*ai,*aj;
349: PetscInt *bi,*bj,*garray,i,j,k,row,*data_i,isz_i;
350: PetscBT table_i;
353: rstart = c->rstartbs;
354: cstart = c->cstartbs;
355: ai = a->i;
356: aj = a->j;
357: bi = b->i;
358: bj = b->j;
359: garray = c->garray;
362: for (i=0; i<imax; i++) {
363: data_i = data[i];
364: table_i = table[i];
365: isz_i = isz[i];
366: for (j=0,max=isz[i]; j<max; j++) {
367: row = data_i[j] - rstart;
368: start = ai[row];
369: end = ai[row+1];
370: for (k=start; k<end; k++) { /* Amat */
371: val = aj[k] + cstart;
372: if (!PetscBTLookupSet(table_i,val)) data_i[isz_i++] = val;
373: }
374: start = bi[row];
375: end = bi[row+1];
376: for (k=start; k<end; k++) { /* Bmat */
377: val = garray[bj[k]];
378: if (!PetscBTLookupSet(table_i,val)) data_i[isz_i++] = val;
379: }
380: }
381: isz[i] = isz_i;
382: }
383: return(0);
384: }
385: /*
386: MatIncreaseOverlap_MPIBAIJ_Receive - Process the recieved messages,
387: and return the output
389: Input:
390: C - the matrix
391: nrqr - no of messages being processed.
392: rbuf - an array of pointers to the recieved requests
394: Output:
395: xdata - array of messages to be sent back
396: isz1 - size of each message
398: For better efficiency perhaps we should malloc separately each xdata[i],
399: then if a remalloc is required we need only copy the data for that one row
400: rather than all previous rows as it is now where a single large chunck of
401: memory is used.
403: */
404: static PetscErrorCode MatIncreaseOverlap_MPIBAIJ_Receive(Mat C,PetscInt nrqr,PetscInt **rbuf,PetscInt **xdata,PetscInt * isz1)
405: {
406: Mat_MPIBAIJ *c = (Mat_MPIBAIJ*)C->data;
407: Mat A = c->A,B = c->B;
408: Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data,*b = (Mat_SeqBAIJ*)B->data;
410: PetscInt rstart,cstart,*ai,*aj,*bi,*bj,*garray,i,j,k;
411: PetscInt row,total_sz,ct,ct1,ct2,ct3,mem_estimate,oct2,l,start,end;
412: PetscInt val,max1,max2,Mbs,no_malloc =0,*tmp,new_estimate,ctr;
413: PetscInt *rbuf_i,kmax,rbuf_0;
414: PetscBT xtable;
417: Mbs = c->Mbs;
418: rstart = c->rstartbs;
419: cstart = c->cstartbs;
420: ai = a->i;
421: aj = a->j;
422: bi = b->i;
423: bj = b->j;
424: garray = c->garray;
427: for (i=0,ct=0,total_sz=0; i<nrqr; ++i) {
428: rbuf_i = rbuf[i];
429: rbuf_0 = rbuf_i[0];
430: ct += rbuf_0;
431: for (j=1; j<=rbuf_0; j++) total_sz += rbuf_i[2*j];
432: }
434: if (c->Mbs) max1 = ct*(a->nz +b->nz)/c->Mbs;
435: else max1 = 1;
436: mem_estimate = 3*((total_sz > max1 ? total_sz : max1)+1);
437: PetscMalloc1(mem_estimate,&xdata[0]);
438: ++no_malloc;
439: PetscBTCreate(Mbs,&xtable);
440: PetscMemzero(isz1,nrqr*sizeof(PetscInt));
442: ct3 = 0;
443: for (i=0; i<nrqr; i++) { /* for easch mesg from proc i */
444: rbuf_i = rbuf[i];
445: rbuf_0 = rbuf_i[0];
446: ct1 = 2*rbuf_0+1;
447: ct2 = ct1;
448: ct3 += ct1;
449: for (j=1; j<=rbuf_0; j++) { /* for each IS from proc i*/
450: PetscBTMemzero(Mbs,xtable);
451: oct2 = ct2;
452: kmax = rbuf_i[2*j];
453: for (k=0; k<kmax; k++,ct1++) {
454: row = rbuf_i[ct1];
455: if (!PetscBTLookupSet(xtable,row)) {
456: if (!(ct3 < mem_estimate)) {
457: new_estimate = (PetscInt)(1.5*mem_estimate)+1;
458: PetscMalloc1(new_estimate,&tmp);
459: PetscMemcpy(tmp,xdata[0],mem_estimate*sizeof(PetscInt));
460: PetscFree(xdata[0]);
461: xdata[0] = tmp;
462: mem_estimate = new_estimate; ++no_malloc;
463: for (ctr=1; ctr<=i; ctr++) xdata[ctr] = xdata[ctr-1] + isz1[ctr-1];
464: }
465: xdata[i][ct2++] = row;
466: ct3++;
467: }
468: }
469: for (k=oct2,max2=ct2; k<max2; k++) {
470: row = xdata[i][k] - rstart;
471: start = ai[row];
472: end = ai[row+1];
473: for (l=start; l<end; l++) {
474: val = aj[l] + cstart;
475: if (!PetscBTLookupSet(xtable,val)) {
476: if (!(ct3 < mem_estimate)) {
477: new_estimate = (PetscInt)(1.5*mem_estimate)+1;
478: PetscMalloc1(new_estimate,&tmp);
479: PetscMemcpy(tmp,xdata[0],mem_estimate*sizeof(PetscInt));
480: PetscFree(xdata[0]);
481: xdata[0] = tmp;
482: mem_estimate = new_estimate; ++no_malloc;
483: for (ctr=1; ctr<=i; ctr++) xdata[ctr] = xdata[ctr-1] + isz1[ctr-1];
484: }
485: xdata[i][ct2++] = val;
486: ct3++;
487: }
488: }
489: start = bi[row];
490: end = bi[row+1];
491: for (l=start; l<end; l++) {
492: val = garray[bj[l]];
493: if (!PetscBTLookupSet(xtable,val)) {
494: if (!(ct3 < mem_estimate)) {
495: new_estimate = (PetscInt)(1.5*mem_estimate)+1;
496: PetscMalloc1(new_estimate,&tmp);
497: PetscMemcpy(tmp,xdata[0],mem_estimate*sizeof(PetscInt));
498: PetscFree(xdata[0]);
499: xdata[0] = tmp;
500: mem_estimate = new_estimate; ++no_malloc;
501: for (ctr =1; ctr <=i; ctr++) xdata[ctr] = xdata[ctr-1] + isz1[ctr-1];
502: }
503: xdata[i][ct2++] = val;
504: ct3++;
505: }
506: }
507: }
508: /* Update the header*/
509: xdata[i][2*j] = ct2 - oct2; /* Undo the vector isz1 and use only a var*/
510: xdata[i][2*j-1] = rbuf_i[2*j-1];
511: }
512: xdata[i][0] = rbuf_0;
513: xdata[i+1] = xdata[i] + ct2;
514: isz1[i] = ct2; /* size of each message */
515: }
516: PetscBTDestroy(&xtable);
517: PetscInfo3(C,"Allocated %D bytes, required %D, no of mallocs = %D\n",mem_estimate,ct3,no_malloc);
518: return(0);
519: }
521: PetscErrorCode MatCreateSubMatrices_MPIBAIJ(Mat C,PetscInt ismax,const IS isrow[],const IS iscol[],MatReuse scall,Mat *submat[])
522: {
523: IS *isrow_block,*iscol_block;
524: Mat_MPIBAIJ *c = (Mat_MPIBAIJ*)C->data;
526: PetscInt nmax,nstages_local,nstages,i,pos,max_no,N=C->cmap->N,bs=C->rmap->bs;
527: Mat_SeqBAIJ *subc;
528: Mat_SubSppt *smat;
531: /* The compression and expansion should be avoided. Doesn't point
532: out errors, might change the indices, hence buggey */
533: PetscMalloc2(ismax+1,&isrow_block,ismax+1,&iscol_block);
534: ISCompressIndicesGeneral(N,C->rmap->n,bs,ismax,isrow,isrow_block);
535: ISCompressIndicesGeneral(N,C->cmap->n,bs,ismax,iscol,iscol_block);
537: /* Determine the number of stages through which submatrices are done */
538: if (!C->cmap->N) nmax=20*1000000/sizeof(PetscInt);
539: else nmax = 20*1000000 / (c->Nbs * sizeof(PetscInt));
540: if (!nmax) nmax = 1;
542: if (scall == MAT_INITIAL_MATRIX) {
543: nstages_local = ismax/nmax + ((ismax % nmax) ? 1 : 0); /* local nstages */
545: /* Make sure every processor loops through the nstages */
546: MPIU_Allreduce(&nstages_local,&nstages,1,MPIU_INT,MPI_MAX,PetscObjectComm((PetscObject)C));
548: /* Allocate memory to hold all the submatrices and dummy submatrices */
549: PetscCalloc1(ismax+nstages,submat);
550: } else { /* MAT_REUSE_MATRIX */
551: if (ismax) {
552: subc = (Mat_SeqBAIJ*)((*submat)[0]->data);
553: smat = subc->submatis1;
554: } else { /* (*submat)[0] is a dummy matrix */
555: smat = (Mat_SubSppt*)(*submat)[0]->data;
556: }
557: if (!smat) {
558: SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"MatCreateSubMatrices(...,MAT_REUSE_MATRIX,...) requires submat");
559: }
560: nstages = smat->nstages;
561: }
563: for (i=0,pos=0; i<nstages; i++) {
564: if (pos+nmax <= ismax) max_no = nmax;
565: else if (pos == ismax) max_no = 0;
566: else max_no = ismax-pos;
568: MatCreateSubMatrices_MPIBAIJ_local(C,max_no,isrow_block+pos,iscol_block+pos,scall,*submat+pos);
569: if (!max_no && scall == MAT_INITIAL_MATRIX) { /* submat[pos] is a dummy matrix */
570: smat = (Mat_SubSppt*)(*submat)[pos]->data;
571: smat->nstages = nstages;
572: }
573: pos += max_no;
574: }
576: if (scall == MAT_INITIAL_MATRIX && ismax) {
577: /* save nstages for reuse */
578: subc = (Mat_SeqBAIJ*)((*submat)[0]->data);
579: smat = subc->submatis1;
580: smat->nstages = nstages;
581: }
583: for (i=0; i<ismax; i++) {
584: ISDestroy(&isrow_block[i]);
585: ISDestroy(&iscol_block[i]);
586: }
587: PetscFree2(isrow_block,iscol_block);
588: return(0);
589: }
591: #if defined(PETSC_USE_CTABLE)
592: PetscErrorCode PetscGetProc(const PetscInt row, const PetscMPIInt size, const PetscInt proc_gnode[], PetscMPIInt *rank)
593: {
594: PetscInt nGlobalNd = proc_gnode[size];
595: PetscMPIInt fproc;
599: PetscMPIIntCast((PetscInt)(((float)row * (float)size / (float)nGlobalNd + 0.5)),&fproc);
600: if (fproc > size) fproc = size;
601: while (row < proc_gnode[fproc] || row >= proc_gnode[fproc+1]) {
602: if (row < proc_gnode[fproc]) fproc--;
603: else fproc++;
604: }
605: *rank = fproc;
606: return(0);
607: }
608: #endif
610: /* -------------------------------------------------------------------------*/
611: /* This code is used for BAIJ and SBAIJ matrices (unfortunate dependency) */
612: PetscErrorCode MatCreateSubMatrices_MPIBAIJ_local(Mat C,PetscInt ismax,const IS isrow[],const IS iscol[],MatReuse scall,Mat *submats)
613: {
614: Mat_MPIBAIJ *c = (Mat_MPIBAIJ*)C->data;
615: Mat A = c->A;
616: Mat_SeqBAIJ *a = (Mat_SeqBAIJ*)A->data,*b = (Mat_SeqBAIJ*)c->B->data,*subc;
617: const PetscInt **icol,**irow;
618: PetscInt *nrow,*ncol,start;
620: PetscMPIInt rank,size,tag0,tag2,tag3,tag4,*w1,*w2,*w3,*w4,nrqr;
621: PetscInt **sbuf1,**sbuf2,*sbuf2_i,i,j,k,l,ct1,ct2,**rbuf1,row,proc=-1;
622: PetscInt nrqs=0,msz,**ptr=NULL,*req_size=NULL,*ctr=NULL,*pa,*tmp=NULL,tcol;
623: PetscInt **rbuf3=NULL,*req_source1=NULL,*req_source2,**sbuf_aj,**rbuf2=NULL,max1,max2;
624: PetscInt **lens,is_no,ncols,*cols,mat_i,*mat_j,tmp2,jmax;
625: #if defined(PETSC_USE_CTABLE)
626: PetscTable *cmap,cmap_i=NULL,*rmap,rmap_i;
627: #else
628: PetscInt **cmap,*cmap_i=NULL,**rmap,*rmap_i;
629: #endif
630: const PetscInt *irow_i,*icol_i;
631: PetscInt ctr_j,*sbuf1_j,*sbuf_aj_i,*rbuf1_i,kmax,*lens_i;
632: MPI_Request *s_waits1,*r_waits1,*s_waits2,*r_waits2,*r_waits3;
633: MPI_Request *r_waits4,*s_waits3,*s_waits4;
634: MPI_Status *r_status1,*r_status2,*s_status1,*s_status3,*s_status2;
635: MPI_Status *r_status3,*r_status4,*s_status4;
636: MPI_Comm comm;
637: PetscScalar **rbuf4,*rbuf4_i=NULL,**sbuf_aa,*vals,*mat_a=NULL,*imat_a,*sbuf_aa_i;
638: PetscMPIInt *onodes1,*olengths1,end;
639: PetscInt **row2proc,*row2proc_i,*imat_ilen,*imat_j,*imat_i;
640: Mat_SubSppt *smat_i;
641: PetscBool *issorted,colflag,iscsorted=PETSC_TRUE;
642: PetscInt *sbuf1_i,*rbuf2_i,*rbuf3_i,ilen;
643: PetscInt bs=C->rmap->bs,bs2=c->bs2,rstart = c->rstartbs;
644: PetscBool ijonly=c->ijonly; /* private flag indicates only matrix data structures are requested */
645: PetscInt nzA,nzB,*a_i=a->i,*b_i=b->i,*a_j = a->j,*b_j = b->j,ctmp,imark,*cworkA,*cworkB;
646: PetscScalar *vworkA=NULL,*vworkB=NULL,*a_a = a->a,*b_a = b->a;
647: PetscInt cstart = c->cstartbs,*bmap = c->garray;
648: PetscBool *allrows,*allcolumns;
651: PetscObjectGetComm((PetscObject)C,&comm);
652: size = c->size;
653: rank = c->rank;
655: PetscMalloc5(ismax,&row2proc,ismax,&cmap,ismax,&rmap,ismax+1,&allcolumns,ismax,&allrows);
656: PetscMalloc5(ismax,(PetscInt***)&irow,ismax,(PetscInt***)&icol,ismax,&nrow,ismax,&ncol,ismax,&issorted);
658: for (i=0; i<ismax; i++) {
659: ISSorted(iscol[i],&issorted[i]);
660: if (!issorted[i]) iscsorted = issorted[i]; /* columns are not sorted! */
661: ISSorted(isrow[i],&issorted[i]);
663: /* Check for special case: allcolumns */
664: ISIdentity(iscol[i],&colflag);
665: ISGetLocalSize(iscol[i],&ncol[i]);
667: if (colflag && ncol[i] == c->Nbs) {
668: allcolumns[i] = PETSC_TRUE;
669: icol[i] = NULL;
670: } else {
671: allcolumns[i] = PETSC_FALSE;
672: ISGetIndices(iscol[i],&icol[i]);
673: }
675: /* Check for special case: allrows */
676: ISIdentity(isrow[i],&colflag);
677: ISGetLocalSize(isrow[i],&nrow[i]);
678: if (colflag && nrow[i] == c->Mbs) {
679: allrows[i] = PETSC_TRUE;
680: irow[i] = NULL;
681: } else {
682: allrows[i] = PETSC_FALSE;
683: ISGetIndices(isrow[i],&irow[i]);
684: }
685: }
687: if (scall == MAT_REUSE_MATRIX) {
688: /* Assumes new rows are same length as the old rows */
689: for (i=0; i<ismax; i++) {
690: subc = (Mat_SeqBAIJ*)(submats[i]->data);
691: if (subc->mbs != nrow[i] || subc->nbs != ncol[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong size");
693: /* Initial matrix as if empty */
694: PetscMemzero(subc->ilen,subc->mbs*sizeof(PetscInt));
696: /* Initial matrix as if empty */
697: submats[i]->factortype = C->factortype;
699: smat_i = subc->submatis1;
701: nrqs = smat_i->nrqs;
702: nrqr = smat_i->nrqr;
703: rbuf1 = smat_i->rbuf1;
704: rbuf2 = smat_i->rbuf2;
705: rbuf3 = smat_i->rbuf3;
706: req_source2 = smat_i->req_source2;
708: sbuf1 = smat_i->sbuf1;
709: sbuf2 = smat_i->sbuf2;
710: ptr = smat_i->ptr;
711: tmp = smat_i->tmp;
712: ctr = smat_i->ctr;
714: pa = smat_i->pa;
715: req_size = smat_i->req_size;
716: req_source1 = smat_i->req_source1;
718: allcolumns[i] = smat_i->allcolumns;
719: allrows[i] = smat_i->allrows;
720: row2proc[i] = smat_i->row2proc;
721: rmap[i] = smat_i->rmap;
722: cmap[i] = smat_i->cmap;
723: }
725: if (!ismax){ /* Get dummy submatrices and retrieve struct submatis1 */
726: if (!submats[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"submats are null, cannot reuse");
727: smat_i = (Mat_SubSppt*)submats[0]->data;
729: nrqs = smat_i->nrqs;
730: nrqr = smat_i->nrqr;
731: rbuf1 = smat_i->rbuf1;
732: rbuf2 = smat_i->rbuf2;
733: rbuf3 = smat_i->rbuf3;
734: req_source2 = smat_i->req_source2;
736: sbuf1 = smat_i->sbuf1;
737: sbuf2 = smat_i->sbuf2;
738: ptr = smat_i->ptr;
739: tmp = smat_i->tmp;
740: ctr = smat_i->ctr;
742: pa = smat_i->pa;
743: req_size = smat_i->req_size;
744: req_source1 = smat_i->req_source1;
746: allcolumns[0] = PETSC_FALSE;
747: }
748: } else { /* scall == MAT_INITIAL_MATRIX */
749: /* Get some new tags to keep the communication clean */
750: PetscObjectGetNewTag((PetscObject)C,&tag2);
751: PetscObjectGetNewTag((PetscObject)C,&tag3);
753: /* evaluate communication - mesg to who, length of mesg, and buffer space
754: required. Based on this, buffers are allocated, and data copied into them*/
755: PetscCalloc4(size,&w1,size,&w2,size,&w3,size,&w4); /* mesg size, initialize work vectors */
757: for (i=0; i<ismax; i++) {
758: jmax = nrow[i];
759: irow_i = irow[i];
761: PetscMalloc1(jmax,&row2proc_i);
762: row2proc[i] = row2proc_i;
764: if (issorted[i]) proc = 0;
765: for (j=0; j<jmax; j++) {
766: if (!issorted[i]) proc = 0;
767: if (allrows[i]) row = j;
768: else row = irow_i[j];
770: while (row >= c->rangebs[proc+1]) proc++;
771: w4[proc]++;
772: row2proc_i[j] = proc; /* map row index to proc */
773: }
774: for (j=0; j<size; j++) {
775: if (w4[j]) { w1[j] += w4[j]; w3[j]++; w4[j] = 0;}
776: }
777: }
779: nrqs = 0; /* no of outgoing messages */
780: msz = 0; /* total mesg length (for all procs) */
781: w1[rank] = 0; /* no mesg sent to self */
782: w3[rank] = 0;
783: for (i=0; i<size; i++) {
784: if (w1[i]) { w2[i] = 1; nrqs++;} /* there exists a message to proc i */
785: }
786: PetscMalloc1(nrqs+1,&pa); /*(proc -array)*/
787: for (i=0,j=0; i<size; i++) {
788: if (w1[i]) { pa[j] = i; j++; }
789: }
791: /* Each message would have a header = 1 + 2*(no of IS) + data */
792: for (i=0; i<nrqs; i++) {
793: j = pa[i];
794: w1[j] += w2[j] + 2* w3[j];
795: msz += w1[j];
796: }
797: PetscInfo2(0,"Number of outgoing messages %D Total message length %D\n",nrqs,msz);
799: /* Determine the number of messages to expect, their lengths, from from-ids */
800: PetscGatherNumberOfMessages(comm,w2,w1,&nrqr);
801: PetscGatherMessageLengths(comm,nrqs,nrqr,w1,&onodes1,&olengths1);
803: /* Now post the Irecvs corresponding to these messages */
804: tag0 = ((PetscObject)C)->tag;
805: PetscPostIrecvInt(comm,tag0,nrqr,onodes1,olengths1,&rbuf1,&r_waits1);
807: PetscFree(onodes1);
808: PetscFree(olengths1);
810: /* Allocate Memory for outgoing messages */
811: PetscMalloc4(size,&sbuf1,size,&ptr,2*msz,&tmp,size,&ctr);
812: PetscMemzero(sbuf1,size*sizeof(PetscInt*));
813: PetscMemzero(ptr,size*sizeof(PetscInt*));
815: {
816: PetscInt *iptr = tmp;
817: k = 0;
818: for (i=0; i<nrqs; i++) {
819: j = pa[i];
820: iptr += k;
821: sbuf1[j] = iptr;
822: k = w1[j];
823: }
824: }
826: /* Form the outgoing messages. Initialize the header space */
827: for (i=0; i<nrqs; i++) {
828: j = pa[i];
829: sbuf1[j][0] = 0;
830: PetscMemzero(sbuf1[j]+1,2*w3[j]*sizeof(PetscInt));
831: ptr[j] = sbuf1[j] + 2*w3[j] + 1;
832: }
834: /* Parse the isrow and copy data into outbuf */
835: for (i=0; i<ismax; i++) {
836: row2proc_i = row2proc[i];
837: PetscMemzero(ctr,size*sizeof(PetscInt));
838: irow_i = irow[i];
839: jmax = nrow[i];
840: for (j=0; j<jmax; j++) { /* parse the indices of each IS */
841: proc = row2proc_i[j];
842: if (allrows[i]) row = j;
843: else row = irow_i[j];
845: if (proc != rank) { /* copy to the outgoing buf*/
846: ctr[proc]++;
847: *ptr[proc] = row;
848: ptr[proc]++;
849: }
850: }
851: /* Update the headers for the current IS */
852: for (j=0; j<size; j++) { /* Can Optimise this loop too */
853: if ((ctr_j = ctr[j])) {
854: sbuf1_j = sbuf1[j];
855: k = ++sbuf1_j[0];
856: sbuf1_j[2*k] = ctr_j;
857: sbuf1_j[2*k-1] = i;
858: }
859: }
860: }
862: /* Now post the sends */
863: PetscMalloc1(nrqs+1,&s_waits1);
864: for (i=0; i<nrqs; ++i) {
865: j = pa[i];
866: MPI_Isend(sbuf1[j],w1[j],MPIU_INT,j,tag0,comm,s_waits1+i);
867: }
869: /* Post Receives to capture the buffer size */
870: PetscMalloc1(nrqs+1,&r_waits2);
871: PetscMalloc3(nrqs+1,&req_source2,nrqs+1,&rbuf2,nrqs+1,&rbuf3);
872: rbuf2[0] = tmp + msz;
873: for (i=1; i<nrqs; ++i) {
874: rbuf2[i] = rbuf2[i-1]+w1[pa[i-1]];
875: }
876: for (i=0; i<nrqs; ++i) {
877: j = pa[i];
878: MPI_Irecv(rbuf2[i],w1[j],MPIU_INT,j,tag2,comm,r_waits2+i);
879: }
881: /* Send to other procs the buf size they should allocate */
882: /* Receive messages*/
883: PetscMalloc1(nrqr+1,&s_waits2);
884: PetscMalloc1(nrqr+1,&r_status1);
885: PetscMalloc3(nrqr,&sbuf2,nrqr,&req_size,nrqr,&req_source1);
887: MPI_Waitall(nrqr,r_waits1,r_status1);
888: for (i=0; i<nrqr; ++i) {
889: req_size[i] = 0;
890: rbuf1_i = rbuf1[i];
891: start = 2*rbuf1_i[0] + 1;
892: MPI_Get_count(r_status1+i,MPIU_INT,&end);
893: PetscMalloc1(end+1,&sbuf2[i]);
894: sbuf2_i = sbuf2[i];
895: for (j=start; j<end; j++) {
896: row = rbuf1_i[j] - rstart;
897: ncols = a_i[row+1] - a_i[row] + b_i[row+1] - b_i[row];
898: sbuf2_i[j] = ncols;
899: req_size[i] += ncols;
900: }
901: req_source1[i] = r_status1[i].MPI_SOURCE;
902: /* form the header */
903: sbuf2_i[0] = req_size[i];
904: for (j=1; j<start; j++) sbuf2_i[j] = rbuf1_i[j];
906: MPI_Isend(sbuf2_i,end,MPIU_INT,req_source1[i],tag2,comm,s_waits2+i);
907: }
909: PetscFree(r_status1);
910: PetscFree(r_waits1);
911: PetscFree4(w1,w2,w3,w4);
913: /* Receive messages*/
914: PetscMalloc1(nrqs+1,&r_waits3);
915: PetscMalloc1(nrqs+1,&r_status2);
917: MPI_Waitall(nrqs,r_waits2,r_status2);
918: for (i=0; i<nrqs; ++i) {
919: PetscMalloc1(rbuf2[i][0]+1,&rbuf3[i]);
920: req_source2[i] = r_status2[i].MPI_SOURCE;
921: MPI_Irecv(rbuf3[i],rbuf2[i][0],MPIU_INT,req_source2[i],tag3,comm,r_waits3+i);
922: }
923: PetscFree(r_status2);
924: PetscFree(r_waits2);
926: /* Wait on sends1 and sends2 */
927: PetscMalloc1(nrqs+1,&s_status1);
928: PetscMalloc1(nrqr+1,&s_status2);
930: if (nrqs) {MPI_Waitall(nrqs,s_waits1,s_status1);}
931: if (nrqr) {MPI_Waitall(nrqr,s_waits2,s_status2);}
932: PetscFree(s_status1);
933: PetscFree(s_status2);
934: PetscFree(s_waits1);
935: PetscFree(s_waits2);
937: /* Now allocate sending buffers for a->j, and send them off */
938: PetscMalloc1(nrqr+1,&sbuf_aj);
939: for (i=0,j=0; i<nrqr; i++) j += req_size[i];
940: PetscMalloc1(j+1,&sbuf_aj[0]);
941: for (i=1; i<nrqr; i++) sbuf_aj[i] = sbuf_aj[i-1] + req_size[i-1];
943: PetscMalloc1(nrqr+1,&s_waits3);
944: {
946: for (i=0; i<nrqr; i++) {
947: rbuf1_i = rbuf1[i];
948: sbuf_aj_i = sbuf_aj[i];
949: ct1 = 2*rbuf1_i[0] + 1;
950: ct2 = 0;
951: for (j=1,max1=rbuf1_i[0]; j<=max1; j++) {
952: kmax = rbuf1[i][2*j];
953: for (k=0; k<kmax; k++,ct1++) {
954: row = rbuf1_i[ct1] - rstart;
955: nzA = a_i[row+1] - a_i[row]; nzB = b_i[row+1] - b_i[row];
956: ncols = nzA + nzB;
957: cworkA = a_j + a_i[row]; cworkB = b_j + b_i[row];
959: /* load the column indices for this row into cols */
960: cols = sbuf_aj_i + ct2;
961: for (l=0; l<nzB; l++) {
962: if ((ctmp = bmap[cworkB[l]]) < cstart) cols[l] = ctmp;
963: else break;
964: }
965: imark = l;
966: for (l=0; l<nzA; l++) {cols[imark+l] = cstart + cworkA[l];}
967: for (l=imark; l<nzB; l++) cols[nzA+l] = bmap[cworkB[l]];
968: ct2 += ncols;
969: }
970: }
971: MPI_Isend(sbuf_aj_i,req_size[i],MPIU_INT,req_source1[i],tag3,comm,s_waits3+i);
972: }
973: }
974: PetscMalloc2(nrqs+1,&r_status3,nrqr+1,&s_status3);
976: /* create col map: global col of C -> local col of submatrices */
977: #if defined(PETSC_USE_CTABLE)
978: for (i=0; i<ismax; i++) {
979: if (!allcolumns[i]) {
980: PetscTableCreate(ncol[i]+1,c->Nbs+1,&cmap[i]);
982: jmax = ncol[i];
983: icol_i = icol[i];
984: cmap_i = cmap[i];
985: for (j=0; j<jmax; j++) {
986: PetscTableAdd(cmap[i],icol_i[j]+1,j+1,INSERT_VALUES);
987: }
988: } else cmap[i] = NULL;
989: }
990: #else
991: for (i=0; i<ismax; i++) {
992: if (!allcolumns[i]) {
993: PetscCalloc1(c->Nbs,&cmap[i]);
994: jmax = ncol[i];
995: icol_i = icol[i];
996: cmap_i = cmap[i];
997: for (j=0; j<jmax; j++) cmap_i[icol_i[j]] = j+1;
998: } else cmap[i] = NULL;
999: }
1000: #endif
1002: /* Create lens which is required for MatCreate... */
1003: for (i=0,j=0; i<ismax; i++) j += nrow[i];
1004: PetscMalloc1(ismax,&lens);
1006: if (ismax) {
1007: PetscCalloc1(j,&lens[0]);
1008: }
1009: for (i=1; i<ismax; i++) lens[i] = lens[i-1] + nrow[i-1];
1011: /* Update lens from local data */
1012: for (i=0; i<ismax; i++) {
1013: row2proc_i = row2proc[i];
1014: jmax = nrow[i];
1015: if (!allcolumns[i]) cmap_i = cmap[i];
1016: irow_i = irow[i];
1017: lens_i = lens[i];
1018: for (j=0; j<jmax; j++) {
1019: if (allrows[i]) row = j;
1020: else row = irow_i[j]; /* global blocked row of C */
1022: proc = row2proc_i[j];
1023: if (proc == rank) {
1024: /* Get indices from matA and then from matB */
1025: #if defined(PETSC_USE_CTABLE)
1026: PetscInt tt;
1027: #endif
1028: row = row - rstart;
1029: nzA = a_i[row+1] - a_i[row];
1030: nzB = b_i[row+1] - b_i[row];
1031: cworkA = a_j + a_i[row];
1032: cworkB = b_j + b_i[row];
1034: if (!allcolumns[i]) {
1035: #if defined(PETSC_USE_CTABLE)
1036: for (k=0; k<nzA; k++) {
1037: PetscTableFind(cmap_i,cstart+cworkA[k]+1,&tt);
1038: if (tt) lens_i[j]++;
1039: }
1040: for (k=0; k<nzB; k++) {
1041: PetscTableFind(cmap_i,bmap[cworkB[k]]+1,&tt);
1042: if (tt) lens_i[j]++;
1043: }
1045: #else
1046: for (k=0; k<nzA; k++) {
1047: if (cmap_i[cstart + cworkA[k]]) lens_i[j]++;
1048: }
1049: for (k=0; k<nzB; k++) {
1050: if (cmap_i[bmap[cworkB[k]]]) lens_i[j]++;
1051: }
1052: #endif
1053: } else { /* allcolumns */
1054: lens_i[j] = nzA + nzB;
1055: }
1056: }
1057: }
1058: }
1060: /* Create row map: global row of C -> local row of submatrices */
1061: for (i=0; i<ismax; i++) {
1062: if (!allrows[i]) {
1063: #if defined(PETSC_USE_CTABLE)
1064: PetscTableCreate(nrow[i]+1,c->Mbs+1,&rmap[i]);
1065: irow_i = irow[i];
1066: jmax = nrow[i];
1067: for (j=0; j<jmax; j++) {
1068: if (allrows[i]) {
1069: PetscTableAdd(rmap[i],j+1,j+1,INSERT_VALUES);
1070: } else {
1071: PetscTableAdd(rmap[i],irow_i[j]+1,j+1,INSERT_VALUES);
1072: }
1073: }
1074: #else
1075: PetscCalloc1(c->Mbs,&rmap[i]);
1076: rmap_i = rmap[i];
1077: irow_i = irow[i];
1078: jmax = nrow[i];
1079: for (j=0; j<jmax; j++) {
1080: if (allrows[i]) rmap_i[j] = j;
1081: else rmap_i[irow_i[j]] = j;
1082: }
1083: #endif
1084: } else rmap[i] = NULL;
1085: }
1087: /* Update lens from offproc data */
1088: {
1089: PetscInt *rbuf2_i,*rbuf3_i,*sbuf1_i;
1091: MPI_Waitall(nrqs,r_waits3,r_status3);
1092: for (tmp2=0; tmp2<nrqs; tmp2++) {
1093: sbuf1_i = sbuf1[pa[tmp2]];
1094: jmax = sbuf1_i[0];
1095: ct1 = 2*jmax+1;
1096: ct2 = 0;
1097: rbuf2_i = rbuf2[tmp2];
1098: rbuf3_i = rbuf3[tmp2];
1099: for (j=1; j<=jmax; j++) {
1100: is_no = sbuf1_i[2*j-1];
1101: max1 = sbuf1_i[2*j];
1102: lens_i = lens[is_no];
1103: if (!allcolumns[is_no]) cmap_i = cmap[is_no];
1104: rmap_i = rmap[is_no];
1105: for (k=0; k<max1; k++,ct1++) {
1106: if (allrows[is_no]) {
1107: row = sbuf1_i[ct1];
1108: } else {
1109: #if defined(PETSC_USE_CTABLE)
1110: PetscTableFind(rmap_i,sbuf1_i[ct1]+1,&row);
1111: row--;
1112: if (row < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"row not found in table");
1113: #else
1114: row = rmap_i[sbuf1_i[ct1]]; /* the val in the new matrix to be */
1115: #endif
1116: }
1117: max2 = rbuf2_i[ct1];
1118: for (l=0; l<max2; l++,ct2++) {
1119: if (!allcolumns[is_no]) {
1120: #if defined(PETSC_USE_CTABLE)
1121: PetscTableFind(cmap_i,rbuf3_i[ct2]+1,&tcol);
1122: #else
1123: tcol = cmap_i[rbuf3_i[ct2]];
1124: #endif
1125: if (tcol) lens_i[row]++;
1126: } else { /* allcolumns */
1127: lens_i[row]++; /* lens_i[row] += max2 ? */
1128: }
1129: }
1130: }
1131: }
1132: }
1133: }
1134: PetscFree(r_waits3);
1135: if (nrqr) {MPI_Waitall(nrqr,s_waits3,s_status3);}
1136: PetscFree2(r_status3,s_status3);
1137: PetscFree(s_waits3);
1139: /* Create the submatrices */
1140: for (i=0; i<ismax; i++) {
1141: PetscInt bs_tmp;
1142: if (ijonly) bs_tmp = 1;
1143: else bs_tmp = bs;
1145: MatCreate(PETSC_COMM_SELF,submats+i);
1146: MatSetSizes(submats[i],nrow[i]*bs_tmp,ncol[i]*bs_tmp,PETSC_DETERMINE,PETSC_DETERMINE);
1148: MatSetType(submats[i],((PetscObject)A)->type_name);
1149: MatSeqBAIJSetPreallocation(submats[i],bs_tmp,0,lens[i]);
1150: MatSeqSBAIJSetPreallocation(submats[i],bs_tmp,0,lens[i]); /* this subroutine is used by SBAIJ routines */
1152: /* create struct Mat_SubSppt and attached it to submat */
1153: PetscNew(&smat_i);
1154: subc = (Mat_SeqBAIJ*)submats[i]->data;
1155: subc->submatis1 = smat_i;
1157: smat_i->destroy = submats[i]->ops->destroy;
1158: submats[i]->ops->destroy = MatDestroySubMatrix_SeqBAIJ;
1159: submats[i]->factortype = C->factortype;
1161: smat_i->id = i;
1162: smat_i->nrqs = nrqs;
1163: smat_i->nrqr = nrqr;
1164: smat_i->rbuf1 = rbuf1;
1165: smat_i->rbuf2 = rbuf2;
1166: smat_i->rbuf3 = rbuf3;
1167: smat_i->sbuf2 = sbuf2;
1168: smat_i->req_source2 = req_source2;
1170: smat_i->sbuf1 = sbuf1;
1171: smat_i->ptr = ptr;
1172: smat_i->tmp = tmp;
1173: smat_i->ctr = ctr;
1175: smat_i->pa = pa;
1176: smat_i->req_size = req_size;
1177: smat_i->req_source1 = req_source1;
1179: smat_i->allcolumns = allcolumns[i];
1180: smat_i->allrows = allrows[i];
1181: smat_i->singleis = PETSC_FALSE;
1182: smat_i->row2proc = row2proc[i];
1183: smat_i->rmap = rmap[i];
1184: smat_i->cmap = cmap[i];
1185: }
1187: if (!ismax) { /* Create dummy submats[0] for reuse struct subc */
1188: MatCreate(PETSC_COMM_SELF,&submats[0]);
1189: MatSetSizes(submats[0],0,0,PETSC_DETERMINE,PETSC_DETERMINE);
1190: MatSetType(submats[0],MATDUMMY);
1192: /* create struct Mat_SubSppt and attached it to submat */
1193: PetscNewLog(submats[0],&smat_i);
1194: submats[0]->data = (void*)smat_i;
1196: smat_i->destroy = submats[0]->ops->destroy;
1197: submats[0]->ops->destroy = MatDestroySubMatrix_Dummy;
1198: submats[0]->factortype = C->factortype;
1200: smat_i->id = 0;
1201: smat_i->nrqs = nrqs;
1202: smat_i->nrqr = nrqr;
1203: smat_i->rbuf1 = rbuf1;
1204: smat_i->rbuf2 = rbuf2;
1205: smat_i->rbuf3 = rbuf3;
1206: smat_i->sbuf2 = sbuf2;
1207: smat_i->req_source2 = req_source2;
1209: smat_i->sbuf1 = sbuf1;
1210: smat_i->ptr = ptr;
1211: smat_i->tmp = tmp;
1212: smat_i->ctr = ctr;
1214: smat_i->pa = pa;
1215: smat_i->req_size = req_size;
1216: smat_i->req_source1 = req_source1;
1218: smat_i->allcolumns = PETSC_FALSE;
1219: smat_i->singleis = PETSC_FALSE;
1220: smat_i->row2proc = NULL;
1221: smat_i->rmap = NULL;
1222: smat_i->cmap = NULL;
1223: }
1226: if (ismax) {PetscFree(lens[0]);}
1227: PetscFree(lens);
1228: PetscFree(sbuf_aj[0]);
1229: PetscFree(sbuf_aj);
1231: } /* endof scall == MAT_INITIAL_MATRIX */
1233: /* Post recv matrix values */
1234: if (!ijonly) {
1235: PetscObjectGetNewTag((PetscObject)C,&tag4);
1236: PetscMalloc1(nrqs+1,&rbuf4);
1237: PetscMalloc1(nrqs+1,&r_waits4);
1238: PetscMalloc1(nrqs+1,&r_status4);
1239: PetscMalloc1(nrqr+1,&s_status4);
1240: for (i=0; i<nrqs; ++i) {
1241: PetscMalloc1(rbuf2[i][0]*bs2,&rbuf4[i]);
1242: MPI_Irecv(rbuf4[i],rbuf2[i][0]*bs2,MPIU_SCALAR,req_source2[i],tag4,comm,r_waits4+i);
1243: }
1245: /* Allocate sending buffers for a->a, and send them off */
1246: PetscMalloc1(nrqr+1,&sbuf_aa);
1247: for (i=0,j=0; i<nrqr; i++) j += req_size[i];
1249: PetscMalloc1((j+1)*bs2,&sbuf_aa[0]);
1250: for (i=1; i<nrqr; i++) sbuf_aa[i] = sbuf_aa[i-1] + req_size[i-1]*bs2;
1252: PetscMalloc1(nrqr+1,&s_waits4);
1254: for (i=0; i<nrqr; i++) {
1255: rbuf1_i = rbuf1[i];
1256: sbuf_aa_i = sbuf_aa[i];
1257: ct1 = 2*rbuf1_i[0]+1;
1258: ct2 = 0;
1259: for (j=1,max1=rbuf1_i[0]; j<=max1; j++) {
1260: kmax = rbuf1_i[2*j];
1261: for (k=0; k<kmax; k++,ct1++) {
1262: row = rbuf1_i[ct1] - rstart;
1263: nzA = a_i[row+1] - a_i[row];
1264: nzB = b_i[row+1] - b_i[row];
1265: ncols = nzA + nzB;
1266: cworkB = b_j + b_i[row];
1267: vworkA = a_a + a_i[row]*bs2;
1268: vworkB = b_a + b_i[row]*bs2;
1270: /* load the column values for this row into vals*/
1271: vals = sbuf_aa_i+ct2*bs2;
1272: for (l=0; l<nzB; l++) {
1273: if ((bmap[cworkB[l]]) < cstart) {
1274: PetscMemcpy(vals+l*bs2,vworkB+l*bs2,bs2*sizeof(MatScalar));
1275: } else break;
1276: }
1277: imark = l;
1278: for (l=0; l<nzA; l++) {
1279: PetscMemcpy(vals+(imark+l)*bs2,vworkA+l*bs2,bs2*sizeof(MatScalar));
1280: }
1281: for (l=imark; l<nzB; l++) {
1282: PetscMemcpy(vals+(nzA+l)*bs2,vworkB+l*bs2,bs2*sizeof(MatScalar));
1283: }
1285: ct2 += ncols;
1286: }
1287: }
1288: MPI_Isend(sbuf_aa_i,req_size[i]*bs2,MPIU_SCALAR,req_source1[i],tag4,comm,s_waits4+i);
1289: }
1290: }
1292: /* Assemble the matrices */
1293: /* First assemble the local rows */
1294: for (i=0; i<ismax; i++) {
1295: row2proc_i = row2proc[i];
1296: subc = (Mat_SeqBAIJ*)submats[i]->data;
1297: imat_ilen = subc->ilen;
1298: imat_j = subc->j;
1299: imat_i = subc->i;
1300: imat_a = subc->a;
1302: if (!allcolumns[i]) cmap_i = cmap[i];
1303: rmap_i = rmap[i];
1304: irow_i = irow[i];
1305: jmax = nrow[i];
1306: for (j=0; j<jmax; j++) {
1307: if (allrows[i]) row = j;
1308: else row = irow_i[j];
1309: proc = row2proc_i[j];
1311: if (proc == rank) {
1313: row = row - rstart;
1314: nzA = a_i[row+1] - a_i[row];
1315: nzB = b_i[row+1] - b_i[row];
1316: cworkA = a_j + a_i[row];
1317: cworkB = b_j + b_i[row];
1318: if (!ijonly) {
1319: vworkA = a_a + a_i[row]*bs2;
1320: vworkB = b_a + b_i[row]*bs2;
1321: }
1323: if (allrows[i]) {
1324: row = row+rstart;
1325: } else {
1326: #if defined(PETSC_USE_CTABLE)
1327: PetscTableFind(rmap_i,row+rstart+1,&row);
1328: row--;
1330: if (row < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"row not found in table");
1331: #else
1332: row = rmap_i[row + rstart];
1333: #endif
1334: }
1335: mat_i = imat_i[row];
1336: if (!ijonly) mat_a = imat_a + mat_i*bs2;
1337: mat_j = imat_j + mat_i;
1338: ilen = imat_ilen[row];
1340: /* load the column indices for this row into cols*/
1341: if (!allcolumns[i]) {
1342: for (l=0; l<nzB; l++) {
1343: if ((ctmp = bmap[cworkB[l]]) < cstart) {
1344: #if defined(PETSC_USE_CTABLE)
1345: PetscTableFind(cmap_i,ctmp+1,&tcol);
1346: if (tcol) {
1347: #else
1348: if ((tcol = cmap_i[ctmp])) {
1349: #endif
1350: *mat_j++ = tcol - 1;
1351: PetscMemcpy(mat_a,vworkB+l*bs2,bs2*sizeof(MatScalar));
1352: mat_a += bs2;
1353: ilen++;
1354: }
1355: } else break;
1356: }
1357: imark = l;
1358: for (l=0; l<nzA; l++) {
1359: #if defined(PETSC_USE_CTABLE)
1360: PetscTableFind(cmap_i,cstart+cworkA[l]+1,&tcol);
1361: if (tcol) {
1362: #else
1363: if ((tcol = cmap_i[cstart + cworkA[l]])) {
1364: #endif
1365: *mat_j++ = tcol - 1;
1366: if (!ijonly) {
1367: PetscMemcpy(mat_a,vworkA+l*bs2,bs2*sizeof(MatScalar));
1368: mat_a += bs2;
1369: }
1370: ilen++;
1371: }
1372: }
1373: for (l=imark; l<nzB; l++) {
1374: #if defined(PETSC_USE_CTABLE)
1375: PetscTableFind(cmap_i,bmap[cworkB[l]]+1,&tcol);
1376: if (tcol) {
1377: #else
1378: if ((tcol = cmap_i[bmap[cworkB[l]]])) {
1379: #endif
1380: *mat_j++ = tcol - 1;
1381: if (!ijonly) {
1382: PetscMemcpy(mat_a,vworkB+l*bs2,bs2*sizeof(MatScalar));
1383: mat_a += bs2;
1384: }
1385: ilen++;
1386: }
1387: }
1388: } else { /* allcolumns */
1389: for (l=0; l<nzB; l++) {
1390: if ((ctmp = bmap[cworkB[l]]) < cstart) {
1391: *mat_j++ = ctmp;
1392: PetscMemcpy(mat_a,vworkB+l*bs2,bs2*sizeof(MatScalar));
1393: mat_a += bs2;
1394: ilen++;
1395: } else break;
1396: }
1397: imark = l;
1398: for (l=0; l<nzA; l++) {
1399: *mat_j++ = cstart+cworkA[l];
1400: if (!ijonly) {
1401: PetscMemcpy(mat_a,vworkA+l*bs2,bs2*sizeof(MatScalar));
1402: mat_a += bs2;
1403: }
1404: ilen++;
1405: }
1406: for (l=imark; l<nzB; l++) {
1407: *mat_j++ = bmap[cworkB[l]];
1408: if (!ijonly) {
1409: PetscMemcpy(mat_a,vworkB+l*bs2,bs2*sizeof(MatScalar));
1410: mat_a += bs2;
1411: }
1412: ilen++;
1413: }
1414: }
1415: imat_ilen[row] = ilen;
1416: }
1417: }
1418: }
1420: /* Now assemble the off proc rows */
1421: if (!ijonly) {
1422: MPI_Waitall(nrqs,r_waits4,r_status4);
1423: }
1424: for (tmp2=0; tmp2<nrqs; tmp2++) {
1425: sbuf1_i = sbuf1[pa[tmp2]];
1426: jmax = sbuf1_i[0];
1427: ct1 = 2*jmax + 1;
1428: ct2 = 0;
1429: rbuf2_i = rbuf2[tmp2];
1430: rbuf3_i = rbuf3[tmp2];
1431: if (!ijonly) rbuf4_i = rbuf4[tmp2];
1432: for (j=1; j<=jmax; j++) {
1433: is_no = sbuf1_i[2*j-1];
1434: rmap_i = rmap[is_no];
1435: if (!allcolumns[is_no]) cmap_i = cmap[is_no];
1436: subc = (Mat_SeqBAIJ*)submats[is_no]->data;
1437: imat_ilen = subc->ilen;
1438: imat_j = subc->j;
1439: imat_i = subc->i;
1440: if (!ijonly) imat_a = subc->a;
1441: max1 = sbuf1_i[2*j];
1442: for (k=0; k<max1; k++,ct1++) { /* for each recved block row */
1443: row = sbuf1_i[ct1];
1445: if (allrows[is_no]) {
1446: row = sbuf1_i[ct1];
1447: } else {
1448: #if defined(PETSC_USE_CTABLE)
1449: PetscTableFind(rmap_i,row+1,&row);
1450: row--;
1451: if (row < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"row not found in table");
1452: #else
1453: row = rmap_i[row];
1454: #endif
1455: }
1456: ilen = imat_ilen[row];
1457: mat_i = imat_i[row];
1458: if (!ijonly) mat_a = imat_a + mat_i*bs2;
1459: mat_j = imat_j + mat_i;
1460: max2 = rbuf2_i[ct1];
1461: if (!allcolumns[is_no]) {
1462: for (l=0; l<max2; l++,ct2++) {
1463: #if defined(PETSC_USE_CTABLE)
1464: PetscTableFind(cmap_i,rbuf3_i[ct2]+1,&tcol);
1465: #else
1466: tcol = cmap_i[rbuf3_i[ct2]];
1467: #endif
1468: if (tcol) {
1469: *mat_j++ = tcol - 1;
1470: if (!ijonly) {
1471: PetscMemcpy(mat_a,rbuf4_i+ct2*bs2,bs2*sizeof(MatScalar));
1472: mat_a += bs2;
1473: }
1474: ilen++;
1475: }
1476: }
1477: } else { /* allcolumns */
1478: for (l=0; l<max2; l++,ct2++) {
1479: *mat_j++ = rbuf3_i[ct2]; /* same global column index of C */
1480: if (!ijonly) {
1481: PetscMemcpy(mat_a,rbuf4_i+ct2*bs2,bs2*sizeof(MatScalar));
1482: mat_a += bs2;
1483: }
1484: ilen++;
1485: }
1486: }
1487: imat_ilen[row] = ilen;
1488: }
1489: }
1490: }
1492: if (!iscsorted) { /* sort column indices of the rows */
1493: MatScalar *work;
1495: PetscMalloc1(bs2,&work);
1496: for (i=0; i<ismax; i++) {
1497: subc = (Mat_SeqBAIJ*)submats[i]->data;
1498: imat_ilen = subc->ilen;
1499: imat_j = subc->j;
1500: imat_i = subc->i;
1501: if (!ijonly) imat_a = subc->a;
1502: if (allcolumns[i]) continue;
1504: jmax = nrow[i];
1505: for (j=0; j<jmax; j++) {
1506: mat_i = imat_i[j];
1507: mat_j = imat_j + mat_i;
1508: ilen = imat_ilen[j];
1509: if (ijonly) {
1510: PetscSortInt(ilen,mat_j);
1511: } else {
1512: mat_a = imat_a + mat_i*bs2;
1513: PetscSortIntWithDataArray(ilen,mat_j,mat_a,bs2*sizeof(MatScalar),work);
1514: }
1515: }
1516: }
1517: PetscFree(work);
1518: }
1520: if (!ijonly) {
1521: PetscFree(r_status4);
1522: PetscFree(r_waits4);
1523: if (nrqr) {MPI_Waitall(nrqr,s_waits4,s_status4);}
1524: PetscFree(s_waits4);
1525: PetscFree(s_status4);
1526: }
1528: /* Restore the indices */
1529: for (i=0; i<ismax; i++) {
1530: if (!allrows[i]) {
1531: ISRestoreIndices(isrow[i],irow+i);
1532: }
1533: if (!allcolumns[i]) {
1534: ISRestoreIndices(iscol[i],icol+i);
1535: }
1536: }
1538: for (i=0; i<ismax; i++) {
1539: MatAssemblyBegin(submats[i],MAT_FINAL_ASSEMBLY);
1540: MatAssemblyEnd(submats[i],MAT_FINAL_ASSEMBLY);
1541: }
1543: PetscFree5(*(PetscInt***)&irow,*(PetscInt***)&icol,nrow,ncol,issorted);
1544: PetscFree5(row2proc,cmap,rmap,allcolumns,allrows);
1546: if (!ijonly) {
1547: PetscFree(sbuf_aa[0]);
1548: PetscFree(sbuf_aa);
1550: for (i=0; i<nrqs; ++i) {
1551: PetscFree(rbuf4[i]);
1552: }
1553: PetscFree(rbuf4);
1554: }
1555: c->ijonly = PETSC_FALSE; /* set back to the default */
1556: return(0);
1557: }