xref: /petsc/src/mat/impls/aij/seq/matmatmult.c (revision b28d80bd5a6d3afe9b178cd0deb1e5452c16d5fc)
1be1d678aSKris Buschelman 
2d50806bdSBarry Smith /*
32499ec78SHong Zhang   Defines matrix-matrix product routines for pairs of SeqAIJ matrices
4d50806bdSBarry Smith           C = A * B
5d50806bdSBarry Smith */
6d50806bdSBarry Smith 
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 <petscbt.h>
10c6db04a5SJed Brown #include <../src/mat/impls/dense/seq/dense.h> /*I "petscmat.h" I*/
116284ec50SHong Zhang 
12ec01deb9SMatthew Knepley EXTERN_C_BEGIN
136284ec50SHong Zhang #undef __FUNCT__
145c66b693SKris Buschelman #define __FUNCT__ "MatMatMult_SeqAIJ_SeqAIJ"
1538baddfdSBarry Smith PetscErrorCode MatMatMult_SeqAIJ_SeqAIJ(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1638baddfdSBarry Smith {
17dfbe8321SBarry Smith   PetscErrorCode ierr;
185c66b693SKris Buschelman 
195c66b693SKris Buschelman   PetscFunctionBegin;
2026be0446SHong Zhang   if (scall == MAT_INITIAL_MATRIX){
2126be0446SHong Zhang     ierr = MatMatMultSymbolic_SeqAIJ_SeqAIJ(A,B,fill,C);CHKERRQ(ierr);
2226be0446SHong Zhang   }
2326be0446SHong Zhang   ierr = MatMatMultNumeric_SeqAIJ_SeqAIJ(A,B,*C);CHKERRQ(ierr);
245c66b693SKris Buschelman   PetscFunctionReturn(0);
255c66b693SKris Buschelman }
26ec01deb9SMatthew Knepley EXTERN_C_END
271c24bd37SHong Zhang 
2826be0446SHong Zhang #undef __FUNCT__
2926be0446SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_SeqAIJ_SeqAIJ"
30dfbe8321SBarry Smith PetscErrorCode MatMatMultSymbolic_SeqAIJ_SeqAIJ(Mat A,Mat B,PetscReal fill,Mat *C)
3158c24d83SHong Zhang {
32dfbe8321SBarry Smith   PetscErrorCode     ierr;
33a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
3458c24d83SHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)B->data,*c;
3538baddfdSBarry Smith   PetscInt           *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bjj,*ci,*cj;
36d0f46423SBarry Smith   PetscInt           am=A->rmap->N,bn=B->cmap->N,bm=B->rmap->N;
3738baddfdSBarry Smith   PetscInt           i,j,anzi,brow,bnzj,cnzi,nlnk,*lnk,nspacedouble=0;
3858c24d83SHong Zhang   MatScalar          *ca;
39be0fcf8dSHong Zhang   PetscBT            lnkbt;
407212da7cSHong Zhang   PetscReal          afill;
4158c24d83SHong Zhang 
4258c24d83SHong Zhang   PetscFunctionBegin;
4358c24d83SHong Zhang   /* Allocate ci array, arrays for fill computation and */
4458c24d83SHong Zhang   /* free space for accumulating nonzero column info */
4538baddfdSBarry Smith   ierr = PetscMalloc(((am+1)+1)*sizeof(PetscInt),&ci);CHKERRQ(ierr);
4658c24d83SHong Zhang   ci[0] = 0;
4758c24d83SHong Zhang 
48be0fcf8dSHong Zhang   /* create and initialize a linked list */
49be0fcf8dSHong Zhang   nlnk = bn+1;
50be0fcf8dSHong Zhang   ierr = PetscLLCreate(bn,bn,nlnk,lnk,lnkbt);CHKERRQ(ierr);
5158c24d83SHong Zhang 
52c5db241fSHong Zhang   /* Initial FreeSpace size is fill*(nnz(A)+nnz(B)) */
53a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+bi[bm])),&free_space);CHKERRQ(ierr);
5458c24d83SHong Zhang   current_space = free_space;
5558c24d83SHong Zhang 
5658c24d83SHong Zhang   /* Determine symbolic info for each row of the product: */
5758c24d83SHong Zhang   for (i=0;i<am;i++) {
5858c24d83SHong Zhang     anzi = ai[i+1] - ai[i];
5958c24d83SHong Zhang     cnzi = 0;
602d09714cSHong Zhang     j    = anzi;
612d09714cSHong Zhang     aj   = a->j + ai[i];
622d09714cSHong Zhang     while (j){/* assume cols are almost in increasing order, starting from its end saves computation */
632d09714cSHong Zhang       j--;
642d09714cSHong Zhang       brow = *(aj + j);
6558c24d83SHong Zhang       bnzj = bi[brow+1] - bi[brow];
6658c24d83SHong Zhang       bjj  = bj + bi[brow];
671c239cc6SHong Zhang       /* add non-zero cols of B into the sorted linked list lnk */
68be0fcf8dSHong Zhang       ierr = PetscLLAdd(bnzj,bjj,bn,nlnk,lnk,lnkbt);CHKERRQ(ierr);
691c239cc6SHong Zhang       cnzi += nlnk;
7058c24d83SHong Zhang     }
7158c24d83SHong Zhang 
7258c24d83SHong Zhang     /* If free space is not available, make more free space */
7358c24d83SHong Zhang     /* Double the amount of total space in the list */
7458c24d83SHong Zhang     if (current_space->local_remaining<cnzi) {
754238b7adSHong Zhang       ierr = PetscFreeSpaceGet(cnzi+current_space->total_array_size,&current_space);CHKERRQ(ierr);
76c5db241fSHong Zhang       nspacedouble++;
7758c24d83SHong Zhang     }
7858c24d83SHong Zhang 
79c5db241fSHong Zhang     /* Copy data into free space, then initialize lnk */
80be0fcf8dSHong Zhang     ierr = PetscLLClean(bn,bn,cnzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
81c5db241fSHong Zhang     current_space->array           += cnzi;
8258c24d83SHong Zhang     current_space->local_used      += cnzi;
8358c24d83SHong Zhang     current_space->local_remaining -= cnzi;
8458c24d83SHong Zhang 
8558c24d83SHong Zhang     ci[i+1] = ci[i] + cnzi;
8658c24d83SHong Zhang   }
8758c24d83SHong Zhang 
8858c24d83SHong Zhang   /* Column indices are in the list of free space */
8958c24d83SHong Zhang   /* Allocate space for cj, initialize cj, and */
9058c24d83SHong Zhang   /* destroy list of free space and other temporary array(s) */
9138baddfdSBarry Smith   ierr = PetscMalloc((ci[am]+1)*sizeof(PetscInt),&cj);CHKERRQ(ierr);
92a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,cj);CHKERRQ(ierr);
93be0fcf8dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
9458c24d83SHong Zhang 
9558c24d83SHong Zhang   /* Allocate space for ca */
9658c24d83SHong Zhang   ierr = PetscMalloc((ci[am]+1)*sizeof(MatScalar),&ca);CHKERRQ(ierr);
9758c24d83SHong Zhang   ierr = PetscMemzero(ca,(ci[am]+1)*sizeof(MatScalar));CHKERRQ(ierr);
9858c24d83SHong Zhang 
9926be0446SHong Zhang   /* put together the new symbolic matrix */
1007adad957SLisandro Dalcin   ierr = MatCreateSeqAIJWithArrays(((PetscObject)A)->comm,am,bn,ci,cj,ca,C);CHKERRQ(ierr);
10158c24d83SHong Zhang 
10258c24d83SHong Zhang   /* MatCreateSeqAIJWithArrays flags matrix so PETSc doesn't free the user's arrays. */
10358c24d83SHong Zhang   /* These are PETSc arrays, so change flags so arrays can be deleted by PETSc */
10458c24d83SHong Zhang   c = (Mat_SeqAIJ *)((*C)->data);
105e6b907acSBarry Smith   c->free_a   = PETSC_TRUE;
106e6b907acSBarry Smith   c->free_ij  = PETSC_TRUE;
10758c24d83SHong Zhang   c->nonew    = 0;
10858c24d83SHong Zhang 
1097212da7cSHong Zhang   /* set MatInfo */
1107212da7cSHong Zhang   afill = (PetscReal)ci[am]/(ai[am]+bi[bm]) + 1.e-5;
111f2b054eeSHong Zhang   if (afill < 1.0) afill = 1.0;
1127212da7cSHong Zhang   c->maxnz                     = ci[am];
1137212da7cSHong Zhang   c->nz                        = ci[am];
1147212da7cSHong Zhang   (*C)->info.mallocs           = nspacedouble;
1157212da7cSHong Zhang   (*C)->info.fill_ratio_given  = fill;
1167212da7cSHong Zhang   (*C)->info.fill_ratio_needed = afill;
1177212da7cSHong Zhang 
1187212da7cSHong Zhang #if defined(PETSC_USE_INFO)
1197212da7cSHong Zhang   if (ci[am]) {
120f2b054eeSHong Zhang     ierr = PetscInfo3((*C),"Reallocs %D; Fill ratio: given %G needed %G.\n",nspacedouble,fill,afill);CHKERRQ(ierr);
121f2b054eeSHong Zhang     ierr = PetscInfo1((*C),"Use MatMatMult(A,B,MatReuse,%G,&C) for best performance.;\n",afill);CHKERRQ(ierr);
122f2b054eeSHong Zhang   } else {
123f2b054eeSHong Zhang     ierr = PetscInfo((*C),"Empty matrix product\n");CHKERRQ(ierr);
124be0fcf8dSHong Zhang   }
125f2b054eeSHong Zhang #endif
12658c24d83SHong Zhang   PetscFunctionReturn(0);
12758c24d83SHong Zhang }
128d50806bdSBarry Smith 
1295c66b693SKris Buschelman 
13026be0446SHong Zhang #undef __FUNCT__
13126be0446SHong Zhang #define __FUNCT__ "MatMatMultNumeric_SeqAIJ_SeqAIJ"
132dfbe8321SBarry Smith PetscErrorCode MatMatMultNumeric_SeqAIJ_SeqAIJ(Mat A,Mat B,Mat C)
133d50806bdSBarry Smith {
134dfbe8321SBarry Smith   PetscErrorCode ierr;
135d13dce4bSSatish Balay   PetscLogDouble flops=0.0;
136d50806bdSBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ *)A->data;
137d50806bdSBarry Smith   Mat_SeqAIJ     *b = (Mat_SeqAIJ *)B->data;
138d50806bdSBarry Smith   Mat_SeqAIJ     *c = (Mat_SeqAIJ *)C->data;
13938baddfdSBarry Smith   PetscInt       *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bjj,*ci=c->i,*cj=c->j;
140d0f46423SBarry Smith   PetscInt       am=A->rmap->N,cm=C->rmap->N;
141c124e916SHong Zhang   PetscInt       i,j,k,anzi,bnzi,cnzi,brow,nextb;
142c124e916SHong Zhang   MatScalar      *aa=a->a,*ba=b->a,*baj,*ca=c->a;
143d50806bdSBarry Smith 
144d50806bdSBarry Smith   PetscFunctionBegin;
145c124e916SHong Zhang   /* clean old values in C */
146c124e916SHong Zhang   ierr = PetscMemzero(ca,ci[cm]*sizeof(MatScalar));CHKERRQ(ierr);
147d50806bdSBarry Smith   /* Traverse A row-wise. */
148d50806bdSBarry Smith   /* Build the ith row in C by summing over nonzero columns in A, */
149d50806bdSBarry Smith   /* the rows of B corresponding to nonzeros of A. */
150d50806bdSBarry Smith   for (i=0;i<am;i++) {
151d50806bdSBarry Smith     anzi = ai[i+1] - ai[i];
152d50806bdSBarry Smith     for (j=0;j<anzi;j++) {
153d50806bdSBarry Smith       brow = *aj++;
154d50806bdSBarry Smith       bnzi = bi[brow+1] - bi[brow];
155d50806bdSBarry Smith       bjj  = bj + bi[brow];
156d50806bdSBarry Smith       baj  = ba + bi[brow];
157c124e916SHong Zhang       nextb = 0;
158c124e916SHong Zhang       for (k=0; nextb<bnzi; k++) {
159c124e916SHong Zhang         if (cj[k] == bjj[nextb]){ /* ccol == bcol */
160c124e916SHong Zhang           ca[k] += (*aa)*baj[nextb++];
161c124e916SHong Zhang         }
162d50806bdSBarry Smith       }
163d50806bdSBarry Smith       flops += 2*bnzi;
164d50806bdSBarry Smith       aa++;
165d50806bdSBarry Smith     }
166d50806bdSBarry Smith     cnzi = ci[i+1] - ci[i];
167d50806bdSBarry Smith     ca += cnzi;
168d50806bdSBarry Smith     cj += cnzi;
169d50806bdSBarry Smith   }
170716bacf3SKris Buschelman   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171716bacf3SKris Buschelman   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
172716bacf3SKris Buschelman 
173d50806bdSBarry Smith   ierr = PetscLogFlops(flops);CHKERRQ(ierr);
174d50806bdSBarry Smith   PetscFunctionReturn(0);
175d50806bdSBarry Smith }
176bc011b1eSHong Zhang 
177bc011b1eSHong Zhang #undef __FUNCT__
178bc011b1eSHong Zhang #define __FUNCT__ "MatMatMultTranspose_SeqAIJ_SeqAIJ"
1795df89d91SHong Zhang PetscErrorCode MatMatMultTranspose_SeqAIJ_SeqAIJ(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1805df89d91SHong Zhang {
181bc011b1eSHong Zhang   PetscErrorCode ierr;
182bc011b1eSHong Zhang 
183bc011b1eSHong Zhang   PetscFunctionBegin;
184bc011b1eSHong Zhang   if (scall == MAT_INITIAL_MATRIX){
185bc011b1eSHong Zhang     ierr = MatMatMultTransposeSymbolic_SeqAIJ_SeqAIJ(A,B,fill,C);CHKERRQ(ierr);
186bc011b1eSHong Zhang   }
187bc011b1eSHong Zhang   ierr = MatMatMultTransposeNumeric_SeqAIJ_SeqAIJ(A,B,*C);CHKERRQ(ierr);
188bc011b1eSHong Zhang   PetscFunctionReturn(0);
189bc011b1eSHong Zhang }
190bc011b1eSHong Zhang 
191bc011b1eSHong Zhang #undef __FUNCT__
1922128a86cSHong Zhang #define __FUNCT__ "PetscContainerDestroy_Mat_MatMatMultTrans"
1932128a86cSHong Zhang PetscErrorCode PetscContainerDestroy_Mat_MatMatMultTrans(void *ptr)
1942128a86cSHong Zhang {
1952128a86cSHong Zhang   PetscErrorCode      ierr;
1962128a86cSHong Zhang   Mat_MatMatMultTrans *multtrans=(Mat_MatMatMultTrans*)ptr;
1972128a86cSHong Zhang 
1982128a86cSHong Zhang   PetscFunctionBegin;
1992128a86cSHong Zhang   ierr = MatMultTransposeColoringDestroy(&multtrans->matcoloring);CHKERRQ(ierr);
2002128a86cSHong Zhang   ierr = MatDestroy(&multtrans->Bt_den);CHKERRQ(ierr);
2012128a86cSHong Zhang   ierr = MatDestroy(&multtrans->ABt_den);CHKERRQ(ierr);
2022128a86cSHong Zhang   ierr = PetscFree(multtrans);CHKERRQ(ierr);
2032128a86cSHong Zhang   PetscFunctionReturn(0);
2042128a86cSHong Zhang }
2052128a86cSHong Zhang 
2062128a86cSHong Zhang #undef __FUNCT__
2072128a86cSHong Zhang #define __FUNCT__ "MatDestroy_SeqAIJ_MatMatMultTrans"
2082128a86cSHong Zhang PetscErrorCode MatDestroy_SeqAIJ_MatMatMultTrans(Mat A)
2092128a86cSHong Zhang {
2102128a86cSHong Zhang   PetscErrorCode      ierr;
2112128a86cSHong Zhang   PetscContainer      container;
2122128a86cSHong Zhang   Mat_MatMatMultTrans *multtrans=PETSC_NULL;
2132128a86cSHong Zhang 
2142128a86cSHong Zhang   PetscFunctionBegin;
2152128a86cSHong Zhang   ierr = PetscObjectQuery((PetscObject)A,"Mat_MatMatMultTrans",(PetscObject *)&container);CHKERRQ(ierr);
2162128a86cSHong Zhang   if (!container) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Container does not exit");
2172128a86cSHong Zhang   ierr = PetscContainerGetPointer(container,(void **)&multtrans);CHKERRQ(ierr);
2182128a86cSHong Zhang   A->ops->destroy   = multtrans->destroy;
2192128a86cSHong Zhang   if (A->ops->destroy) {
2202128a86cSHong Zhang     ierr = (*A->ops->destroy)(A);CHKERRQ(ierr);
2212128a86cSHong Zhang   }
2222128a86cSHong Zhang   ierr = PetscObjectCompose((PetscObject)A,"Mat_MatMatMultTrans",0);CHKERRQ(ierr);
2232128a86cSHong Zhang   PetscFunctionReturn(0);
2242128a86cSHong Zhang }
2252128a86cSHong Zhang 
2262128a86cSHong Zhang #undef __FUNCT__
227bc011b1eSHong Zhang #define __FUNCT__ "MatMatMultTransposeSymbolic_SeqAIJ_SeqAIJ"
228bc011b1eSHong Zhang PetscErrorCode MatMatMultTransposeSymbolic_SeqAIJ_SeqAIJ(Mat A,Mat B,PetscReal fill,Mat *C)
229bc011b1eSHong Zhang {
2305df89d91SHong Zhang   PetscErrorCode      ierr;
23181d82fe4SHong Zhang   Mat                 Bt;
23281d82fe4SHong Zhang   PetscInt            *bti,*btj;
2332128a86cSHong Zhang   Mat_MatMatMultTrans *multtrans;
2342128a86cSHong Zhang   PetscContainer      container;
23581d82fe4SHong Zhang 
23681d82fe4SHong Zhang   PetscFunctionBegin;
23781d82fe4SHong Zhang    /* create symbolic Bt */
23881d82fe4SHong Zhang   ierr = MatGetSymbolicTranspose_SeqAIJ(B,&bti,&btj);CHKERRQ(ierr);
23981d82fe4SHong Zhang   ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,B->cmap->n,B->rmap->n,bti,btj,PETSC_NULL,&Bt);CHKERRQ(ierr);
24081d82fe4SHong Zhang 
24181d82fe4SHong Zhang   /* get symbolic C=A*Bt */
24281d82fe4SHong Zhang   ierr = MatMatMultSymbolic_SeqAIJ_SeqAIJ(A,Bt,fill,C);CHKERRQ(ierr);
24381d82fe4SHong Zhang 
24481d82fe4SHong Zhang   /* clean up */
24581d82fe4SHong Zhang   ierr = MatDestroy(&Bt);CHKERRQ(ierr);
24681d82fe4SHong Zhang   ierr = MatRestoreSymbolicTranspose_SeqAIJ(B,&bti,&btj);CHKERRQ(ierr);
24781d82fe4SHong Zhang 
2482128a86cSHong Zhang   /* create a supporting struct for reuse intermidiate dense matrices with matcoloring */
2492128a86cSHong Zhang   ierr = PetscNew(Mat_MatMatMultTrans,&multtrans);CHKERRQ(ierr);
2502128a86cSHong Zhang 
2512128a86cSHong Zhang   /* attach the supporting struct to C */
2522128a86cSHong Zhang   ierr = PetscContainerCreate(PETSC_COMM_SELF,&container);CHKERRQ(ierr);
2532128a86cSHong Zhang   ierr = PetscContainerSetPointer(container,multtrans);CHKERRQ(ierr);
2542128a86cSHong Zhang   ierr = PetscContainerSetUserDestroy(container,PetscContainerDestroy_Mat_MatMatMultTrans);CHKERRQ(ierr);
2552128a86cSHong Zhang   ierr = PetscObjectCompose((PetscObject)(*C),"Mat_MatMatMultTrans",(PetscObject)container);CHKERRQ(ierr);
2562128a86cSHong Zhang   ierr = PetscContainerDestroy(&container);CHKERRQ(ierr);
2572128a86cSHong Zhang 
2582128a86cSHong Zhang   multtrans->usecoloring = PETSC_FALSE;
2592128a86cSHong Zhang   multtrans->destroy = (*C)->ops->destroy;
2602128a86cSHong Zhang   (*C)->ops->destroy = MatDestroy_SeqAIJ_MatMatMultTrans;
2612128a86cSHong Zhang 
2622128a86cSHong Zhang   ierr = PetscOptionsGetBool(PETSC_NULL,"-matmulttrans_color",&multtrans->usecoloring,PETSC_NULL);CHKERRQ(ierr);
2632128a86cSHong Zhang   if (multtrans->usecoloring){
2642128a86cSHong Zhang     /* Create MatMultTransposeColoring from symbolic C=A*B^T */
2652128a86cSHong Zhang     MatMultTransposeColoring  matcoloring;
2662128a86cSHong Zhang     ISColoring                iscoloring;
2672128a86cSHong Zhang     Mat                       Bt_dense,C_dense;
268e8354b3eSHong Zhang     PetscLogDouble            t0,tf,etime0=0.0,etime1=0.0;
269e8354b3eSHong Zhang 
270e8354b3eSHong Zhang     ierr = PetscGetTime(&t0);CHKERRQ(ierr);
2712128a86cSHong Zhang     ierr = MatGetColoring(*C,MATCOLORINGSL,&iscoloring);CHKERRQ(ierr);
2722128a86cSHong Zhang     ierr = MatMultTransposeColoringCreate(*C,iscoloring,&matcoloring);CHKERRQ(ierr);
2732128a86cSHong Zhang     multtrans->matcoloring = matcoloring;
2742128a86cSHong Zhang     ierr = ISColoringDestroy(&iscoloring);CHKERRQ(ierr);
275e8354b3eSHong Zhang     ierr = PetscGetTime(&tf);CHKERRQ(ierr);
276e8354b3eSHong Zhang     etime0 += tf - t0;
2772128a86cSHong Zhang 
278e8354b3eSHong Zhang     ierr = PetscGetTime(&t0);CHKERRQ(ierr);
2792128a86cSHong Zhang     /* Create Bt_dense and C_dense = A*Bt_dense */
2802128a86cSHong Zhang     ierr = MatCreate(PETSC_COMM_SELF,&Bt_dense);CHKERRQ(ierr);
2812128a86cSHong Zhang     ierr = MatSetSizes(Bt_dense,A->cmap->n,matcoloring->ncolors,A->cmap->n,matcoloring->ncolors);CHKERRQ(ierr);
2822128a86cSHong Zhang     ierr = MatSetType(Bt_dense,MATSEQDENSE);CHKERRQ(ierr);
2832128a86cSHong Zhang     ierr = MatSeqDenseSetPreallocation(Bt_dense,PETSC_NULL);CHKERRQ(ierr);
2842128a86cSHong Zhang     Bt_dense->assembled = PETSC_TRUE;
2852128a86cSHong Zhang     multtrans->Bt_den = Bt_dense;
2862128a86cSHong Zhang 
2872128a86cSHong Zhang     ierr = MatCreate(PETSC_COMM_SELF,&C_dense);CHKERRQ(ierr);
2882128a86cSHong Zhang     ierr = MatSetSizes(C_dense,A->rmap->n,matcoloring->ncolors,A->rmap->n,matcoloring->ncolors);CHKERRQ(ierr);
2892128a86cSHong Zhang     ierr = MatSetType(C_dense,MATSEQDENSE);CHKERRQ(ierr);
2902128a86cSHong Zhang     ierr = MatSeqDenseSetPreallocation(C_dense,PETSC_NULL);CHKERRQ(ierr);
2912128a86cSHong Zhang     Bt_dense->assembled = PETSC_TRUE;
2922128a86cSHong Zhang     multtrans->ABt_den = C_dense;
293e8354b3eSHong Zhang     ierr = PetscGetTime(&tf);CHKERRQ(ierr);
294e8354b3eSHong Zhang     etime1 += tf - t0;
295e8354b3eSHong Zhang     printf("MatMultTransColorCreate %g, MatDenseCreate %g\n",etime0,etime1);
2962128a86cSHong Zhang   }
2972128a86cSHong Zhang 
29881d82fe4SHong Zhang #if defined(INEFFICIENT_ALGORITHM)
29981d82fe4SHong Zhang   /* The algorithm below computes am*bm sparse inner-product - inefficient! It will be deleted later. */
3001fa1209cSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
3011fa1209cSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)B->data,*c;
3021fa1209cSHong Zhang   PetscInt           *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*ci,*cj,*acol,*bcol;
3031fa1209cSHong Zhang   PetscInt           am=A->rmap->N,bm=B->rmap->N;
3041fa1209cSHong Zhang   PetscInt           i,j,anzi,bnzj,cnzi,nlnk,*lnk,nspacedouble=0,ka,kb,index[1];
3051fa1209cSHong Zhang   MatScalar          *ca;
3061fa1209cSHong Zhang   PetscBT            lnkbt;
3071fa1209cSHong Zhang   PetscReal          afill;
3085df89d91SHong Zhang 
3091fa1209cSHong Zhang   /* Allocate row pointer array ci  */
3101fa1209cSHong Zhang   ierr = PetscMalloc(((am+1)+1)*sizeof(PetscInt),&ci);CHKERRQ(ierr);
3111fa1209cSHong Zhang   ci[0] = 0;
3121fa1209cSHong Zhang 
3131fa1209cSHong Zhang   /* Create and initialize a linked list for C columns */
3141fa1209cSHong Zhang   nlnk = bm+1;
3151fa1209cSHong Zhang   ierr = PetscLLCreate(bm,bm,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3161fa1209cSHong Zhang 
3171fa1209cSHong Zhang   /* Initial FreeSpace with size fill*(nnz(A)+nnz(B)) */
3181fa1209cSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+bi[bm])),&free_space);CHKERRQ(ierr);
3191fa1209cSHong Zhang   current_space = free_space;
3201fa1209cSHong Zhang 
3211fa1209cSHong Zhang   /* Determine symbolic info for each row of the product A*B^T: */
3221fa1209cSHong Zhang   for (i=0; i<am; i++) {
3231fa1209cSHong Zhang     anzi = ai[i+1] - ai[i];
3241fa1209cSHong Zhang     cnzi = 0;
3251fa1209cSHong Zhang     acol = aj + ai[i];
3261fa1209cSHong Zhang     for (j=0; j<bm; j++){
3271fa1209cSHong Zhang       bnzj = bi[j+1] - bi[j];
3281fa1209cSHong Zhang       bcol= bj + bi[j];
32981d82fe4SHong Zhang       /* sparse inner-product c(i,j)=A[i,:]*B[j,:]^T */
3301fa1209cSHong Zhang       ka = 0; kb = 0;
3311fa1209cSHong Zhang       while (ka < anzi && kb < bnzj){
33281d82fe4SHong Zhang         while (acol[ka] < bcol[kb] && ka < anzi) ka++;
33381d82fe4SHong Zhang         if (ka == anzi) break;
33481d82fe4SHong Zhang         while (acol[ka] > bcol[kb] && kb < bnzj) kb++;
33581d82fe4SHong Zhang         if (kb == bnzj) break;
3361fa1209cSHong Zhang         if (acol[ka] == bcol[kb]){ /* add nonzero c(i,j) to lnk */
3371fa1209cSHong Zhang           index[0] = j;
3381fa1209cSHong Zhang           ierr = PetscLLAdd(1,index,bm,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3391fa1209cSHong Zhang           cnzi++;
3401fa1209cSHong Zhang           break;
3411fa1209cSHong Zhang         }
3421fa1209cSHong Zhang       }
3431fa1209cSHong Zhang     }
3441fa1209cSHong Zhang 
3451fa1209cSHong Zhang     /* If free space is not available, make more free space */
3461fa1209cSHong Zhang     /* Double the amount of total space in the list */
3471fa1209cSHong Zhang     if (current_space->local_remaining<cnzi) {
3481fa1209cSHong Zhang       ierr = PetscFreeSpaceGet(cnzi+current_space->total_array_size,&current_space);CHKERRQ(ierr);
3491fa1209cSHong Zhang       nspacedouble++;
3501fa1209cSHong Zhang     }
3511fa1209cSHong Zhang 
3521fa1209cSHong Zhang     /* Copy data into free space, then initialize lnk */
3531fa1209cSHong Zhang     ierr = PetscLLClean(bm,bm,cnzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3541fa1209cSHong Zhang     current_space->array           += cnzi;
3551fa1209cSHong Zhang     current_space->local_used      += cnzi;
3561fa1209cSHong Zhang     current_space->local_remaining -= cnzi;
3571fa1209cSHong Zhang 
3581fa1209cSHong Zhang     ci[i+1] = ci[i] + cnzi;
3591fa1209cSHong Zhang   }
3601fa1209cSHong Zhang 
3611fa1209cSHong Zhang 
3621fa1209cSHong Zhang   /* Column indices are in the list of free space.
3631fa1209cSHong Zhang      Allocate array cj, copy column indices to cj, and destroy list of free space */
3641fa1209cSHong Zhang   ierr = PetscMalloc((ci[am]+1)*sizeof(PetscInt),&cj);CHKERRQ(ierr);
3651fa1209cSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,cj);CHKERRQ(ierr);
3661fa1209cSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3671fa1209cSHong Zhang 
3681fa1209cSHong Zhang   /* Allocate space for ca */
3691fa1209cSHong Zhang   ierr = PetscMalloc((ci[am]+1)*sizeof(MatScalar),&ca);CHKERRQ(ierr);
3701fa1209cSHong Zhang   ierr = PetscMemzero(ca,(ci[am]+1)*sizeof(MatScalar));CHKERRQ(ierr);
3711fa1209cSHong Zhang 
3721fa1209cSHong Zhang   /* put together the new symbolic matrix */
3731fa1209cSHong Zhang   ierr = MatCreateSeqAIJWithArrays(((PetscObject)A)->comm,am,bm,ci,cj,ca,C);CHKERRQ(ierr);
3741fa1209cSHong Zhang 
3751fa1209cSHong Zhang   /* MatCreateSeqAIJWithArrays flags matrix so PETSc doesn't free the user's arrays. */
3761fa1209cSHong Zhang   /* These are PETSc arrays, so change flags so arrays can be deleted by PETSc */
3771fa1209cSHong Zhang   c = (Mat_SeqAIJ *)((*C)->data);
3781fa1209cSHong Zhang   c->free_a   = PETSC_TRUE;
3791fa1209cSHong Zhang   c->free_ij  = PETSC_TRUE;
3801fa1209cSHong Zhang   c->nonew    = 0;
3811fa1209cSHong Zhang 
3821fa1209cSHong Zhang   /* set MatInfo */
3831fa1209cSHong Zhang   afill = (PetscReal)ci[am]/(ai[am]+bi[bm]) + 1.e-5;
3841fa1209cSHong Zhang   if (afill < 1.0) afill = 1.0;
3851fa1209cSHong Zhang   c->maxnz                     = ci[am];
3861fa1209cSHong Zhang   c->nz                        = ci[am];
3871fa1209cSHong Zhang   (*C)->info.mallocs           = nspacedouble;
3881fa1209cSHong Zhang   (*C)->info.fill_ratio_given  = fill;
3891fa1209cSHong Zhang   (*C)->info.fill_ratio_needed = afill;
3901fa1209cSHong Zhang 
3911fa1209cSHong Zhang #if defined(PETSC_USE_INFO)
3921fa1209cSHong Zhang   if (ci[am]) {
3931fa1209cSHong Zhang     ierr = PetscInfo3((*C),"Reallocs %D; Fill ratio: given %G needed %G.\n",nspacedouble,fill,afill);CHKERRQ(ierr);
3941fa1209cSHong Zhang     ierr = PetscInfo1((*C),"Use MatMatMultTranspose(A,B,MatReuse,%G,&C) for best performance.;\n",afill);CHKERRQ(ierr);
3951fa1209cSHong Zhang   } else {
3961fa1209cSHong Zhang     ierr = PetscInfo((*C),"Empty matrix product\n");CHKERRQ(ierr);
3971fa1209cSHong Zhang   }
3981fa1209cSHong Zhang #endif
39981d82fe4SHong Zhang #endif
4005df89d91SHong Zhang   PetscFunctionReturn(0);
4015df89d91SHong Zhang }
4025df89d91SHong Zhang 
4036973769bSHong Zhang /* #define USE_ARRAY - for sparse dot product. Slower than !USE_ARRAY */
4045df89d91SHong Zhang #undef __FUNCT__
4055df89d91SHong Zhang #define __FUNCT__ "MatMatMultTransposeNumeric_SeqAIJ_SeqAIJ"
4065df89d91SHong Zhang PetscErrorCode MatMatMultTransposeNumeric_SeqAIJ_SeqAIJ(Mat A,Mat B,Mat C)
4075df89d91SHong Zhang {
4085df89d91SHong Zhang   PetscErrorCode ierr;
4095df89d91SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)B->data,*c=(Mat_SeqAIJ*)C->data;
410e2cac8caSJed Brown   PetscInt       *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,anzi,bnzj,nexta,nextb,*acol,*bcol,brow;
41181d82fe4SHong Zhang   PetscInt       cm=C->rmap->n,*ci=c->i,*cj=c->j,i,j,cnzi,*ccol;
4125df89d91SHong Zhang   PetscLogDouble flops=0.0;
4136973769bSHong Zhang   MatScalar      *aa=a->a,*aval,*ba=b->a,*bval,*ca=c->a,*cval;
4142128a86cSHong Zhang   Mat_MatMatMultTrans *multtrans;
4152128a86cSHong Zhang   PetscContainer      container;
4166973769bSHong Zhang #if defined(USE_ARRAY)
4176973769bSHong Zhang   MatScalar      *spdot;
4186973769bSHong Zhang #endif
4195df89d91SHong Zhang 
4205df89d91SHong Zhang   PetscFunctionBegin;
4212128a86cSHong Zhang   ierr = PetscObjectQuery((PetscObject)C,"Mat_MatMatMultTrans",(PetscObject *)&container);CHKERRQ(ierr);
4222128a86cSHong Zhang   if (!container) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Container does not exit");
4232128a86cSHong Zhang   ierr  = PetscContainerGetPointer(container,(void **)&multtrans);CHKERRQ(ierr);
4242128a86cSHong Zhang   if (multtrans->usecoloring){
4252128a86cSHong Zhang     MatMultTransposeColoring  matcoloring = multtrans->matcoloring;
426c8db22f6SHong Zhang     Mat                       Bt_dense;
427c8db22f6SHong Zhang     PetscInt                  m,n;
428b2d2b04fSHong Zhang     PetscLogDouble t0,tf,etime0=0.0,etime1=0.0,etime2=0.0;
429a0b95ffbSSatish Balay     Mat C_dense = multtrans->ABt_den;
430a0b95ffbSSatish Balay 
4312128a86cSHong Zhang     Bt_dense = multtrans->Bt_den;
432c8db22f6SHong Zhang     ierr = MatGetLocalSize(Bt_dense,&m,&n);CHKERRQ(ierr);
433c8db22f6SHong Zhang     printf("Bt_dense: %d,%d\n",m,n);
434c8db22f6SHong Zhang 
435c8db22f6SHong Zhang     /* Get Bt_dense by Apply MatMultTransposeColoring to B */
4362128a86cSHong Zhang     ierr = PetscGetTime(&t0);CHKERRQ(ierr);
4372128a86cSHong Zhang     ierr = MatMultTransposeColoringApply(B,Bt_dense,matcoloring);CHKERRQ(ierr);
4382128a86cSHong Zhang     ierr = PetscGetTime(&tf);CHKERRQ(ierr);
439b2d2b04fSHong Zhang     etime0 += tf - t0;
440c8db22f6SHong Zhang 
441c8db22f6SHong Zhang     /* C_dense = A*Bt_dense */
4422128a86cSHong Zhang     ierr = PetscGetTime(&t0);CHKERRQ(ierr);
4432128a86cSHong Zhang     ierr = MatMatMultNumeric_SeqAIJ_SeqDense(A,Bt_dense,C_dense);CHKERRQ(ierr);
4442128a86cSHong Zhang     ierr = PetscGetTime(&tf);CHKERRQ(ierr);
4452128a86cSHong Zhang     etime2 += tf - t0;
446c8db22f6SHong Zhang 
4472128a86cSHong Zhang     /* Recover C from C_dense */
4482128a86cSHong Zhang     ierr = PetscGetTime(&t0);CHKERRQ(ierr);
4492128a86cSHong Zhang     ierr = MatMultTransColoringApplyDenToSp(matcoloring,C_dense,C);CHKERRQ(ierr);
4502128a86cSHong Zhang     ierr = PetscGetTime(&tf);CHKERRQ(ierr);
4512128a86cSHong Zhang     etime1 += tf - t0;
452b2d2b04fSHong Zhang     printf("etime ColoringApply: %g %g; MatMatMultNumeric_sp_dense: %g\n",etime0,etime1,etime2);
453c8db22f6SHong Zhang     PetscFunctionReturn(0);
454c8db22f6SHong Zhang   }
455c8db22f6SHong Zhang 
4566973769bSHong Zhang #if defined(USE_ARRAY)
4576973769bSHong Zhang   /* allocate an array for implementing sparse inner-product */
458e2cac8caSJed Brown   ierr = PetscMalloc((A->cmap->n+1)*sizeof(MatScalar),&spdot);CHKERRQ(ierr);
459e2cac8caSJed Brown   ierr = PetscMemzero(spdot,(A->cmap->n+1)*sizeof(MatScalar));CHKERRQ(ierr);
4606973769bSHong Zhang #endif
4616973769bSHong Zhang 
46281d82fe4SHong Zhang   /* clear old values in C */
46381d82fe4SHong Zhang   ierr = PetscMemzero(ca,ci[cm]*sizeof(MatScalar));CHKERRQ(ierr);
4645df89d91SHong Zhang 
4651fa1209cSHong Zhang   for (i=0; i<cm; i++) {
46681d82fe4SHong Zhang     anzi = ai[i+1] - ai[i];
46781d82fe4SHong Zhang     acol = aj + ai[i];
4686973769bSHong Zhang     aval = aa + ai[i];
4691fa1209cSHong Zhang     cnzi = ci[i+1] - ci[i];
4701fa1209cSHong Zhang     ccol = cj + ci[i];
4716973769bSHong Zhang     cval = ca + ci[i];
4721fa1209cSHong Zhang     for (j=0; j<cnzi; j++){
47381d82fe4SHong Zhang       brow = ccol[j];
47481d82fe4SHong Zhang       bnzj = bi[brow+1] - bi[brow];
47581d82fe4SHong Zhang       bcol = bj + bi[brow];
4766973769bSHong Zhang       bval = ba + bi[brow];
4776973769bSHong Zhang 
47881d82fe4SHong Zhang       /* perform sparse inner-product c(i,j)=A[i,:]*B[j,:]^T */
4796973769bSHong Zhang #if defined(USE_ARRAY)
4806973769bSHong Zhang       /* put ba to spdot array */
4816973769bSHong Zhang       for (nextb=0; nextb<bnzj; nextb++) spdot[bcol[nextb]] = bval[nextb];
4826973769bSHong Zhang       /* c(i,j)=A[i,:]*B[j,:]^T */
4836973769bSHong Zhang       for (nexta=0; nexta<anzi; nexta++){
4846973769bSHong Zhang         cval[j] += spdot[acol[nexta]]*aval[nexta];
4856973769bSHong Zhang       }
4866973769bSHong Zhang       /* zero spdot array */
4876973769bSHong Zhang       for (nextb=0; nextb<bnzj; nextb++) spdot[bcol[nextb]] = 0.0;
4886973769bSHong Zhang #else
48981d82fe4SHong Zhang       nexta = 0; nextb = 0;
49081d82fe4SHong Zhang       while (nexta<anzi && nextb<bnzj){
49181d82fe4SHong Zhang         while (acol[nexta] < bcol[nextb] && nexta < anzi) nexta++;
49281d82fe4SHong Zhang         if (nexta == anzi) break;
49381d82fe4SHong Zhang         while (acol[nexta] > bcol[nextb] && nextb < bnzj) nextb++;
49481d82fe4SHong Zhang         if (nextb == bnzj) break;
49581d82fe4SHong Zhang         if (acol[nexta] == bcol[nextb]){
4966973769bSHong Zhang           cval[j] += aval[nexta]*bval[nextb];
49781d82fe4SHong Zhang           nexta++; nextb++;
49881d82fe4SHong Zhang           flops += 2;
4991fa1209cSHong Zhang         }
5001fa1209cSHong Zhang       }
5016973769bSHong Zhang #endif
50281d82fe4SHong Zhang     }
50381d82fe4SHong Zhang   }
50481d82fe4SHong Zhang   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
50581d82fe4SHong Zhang   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
50681d82fe4SHong Zhang   ierr = PetscLogFlops(flops);CHKERRQ(ierr);
5076973769bSHong Zhang #if defined(USE_ARRAY)
5086973769bSHong Zhang   ierr = PetscFree(spdot);
5096973769bSHong Zhang #endif
5105df89d91SHong Zhang   PetscFunctionReturn(0);
5115df89d91SHong Zhang }
5125df89d91SHong Zhang 
5135df89d91SHong Zhang #undef __FUNCT__
5145df89d91SHong Zhang #define __FUNCT__ "MatMatTransposeMult_SeqAIJ_SeqAIJ"
5155df89d91SHong Zhang PetscErrorCode MatMatTransposeMult_SeqAIJ_SeqAIJ(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) {
5165df89d91SHong Zhang   PetscErrorCode ierr;
5175df89d91SHong Zhang 
5185df89d91SHong Zhang   PetscFunctionBegin;
5195df89d91SHong Zhang   if (scall == MAT_INITIAL_MATRIX){
5205df89d91SHong Zhang     ierr = MatMatTransposeMultSymbolic_SeqAIJ_SeqAIJ(A,B,fill,C);CHKERRQ(ierr);
5215df89d91SHong Zhang   }
5225df89d91SHong Zhang   ierr = MatMatTransposeMultNumeric_SeqAIJ_SeqAIJ(A,B,*C);CHKERRQ(ierr);
5235df89d91SHong Zhang   PetscFunctionReturn(0);
5245df89d91SHong Zhang }
5255df89d91SHong Zhang 
5265df89d91SHong Zhang #undef __FUNCT__
5275df89d91SHong Zhang #define __FUNCT__ "MatMatTransposeMultSymbolic_SeqAIJ_SeqAIJ"
5285df89d91SHong Zhang PetscErrorCode MatMatTransposeMultSymbolic_SeqAIJ_SeqAIJ(Mat A,Mat B,PetscReal fill,Mat *C)
5295df89d91SHong Zhang {
530bc011b1eSHong Zhang   PetscErrorCode ierr;
531bc011b1eSHong Zhang   Mat            At;
53238baddfdSBarry Smith   PetscInt       *ati,*atj;
533bc011b1eSHong Zhang 
534bc011b1eSHong Zhang   PetscFunctionBegin;
535bc011b1eSHong Zhang   /* create symbolic At */
536bc011b1eSHong Zhang   ierr = MatGetSymbolicTranspose_SeqAIJ(A,&ati,&atj);CHKERRQ(ierr);
537d0f46423SBarry Smith   ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,A->cmap->n,A->rmap->n,ati,atj,PETSC_NULL,&At);CHKERRQ(ierr);
538bc011b1eSHong Zhang 
539bc011b1eSHong Zhang   /* get symbolic C=At*B */
540bc011b1eSHong Zhang   ierr = MatMatMultSymbolic_SeqAIJ_SeqAIJ(At,B,fill,C);CHKERRQ(ierr);
541bc011b1eSHong Zhang 
542bc011b1eSHong Zhang   /* clean up */
5436bf464f9SBarry Smith   ierr = MatDestroy(&At);CHKERRQ(ierr);
544bc011b1eSHong Zhang   ierr = MatRestoreSymbolicTranspose_SeqAIJ(A,&ati,&atj);CHKERRQ(ierr);
545bc011b1eSHong Zhang   PetscFunctionReturn(0);
546bc011b1eSHong Zhang }
547bc011b1eSHong Zhang 
548bc011b1eSHong Zhang #undef __FUNCT__
5495df89d91SHong Zhang #define __FUNCT__ "MatMatTransposeMultNumeric_SeqAIJ_SeqAIJ"
5505df89d91SHong Zhang PetscErrorCode MatMatTransposeMultNumeric_SeqAIJ_SeqAIJ(Mat A,Mat B,Mat C)
551bc011b1eSHong Zhang {
552bc011b1eSHong Zhang   PetscErrorCode ierr;
5530fbc74f4SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)B->data,*c=(Mat_SeqAIJ*)C->data;
554d0f46423SBarry Smith   PetscInt       am=A->rmap->n,anzi,*ai=a->i,*aj=a->j,*bi=b->i,*bj,bnzi,nextb;
555d0f46423SBarry Smith   PetscInt       cm=C->rmap->n,*ci=c->i,*cj=c->j,crow,*cjj,i,j,k;
556d13dce4bSSatish Balay   PetscLogDouble flops=0.0;
5570fbc74f4SHong Zhang   MatScalar      *aa=a->a,*ba,*ca=c->a,*caj;
558bc011b1eSHong Zhang 
559bc011b1eSHong Zhang   PetscFunctionBegin;
560bc011b1eSHong Zhang   /* clear old values in C */
561bc011b1eSHong Zhang   ierr = PetscMemzero(ca,ci[cm]*sizeof(MatScalar));CHKERRQ(ierr);
562bc011b1eSHong Zhang 
563bc011b1eSHong Zhang   /* compute A^T*B using outer product (A^T)[:,i]*B[i,:] */
564bc011b1eSHong Zhang   for (i=0;i<am;i++) {
565bc011b1eSHong Zhang     bj   = b->j + bi[i];
566bc011b1eSHong Zhang     ba   = b->a + bi[i];
567bc011b1eSHong Zhang     bnzi = bi[i+1] - bi[i];
568bc011b1eSHong Zhang     anzi = ai[i+1] - ai[i];
569bc011b1eSHong Zhang     for (j=0; j<anzi; j++) {
570bc011b1eSHong Zhang       nextb = 0;
5710fbc74f4SHong Zhang       crow  = *aj++;
572bc011b1eSHong Zhang       cjj   = cj + ci[crow];
573bc011b1eSHong Zhang       caj   = ca + ci[crow];
574bc011b1eSHong Zhang       /* perform sparse axpy operation.  Note cjj includes bj. */
575bc011b1eSHong Zhang       for (k=0; nextb<bnzi; k++) {
5760fbc74f4SHong Zhang         if (cjj[k] == *(bj+nextb)) { /* ccol == bcol */
5770fbc74f4SHong Zhang           caj[k] += (*aa)*(*(ba+nextb));
578bc011b1eSHong Zhang           nextb++;
579bc011b1eSHong Zhang         }
580bc011b1eSHong Zhang       }
581bc011b1eSHong Zhang       flops += 2*bnzi;
5820fbc74f4SHong Zhang       aa++;
583bc011b1eSHong Zhang     }
584bc011b1eSHong Zhang   }
585bc011b1eSHong Zhang 
586bc011b1eSHong Zhang   /* Assemble the final matrix and clean up */
587bc011b1eSHong Zhang   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
588bc011b1eSHong Zhang   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
589bc011b1eSHong Zhang   ierr = PetscLogFlops(flops);CHKERRQ(ierr);
590bc011b1eSHong Zhang   PetscFunctionReturn(0);
591bc011b1eSHong Zhang }
5929123193aSHong Zhang 
593ec01deb9SMatthew Knepley EXTERN_C_BEGIN
5949123193aSHong Zhang #undef __FUNCT__
5959123193aSHong Zhang #define __FUNCT__ "MatMatMult_SeqAIJ_SeqDense"
5969123193aSHong Zhang PetscErrorCode MatMatMult_SeqAIJ_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
5979123193aSHong Zhang {
5989123193aSHong Zhang   PetscErrorCode ierr;
5999123193aSHong Zhang 
6009123193aSHong Zhang   PetscFunctionBegin;
6019123193aSHong Zhang   if (scall == MAT_INITIAL_MATRIX){
6029123193aSHong Zhang     ierr = MatMatMultSymbolic_SeqAIJ_SeqDense(A,B,fill,C);CHKERRQ(ierr);
6039123193aSHong Zhang   }
6049123193aSHong Zhang   ierr = MatMatMultNumeric_SeqAIJ_SeqDense(A,B,*C);CHKERRQ(ierr);
6059123193aSHong Zhang   PetscFunctionReturn(0);
6069123193aSHong Zhang }
607ec01deb9SMatthew Knepley EXTERN_C_END
608edd81eecSMatthew Knepley 
6099123193aSHong Zhang #undef __FUNCT__
6109123193aSHong Zhang #define __FUNCT__ "MatMatMultSymbolic_SeqAIJ_SeqDense"
6119123193aSHong Zhang PetscErrorCode MatMatMultSymbolic_SeqAIJ_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
6129123193aSHong Zhang {
6139123193aSHong Zhang   PetscErrorCode ierr;
6149123193aSHong Zhang 
6159123193aSHong Zhang   PetscFunctionBegin;
6165a586d82SBarry Smith   ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,0.0,C);CHKERRQ(ierr);
6179123193aSHong Zhang   PetscFunctionReturn(0);
6189123193aSHong Zhang }
6199123193aSHong Zhang 
6209123193aSHong Zhang #undef __FUNCT__
6219123193aSHong Zhang #define __FUNCT__ "MatMatMultNumeric_SeqAIJ_SeqDense"
6229123193aSHong Zhang PetscErrorCode MatMatMultNumeric_SeqAIJ_SeqDense(Mat A,Mat B,Mat C)
6239123193aSHong Zhang {
624f32d5d43SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
6259123193aSHong Zhang   PetscErrorCode ierr;
626dd6ea824SBarry Smith   PetscScalar    *b,*c,r1,r2,r3,r4,*b1,*b2,*b3,*b4;
627dd6ea824SBarry Smith   MatScalar      *aa;
628d0f46423SBarry Smith   PetscInt       cm=C->rmap->n, cn=B->cmap->n, bm=B->rmap->n, col, i,j,n,*aj, am = A->rmap->n;
629f32d5d43SBarry Smith   PetscInt       am2 = 2*am, am3 = 3*am,  bm4 = 4*bm,colam;
6309123193aSHong Zhang 
6319123193aSHong Zhang   PetscFunctionBegin;
632f32d5d43SBarry Smith   if (!cm || !cn) PetscFunctionReturn(0);
633e32f2f54SBarry Smith   if (bm != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number columns in A %D not equal rows in B %D\n",A->cmap->n,bm);
634e32f2f54SBarry Smith   if (A->rmap->n != C->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows in C %D not equal rows in A %D\n",C->rmap->n,A->rmap->n);
635e32f2f54SBarry Smith   if (B->cmap->n != C->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number columns in B %D not equal columns in C %D\n",B->cmap->n,C->cmap->n);
636f32d5d43SBarry Smith   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
637f32d5d43SBarry Smith   ierr = MatGetArray(C,&c);CHKERRQ(ierr);
638f32d5d43SBarry Smith   b1 = b; b2 = b1 + bm; b3 = b2 + bm; b4 = b3 + bm;
639f32d5d43SBarry Smith   for (col=0; col<cn-4; col += 4){  /* over columns of C */
640f32d5d43SBarry Smith     colam = col*am;
641f32d5d43SBarry Smith     for (i=0; i<am; i++) {        /* over rows of C in those columns */
642f32d5d43SBarry Smith       r1 = r2 = r3 = r4 = 0.0;
643f32d5d43SBarry Smith       n   = a->i[i+1] - a->i[i];
644f32d5d43SBarry Smith       aj  = a->j + a->i[i];
645f32d5d43SBarry Smith       aa  = a->a + a->i[i];
646f32d5d43SBarry Smith       for (j=0; j<n; j++) {
647f32d5d43SBarry Smith         r1 += (*aa)*b1[*aj];
648f32d5d43SBarry Smith         r2 += (*aa)*b2[*aj];
649f32d5d43SBarry Smith         r3 += (*aa)*b3[*aj];
650f32d5d43SBarry Smith         r4 += (*aa++)*b4[*aj++];
6519123193aSHong Zhang       }
652f32d5d43SBarry Smith       c[colam + i]       = r1;
653f32d5d43SBarry Smith       c[colam + am + i]  = r2;
654f32d5d43SBarry Smith       c[colam + am2 + i] = r3;
655f32d5d43SBarry Smith       c[colam + am3 + i] = r4;
656f32d5d43SBarry Smith     }
657f32d5d43SBarry Smith     b1 += bm4;
658f32d5d43SBarry Smith     b2 += bm4;
659f32d5d43SBarry Smith     b3 += bm4;
660f32d5d43SBarry Smith     b4 += bm4;
661f32d5d43SBarry Smith   }
662f32d5d43SBarry Smith   for (;col<cn; col++){     /* over extra columns of C */
663f32d5d43SBarry Smith     for (i=0; i<am; i++) {  /* over rows of C in those columns */
664f32d5d43SBarry Smith       r1 = 0.0;
665f32d5d43SBarry Smith       n   = a->i[i+1] - a->i[i];
666f32d5d43SBarry Smith       aj  = a->j + a->i[i];
667f32d5d43SBarry Smith       aa  = a->a + a->i[i];
668f32d5d43SBarry Smith 
669f32d5d43SBarry Smith       for (j=0; j<n; j++) {
670f32d5d43SBarry Smith         r1 += (*aa++)*b1[*aj++];
671f32d5d43SBarry Smith       }
672f32d5d43SBarry Smith       c[col*am + i]     = r1;
673f32d5d43SBarry Smith     }
674f32d5d43SBarry Smith     b1 += bm;
675f32d5d43SBarry Smith   }
676dc0b31edSSatish Balay   ierr = PetscLogFlops(cn*(2.0*a->nz));CHKERRQ(ierr);
677f32d5d43SBarry Smith   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
678f32d5d43SBarry Smith   ierr = MatRestoreArray(C,&c);CHKERRQ(ierr);
679f32d5d43SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
680f32d5d43SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
681f32d5d43SBarry Smith   PetscFunctionReturn(0);
682f32d5d43SBarry Smith }
683f32d5d43SBarry Smith 
684f32d5d43SBarry Smith /*
6854324174eSBarry Smith    Note very similar to MatMult_SeqAIJ(), should generate both codes from same base
686f32d5d43SBarry Smith */
687f32d5d43SBarry Smith #undef __FUNCT__
688f32d5d43SBarry Smith #define __FUNCT__ "MatMatMultNumericAdd_SeqAIJ_SeqDense"
689f32d5d43SBarry Smith PetscErrorCode MatMatMultNumericAdd_SeqAIJ_SeqDense(Mat A,Mat B,Mat C)
690f32d5d43SBarry Smith {
691f32d5d43SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
692f32d5d43SBarry Smith   PetscErrorCode ierr;
693dd6ea824SBarry Smith   PetscScalar    *b,*c,r1,r2,r3,r4,*b1,*b2,*b3,*b4;
694dd6ea824SBarry Smith   MatScalar      *aa;
695d0f46423SBarry Smith   PetscInt       cm=C->rmap->n, cn=B->cmap->n, bm=B->rmap->n, col, i,j,n,*aj, am = A->rmap->n,*ii,arm;
6964324174eSBarry Smith   PetscInt       am2 = 2*am, am3 = 3*am,  bm4 = 4*bm,colam,*ridx;
697f32d5d43SBarry Smith 
698f32d5d43SBarry Smith   PetscFunctionBegin;
699f32d5d43SBarry Smith   if (!cm || !cn) PetscFunctionReturn(0);
700f32d5d43SBarry Smith   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
701f32d5d43SBarry Smith   ierr = MatGetArray(C,&c);CHKERRQ(ierr);
702f32d5d43SBarry Smith   b1 = b; b2 = b1 + bm; b3 = b2 + bm; b4 = b3 + bm;
7034324174eSBarry Smith 
7044324174eSBarry Smith   if (a->compressedrow.use){ /* use compressed row format */
7054324174eSBarry Smith     for (col=0; col<cn-4; col += 4){  /* over columns of C */
7064324174eSBarry Smith       colam = col*am;
7074324174eSBarry Smith       arm   = a->compressedrow.nrows;
7084324174eSBarry Smith       ii    = a->compressedrow.i;
7094324174eSBarry Smith       ridx  = a->compressedrow.rindex;
7104324174eSBarry Smith       for (i=0; i<arm; i++) {        /* over rows of C in those columns */
7114324174eSBarry Smith 	r1 = r2 = r3 = r4 = 0.0;
7124324174eSBarry Smith 	n   = ii[i+1] - ii[i];
7134324174eSBarry Smith 	aj  = a->j + ii[i];
7144324174eSBarry Smith 	aa  = a->a + ii[i];
7154324174eSBarry Smith 	for (j=0; j<n; j++) {
7164324174eSBarry Smith 	  r1 += (*aa)*b1[*aj];
7174324174eSBarry Smith 	  r2 += (*aa)*b2[*aj];
7184324174eSBarry Smith 	  r3 += (*aa)*b3[*aj];
7194324174eSBarry Smith 	  r4 += (*aa++)*b4[*aj++];
7204324174eSBarry Smith 	}
7214324174eSBarry Smith 	c[colam       + ridx[i]] += r1;
7224324174eSBarry Smith 	c[colam + am  + ridx[i]] += r2;
7234324174eSBarry Smith 	c[colam + am2 + ridx[i]] += r3;
7244324174eSBarry Smith 	c[colam + am3 + ridx[i]] += r4;
7254324174eSBarry Smith       }
7264324174eSBarry Smith       b1 += bm4;
7274324174eSBarry Smith       b2 += bm4;
7284324174eSBarry Smith       b3 += bm4;
7294324174eSBarry Smith       b4 += bm4;
7304324174eSBarry Smith     }
7314324174eSBarry Smith     for (;col<cn; col++){     /* over extra columns of C */
7324324174eSBarry Smith       colam = col*am;
7334324174eSBarry Smith       arm   = a->compressedrow.nrows;
7344324174eSBarry Smith       ii    = a->compressedrow.i;
7354324174eSBarry Smith       ridx  = a->compressedrow.rindex;
7364324174eSBarry Smith       for (i=0; i<arm; i++) {  /* over rows of C in those columns */
7374324174eSBarry Smith 	r1 = 0.0;
7384324174eSBarry Smith 	n   = ii[i+1] - ii[i];
7394324174eSBarry Smith 	aj  = a->j + ii[i];
7404324174eSBarry Smith 	aa  = a->a + ii[i];
7414324174eSBarry Smith 
7424324174eSBarry Smith 	for (j=0; j<n; j++) {
7434324174eSBarry Smith 	  r1 += (*aa++)*b1[*aj++];
7444324174eSBarry Smith 	}
7454324174eSBarry Smith 	c[col*am + ridx[i]] += r1;
7464324174eSBarry Smith       }
7474324174eSBarry Smith       b1 += bm;
7484324174eSBarry Smith     }
7494324174eSBarry Smith   } else {
750f32d5d43SBarry Smith     for (col=0; col<cn-4; col += 4){  /* over columns of C */
751f32d5d43SBarry Smith       colam = col*am;
752f32d5d43SBarry Smith       for (i=0; i<am; i++) {        /* over rows of C in those columns */
753f32d5d43SBarry Smith 	r1 = r2 = r3 = r4 = 0.0;
754f32d5d43SBarry Smith 	n   = a->i[i+1] - a->i[i];
755f32d5d43SBarry Smith 	aj  = a->j + a->i[i];
756f32d5d43SBarry Smith 	aa  = a->a + a->i[i];
757f32d5d43SBarry Smith 	for (j=0; j<n; j++) {
758f32d5d43SBarry Smith 	  r1 += (*aa)*b1[*aj];
759f32d5d43SBarry Smith 	  r2 += (*aa)*b2[*aj];
760f32d5d43SBarry Smith 	  r3 += (*aa)*b3[*aj];
761f32d5d43SBarry Smith 	  r4 += (*aa++)*b4[*aj++];
762f32d5d43SBarry Smith 	}
763f32d5d43SBarry Smith 	c[colam + i]       += r1;
764f32d5d43SBarry Smith 	c[colam + am + i]  += r2;
765f32d5d43SBarry Smith 	c[colam + am2 + i] += r3;
766f32d5d43SBarry Smith 	c[colam + am3 + i] += r4;
767f32d5d43SBarry Smith       }
768f32d5d43SBarry Smith       b1 += bm4;
769f32d5d43SBarry Smith       b2 += bm4;
770f32d5d43SBarry Smith       b3 += bm4;
771f32d5d43SBarry Smith       b4 += bm4;
772f32d5d43SBarry Smith     }
773f32d5d43SBarry Smith     for (;col<cn; col++){     /* over extra columns of C */
774f32d5d43SBarry Smith       for (i=0; i<am; i++) {  /* over rows of C in those columns */
775f32d5d43SBarry Smith 	r1 = 0.0;
776f32d5d43SBarry Smith 	n   = a->i[i+1] - a->i[i];
777f32d5d43SBarry Smith 	aj  = a->j + a->i[i];
778f32d5d43SBarry Smith 	aa  = a->a + a->i[i];
779f32d5d43SBarry Smith 
780f32d5d43SBarry Smith 	for (j=0; j<n; j++) {
781f32d5d43SBarry Smith 	  r1 += (*aa++)*b1[*aj++];
782f32d5d43SBarry Smith 	}
783f32d5d43SBarry Smith 	c[col*am + i]     += r1;
784f32d5d43SBarry Smith       }
785f32d5d43SBarry Smith       b1 += bm;
786f32d5d43SBarry Smith     }
7874324174eSBarry Smith   }
788dc0b31edSSatish Balay   ierr = PetscLogFlops(cn*2.0*a->nz);CHKERRQ(ierr);
789f32d5d43SBarry Smith   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
790f32d5d43SBarry Smith   ierr = MatRestoreArray(C,&c);CHKERRQ(ierr);
7919123193aSHong Zhang   PetscFunctionReturn(0);
7929123193aSHong Zhang }
793b1683b59SHong Zhang 
794b1683b59SHong Zhang #undef __FUNCT__
795c8db22f6SHong Zhang #define __FUNCT__ "MatMultTransposeColoringApply_SeqAIJ"
796c8db22f6SHong Zhang PetscErrorCode  MatMultTransposeColoringApply_SeqAIJ(Mat B,Mat Btdense,MatMultTransposeColoring coloring)
797c8db22f6SHong Zhang {
798c8db22f6SHong Zhang   PetscErrorCode ierr;
799c8db22f6SHong Zhang   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)B->data;
800c8db22f6SHong Zhang   Mat_SeqDense   *atdense = (Mat_SeqDense*)Btdense->data;
801c8db22f6SHong Zhang   PetscInt       m=Btdense->rmap->n,n=Btdense->cmap->n,j,k,l,col,anz,*atcol,brow,bcol;
802c8db22f6SHong Zhang   MatScalar      *atval,*bval;
803c8db22f6SHong Zhang 
804c8db22f6SHong Zhang   PetscFunctionBegin;
805c8db22f6SHong Zhang   ierr = PetscMemzero(atdense->v,(m*n)*sizeof(MatScalar));CHKERRQ(ierr);
806c8db22f6SHong Zhang   for (k=0; k<coloring->ncolors; k++) {
807c8db22f6SHong Zhang     for (l=0; l<coloring->ncolumns[k]; l++) { /* insert a row of B to a column of Btdense */
808c8db22f6SHong Zhang       col = coloring->columns[k][l];   /* =row of B */
809c8db22f6SHong Zhang       anz = a->i[col+1] - a->i[col];
810c8db22f6SHong Zhang       for (j=0; j<anz; j++){
811c8db22f6SHong Zhang         atcol = a->j + a->i[col];
812c8db22f6SHong Zhang         atval = a->a + a->i[col];
813c8db22f6SHong Zhang         bval  = atdense->v;
814c8db22f6SHong Zhang         brow  = atcol[j];
815c8db22f6SHong Zhang         bcol  = k;
816c8db22f6SHong Zhang         bval[bcol*m+brow] = atval[j];
817c8db22f6SHong Zhang       }
818c8db22f6SHong Zhang     }
819c8db22f6SHong Zhang   }
820c8db22f6SHong Zhang   PetscFunctionReturn(0);
821c8db22f6SHong Zhang }
822c8db22f6SHong Zhang 
823c8db22f6SHong Zhang #undef __FUNCT__
8248972f759SHong Zhang #define __FUNCT__ "MatMultTransColoringApplyDenToSp_SeqAIJ"
8258972f759SHong Zhang PetscErrorCode MatMultTransColoringApplyDenToSp_SeqAIJ(MatMultTransposeColoring matcoloring,Mat Cden,Mat Csp)
8268972f759SHong Zhang {
8278972f759SHong Zhang   PetscErrorCode ierr;
828b2d2b04fSHong Zhang   Mat_SeqAIJ     *csp = (Mat_SeqAIJ*)Csp->data;
829*b28d80bdSHong Zhang   PetscInt       k,l,row,m,**rows=matcoloring->rows,**spidx=matcoloring->columnsforspidx;
830b2d2b04fSHong Zhang   PetscScalar    *ca_den,*cp_den,*ca=csp->a;
8318972f759SHong Zhang 
8328972f759SHong Zhang   PetscFunctionBegin;
8338972f759SHong Zhang   ierr = MatGetLocalSize(Csp,&m,PETSC_NULL);CHKERRQ(ierr);
834b2d2b04fSHong Zhang   ierr = MatGetArray(Cden,&ca_den);CHKERRQ(ierr);
835b2d2b04fSHong Zhang   cp_den = ca_den;
8368972f759SHong Zhang   for (k=0; k<matcoloring->ncolors; k++) {
8378972f759SHong Zhang     for (l=0; l<matcoloring->nrows[k]; l++){
838*b28d80bdSHong Zhang       row             = rows[k][l];
839*b28d80bdSHong Zhang       ca[spidx[k][l]] = cp_den[row];
840b2d2b04fSHong Zhang     }
841b2d2b04fSHong Zhang     cp_den += m;
842b2d2b04fSHong Zhang   }
843b2d2b04fSHong Zhang   ierr = MatRestoreArray(Cden,&ca_den);CHKERRQ(ierr);
8448972f759SHong Zhang   PetscFunctionReturn(0);
8458972f759SHong Zhang }
8468972f759SHong Zhang 
8478972f759SHong Zhang #undef __FUNCT__
848b1683b59SHong Zhang #define __FUNCT__ "MatMultTransposeColoringCreate_SeqAIJ"
849b1683b59SHong Zhang PetscErrorCode MatMultTransposeColoringCreate_SeqAIJ(Mat mat,ISColoring iscoloring,MatMultTransposeColoring c)
850b1683b59SHong Zhang {
851b1683b59SHong Zhang   PetscErrorCode ierr;
852e8354b3eSHong Zhang   PetscInt       i,n,nrows,N,j,k,m,*rows,*ci,*cj,ncols,col,cm;
853b1683b59SHong Zhang   const PetscInt *is;
854*b28d80bdSHong Zhang   PetscInt       nis = iscoloring->n,*rowhit,bs = 1;
855b1683b59SHong Zhang   IS             *isa;
856b1683b59SHong Zhang   PetscBool      done;
857b1683b59SHong Zhang   PetscBool      flg1,flg2;
858*b28d80bdSHong Zhang   Mat_SeqAIJ     *csp = (Mat_SeqAIJ*)mat->data;
859*b28d80bdSHong Zhang   PetscInt       l,row,*rp,low,high,*cilen = csp->ilen,t;
860*b28d80bdSHong Zhang #if defined(PETSC_USE_DEBUG)
861*b28d80bdSHong Zhang   PetscBool      found;
862*b28d80bdSHong Zhang #endif
863b1683b59SHong Zhang 
864b1683b59SHong Zhang   PetscFunctionBegin;
865b1683b59SHong Zhang   ierr = ISColoringGetIS(iscoloring,PETSC_IGNORE,&isa);CHKERRQ(ierr);
866b1683b59SHong Zhang   /* this is ugly way to get blocksize but cannot call MatGetBlockSize() because AIJ can have bs > 1 */
867b1683b59SHong Zhang   ierr = PetscTypeCompare((PetscObject)mat,MATSEQBAIJ,&flg1);CHKERRQ(ierr);
868b1683b59SHong Zhang   ierr = PetscTypeCompare((PetscObject)mat,MATMPIBAIJ,&flg2);CHKERRQ(ierr);
869b1683b59SHong Zhang   if (flg1 || flg2) {
870b1683b59SHong Zhang     ierr = MatGetBlockSize(mat,&bs);CHKERRQ(ierr);
871b1683b59SHong Zhang   }
872b1683b59SHong Zhang 
873b1683b59SHong Zhang   N          = mat->cmap->N/bs;
874b1683b59SHong Zhang   c->M       = mat->rmap->N/bs;  /* set total rows, columns and local rows */
875b1683b59SHong Zhang   c->N       = mat->cmap->N/bs;
876b1683b59SHong Zhang   c->m       = mat->rmap->N/bs;
877b1683b59SHong Zhang   c->rstart  = 0;
878b1683b59SHong Zhang 
879b1683b59SHong Zhang   c->ncolors = nis;
880b1683b59SHong Zhang   ierr       = PetscMalloc(nis*sizeof(PetscInt),&c->ncolumns);CHKERRQ(ierr);
881b1683b59SHong Zhang   ierr       = PetscMalloc(nis*sizeof(PetscInt*),&c->columns);CHKERRQ(ierr);
882*b28d80bdSHong Zhang   ierr       = PetscMalloc(nis*sizeof(PetscInt),&c->nrows);CHKERRQ(ierr);
883*b28d80bdSHong Zhang   ierr       = PetscMalloc(nis*sizeof(PetscInt*),&c->rows);CHKERRQ(ierr);
884*b28d80bdSHong Zhang   ierr       = PetscMalloc(nis*sizeof(PetscInt*),&c->columnsforrow);CHKERRQ(ierr);
885*b28d80bdSHong Zhang   ierr       = PetscMalloc(nis*sizeof(PetscInt*),&c->columnsforspidx);CHKERRQ(ierr);
886b1683b59SHong Zhang 
887*b28d80bdSHong Zhang   ierr = MatGetColumnIJ(mat,0,PETSC_FALSE,PETSC_FALSE,&ncols,&ci,&cj,&done);CHKERRQ(ierr); // column-wise storage!!!
888b1683b59SHong Zhang   if (!done) SETERRQ1(((PetscObject)mat)->comm,PETSC_ERR_SUP,"MatGetColumnIJ() not supported for matrix type %s",((PetscObject)mat)->type_name);
889b1683b59SHong Zhang 
890*b28d80bdSHong Zhang   cm = c->m;
891*b28d80bdSHong Zhang   ierr = PetscMalloc((cm+1)*sizeof(PetscInt),&rowhit);CHKERRQ(ierr);
892b1683b59SHong Zhang   for (i=0; i<nis; i++) {
893b1683b59SHong Zhang     ierr = ISGetLocalSize(isa[i],&n);CHKERRQ(ierr);
894b1683b59SHong Zhang     ierr = ISGetIndices(isa[i],&is);CHKERRQ(ierr);
895b1683b59SHong Zhang     c->ncolumns[i] = n;
896b1683b59SHong Zhang     if (n) {
897b1683b59SHong Zhang       ierr = PetscMalloc(n*sizeof(PetscInt),&c->columns[i]);CHKERRQ(ierr);
898b1683b59SHong Zhang       ierr = PetscMemcpy(c->columns[i],is,n*sizeof(PetscInt));CHKERRQ(ierr);
899b1683b59SHong Zhang     } else {
900b1683b59SHong Zhang       c->columns[i]  = 0;
901b1683b59SHong Zhang     }
902b1683b59SHong Zhang 
903b1683b59SHong Zhang     /* fast, crude version requires O(N*N) work */
904e8354b3eSHong Zhang     ierr = PetscMemzero(rowhit,cm*sizeof(PetscInt));CHKERRQ(ierr);
905b1683b59SHong Zhang     /* loop over columns*/
906b1683b59SHong Zhang     for (j=0; j<n; j++) {
907b1683b59SHong Zhang       col  = is[j];
908b1683b59SHong Zhang       rows = cj + ci[col];
909b1683b59SHong Zhang       m    = ci[col+1] - ci[col];
910b1683b59SHong Zhang       /* loop over columns marking them in rowhit */
911b1683b59SHong Zhang       for (k=0; k<m; k++) {
912b1683b59SHong Zhang         rowhit[*rows++] = col + 1;
913b1683b59SHong Zhang       }
914b1683b59SHong Zhang     }
915b1683b59SHong Zhang     /* count the number of hits */
916b1683b59SHong Zhang     nrows = 0;
917e8354b3eSHong Zhang     for (j=0; j<cm; j++) {
918b1683b59SHong Zhang       if (rowhit[j]) nrows++;
919b1683b59SHong Zhang     }
920b1683b59SHong Zhang     c->nrows[i] = nrows;
921*b28d80bdSHong Zhang     ierr = PetscMalloc3(nrows+1,PetscInt,&c->rows[i],nrows+1,PetscInt,&c->columnsforrow[i],nrows+1,PetscInt,&c->columnsforspidx[i]);CHKERRQ(ierr);
922b1683b59SHong Zhang     nrows       = 0;
923e8354b3eSHong Zhang     for (j=0; j<cm; j++) {
924b1683b59SHong Zhang       if (rowhit[j]) {
925b1683b59SHong Zhang         c->rows[i][nrows]          = j;
926b1683b59SHong Zhang         c->columnsforrow[i][nrows] = rowhit[j] - 1;
927b1683b59SHong Zhang         nrows++;
928b1683b59SHong Zhang       }
929b1683b59SHong Zhang     }
930b1683b59SHong Zhang     ierr = ISRestoreIndices(isa[i],&is);CHKERRQ(ierr);
931b1683b59SHong Zhang   }
932b1683b59SHong Zhang   ierr = MatRestoreColumnIJ(mat,0,PETSC_FALSE,PETSC_FALSE,&ncols,&ci,&cj,&done);CHKERRQ(ierr);
933*b28d80bdSHong Zhang   ierr = PetscFree(rowhit);CHKERRQ(ierr);
934b1683b59SHong Zhang   ierr = ISColoringRestoreIS(iscoloring,&isa);CHKERRQ(ierr);
935*b28d80bdSHong Zhang 
936*b28d80bdSHong Zhang   /* set c->columnsforspidx to be used by MatMultTransColoringApplyDenToSp() in MatMatMultTransposeNumeric()  */
937*b28d80bdSHong Zhang   ci = csp->i; cj = csp->j;
938*b28d80bdSHong Zhang   for (k=0; k<c->ncolors; k++) {
939*b28d80bdSHong Zhang     for (l=0; l<c->nrows[k]; l++){
940*b28d80bdSHong Zhang       row = c->rows[k][l];             /* local row index */
941*b28d80bdSHong Zhang       col = c->columnsforrow[k][l];    /* global column index */
942*b28d80bdSHong Zhang       /* below is optimized from  MatSetValues(Csp,1,&row,1,&col,cval+row,INSERT_VALUES); */
943*b28d80bdSHong Zhang       rp   = cj + ci[row];
944*b28d80bdSHong Zhang       low  = 0; high = cilen[row];
945*b28d80bdSHong Zhang       while (high-low > 5) {
946*b28d80bdSHong Zhang         t = (low+high)/2;
947*b28d80bdSHong Zhang         if (rp[t] > col) high = t;
948*b28d80bdSHong Zhang         else             low  = t;
949*b28d80bdSHong Zhang       }
950*b28d80bdSHong Zhang #if defined(PETSC_USE_DEBUG)
951*b28d80bdSHong Zhang       found = PETSC_FALSE;
952*b28d80bdSHong Zhang #endif
953*b28d80bdSHong Zhang       for (i=low; i<high; i++) {
954*b28d80bdSHong Zhang         if (rp[i] == col) {
955*b28d80bdSHong Zhang           c->columnsforspidx[k][l] = ci[row] + i; /* index of mat(row,col) in csp->j and csp->a arrays */
956*b28d80bdSHong Zhang #if defined(PETSC_USE_DEBUG)
957*b28d80bdSHong Zhang           found = PETSC_TRUE;
958*b28d80bdSHong Zhang #endif
959*b28d80bdSHong Zhang           break;
960*b28d80bdSHong Zhang         }
961*b28d80bdSHong Zhang       }
962*b28d80bdSHong Zhang #if defined(PETSC_USE_DEBUG)
963*b28d80bdSHong Zhang       if (!found) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"%d %d is not inserted",row,col);
964*b28d80bdSHong Zhang #endif
965*b28d80bdSHong Zhang     }
966*b28d80bdSHong Zhang   }
967b1683b59SHong Zhang   PetscFunctionReturn(0);
968b1683b59SHong Zhang }
969