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,¤t_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,¤t_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