149b5e25fSSatish Balay 249b5e25fSSatish Balay /* 3a1373b80SHong Zhang Defines the basic matrix operations for the SBAIJ (compressed row) 449b5e25fSSatish Balay matrix storage format. 549b5e25fSSatish Balay */ 6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h> /*I "petscmat.h" I*/ 7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h> 8c6db04a5SJed Brown #include <petscblaslapack.h> 949b5e25fSSatish Balay 10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h> 1170dcbbb9SBarry Smith #define USESHORT 12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h> 1370dcbbb9SBarry Smith 14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool); 15b5b17502SBarry Smith 1649b5e25fSSatish Balay /* 1749b5e25fSSatish Balay Checks for missing diagonals 1849b5e25fSSatish Balay */ 194a2ae208SSatish Balay #undef __FUNCT__ 204a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ" 21ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool *missing,PetscInt *dd) 2249b5e25fSSatish Balay { 23045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 246849ba73SBarry Smith PetscErrorCode ierr; 2513f74950SBarry Smith PetscInt *diag,*jj = a->j,i; 2649b5e25fSSatish Balay 2749b5e25fSSatish Balay PetscFunctionBegin; 28045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 292af78befSBarry Smith *missing = PETSC_FALSE; 30358d2f5dSShri Abhyankar if (A->rmap->n > 0 && !jj) { 31358d2f5dSShri Abhyankar *missing = PETSC_TRUE; 32358d2f5dSShri Abhyankar if (dd) *dd = 0; 33358d2f5dSShri Abhyankar PetscInfo(A,"Matrix has no entries therefore is missing diagonal"); 34358d2f5dSShri Abhyankar } else { 35358d2f5dSShri Abhyankar diag = a->diag; 3649b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 372af78befSBarry Smith if (jj[diag[i]] != i) { 382af78befSBarry Smith *missing = PETSC_TRUE; 392af78befSBarry Smith if (dd) *dd = i; 402af78befSBarry Smith break; 412af78befSBarry Smith } 4249b5e25fSSatish Balay } 43358d2f5dSShri Abhyankar } 4449b5e25fSSatish Balay PetscFunctionReturn(0); 4549b5e25fSSatish Balay } 4649b5e25fSSatish Balay 474a2ae208SSatish Balay #undef __FUNCT__ 484a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ" 49dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 5049b5e25fSSatish Balay { 51045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 526849ba73SBarry Smith PetscErrorCode ierr; 5309f38230SBarry Smith PetscInt i; 5449b5e25fSSatish Balay 5549b5e25fSSatish Balay PetscFunctionBegin; 5609f38230SBarry Smith if (!a->diag) { 57785e854fSJed Brown ierr = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr); 583bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 59c760cd28SBarry Smith a->free_diag = PETSC_TRUE; 6009f38230SBarry Smith } 6109f38230SBarry Smith for (i=0; i<a->mbs; i++) a->diag[i] = a->i[i]; 6249b5e25fSSatish Balay PetscFunctionReturn(0); 6349b5e25fSSatish Balay } 6449b5e25fSSatish Balay 654a2ae208SSatish Balay #undef __FUNCT__ 664a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ" 671a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool *done) 6849b5e25fSSatish Balay { 69a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 70d0f46423SBarry Smith PetscInt i,j,n = a->mbs,nz = a->i[n],bs = A->rmap->bs; 711a83f524SJed Brown PetscInt **ia = (PetscInt**)inia,**ja = (PetscInt**)inja; 728f7157efSSatish Balay PetscErrorCode ierr; 7349b5e25fSSatish Balay 7449b5e25fSSatish Balay PetscFunctionBegin; 75d3e5a4abSHong Zhang *nn = n; 76a1373b80SHong Zhang if (!ia) PetscFunctionReturn(0); 778f7157efSSatish Balay if (!blockcompressed) { 788f7157efSSatish Balay /* malloc & create the natural set of indices */ 79dcca6d9dSJed Brown ierr = PetscMalloc2((n+1)*bs,ia,nz*bs,ja);CHKERRQ(ierr); 808f7157efSSatish Balay for (i=0; i<n+1; i++) { 818f7157efSSatish Balay for (j=0; j<bs; j++) { 828f7157efSSatish Balay *ia[i*bs+j] = a->i[i]*bs+j+oshift; 838f7157efSSatish Balay } 848f7157efSSatish Balay } 858f7157efSSatish Balay for (i=0; i<nz; i++) { 868f7157efSSatish Balay for (j=0; j<bs; j++) { 878f7157efSSatish Balay *ja[i*bs+j] = a->j[i]*bs+j+oshift; 888f7157efSSatish Balay } 898f7157efSSatish Balay } 908f7157efSSatish Balay } else { /* blockcompressed */ 91a6ece127SHong Zhang if (oshift == 1) { 92a6ece127SHong Zhang /* temporarily add 1 to i and j indices */ 936c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]++; 94a1373b80SHong Zhang for (i=0; i<n+1; i++) a->i[i]++; 958f7157efSSatish Balay } 96a1373b80SHong Zhang *ia = a->i; *ja = a->j; 97a6ece127SHong Zhang } 9849b5e25fSSatish Balay PetscFunctionReturn(0); 9949b5e25fSSatish Balay } 10049b5e25fSSatish Balay 1014a2ae208SSatish Balay #undef __FUNCT__ 1024a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ" 1031a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 10449b5e25fSSatish Balay { 105b7aaefc3SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1068f7157efSSatish Balay PetscInt i,n = a->mbs,nz = a->i[n]; 1078f7157efSSatish Balay PetscErrorCode ierr; 108a6ece127SHong Zhang 10949b5e25fSSatish Balay PetscFunctionBegin; 11049b5e25fSSatish Balay if (!ia) PetscFunctionReturn(0); 111a6ece127SHong Zhang 1128f7157efSSatish Balay if (!blockcompressed) { 1138f7157efSSatish Balay ierr = PetscFree2(*ia,*ja);CHKERRQ(ierr); 1148f7157efSSatish Balay } else if (oshift == 1) { /* blockcompressed */ 1156c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]--; 116a6ece127SHong Zhang for (i=0; i<n+1; i++) a->i[i]--; 117a6ece127SHong Zhang } 118a6ece127SHong Zhang PetscFunctionReturn(0); 11949b5e25fSSatish Balay } 12049b5e25fSSatish Balay 1214a2ae208SSatish Balay #undef __FUNCT__ 1224a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ" 123dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A) 12449b5e25fSSatish Balay { 12549b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 126dfbe8321SBarry Smith PetscErrorCode ierr; 12749b5e25fSSatish Balay 12849b5e25fSSatish Balay PetscFunctionBegin; 129a9f03627SSatish Balay #if defined(PETSC_USE_LOG) 130d0f46423SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz); 131a9f03627SSatish Balay #endif 132e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 1337f53bb6cSHong Zhang if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);} 1346bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1356bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1366bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 137c31cb41cSBarry Smith ierr = PetscFree(a->idiag);CHKERRQ(ierr); 138c31cb41cSBarry Smith ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 139c760cd28SBarry Smith if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);} 14005b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 14141f059aeSBarry Smith ierr = PetscFree(a->sor_work);CHKERRQ(ierr); 14205b42c5fSBarry Smith ierr = PetscFree(a->solves_work);CHKERRQ(ierr); 14305b42c5fSBarry Smith ierr = PetscFree(a->mult_work);CHKERRQ(ierr); 14405b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 14505b42c5fSBarry Smith ierr = PetscFree(a->xtoy);CHKERRQ(ierr); 1464da8f245SBarry Smith if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);} 1471a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1486bf464f9SBarry Smith ierr = MatDestroy(&a->parent);CHKERRQ(ierr); 149bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 150901853e0SKris Buschelman 151dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 152bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 153bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 154bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr); 155bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr); 156bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr); 157bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr); 15838f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr); 159bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr); 16049b5e25fSSatish Balay PetscFunctionReturn(0); 16149b5e25fSSatish Balay } 16249b5e25fSSatish Balay 1634a2ae208SSatish Balay #undef __FUNCT__ 1644a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ" 165ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg) 16649b5e25fSSatish Balay { 167045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 16863ba0a88SBarry Smith PetscErrorCode ierr; 16949b5e25fSSatish Balay 17049b5e25fSSatish Balay PetscFunctionBegin; 1714d9d31abSKris Buschelman switch (op) { 1724d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 1734e0d8c25SBarry Smith a->roworiented = flg; 1744d9d31abSKris Buschelman break; 175a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 176a9817697SBarry Smith a->keepnonzeropattern = flg; 1774d9d31abSKris Buschelman break; 178512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 179512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 1804d9d31abSKris Buschelman break; 1814d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 1824e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 1834d9d31abSKris Buschelman break; 1844d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 1854e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 1864d9d31abSKris Buschelman break; 18728b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 18828b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 18928b2fa4aSMatthew Knepley break; 1904e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1914d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1924d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 193290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 1944d9d31abSKris Buschelman break; 1959a4540c5SBarry Smith case MAT_HERMITIAN: 196e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first"); 1970bc54ff2SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 198eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort; 1990bc54ff2SBarry Smith } else if (A->cmap->bs == 1) { 200eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian; 201e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1"); 202eeffb40dSHong Zhang break; 2033d472b54SHong Zhang case MAT_SPD: 2045021d80fSJed Brown /* These options are handled directly by MatSetOption() */ 2053d472b54SHong Zhang break; 20677e54ba9SKris Buschelman case MAT_SYMMETRIC: 20777e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 2089a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 209e32f2f54SBarry Smith if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix must be symmetric"); 210290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr); 211290bbb0aSBarry Smith break; 212941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 2134e0d8c25SBarry Smith a->ignore_ltriangular = flg; 214941593c8SHong Zhang break; 215941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 2164e0d8c25SBarry Smith a->ignore_ltriangular = flg; 21777e54ba9SKris Buschelman break; 218f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 2194e0d8c25SBarry Smith a->getrow_utriangular = flg; 220f5edf698SHong Zhang break; 2214d9d31abSKris Buschelman default: 222e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 22349b5e25fSSatish Balay } 22449b5e25fSSatish Balay PetscFunctionReturn(0); 22549b5e25fSSatish Balay } 22649b5e25fSSatish Balay 2274a2ae208SSatish Balay #undef __FUNCT__ 2284a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ" 22913f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v) 23049b5e25fSSatish Balay { 23149b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2326849ba73SBarry Smith PetscErrorCode ierr; 23313f74950SBarry Smith PetscInt itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2; 23449b5e25fSSatish Balay MatScalar *aa,*aa_i; 23587828ca2SBarry Smith PetscScalar *v_i; 23649b5e25fSSatish Balay 23749b5e25fSSatish Balay PetscFunctionBegin; 238e32f2f54SBarry Smith if (A && !a->getrow_utriangular) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRow is not supported for SBAIJ matrix format. Getting the upper triangular part of row, run with -mat_getrow_uppertriangular, call MatSetOption(mat,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE) or MatGetRowUpperTriangular()"); 239f5edf698SHong Zhang /* Get the upper triangular part of the row */ 240d0f46423SBarry Smith bs = A->rmap->bs; 24149b5e25fSSatish Balay ai = a->i; 24249b5e25fSSatish Balay aj = a->j; 24349b5e25fSSatish Balay aa = a->a; 24449b5e25fSSatish Balay bs2 = a->bs2; 24549b5e25fSSatish Balay 246e32f2f54SBarry Smith if (row < 0 || row >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row); 24749b5e25fSSatish Balay 24849b5e25fSSatish Balay bn = row/bs; /* Block number */ 24949b5e25fSSatish Balay bp = row % bs; /* Block position */ 25049b5e25fSSatish Balay M = ai[bn+1] - ai[bn]; 25149b5e25fSSatish Balay *ncols = bs*M; 25249b5e25fSSatish Balay 25349b5e25fSSatish Balay if (v) { 25449b5e25fSSatish Balay *v = 0; 25549b5e25fSSatish Balay if (*ncols) { 256785e854fSJed Brown ierr = PetscMalloc1((*ncols+row),v);CHKERRQ(ierr); 25749b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 25849b5e25fSSatish Balay v_i = *v + i*bs; 25949b5e25fSSatish Balay aa_i = aa + bs2*(ai[bn] + i); 26026fbe8dcSKarl Rupp for (j=bp,k=0; j<bs2; j+=bs,k++) v_i[k] = aa_i[j]; 26149b5e25fSSatish Balay } 26249b5e25fSSatish Balay } 26349b5e25fSSatish Balay } 26449b5e25fSSatish Balay 26549b5e25fSSatish Balay if (cols) { 26649b5e25fSSatish Balay *cols = 0; 26749b5e25fSSatish Balay if (*ncols) { 268785e854fSJed Brown ierr = PetscMalloc1((*ncols+row),cols);CHKERRQ(ierr); 26949b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 27049b5e25fSSatish Balay cols_i = *cols + i*bs; 27149b5e25fSSatish Balay itmp = bs*aj[ai[bn] + i]; 27226fbe8dcSKarl Rupp for (j=0; j<bs; j++) cols_i[j] = itmp++; 27349b5e25fSSatish Balay } 27449b5e25fSSatish Balay } 27549b5e25fSSatish Balay } 27649b5e25fSSatish Balay 27749b5e25fSSatish Balay /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */ 2785ddb2528SHong Zhang /* this segment is currently removed, so only entries in the upper triangle are obtained */ 279519f805aSKarl Rupp #if defined(column_search) 28049b5e25fSSatish Balay v_i = *v + M*bs; 28149b5e25fSSatish Balay cols_i = *cols + M*bs; 28249b5e25fSSatish Balay for (i=0; i<bn; i++) { /* for each block row */ 28349b5e25fSSatish Balay M = ai[i+1] - ai[i]; 28449b5e25fSSatish Balay for (j=0; j<M; j++) { 28549b5e25fSSatish Balay itmp = aj[ai[i] + j]; /* block column value */ 28649b5e25fSSatish Balay if (itmp == bn) { 28749b5e25fSSatish Balay aa_i = aa + bs2*(ai[i] + j) + bs*bp; 28849b5e25fSSatish Balay for (k=0; k<bs; k++) { 28949b5e25fSSatish Balay *cols_i++ = i*bs+k; 29049b5e25fSSatish Balay *v_i++ = aa_i[k]; 29149b5e25fSSatish Balay } 29249b5e25fSSatish Balay *ncols += bs; 29349b5e25fSSatish Balay break; 29449b5e25fSSatish Balay } 29549b5e25fSSatish Balay } 29649b5e25fSSatish Balay } 2975ddb2528SHong Zhang #endif 29849b5e25fSSatish Balay PetscFunctionReturn(0); 29949b5e25fSSatish Balay } 30049b5e25fSSatish Balay 3014a2ae208SSatish Balay #undef __FUNCT__ 3024a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ" 30313f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 30449b5e25fSSatish Balay { 305dfbe8321SBarry Smith PetscErrorCode ierr; 30649b5e25fSSatish Balay 30749b5e25fSSatish Balay PetscFunctionBegin; 30805b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 30905b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 31049b5e25fSSatish Balay PetscFunctionReturn(0); 31149b5e25fSSatish Balay } 31249b5e25fSSatish Balay 3134a2ae208SSatish Balay #undef __FUNCT__ 314f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ" 315f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 316f5edf698SHong Zhang { 317f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 318f5edf698SHong Zhang 319f5edf698SHong Zhang PetscFunctionBegin; 320f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 321f5edf698SHong Zhang PetscFunctionReturn(0); 322f5edf698SHong Zhang } 323f5edf698SHong Zhang #undef __FUNCT__ 324f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ" 325f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 326f5edf698SHong Zhang { 327f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 328f5edf698SHong Zhang 329f5edf698SHong Zhang PetscFunctionBegin; 330f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 331f5edf698SHong Zhang PetscFunctionReturn(0); 332f5edf698SHong Zhang } 333f5edf698SHong Zhang 334f5edf698SHong Zhang #undef __FUNCT__ 3354a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ" 336fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B) 33749b5e25fSSatish Balay { 338dfbe8321SBarry Smith PetscErrorCode ierr; 3395fd66863SKarl Rupp 34049b5e25fSSatish Balay PetscFunctionBegin; 341815cbec1SBarry Smith if (reuse == MAT_INITIAL_MATRIX || *B != A) { 342999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 343fc4dec0aSBarry Smith } 3448115998fSBarry Smith PetscFunctionReturn(0); 34549b5e25fSSatish Balay } 34649b5e25fSSatish Balay 3474a2ae208SSatish Balay #undef __FUNCT__ 3484a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII" 349*7da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 35049b5e25fSSatish Balay { 35149b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 352dfbe8321SBarry Smith PetscErrorCode ierr; 353d0f46423SBarry Smith PetscInt i,j,bs = A->rmap->bs,k,l,bs2=a->bs2; 354f3ef73ceSBarry Smith PetscViewerFormat format; 355121deb67SSatish Balay PetscInt *diag; 35649b5e25fSSatish Balay 35749b5e25fSSatish Balay PetscFunctionBegin; 358b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 359456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 36077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 361fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 362d2507d54SMatthew Knepley Mat aij; 363d5f3da31SBarry Smith if (A->factortype && bs>1) { 36470d5e725SHong Zhang ierr = PetscPrintf(PETSC_COMM_SELF,"Warning: matrix is factored with bs>1. MatView() with PETSC_VIEWER_ASCII_MATLAB is not supported and ignored!\n");CHKERRQ(ierr); 36570d5e725SHong Zhang PetscFunctionReturn(0); 36670d5e725SHong Zhang } 367c9f458caSMatthew Knepley ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr); 368c9f458caSMatthew Knepley ierr = MatView(aij,viewer);CHKERRQ(ierr); 3696bf464f9SBarry Smith ierr = MatDestroy(&aij);CHKERRQ(ierr); 370fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 371d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 37249b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 37349b5e25fSSatish Balay for (j=0; j<bs; j++) { 37477431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 37549b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 37649b5e25fSSatish Balay for (l=0; l<bs; l++) { 37749b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 37849b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 37957622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 38057622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38149b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 38257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 38357622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38449b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 38557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38649b5e25fSSatish Balay } 38749b5e25fSSatish Balay #else 38849b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 38957622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 39049b5e25fSSatish Balay } 39149b5e25fSSatish Balay #endif 39249b5e25fSSatish Balay } 39349b5e25fSSatish Balay } 394b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 39549b5e25fSSatish Balay } 39649b5e25fSSatish Balay } 397d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 398c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 399c1490034SHong Zhang PetscFunctionReturn(0); 40049b5e25fSSatish Balay } else { 401d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 4022c990fa1SHong Zhang if (A->factortype) { /* for factored matrix */ 4032c990fa1SHong Zhang if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet"); 4042c990fa1SHong Zhang 405121deb67SSatish Balay diag=a->diag; 406121deb67SSatish Balay for (i=0; i<a->mbs; i++) { /* for row block i */ 4072c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 4082c990fa1SHong Zhang /* diagonal entry */ 4092c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 4102c990fa1SHong Zhang if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) { 41157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]),(double)PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4122c990fa1SHong Zhang } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) { 41357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]),-(double)PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4142c990fa1SHong Zhang } else { 41557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4162c990fa1SHong Zhang } 4172c990fa1SHong Zhang #else 4186712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4192c990fa1SHong Zhang #endif 4202c990fa1SHong Zhang /* off-diagonal entries */ 4212c990fa1SHong Zhang for (k=a->i[i]; k<a->i[i+1]-1; k++) { 4222c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 423ca0704adSBarry Smith if (PetscImaginaryPart(a->a[k]) > 0.0) { 42457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 425ca0704adSBarry Smith } else if (PetscImaginaryPart(a->a[k]) < 0.0) { 42657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 4272c990fa1SHong Zhang } else { 42857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr); 4292c990fa1SHong Zhang } 4302c990fa1SHong Zhang #else 43157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr); 4322c990fa1SHong Zhang #endif 4332c990fa1SHong Zhang } 4342c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 4352c990fa1SHong Zhang } 4362c990fa1SHong Zhang 4372c990fa1SHong Zhang } else { /* for non-factored matrix */ 4380c74a584SJed Brown for (i=0; i<a->mbs; i++) { /* for row block i */ 4390c74a584SJed Brown for (j=0; j<bs; j++) { /* for row bs*i + j */ 44077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 4410c74a584SJed Brown for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */ 4420c74a584SJed Brown for (l=0; l<bs; l++) { /* for column */ 44349b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 44449b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 44557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 44657622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 44749b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 44857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 44957622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 45049b5e25fSSatish Balay } else { 45157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 45249b5e25fSSatish Balay } 45349b5e25fSSatish Balay #else 45457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 45549b5e25fSSatish Balay #endif 45649b5e25fSSatish Balay } 45749b5e25fSSatish Balay } 458b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 45949b5e25fSSatish Balay } 46049b5e25fSSatish Balay } 4612c990fa1SHong Zhang } 462d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 46349b5e25fSSatish Balay } 464b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 46549b5e25fSSatish Balay PetscFunctionReturn(0); 46649b5e25fSSatish Balay } 46749b5e25fSSatish Balay 4689804daf3SBarry Smith #include <petscdraw.h> 4694a2ae208SSatish Balay #undef __FUNCT__ 4704a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom" 4716849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 47249b5e25fSSatish Balay { 47349b5e25fSSatish Balay Mat A = (Mat) Aa; 47449b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 4756849ba73SBarry Smith PetscErrorCode ierr; 476d0f46423SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2; 47713f74950SBarry Smith PetscMPIInt rank; 47849b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 47949b5e25fSSatish Balay MatScalar *aa; 48049b5e25fSSatish Balay MPI_Comm comm; 481b0a32e0cSBarry Smith PetscViewer viewer; 48249b5e25fSSatish Balay 48349b5e25fSSatish Balay PetscFunctionBegin; 48449b5e25fSSatish Balay /* 48549b5e25fSSatish Balay This is nasty. If this is called from an originally parallel matrix 48649b5e25fSSatish Balay then all processes call this,but only the first has the matrix so the 48749b5e25fSSatish Balay rest should return immediately. 48849b5e25fSSatish Balay */ 48949b5e25fSSatish Balay ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr); 49049b5e25fSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 49149b5e25fSSatish Balay if (rank) PetscFunctionReturn(0); 49249b5e25fSSatish Balay 49349b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 49449b5e25fSSatish Balay 495b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 496b0a32e0cSBarry Smith PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric"); 49749b5e25fSSatish Balay 49849b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 499b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 50049b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 50149b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 502d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 50349b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 50449b5e25fSSatish Balay aa = a->a + j*bs2; 50549b5e25fSSatish Balay for (k=0; k<bs; k++) { 50649b5e25fSSatish Balay for (l=0; l<bs; l++) { 50749b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 508b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 50949b5e25fSSatish Balay } 51049b5e25fSSatish Balay } 51149b5e25fSSatish Balay } 51249b5e25fSSatish Balay } 513b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 51449b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 51549b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 516d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 51749b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 51849b5e25fSSatish Balay aa = a->a + j*bs2; 51949b5e25fSSatish Balay for (k=0; k<bs; k++) { 52049b5e25fSSatish Balay for (l=0; l<bs; l++) { 52149b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 522b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 52349b5e25fSSatish Balay } 52449b5e25fSSatish Balay } 52549b5e25fSSatish Balay } 52649b5e25fSSatish Balay } 52749b5e25fSSatish Balay 528b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 52949b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 53049b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 531d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 53249b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 53349b5e25fSSatish Balay aa = a->a + j*bs2; 53449b5e25fSSatish Balay for (k=0; k<bs; k++) { 53549b5e25fSSatish Balay for (l=0; l<bs; l++) { 53649b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 537b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 53849b5e25fSSatish Balay } 53949b5e25fSSatish Balay } 54049b5e25fSSatish Balay } 54149b5e25fSSatish Balay } 54249b5e25fSSatish Balay PetscFunctionReturn(0); 54349b5e25fSSatish Balay } 54449b5e25fSSatish Balay 5454a2ae208SSatish Balay #undef __FUNCT__ 5464a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw" 5476849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 54849b5e25fSSatish Balay { 549dfbe8321SBarry Smith PetscErrorCode ierr; 55049b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 551b0a32e0cSBarry Smith PetscDraw draw; 552ace3abfcSBarry Smith PetscBool isnull; 55349b5e25fSSatish Balay 55449b5e25fSSatish Balay PetscFunctionBegin; 555b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 556b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 55749b5e25fSSatish Balay 55849b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 559d0f46423SBarry Smith xr = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0; 56049b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 561b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 562b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 5630298fd71SBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 56449b5e25fSSatish Balay PetscFunctionReturn(0); 56549b5e25fSSatish Balay } 56649b5e25fSSatish Balay 5674a2ae208SSatish Balay #undef __FUNCT__ 5684a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ" 569dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 57049b5e25fSSatish Balay { 571dfbe8321SBarry Smith PetscErrorCode ierr; 572ace3abfcSBarry Smith PetscBool iascii,isdraw; 57308917f38SBarry Smith FILE *file = 0; 57449b5e25fSSatish Balay 57549b5e25fSSatish Balay PetscFunctionBegin; 576251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 577251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 57832077d6dSBarry Smith if (iascii) { 57949b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 58049b5e25fSSatish Balay } else if (isdraw) { 58149b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 58249b5e25fSSatish Balay } else { 583a5e6ed63SBarry Smith Mat B; 584ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 585a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 5866bf464f9SBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 58708917f38SBarry Smith ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr); 58808917f38SBarry Smith if (file) { 58908917f38SBarry Smith fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs); 59008917f38SBarry Smith } 59149b5e25fSSatish Balay } 59249b5e25fSSatish Balay PetscFunctionReturn(0); 59349b5e25fSSatish Balay } 59449b5e25fSSatish Balay 59549b5e25fSSatish Balay 5964a2ae208SSatish Balay #undef __FUNCT__ 5974a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ" 59813f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 59949b5e25fSSatish Balay { 600045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 60113f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 60213f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 603d0f46423SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2; 60497e567efSBarry Smith MatScalar *ap,*aa = a->a; 60549b5e25fSSatish Balay 60649b5e25fSSatish Balay PetscFunctionBegin; 60749b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 60849b5e25fSSatish Balay row = im[k]; brow = row/bs; 609e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 610e32f2f54SBarry Smith if (row >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1); 61149b5e25fSSatish Balay rp = aj + ai[brow]; ap = aa + bs2*ai[brow]; 61249b5e25fSSatish Balay nrow = ailen[brow]; 61349b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 614e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 615e32f2f54SBarry Smith if (in[l] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap->n-1); 61649b5e25fSSatish Balay col = in[l]; 61749b5e25fSSatish Balay bcol = col/bs; 61849b5e25fSSatish Balay cidx = col%bs; 61949b5e25fSSatish Balay ridx = row%bs; 62049b5e25fSSatish Balay high = nrow; 62149b5e25fSSatish Balay low = 0; /* assume unsorted */ 62249b5e25fSSatish Balay while (high-low > 5) { 62349b5e25fSSatish Balay t = (low+high)/2; 62449b5e25fSSatish Balay if (rp[t] > bcol) high = t; 62549b5e25fSSatish Balay else low = t; 62649b5e25fSSatish Balay } 62749b5e25fSSatish Balay for (i=low; i<high; i++) { 62849b5e25fSSatish Balay if (rp[i] > bcol) break; 62949b5e25fSSatish Balay if (rp[i] == bcol) { 63049b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 63149b5e25fSSatish Balay goto finished; 63249b5e25fSSatish Balay } 63349b5e25fSSatish Balay } 63497e567efSBarry Smith *v++ = 0.0; 63549b5e25fSSatish Balay finished:; 63649b5e25fSSatish Balay } 63749b5e25fSSatish Balay } 63849b5e25fSSatish Balay PetscFunctionReturn(0); 63949b5e25fSSatish Balay } 64049b5e25fSSatish Balay 64149b5e25fSSatish Balay 6424a2ae208SSatish Balay #undef __FUNCT__ 6434a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ" 64413f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 64549b5e25fSSatish Balay { 6460880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 6476849ba73SBarry Smith PetscErrorCode ierr; 648e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 64913f74950SBarry Smith PetscInt *imax =a->imax,*ai=a->i,*ailen=a->ilen; 650d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval; 651ace3abfcSBarry Smith PetscBool roworiented=a->roworiented; 652dd6ea824SBarry Smith const PetscScalar *value = v; 653f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 6540880e062SHong Zhang 65549b5e25fSSatish Balay PetscFunctionBegin; 65626fbe8dcSKarl Rupp if (roworiented) stepval = (n-1)*bs; 65726fbe8dcSKarl Rupp else stepval = (m-1)*bs; 65826fbe8dcSKarl Rupp 6590880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 6600880e062SHong Zhang row = im[k]; 6610880e062SHong Zhang if (row < 0) continue; 6622515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 663e32f2f54SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1); 6640880e062SHong Zhang #endif 6650880e062SHong Zhang rp = aj + ai[row]; 6660880e062SHong Zhang ap = aa + bs2*ai[row]; 6670880e062SHong Zhang rmax = imax[row]; 6680880e062SHong Zhang nrow = ailen[row]; 6690880e062SHong Zhang low = 0; 670818f2c47SBarry Smith high = nrow; 6710880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 6720880e062SHong Zhang if (in[l] < 0) continue; 6730880e062SHong Zhang col = in[l]; 6742515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 675e32f2f54SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1); 676b1823623SSatish Balay #endif 677b98bf0e1SJed Brown if (col < row) { 67826fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular block */ 67926fbe8dcSKarl Rupp else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_USER,"Lower triangular value cannot be set for sbaij format. Ignoring these values, run with -mat_ignore_lower_triangular or call MatSetOption(mat,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_TRUE)"); 680b98bf0e1SJed Brown } 68126fbe8dcSKarl Rupp if (roworiented) value = v + k*(stepval+bs)*bs + l*bs; 68226fbe8dcSKarl Rupp else value = v + l*(stepval+bs)*bs + k*bs; 68326fbe8dcSKarl Rupp 68426fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 68526fbe8dcSKarl Rupp else high = nrow; 68626fbe8dcSKarl Rupp 687e2ee6c50SBarry Smith lastcol = col; 6880880e062SHong Zhang while (high-low > 7) { 6890880e062SHong Zhang t = (low+high)/2; 6900880e062SHong Zhang if (rp[t] > col) high = t; 6910880e062SHong Zhang else low = t; 6920880e062SHong Zhang } 6930880e062SHong Zhang for (i=low; i<high; i++) { 6940880e062SHong Zhang if (rp[i] > col) break; 6950880e062SHong Zhang if (rp[i] == col) { 6960880e062SHong Zhang bap = ap + bs2*i; 6970880e062SHong Zhang if (roworiented) { 6980880e062SHong Zhang if (is == ADD_VALUES) { 6990880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7000880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7010880e062SHong Zhang bap[jj] += *value++; 7020880e062SHong Zhang } 7030880e062SHong Zhang } 7040880e062SHong Zhang } else { 7050880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7060880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7070880e062SHong Zhang bap[jj] = *value++; 7080880e062SHong Zhang } 7090880e062SHong Zhang } 7100880e062SHong Zhang } 7110880e062SHong Zhang } else { 7120880e062SHong Zhang if (is == ADD_VALUES) { 7130880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7140880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7150880e062SHong Zhang *bap++ += *value++; 7160880e062SHong Zhang } 7170880e062SHong Zhang } 7180880e062SHong Zhang } else { 7190880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7200880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7210880e062SHong Zhang *bap++ = *value++; 7220880e062SHong Zhang } 7230880e062SHong Zhang } 7240880e062SHong Zhang } 7250880e062SHong Zhang } 7260880e062SHong Zhang goto noinsert2; 7270880e062SHong Zhang } 7280880e062SHong Zhang } 7290880e062SHong Zhang if (nonew == 1) goto noinsert2; 730e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 731fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 732c03d1d03SSatish Balay N = nrow++ - 1; high++; 7330880e062SHong Zhang /* shift up all the later entries in this row */ 7340880e062SHong Zhang for (ii=N; ii>=i; ii--) { 7350880e062SHong Zhang rp[ii+1] = rp[ii]; 7360880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 7370880e062SHong Zhang } 7380880e062SHong Zhang if (N >= i) { 7390880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 7400880e062SHong Zhang } 7410880e062SHong Zhang rp[i] = col; 7420880e062SHong Zhang bap = ap + bs2*i; 7430880e062SHong Zhang if (roworiented) { 7440880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7450880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7460880e062SHong Zhang bap[jj] = *value++; 7470880e062SHong Zhang } 7480880e062SHong Zhang } 7490880e062SHong Zhang } else { 7500880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7510880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7520880e062SHong Zhang *bap++ = *value++; 7530880e062SHong Zhang } 7540880e062SHong Zhang } 7550880e062SHong Zhang } 7560880e062SHong Zhang noinsert2:; 7570880e062SHong Zhang low = i; 7580880e062SHong Zhang } 7590880e062SHong Zhang ailen[row] = nrow; 7600880e062SHong Zhang } 7610880e062SHong Zhang PetscFunctionReturn(0); 76249b5e25fSSatish Balay } 76349b5e25fSSatish Balay 76464831d72SBarry Smith /* 76564831d72SBarry Smith This is not yet used 76664831d72SBarry Smith */ 7674a2ae208SSatish Balay #undef __FUNCT__ 7684108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode" 7694108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A) 7700def2e27SBarry Smith { 7710def2e27SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7720def2e27SBarry Smith PetscErrorCode ierr; 7730def2e27SBarry Smith const PetscInt *ai = a->i, *aj = a->j,*cols; 7740def2e27SBarry Smith PetscInt i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts; 775ace3abfcSBarry Smith PetscBool flag; 7760def2e27SBarry Smith 7770def2e27SBarry Smith PetscFunctionBegin; 778785e854fSJed Brown ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr); 7790def2e27SBarry Smith while (i < m) { 7800def2e27SBarry Smith nzx = ai[i+1] - ai[i]; /* Number of nonzeros */ 7810def2e27SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 7820def2e27SBarry Smith for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 7830def2e27SBarry Smith nzy = ai[j+1] - ai[j]; 7840def2e27SBarry Smith if (nzy != (nzx - j + i)) break; 7850def2e27SBarry Smith ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr); 7860def2e27SBarry Smith if (!flag) break; 7870def2e27SBarry Smith } 7880def2e27SBarry Smith ns[node_count++] = blk_size; 78926fbe8dcSKarl Rupp 7900def2e27SBarry Smith i = j; 7910def2e27SBarry Smith } 7920def2e27SBarry Smith if (!a->inode.size && m && node_count > .9*m) { 7930def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7940def2e27SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 7950def2e27SBarry Smith } else { 7960def2e27SBarry Smith a->inode.node_count = node_count; 79726fbe8dcSKarl Rupp 798785e854fSJed Brown ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr); 7993bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr); 80022d28d08SBarry Smith ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr); 8010def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 8020def2e27SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 8030def2e27SBarry Smith 8040def2e27SBarry Smith /* count collections of adjacent columns in each inode */ 8050def2e27SBarry Smith row = 0; 8060def2e27SBarry Smith cnt = 0; 8070def2e27SBarry Smith for (i=0; i<node_count; i++) { 8080def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 8090def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 8100def2e27SBarry Smith for (j=1; j<nz; j++) { 81126fbe8dcSKarl Rupp if (cols[j] != cols[j-1]+1) cnt++; 8120def2e27SBarry Smith } 8130def2e27SBarry Smith cnt++; 8140def2e27SBarry Smith row += a->inode.size[i]; 8150def2e27SBarry Smith } 816785e854fSJed Brown ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr); 8170def2e27SBarry Smith cnt = 0; 8180def2e27SBarry Smith row = 0; 8190def2e27SBarry Smith for (i=0; i<node_count; i++) { 8200def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 8210def2e27SBarry Smith counts[2*cnt] = cols[0]; 8220def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 8230def2e27SBarry Smith cnt2 = 1; 8240def2e27SBarry Smith for (j=1; j<nz; j++) { 8250def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 8260def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 8270def2e27SBarry Smith counts[2*cnt] = cols[j]; 8280def2e27SBarry Smith cnt2 = 1; 8290def2e27SBarry Smith } else cnt2++; 8300def2e27SBarry Smith } 8310def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 8320def2e27SBarry Smith row += a->inode.size[i]; 8330def2e27SBarry Smith } 83422d28d08SBarry Smith ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr); 8350def2e27SBarry Smith } 83638702af4SBarry Smith PetscFunctionReturn(0); 83738702af4SBarry Smith } 83838702af4SBarry Smith 83938702af4SBarry Smith #undef __FUNCT__ 8404a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ" 841dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 84249b5e25fSSatish Balay { 84349b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 8446849ba73SBarry Smith PetscErrorCode ierr; 84513f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 846d0f46423SBarry Smith PetscInt m = A->rmap->N,*ip,N,*ailen = a->ilen; 84713f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 84849b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 84949b5e25fSSatish Balay 85049b5e25fSSatish Balay PetscFunctionBegin; 85149b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 85249b5e25fSSatish Balay 85349b5e25fSSatish Balay if (m) rmax = ailen[0]; 85449b5e25fSSatish Balay for (i=1; i<mbs; i++) { 85549b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 85649b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 85749b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 85849b5e25fSSatish Balay if (fshift) { 85949b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 86049b5e25fSSatish Balay N = ailen[i]; 86149b5e25fSSatish Balay for (j=0; j<N; j++) { 86249b5e25fSSatish Balay ip[j-fshift] = ip[j]; 86349b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 86449b5e25fSSatish Balay } 86549b5e25fSSatish Balay } 86649b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 86749b5e25fSSatish Balay } 86849b5e25fSSatish Balay if (mbs) { 86949b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 87049b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 87149b5e25fSSatish Balay } 87249b5e25fSSatish Balay /* reset ilen and imax for each row */ 87349b5e25fSSatish Balay for (i=0; i<mbs; i++) { 87449b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 87549b5e25fSSatish Balay } 8766c6c5352SBarry Smith a->nz = ai[mbs]; 87749b5e25fSSatish Balay 878b424e231SHong Zhang /* diagonals may have moved, reset it */ 879b424e231SHong Zhang if (a->diag) { 8802ed38d0bSJed Brown ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr); 88149b5e25fSSatish Balay } 88226fbe8dcSKarl Rupp if (fshift && a->nounused == -1) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_PLIB, "Unused space detected in matrix: %D X %D block size %D, %D unneeded", m, A->cmap->n, A->rmap->bs, fshift*bs2); 88326fbe8dcSKarl Rupp 884d0f46423SBarry Smith ierr = PetscInfo5(A,"Matrix size: %D X %D, block size %D; storage space: %D unneeded, %D used\n",m,A->rmap->N,A->rmap->bs,fshift*bs2,a->nz*bs2);CHKERRQ(ierr); 885ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 886ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 88726fbe8dcSKarl Rupp 8888e58a170SBarry Smith A->info.mallocs += a->reallocs; 88949b5e25fSSatish Balay a->reallocs = 0; 89049b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 891061b2667SBarry Smith a->idiagvalid = PETSC_FALSE; 89238702af4SBarry Smith 89338702af4SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 89444e1c64aSLisandro Dalcin if (a->jshort && a->free_jshort) { 89517803ae8SHong Zhang /* when matrix data structure is changed, previous jshort must be replaced */ 89617803ae8SHong Zhang ierr = PetscFree(a->jshort);CHKERRQ(ierr); 89717803ae8SHong Zhang } 898785e854fSJed Brown ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr); 8993bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr); 90038702af4SBarry Smith for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i]; 90138702af4SBarry Smith A->ops->mult = MatMult_SeqSBAIJ_1_ushort; 90241f059aeSBarry Smith A->ops->sor = MatSOR_SeqSBAIJ_ushort; 9034da8f245SBarry Smith a->free_jshort = PETSC_TRUE; 90438702af4SBarry Smith } 90549b5e25fSSatish Balay PetscFunctionReturn(0); 90649b5e25fSSatish Balay } 90749b5e25fSSatish Balay 90849b5e25fSSatish Balay /* 90949b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 91049b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 91149b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 91249b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 91349b5e25fSSatish Balay */ 9144a2ae208SSatish Balay #undef __FUNCT__ 9154a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks" 91613f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 91749b5e25fSSatish Balay { 91813f74950SBarry Smith PetscInt i,j,k,row; 919ace3abfcSBarry Smith PetscBool flg; 92049b5e25fSSatish Balay 92149b5e25fSSatish Balay PetscFunctionBegin; 92249b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 92349b5e25fSSatish Balay row = idx[i]; 92449b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 92549b5e25fSSatish Balay sizes[j] = 1; 92649b5e25fSSatish Balay i++; 92749b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 92849b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 92949b5e25fSSatish Balay i++; 93049b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 93149b5e25fSSatish Balay flg = PETSC_TRUE; 93249b5e25fSSatish Balay for (k=1; k<bs; k++) { 93349b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 93449b5e25fSSatish Balay flg = PETSC_FALSE; 93549b5e25fSSatish Balay break; 93649b5e25fSSatish Balay } 93749b5e25fSSatish Balay } 938abc0a331SBarry Smith if (flg) { /* No break in the bs */ 93949b5e25fSSatish Balay sizes[j] = bs; 94049b5e25fSSatish Balay i += bs; 94149b5e25fSSatish Balay } else { 94249b5e25fSSatish Balay sizes[j] = 1; 94349b5e25fSSatish Balay i++; 94449b5e25fSSatish Balay } 94549b5e25fSSatish Balay } 94649b5e25fSSatish Balay } 94749b5e25fSSatish Balay *bs_max = j; 94849b5e25fSSatish Balay PetscFunctionReturn(0); 94949b5e25fSSatish Balay } 95049b5e25fSSatish Balay 95149b5e25fSSatish Balay 95249b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 95349b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 95449b5e25fSSatish Balay */ 95549b5e25fSSatish Balay 9564a2ae208SSatish Balay #undef __FUNCT__ 9574a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ" 95813f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 95949b5e25fSSatish Balay { 96049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 9616849ba73SBarry Smith PetscErrorCode ierr; 962e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 96313f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 964d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol; 96513f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 96649b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 96749b5e25fSSatish Balay 96849b5e25fSSatish Balay PetscFunctionBegin; 96971fd2e92SBarry Smith if (v) PetscValidScalarPointer(v,6); 97049b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 97149b5e25fSSatish Balay row = im[k]; /* row number */ 97249b5e25fSSatish Balay brow = row/bs; /* block row number */ 97349b5e25fSSatish Balay if (row < 0) continue; 9742515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 975e32f2f54SBarry Smith if (row >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1); 97649b5e25fSSatish Balay #endif 97749b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 97849b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 97949b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 98049b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 98149b5e25fSSatish Balay low = 0; 98249b5e25fSSatish Balay 98349b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 98449b5e25fSSatish Balay if (in[l] < 0) continue; 9852515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 986e32f2f54SBarry Smith if (in[l] >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->rmap->N-1); 98749b5e25fSSatish Balay #endif 98849b5e25fSSatish Balay col = in[l]; 98949b5e25fSSatish Balay bcol = col/bs; /* block col number */ 99049b5e25fSSatish Balay 991941593c8SHong Zhang if (brow > bcol) { 99226fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular values */ 99326fbe8dcSKarl Rupp else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_USER,"Lower triangular value cannot be set for sbaij format. Ignoring these values, run with -mat_ignore_lower_triangular or call MatSetOption(mat,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_TRUE)"); 994941593c8SHong Zhang } 995f4989cb3SHong Zhang 99649b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 9978549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)) { 99849b5e25fSSatish Balay /* element value a(k,l) */ 99926fbe8dcSKarl Rupp if (roworiented) value = v[l + k*n]; 100026fbe8dcSKarl Rupp else value = v[k + l*m]; 100149b5e25fSSatish Balay 100249b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 100326fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 100426fbe8dcSKarl Rupp high = nrow; 1005e2ee6c50SBarry Smith lastcol = col; 100649b5e25fSSatish Balay while (high-low > 7) { 100749b5e25fSSatish Balay t = (low+high)/2; 100849b5e25fSSatish Balay if (rp[t] > bcol) high = t; 100949b5e25fSSatish Balay else low = t; 101049b5e25fSSatish Balay } 101149b5e25fSSatish Balay for (i=low; i<high; i++) { 101249b5e25fSSatish Balay if (rp[i] > bcol) break; 101349b5e25fSSatish Balay if (rp[i] == bcol) { 101449b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 101549b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 101649b5e25fSSatish Balay else *bap = value; 10178549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 10188549e402SHong Zhang if (brow == bcol && ridx < cidx) { 10198549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 10208549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 10218549e402SHong Zhang else *bap = value; 10228549e402SHong Zhang } 102349b5e25fSSatish Balay goto noinsert1; 102449b5e25fSSatish Balay } 102549b5e25fSSatish Balay } 102649b5e25fSSatish Balay 102749b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 1028e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 1029fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 103049b5e25fSSatish Balay 1031c03d1d03SSatish Balay N = nrow++ - 1; high++; 103249b5e25fSSatish Balay /* shift up all the later entries in this row */ 103349b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 103449b5e25fSSatish Balay rp[ii+1] = rp[ii]; 103549b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 103649b5e25fSSatish Balay } 103749b5e25fSSatish Balay if (N>=i) { 103849b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 103949b5e25fSSatish Balay } 104049b5e25fSSatish Balay rp[i] = bcol; 104149b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 1042e56f5c9eSBarry Smith A->nonzerostate++; 104349b5e25fSSatish Balay noinsert1:; 104449b5e25fSSatish Balay low = i; 10458549e402SHong Zhang } 104649b5e25fSSatish Balay } /* end of loop over added columns */ 104749b5e25fSSatish Balay ailen[brow] = nrow; 104849b5e25fSSatish Balay } /* end of loop over added rows */ 104949b5e25fSSatish Balay PetscFunctionReturn(0); 105049b5e25fSSatish Balay } 105149b5e25fSSatish Balay 10524a2ae208SSatish Balay #undef __FUNCT__ 10534d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ" 10540481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info) 105549b5e25fSSatish Balay { 10564ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 105749b5e25fSSatish Balay Mat outA; 1058dfbe8321SBarry Smith PetscErrorCode ierr; 1059ace3abfcSBarry Smith PetscBool row_identity; 106049b5e25fSSatish Balay 106149b5e25fSSatish Balay PetscFunctionBegin; 1062e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 1063c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 1064e32f2f54SBarry Smith if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported"); 1065e32f2f54SBarry Smith if (inA->rmap->bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix block size %D is not supported",inA->rmap->bs); /* Need to replace MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()! */ 1066c84f5b01SHong Zhang 106749b5e25fSSatish Balay outA = inA; 1068d5f3da31SBarry Smith inA->factortype = MAT_FACTOR_ICC; 106949b5e25fSSatish Balay 10701a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 1071d595f711SHong Zhang ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr); 107249b5e25fSSatish Balay 1073c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10746bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1075c84f5b01SHong Zhang a->row = row; 1076c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10776bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1078c84f5b01SHong Zhang a->col = row; 1079c84f5b01SHong Zhang 1080c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1081c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 10823bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr); 108349b5e25fSSatish Balay 108449b5e25fSSatish Balay if (!a->solve_work) { 1085785e854fSJed Brown ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr); 10863bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 108749b5e25fSSatish Balay } 108849b5e25fSSatish Balay 1089719d5645SBarry Smith ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr); 109049b5e25fSSatish Balay PetscFunctionReturn(0); 109149b5e25fSSatish Balay } 1092950f1e5bSHong Zhang 10934a2ae208SSatish Balay #undef __FUNCT__ 10944a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ" 10957087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 109649b5e25fSSatish Balay { 1097045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ*)mat->data; 109813f74950SBarry Smith PetscInt i,nz,n; 10997827cd58SJed Brown PetscErrorCode ierr; 110049b5e25fSSatish Balay 110149b5e25fSSatish Balay PetscFunctionBegin; 11026c6c5352SBarry Smith nz = baij->maxnz; 1103d0f46423SBarry Smith n = mat->cmap->n; 110426fbe8dcSKarl Rupp for (i=0; i<nz; i++) baij->j[i] = indices[i]; 110526fbe8dcSKarl Rupp 11066c6c5352SBarry Smith baij->nz = nz; 110726fbe8dcSKarl Rupp for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i]; 110826fbe8dcSKarl Rupp 11097827cd58SJed Brown ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 111049b5e25fSSatish Balay PetscFunctionReturn(0); 111149b5e25fSSatish Balay } 111249b5e25fSSatish Balay 11134a2ae208SSatish Balay #undef __FUNCT__ 11144a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices" 111549b5e25fSSatish Balay /*@ 111619585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 111749b5e25fSSatish Balay in the matrix. 111849b5e25fSSatish Balay 111949b5e25fSSatish Balay Input Parameters: 112019585528SSatish Balay + mat - the SeqSBAIJ matrix 112149b5e25fSSatish Balay - indices - the column indices 112249b5e25fSSatish Balay 112349b5e25fSSatish Balay Level: advanced 112449b5e25fSSatish Balay 112549b5e25fSSatish Balay Notes: 112649b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 112749b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 112849b5e25fSSatish Balay of the MatSetValues() operation. 112949b5e25fSSatish Balay 113049b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1131d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 113249b5e25fSSatish Balay 1133ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 113449b5e25fSSatish Balay 1135ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 113649b5e25fSSatish Balay @*/ 11377087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 113849b5e25fSSatish Balay { 11394ac538c5SBarry Smith PetscErrorCode ierr; 114049b5e25fSSatish Balay 114149b5e25fSSatish Balay PetscFunctionBegin; 11420700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 11434482741eSBarry Smith PetscValidPointer(indices,2); 11444ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr); 114549b5e25fSSatish Balay PetscFunctionReturn(0); 114649b5e25fSSatish Balay } 114749b5e25fSSatish Balay 11484a2ae208SSatish Balay #undef __FUNCT__ 11493c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ" 11503c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 11513c896bc6SHong Zhang { 11523c896bc6SHong Zhang PetscErrorCode ierr; 11533c896bc6SHong Zhang 11543c896bc6SHong Zhang PetscFunctionBegin; 11553c896bc6SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 11563c896bc6SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 11573c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11583c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11593c896bc6SHong Zhang 1160e7e72b3dSBarry Smith if (a->i[A->rmap->N] != b->i[B->rmap->N]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 1161d0f46423SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr); 11623c896bc6SHong Zhang } else { 1163f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11643c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1165f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11663c896bc6SHong Zhang } 11673c896bc6SHong Zhang PetscFunctionReturn(0); 11683c896bc6SHong Zhang } 11693c896bc6SHong Zhang 11703c896bc6SHong Zhang #undef __FUNCT__ 11714994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ" 11724994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A) 1173273d9f13SBarry Smith { 1174dfbe8321SBarry Smith PetscErrorCode ierr; 1175273d9f13SBarry Smith 1176273d9f13SBarry Smith PetscFunctionBegin; 1177535b19f3SBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr); 1178273d9f13SBarry Smith PetscFunctionReturn(0); 1179273d9f13SBarry Smith } 1180273d9f13SBarry Smith 1181a6ece127SHong Zhang #undef __FUNCT__ 11828c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ" 11838c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1184a6ece127SHong Zhang { 1185a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11865fd66863SKarl Rupp 1187a6ece127SHong Zhang PetscFunctionBegin; 1188a6ece127SHong Zhang *array = a->a; 1189a6ece127SHong Zhang PetscFunctionReturn(0); 1190a6ece127SHong Zhang } 1191a6ece127SHong Zhang 1192a6ece127SHong Zhang #undef __FUNCT__ 11938c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ" 11948c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1195a6ece127SHong Zhang { 1196a6ece127SHong Zhang PetscFunctionBegin; 1197a6ece127SHong Zhang PetscFunctionReturn(0); 1198a6ece127SHong Zhang } 1199a6ece127SHong Zhang 120042ee4b1aSHong Zhang #undef __FUNCT__ 120142ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ" 1202f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 120342ee4b1aSHong Zhang { 120442ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data; 1205dfbe8321SBarry Smith PetscErrorCode ierr; 1206e838b9e7SJed Brown PetscInt i,bs=Y->rmap->bs,bs2=bs*bs,j; 1207e838b9e7SJed Brown PetscBLASInt one = 1; 120842ee4b1aSHong Zhang 120942ee4b1aSHong Zhang PetscFunctionBegin; 121042ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1211f4df32b1SMatthew Knepley PetscScalar alpha = a; 1212c5df96a5SBarry Smith PetscBLASInt bnz; 1213c5df96a5SBarry Smith ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr); 12148b83055fSJed Brown PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one)); 1215c537a176SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1216c4319e64SHong Zhang if (y->xtoy && y->XtoY != X) { 1217c4319e64SHong Zhang ierr = PetscFree(y->xtoy);CHKERRQ(ierr); 12186bf464f9SBarry Smith ierr = MatDestroy(&y->XtoY);CHKERRQ(ierr); 1219c537a176SHong Zhang } 1220c4319e64SHong Zhang if (!y->xtoy) { /* get xtoy */ 12210298fd71SBarry Smith ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,NULL, y->i,y->j,NULL, &y->xtoy);CHKERRQ(ierr); 1222c4319e64SHong Zhang y->XtoY = X; 1223c537a176SHong Zhang } 12246c6c5352SBarry Smith for (i=0; i<x->nz; i++) { 1225c4319e64SHong Zhang j = 0; 1226c4319e64SHong Zhang while (j < bs2) { 1227f4df32b1SMatthew Knepley y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]); 1228c4319e64SHong Zhang j++; 1229c537a176SHong Zhang } 1230c4319e64SHong Zhang } 12316712e2f1SBarry Smith ierr = PetscInfo3(Y,"ratio of nnz_s(X)/nnz_s(Y): %D/%D = %g\n",bs2*x->nz,bs2*y->nz,(double)((PetscReal)(bs2*x->nz)/(PetscReal)(bs2*y->nz)));CHKERRQ(ierr); 123242ee4b1aSHong Zhang } else { 1233f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 1234f4df32b1SMatthew Knepley ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1235f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 123642ee4b1aSHong Zhang } 123742ee4b1aSHong Zhang PetscFunctionReturn(0); 123842ee4b1aSHong Zhang } 123942ee4b1aSHong Zhang 1240efcf0fc3SBarry Smith #undef __FUNCT__ 1241efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ" 1242ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1243efcf0fc3SBarry Smith { 1244efcf0fc3SBarry Smith PetscFunctionBegin; 1245efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1246efcf0fc3SBarry Smith PetscFunctionReturn(0); 1247efcf0fc3SBarry Smith } 1248efcf0fc3SBarry Smith 1249efcf0fc3SBarry Smith #undef __FUNCT__ 1250efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ" 1251ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool *flg) 1252efcf0fc3SBarry Smith { 1253efcf0fc3SBarry Smith PetscFunctionBegin; 1254efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1255efcf0fc3SBarry Smith PetscFunctionReturn(0); 1256efcf0fc3SBarry Smith } 1257efcf0fc3SBarry Smith 1258efcf0fc3SBarry Smith #undef __FUNCT__ 1259efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ" 1260ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1261efcf0fc3SBarry Smith { 1262efcf0fc3SBarry Smith PetscFunctionBegin; 1263efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1264efcf0fc3SBarry Smith PetscFunctionReturn(0); 1265efcf0fc3SBarry Smith } 1266efcf0fc3SBarry Smith 126799cafbc1SBarry Smith #undef __FUNCT__ 126899cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ" 126999cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 127099cafbc1SBarry Smith { 127199cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 127299cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1273dd6ea824SBarry Smith MatScalar *aa = a->a; 127499cafbc1SBarry Smith 127599cafbc1SBarry Smith PetscFunctionBegin; 127699cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 127799cafbc1SBarry Smith PetscFunctionReturn(0); 127899cafbc1SBarry Smith } 127999cafbc1SBarry Smith 128099cafbc1SBarry Smith #undef __FUNCT__ 128199cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ" 128299cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A) 128399cafbc1SBarry Smith { 128499cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 128599cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1286dd6ea824SBarry Smith MatScalar *aa = a->a; 128799cafbc1SBarry Smith 128899cafbc1SBarry Smith PetscFunctionBegin; 128999cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 129099cafbc1SBarry Smith PetscFunctionReturn(0); 129199cafbc1SBarry Smith } 129299cafbc1SBarry Smith 12933bededecSBarry Smith #undef __FUNCT__ 12943bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ" 12953bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b) 12963bededecSBarry Smith { 12973bededecSBarry Smith Mat_SeqSBAIJ *baij=(Mat_SeqSBAIJ*)A->data; 12983bededecSBarry Smith PetscErrorCode ierr; 12993bededecSBarry Smith PetscInt i,j,k,count; 13003bededecSBarry Smith PetscInt bs =A->rmap->bs,bs2=baij->bs2,row,col; 13013bededecSBarry Smith PetscScalar zero = 0.0; 13023bededecSBarry Smith MatScalar *aa; 13033bededecSBarry Smith const PetscScalar *xx; 13043bededecSBarry Smith PetscScalar *bb; 130556777dd2SBarry Smith PetscBool *zeroed,vecs = PETSC_FALSE; 13063bededecSBarry Smith 13073bededecSBarry Smith PetscFunctionBegin; 13083bededecSBarry Smith /* fix right hand side if needed */ 13093bededecSBarry Smith if (x && b) { 13103bededecSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 13113bededecSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 131256777dd2SBarry Smith vecs = PETSC_TRUE; 13133bededecSBarry Smith } 13143bededecSBarry Smith 13153bededecSBarry Smith /* zero the columns */ 13161795a4d1SJed Brown ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr); 13173bededecSBarry Smith for (i=0; i<is_n; i++) { 13183bededecSBarry Smith if (is_idx[i] < 0 || is_idx[i] >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range",is_idx[i]); 13193bededecSBarry Smith zeroed[is_idx[i]] = PETSC_TRUE; 13203bededecSBarry Smith } 132156777dd2SBarry Smith if (vecs) { 132256777dd2SBarry Smith for (i=0; i<A->rmap->N; i++) { 132356777dd2SBarry Smith row = i/bs; 132456777dd2SBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 132556777dd2SBarry Smith for (k=0; k<bs; k++) { 132656777dd2SBarry Smith col = bs*baij->j[j] + k; 132756777dd2SBarry Smith if (col <= i) continue; 132856777dd2SBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 132926fbe8dcSKarl Rupp if (!zeroed[i] && zeroed[col]) bb[i] -= aa[0]*xx[col]; 133026fbe8dcSKarl Rupp if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i]; 133156777dd2SBarry Smith } 133256777dd2SBarry Smith } 133356777dd2SBarry Smith } 133426fbe8dcSKarl Rupp for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]]; 133556777dd2SBarry Smith } 133656777dd2SBarry Smith 13373bededecSBarry Smith for (i=0; i<A->rmap->N; i++) { 13383bededecSBarry Smith if (!zeroed[i]) { 13393bededecSBarry Smith row = i/bs; 13403bededecSBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 13413bededecSBarry Smith for (k=0; k<bs; k++) { 13423bededecSBarry Smith col = bs*baij->j[j] + k; 13433bededecSBarry Smith if (zeroed[col]) { 13443bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 13453bededecSBarry Smith aa[0] = 0.0; 13463bededecSBarry Smith } 13473bededecSBarry Smith } 13483bededecSBarry Smith } 13493bededecSBarry Smith } 13503bededecSBarry Smith } 13513bededecSBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 135256777dd2SBarry Smith if (vecs) { 135356777dd2SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 135456777dd2SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 135556777dd2SBarry Smith } 13563bededecSBarry Smith 13573bededecSBarry Smith /* zero the rows */ 13583bededecSBarry Smith for (i=0; i<is_n; i++) { 13593bededecSBarry Smith row = is_idx[i]; 13603bededecSBarry Smith count = (baij->i[row/bs +1] - baij->i[row/bs])*bs; 13613bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs); 13623bededecSBarry Smith for (k=0; k<count; k++) { 13633bededecSBarry Smith aa[0] = zero; 13643bededecSBarry Smith aa += bs; 13653bededecSBarry Smith } 13663bededecSBarry Smith if (diag != 0.0) { 13673bededecSBarry Smith ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr); 13683bededecSBarry Smith } 13693bededecSBarry Smith } 13703bededecSBarry Smith ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 13713bededecSBarry Smith PetscFunctionReturn(0); 13723bededecSBarry Smith } 13733bededecSBarry Smith 137449b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 13753964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 137649b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 137749b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 137849b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 137997304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1380431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1381e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 1382db4efbfdSBarry Smith 0, 138349b5e25fSSatish Balay 0, 138449b5e25fSSatish Balay 0, 138597304618SKris Buschelman /* 10*/ 0, 138649b5e25fSSatish Balay 0, 1387c078aec8SLisandro Dalcin MatCholeskyFactor_SeqSBAIJ, 138841f059aeSBarry Smith MatSOR_SeqSBAIJ, 138949b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 139097304618SKris Buschelman /* 15*/ MatGetInfo_SeqSBAIJ, 139149b5e25fSSatish Balay MatEqual_SeqSBAIJ, 139249b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 139349b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 139449b5e25fSSatish Balay MatNorm_SeqSBAIJ, 139597304618SKris Buschelman /* 20*/ 0, 139649b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 139749b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 139849b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1399d519adbfSMatthew Knepley /* 24*/ 0, 140049b5e25fSSatish Balay 0, 140149b5e25fSSatish Balay 0, 1402db4efbfdSBarry Smith 0, 1403db4efbfdSBarry Smith 0, 14044994cf47SJed Brown /* 29*/ MatSetUp_SeqSBAIJ, 1405c464158bSHong Zhang 0, 1406db4efbfdSBarry Smith 0, 14078c778c55SBarry Smith 0, 14088c778c55SBarry Smith 0, 1409d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqSBAIJ, 1410719d5645SBarry Smith 0, 1411719d5645SBarry Smith 0, 141249b5e25fSSatish Balay 0, 1413c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 1414d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqSBAIJ, 141549b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 141649b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 141749b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 14183c896bc6SHong Zhang MatCopy_SeqSBAIJ, 1419d519adbfSMatthew Knepley /* 44*/ 0, 142049b5e25fSSatish Balay MatScale_SeqSBAIJ, 142149b5e25fSSatish Balay 0, 142249b5e25fSSatish Balay 0, 14233bededecSBarry Smith MatZeroRowsColumns_SeqSBAIJ, 1424f73d5cc4SBarry Smith /* 49*/ 0, 142549b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 142649b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 142749b5e25fSSatish Balay 0, 142849b5e25fSSatish Balay 0, 1429d519adbfSMatthew Knepley /* 54*/ 0, 143049b5e25fSSatish Balay 0, 143149b5e25fSSatish Balay 0, 143249b5e25fSSatish Balay 0, 143349b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 1434d519adbfSMatthew Knepley /* 59*/ MatGetSubMatrix_SeqSBAIJ, 143549b5e25fSSatish Balay 0, 143649b5e25fSSatish Balay 0, 1437357abbc8SBarry Smith 0, 1438d959ec07SHong Zhang 0, 1439d519adbfSMatthew Knepley /* 64*/ 0, 1440d959ec07SHong Zhang 0, 1441d959ec07SHong Zhang 0, 1442d959ec07SHong Zhang 0, 1443d959ec07SHong Zhang 0, 1444d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqSBAIJ, 14453e0d88b5SBarry Smith 0, 14463e0d88b5SBarry Smith 0, 14473e0d88b5SBarry Smith 0, 14483e0d88b5SBarry Smith 0, 1449d519adbfSMatthew Knepley /* 74*/ 0, 14503e0d88b5SBarry Smith 0, 14513e0d88b5SBarry Smith 0, 14523e0d88b5SBarry Smith 0, 14533e0d88b5SBarry Smith 0, 1454d519adbfSMatthew Knepley /* 79*/ 0, 14553e0d88b5SBarry Smith 0, 14563e0d88b5SBarry Smith 0, 145797304618SKris Buschelman MatGetInertia_SeqSBAIJ, 14585bba2384SShri Abhyankar MatLoad_SeqSBAIJ, 1459d519adbfSMatthew Knepley /* 84*/ MatIsSymmetric_SeqSBAIJ, 1460865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1461efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1462865e5f61SKris Buschelman 0, 1463865e5f61SKris Buschelman 0, 1464d519adbfSMatthew Knepley /* 89*/ 0, 1465865e5f61SKris Buschelman 0, 1466865e5f61SKris Buschelman 0, 1467865e5f61SKris Buschelman 0, 1468865e5f61SKris Buschelman 0, 1469d519adbfSMatthew Knepley /* 94*/ 0, 1470865e5f61SKris Buschelman 0, 1471865e5f61SKris Buschelman 0, 147299cafbc1SBarry Smith 0, 147399cafbc1SBarry Smith 0, 1474d519adbfSMatthew Knepley /* 99*/ 0, 147599cafbc1SBarry Smith 0, 147699cafbc1SBarry Smith 0, 147799cafbc1SBarry Smith 0, 147899cafbc1SBarry Smith 0, 1479d519adbfSMatthew Knepley /*104*/ 0, 148099cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1481f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1482f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 14832af78befSBarry Smith MatRestoreRowUpperTriangular_SeqSBAIJ, 1484d519adbfSMatthew Knepley /*109*/ 0, 14852af78befSBarry Smith 0, 14862af78befSBarry Smith 0, 14872af78befSBarry Smith 0, 1488547795f9SHong Zhang MatMissingDiagonal_SeqSBAIJ, 1489547795f9SHong Zhang /*114*/ 0, 1490547795f9SHong Zhang 0, 1491547795f9SHong Zhang 0, 1492547795f9SHong Zhang 0, 1493547795f9SHong Zhang 0, 1494547795f9SHong Zhang /*119*/ 0, 1495547795f9SHong Zhang 0, 14962f480046SShri Abhyankar 0, 14973964eb88SJed Brown 0, 14983964eb88SJed Brown 0, 14993964eb88SJed Brown /*124*/ 0, 15003964eb88SJed Brown 0, 15013964eb88SJed Brown 0, 15023964eb88SJed Brown 0, 15033964eb88SJed Brown 0, 15043964eb88SJed Brown /*129*/ 0, 15053964eb88SJed Brown 0, 15063964eb88SJed Brown 0, 15073964eb88SJed Brown 0, 15083964eb88SJed Brown 0, 15093964eb88SJed Brown /*134*/ 0, 15103964eb88SJed Brown 0, 15113964eb88SJed Brown 0, 15123964eb88SJed Brown 0, 15133964eb88SJed Brown 0, 15143964eb88SJed Brown /*139*/ 0, 1515f9426fe0SMark Adams 0, 15165bba2384SShri Abhyankar 0 151799cafbc1SBarry Smith }; 1518be1d678aSKris Buschelman 15194a2ae208SSatish Balay #undef __FUNCT__ 15204a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ" 15217087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 152249b5e25fSSatish Balay { 15234afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 1524d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 1525dfbe8321SBarry Smith PetscErrorCode ierr; 152649b5e25fSSatish Balay 152749b5e25fSSatish Balay PetscFunctionBegin; 1528e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 152949b5e25fSSatish Balay 153049b5e25fSSatish Balay /* allocate space for values if not already there */ 153149b5e25fSSatish Balay if (!aij->saved_values) { 1532785e854fSJed Brown ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr); 153349b5e25fSSatish Balay } 153449b5e25fSSatish Balay 153549b5e25fSSatish Balay /* copy values over */ 153687828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 153749b5e25fSSatish Balay PetscFunctionReturn(0); 153849b5e25fSSatish Balay } 153949b5e25fSSatish Balay 15404a2ae208SSatish Balay #undef __FUNCT__ 15414a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ" 15427087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 154349b5e25fSSatish Balay { 15444afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 15456849ba73SBarry Smith PetscErrorCode ierr; 1546d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 154749b5e25fSSatish Balay 154849b5e25fSSatish Balay PetscFunctionBegin; 1549e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1550e7e72b3dSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 155149b5e25fSSatish Balay 155249b5e25fSSatish Balay /* copy values over */ 155387828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 155449b5e25fSSatish Balay PetscFunctionReturn(0); 155549b5e25fSSatish Balay } 155649b5e25fSSatish Balay 15574a2ae208SSatish Balay #undef __FUNCT__ 1558a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ" 15597087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 156049b5e25fSSatish Balay { 1561c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 15626849ba73SBarry Smith PetscErrorCode ierr; 1563535b19f3SBarry Smith PetscInt i,mbs,bs2; 15642576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE; 156549b5e25fSSatish Balay 156649b5e25fSSatish Balay PetscFunctionBegin; 15672576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 1568273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1569db4efbfdSBarry Smith 157033d57670SJed Brown ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr); 157126283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 157226283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1573e02043d6SBarry Smith ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 1574899cda47SBarry Smith 1575d0f46423SBarry Smith mbs = B->rmap->N/bs; 157649b5e25fSSatish Balay bs2 = bs*bs; 157749b5e25fSSatish Balay 1578e7e72b3dSBarry Smith if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize"); 157949b5e25fSSatish Balay 1580ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1581ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1582ab93d7beSBarry Smith nz = 0; 1583ab93d7beSBarry Smith } 1584ab93d7beSBarry Smith 1585435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 1586e32f2f54SBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 158749b5e25fSSatish Balay if (nnz) { 158849b5e25fSSatish Balay for (i=0; i<mbs; i++) { 1589e32f2f54SBarry Smith if (nnz[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]); 1590e32f2f54SBarry Smith if (nnz[i] > mbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs); 159149b5e25fSSatish Balay } 159249b5e25fSSatish Balay } 159349b5e25fSSatish Balay 1594db4efbfdSBarry Smith B->ops->mult = MatMult_SeqSBAIJ_N; 1595db4efbfdSBarry Smith B->ops->multadd = MatMultAdd_SeqSBAIJ_N; 1596db4efbfdSBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_N; 1597db4efbfdSBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N; 159826fbe8dcSKarl Rupp 15990298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 160049b5e25fSSatish Balay if (!flg) { 160149b5e25fSSatish Balay switch (bs) { 160249b5e25fSSatish Balay case 1: 160349b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 160449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1605431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1606431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 160749b5e25fSSatish Balay break; 160849b5e25fSSatish Balay case 2: 160949b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 161049b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1611431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1612431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 161349b5e25fSSatish Balay break; 161449b5e25fSSatish Balay case 3: 161549b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 161649b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1617431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1618431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 161949b5e25fSSatish Balay break; 162049b5e25fSSatish Balay case 4: 162149b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 162249b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1623431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1624431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 162549b5e25fSSatish Balay break; 162649b5e25fSSatish Balay case 5: 162749b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 162849b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1629431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1630431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 163149b5e25fSSatish Balay break; 163249b5e25fSSatish Balay case 6: 163349b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 163449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1635431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1636431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 163749b5e25fSSatish Balay break; 163849b5e25fSSatish Balay case 7: 1639de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 164049b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1641431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1642431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 164349b5e25fSSatish Balay break; 164449b5e25fSSatish Balay } 164549b5e25fSSatish Balay } 164649b5e25fSSatish Balay 164749b5e25fSSatish Balay b->mbs = mbs; 16484afc71dfSHong Zhang b->nbs = mbs; 1649ab93d7beSBarry Smith if (!skipallocation) { 16502ee49352SLisandro Dalcin if (!b->imax) { 1651dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr); 165226fbe8dcSKarl Rupp 1653c760cd28SBarry Smith b->free_imax_ilen = PETSC_TRUE; 165426fbe8dcSKarl Rupp 16553bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr); 16562ee49352SLisandro Dalcin } 165749b5e25fSSatish Balay if (!nnz) { 1658435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 165949b5e25fSSatish Balay else if (nz <= 0) nz = 1; 166026fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->imax[i] = nz; 1661153ea458SHong Zhang nz = nz*mbs; /* total nz */ 166249b5e25fSSatish Balay } else { 166349b5e25fSSatish Balay nz = 0; 16648cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 166549b5e25fSSatish Balay } 16662ee49352SLisandro Dalcin /* b->ilen will count nonzeros in each block row so far. */ 166726fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->ilen[i] = 0; 16686c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 166949b5e25fSSatish Balay 167049b5e25fSSatish Balay /* allocate the matrix space */ 16712ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 1672dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr); 16733bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 16746c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 167513f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 167626fbe8dcSKarl Rupp 167749b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 167849b5e25fSSatish Balay 167949b5e25fSSatish Balay /* pointer to beginning of each row */ 1680e60cf9a0SBarry Smith b->i[0] = 0; 168126fbe8dcSKarl Rupp for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1]; 168226fbe8dcSKarl Rupp 1683e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1684e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1685e811da20SHong Zhang } else { 1686e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1687e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1688ab93d7beSBarry Smith } 168949b5e25fSSatish Balay 1690d0f46423SBarry Smith B->rmap->bs = bs; 169149b5e25fSSatish Balay b->bs2 = bs2; 16926c6c5352SBarry Smith b->nz = 0; 1693b32cb4a7SJed Brown b->maxnz = nz; 1694153ea458SHong Zhang 169516cdd363SHong Zhang b->inew = 0; 169616cdd363SHong Zhang b->jnew = 0; 169716cdd363SHong Zhang b->anew = 0; 169816cdd363SHong Zhang b->a2anew = 0; 16991a3463dfSHong Zhang b->permute = PETSC_FALSE; 17002576faa2SJed Brown if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);} 1701c464158bSHong Zhang PetscFunctionReturn(0); 1702c464158bSHong Zhang } 1703153ea458SHong Zhang 170438f409ebSLisandro Dalcin #undef __FUNCT__ 170538f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ" 170638f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[]) 170738f409ebSLisandro Dalcin { 170838f409ebSLisandro Dalcin PetscInt i,j,m,nz,nz_max=0,*nnz; 170938f409ebSLisandro Dalcin PetscScalar *values=0; 171038f409ebSLisandro Dalcin PetscBool roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented; 171138f409ebSLisandro Dalcin PetscErrorCode ierr; 171238f409ebSLisandro Dalcin PetscFunctionBegin; 171338f409ebSLisandro Dalcin if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs); 171438f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr); 171538f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr); 171638f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 171738f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 171838f409ebSLisandro Dalcin ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 171938f409ebSLisandro Dalcin m = B->rmap->n/bs; 172038f409ebSLisandro Dalcin 172138f409ebSLisandro Dalcin if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]); 1722785e854fSJed Brown ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr); 172338f409ebSLisandro Dalcin for (i=0; i<m; i++) { 172438f409ebSLisandro Dalcin nz = ii[i+1] - ii[i]; 172538f409ebSLisandro Dalcin if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz); 172638f409ebSLisandro Dalcin nz_max = PetscMax(nz_max,nz); 172738f409ebSLisandro Dalcin nnz[i] = nz; 172838f409ebSLisandro Dalcin } 172938f409ebSLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 173038f409ebSLisandro Dalcin ierr = PetscFree(nnz);CHKERRQ(ierr); 173138f409ebSLisandro Dalcin 173238f409ebSLisandro Dalcin values = (PetscScalar*)V; 173338f409ebSLisandro Dalcin if (!values) { 17341795a4d1SJed Brown ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr); 173538f409ebSLisandro Dalcin } 173638f409ebSLisandro Dalcin for (i=0; i<m; i++) { 173738f409ebSLisandro Dalcin PetscInt ncols = ii[i+1] - ii[i]; 173838f409ebSLisandro Dalcin const PetscInt *icols = jj + ii[i]; 173938f409ebSLisandro Dalcin if (!roworiented || bs == 1) { 174038f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0); 174138f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr); 174238f409ebSLisandro Dalcin } else { 174338f409ebSLisandro Dalcin for (j=0; j<ncols; j++) { 174438f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0); 174538f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr); 174638f409ebSLisandro Dalcin } 174738f409ebSLisandro Dalcin } 174838f409ebSLisandro Dalcin } 174938f409ebSLisandro Dalcin if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); } 175038f409ebSLisandro Dalcin ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 175138f409ebSLisandro Dalcin ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 175238f409ebSLisandro Dalcin ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 175338f409ebSLisandro Dalcin PetscFunctionReturn(0); 175438f409ebSLisandro Dalcin } 175538f409ebSLisandro Dalcin 1756db4efbfdSBarry Smith /* 1757db4efbfdSBarry Smith This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization 1758db4efbfdSBarry Smith */ 17598b1456e3SHong Zhang #undef __FUNCT__ 1760d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace" 1761ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural) 1762db4efbfdSBarry Smith { 1763db4efbfdSBarry Smith PetscErrorCode ierr; 1764ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 1765db4efbfdSBarry Smith PetscInt bs = B->rmap->bs; 1766db4efbfdSBarry Smith 1767db4efbfdSBarry Smith PetscFunctionBegin; 17680298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 1769db4efbfdSBarry Smith if (flg) bs = 8; 1770db4efbfdSBarry Smith 1771db4efbfdSBarry Smith if (!natural) { 1772db4efbfdSBarry Smith switch (bs) { 1773db4efbfdSBarry Smith case 1: 1774d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace; 1775db4efbfdSBarry Smith break; 1776db4efbfdSBarry Smith case 2: 1777db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 1778db4efbfdSBarry Smith break; 1779db4efbfdSBarry Smith case 3: 1780db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 1781db4efbfdSBarry Smith break; 1782db4efbfdSBarry Smith case 4: 1783db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 1784db4efbfdSBarry Smith break; 1785db4efbfdSBarry Smith case 5: 1786db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 1787db4efbfdSBarry Smith break; 1788db4efbfdSBarry Smith case 6: 1789db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 1790db4efbfdSBarry Smith break; 1791db4efbfdSBarry Smith case 7: 1792db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 1793db4efbfdSBarry Smith break; 1794db4efbfdSBarry Smith default: 1795db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N; 1796db4efbfdSBarry Smith break; 1797db4efbfdSBarry Smith } 1798db4efbfdSBarry Smith } else { 1799db4efbfdSBarry Smith switch (bs) { 1800db4efbfdSBarry Smith case 1: 1801d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace; 1802db4efbfdSBarry Smith break; 1803db4efbfdSBarry Smith case 2: 1804db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 1805db4efbfdSBarry Smith break; 1806db4efbfdSBarry Smith case 3: 1807db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 1808db4efbfdSBarry Smith break; 1809db4efbfdSBarry Smith case 4: 1810db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 1811db4efbfdSBarry Smith break; 1812db4efbfdSBarry Smith case 5: 1813db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 1814db4efbfdSBarry Smith break; 1815db4efbfdSBarry Smith case 6: 1816db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 1817db4efbfdSBarry Smith break; 1818db4efbfdSBarry Smith case 7: 1819db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 1820db4efbfdSBarry Smith break; 1821db4efbfdSBarry Smith default: 1822db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1823db4efbfdSBarry Smith break; 1824db4efbfdSBarry Smith } 1825db4efbfdSBarry Smith } 1826db4efbfdSBarry Smith PetscFunctionReturn(0); 1827db4efbfdSBarry Smith } 1828db4efbfdSBarry Smith 18298cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*); 18308cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*); 1831d769727bSBarry Smith 18325c9eb25fSBarry Smith #undef __FUNCT__ 18335c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc" 18348cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B) 18355c9eb25fSBarry Smith { 1836d0f46423SBarry Smith PetscInt n = A->rmap->n; 18375c9eb25fSBarry Smith PetscErrorCode ierr; 18385c9eb25fSBarry Smith 18395c9eb25fSBarry Smith PetscFunctionBegin; 18400e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX) 18410e92d65fSHong Zhang if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 18420e92d65fSHong Zhang #endif 1843ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 18445c9eb25fSBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 18455c9eb25fSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 18465c9eb25fSBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 18470298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 184826fbe8dcSKarl Rupp 18497b056e98SHong Zhang (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ; 1850c6d0d4f0SHong Zhang (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqSBAIJ; 1851e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 1852d5f3da31SBarry Smith (*B)->factortype = ftype; 18535c9eb25fSBarry Smith PetscFunctionReturn(0); 18545c9eb25fSBarry Smith } 18555c9eb25fSBarry Smith 1856db4efbfdSBarry Smith #undef __FUNCT__ 1857db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc" 1858ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool *flg) 1859db4efbfdSBarry Smith { 1860db4efbfdSBarry Smith PetscFunctionBegin; 1861db4efbfdSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 1862db4efbfdSBarry Smith *flg = PETSC_TRUE; 1863db4efbfdSBarry Smith } else { 1864db4efbfdSBarry Smith *flg = PETSC_FALSE; 1865db4efbfdSBarry Smith } 1866db4efbfdSBarry Smith PetscFunctionReturn(0); 1867db4efbfdSBarry Smith } 1868db4efbfdSBarry Smith 1869611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS) 18708cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*); 1871611f576cSBarry Smith #endif 1872b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX) 18738cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*); 1874b5e56a35SBarry Smith #endif 1875913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE) 18768cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*); 187720db9a53SJed Brown #endif 18788cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*); 18795c9eb25fSBarry Smith 18800bad9183SKris Buschelman /*MC 1881fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 18820bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 18830bad9183SKris Buschelman 1884828413b8SBarry Smith For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you 188571dad5bbSBarry Smith can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd() 1886828413b8SBarry Smith 18870bad9183SKris Buschelman Options Database Keys: 18880bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 18890bad9183SKris Buschelman 189071dad5bbSBarry Smith Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not 189171dad5bbSBarry Smith stored and it is assumed they symmetric to the upper triangular). If you call MatSetOption(Mat,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_FALSE) or use 189271dad5bbSBarry Smith the options database -mat_ignore_lower_triangular false it will generate an error if you try to set a value in the lower triangular portion. 189371dad5bbSBarry Smith 189471dad5bbSBarry Smith 18950bad9183SKris Buschelman Level: beginner 18960bad9183SKris Buschelman 18970bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 18980bad9183SKris Buschelman M*/ 18990bad9183SKris Buschelman 19008cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*); 1901aa5a9175SDahai Guo 1902a23d5eceSKris Buschelman #undef __FUNCT__ 1903a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ" 19048cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1905a23d5eceSKris Buschelman { 1906a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1907dfbe8321SBarry Smith PetscErrorCode ierr; 190813f74950SBarry Smith PetscMPIInt size; 1909ace3abfcSBarry Smith PetscBool no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE; 1910a23d5eceSKris Buschelman 1911a23d5eceSKris Buschelman PetscFunctionBegin; 1912ce94432eSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 1913e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1914a23d5eceSKris Buschelman 1915b00a9115SJed Brown ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1916a23d5eceSKris Buschelman B->data = (void*)b; 1917a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 191826fbe8dcSKarl Rupp 1919a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1920a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1921a23d5eceSKris Buschelman b->row = 0; 1922a23d5eceSKris Buschelman b->icol = 0; 1923a23d5eceSKris Buschelman b->reallocs = 0; 1924a23d5eceSKris Buschelman b->saved_values = 0; 19250def2e27SBarry Smith b->inode.limit = 5; 19260def2e27SBarry Smith b->inode.max_limit = 5; 1927a23d5eceSKris Buschelman 1928a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1929a23d5eceSKris Buschelman b->nonew = 0; 1930a23d5eceSKris Buschelman b->diag = 0; 1931a23d5eceSKris Buschelman b->solve_work = 0; 1932a23d5eceSKris Buschelman b->mult_work = 0; 1933a23d5eceSKris Buschelman B->spptr = 0; 1934f2cbd3d5SJed Brown B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2; 1935a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 1936a23d5eceSKris Buschelman b->xtoy = 0; 1937a23d5eceSKris Buschelman b->XtoY = 0; 1938a23d5eceSKris Buschelman 1939a23d5eceSKris Buschelman b->inew = 0; 1940a23d5eceSKris Buschelman b->jnew = 0; 1941a23d5eceSKris Buschelman b->anew = 0; 1942a23d5eceSKris Buschelman b->a2anew = 0; 1943a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1944a23d5eceSKris Buschelman 194571dad5bbSBarry Smith b->ignore_ltriangular = PETSC_TRUE; 194626fbe8dcSKarl Rupp 19470298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr); 1948941593c8SHong Zhang 1949f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 195026fbe8dcSKarl Rupp 19510298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr); 1952f5edf698SHong Zhang 1953b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX) 1954bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_pastix_C",MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr); 1955b5e56a35SBarry Smith #endif 1956611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS) 1957bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_mumps_C",MatGetFactor_sbaij_mumps);CHKERRQ(ierr); 1958611f576cSBarry Smith #endif 1959913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE) 1960bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_cholmod_C",MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr); 196120db9a53SJed Brown #endif 1962bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactorAvailable_petsc_C",MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr); 1963bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr); 1964bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_sbstrm_C",MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr); 1965bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1966bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1967bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 1968bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1969bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1970bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 197138f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr); 1972bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr); 197323ce1328SBarry Smith 197423ce1328SBarry Smith B->symmetric = PETSC_TRUE; 197523ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 197623ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 197723ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 197826fbe8dcSKarl Rupp 197917667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 19800def2e27SBarry Smith 1981ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr); 19820298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr); 198326fbe8dcSKarl Rupp if (no_unroll) { 198426fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr); 198526fbe8dcSKarl Rupp } 19860298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr); 198726fbe8dcSKarl Rupp if (no_inode) { 198826fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr); 198926fbe8dcSKarl Rupp } 19900298fd71SBarry Smith ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr); 19910def2e27SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 1992ace3abfcSBarry Smith b->inode.use = (PetscBool)(!(no_unroll || no_inode)); 19930def2e27SBarry Smith if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit; 1994a23d5eceSKris Buschelman PetscFunctionReturn(0); 1995a23d5eceSKris Buschelman } 1996a23d5eceSKris Buschelman 1997a23d5eceSKris Buschelman #undef __FUNCT__ 1998a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation" 1999a23d5eceSKris Buschelman /*@C 2000a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 2001a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 2002a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 2003a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 2004a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 2005a23d5eceSKris Buschelman 2006a23d5eceSKris Buschelman Collective on Mat 2007a23d5eceSKris Buschelman 2008a23d5eceSKris Buschelman Input Parameters: 2009a23d5eceSKris Buschelman + A - the symmetric matrix 2010a23d5eceSKris Buschelman . bs - size of block 2011a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 2012a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 20130298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2014a23d5eceSKris Buschelman 2015a23d5eceSKris Buschelman Options Database Keys: 2016a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 2017a23d5eceSKris Buschelman block calculations (much slower) 2018db4efbfdSBarry Smith . -mat_block_size - size of the blocks to use (only works if a negative bs is passed in 2019a23d5eceSKris Buschelman 2020a23d5eceSKris Buschelman Level: intermediate 2021a23d5eceSKris Buschelman 2022a23d5eceSKris Buschelman Notes: 2023a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 20240298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 20250598bfebSBarry Smith allocation. See the <a href="../../docs/manual.pdf#nameddest=ch_mat">Mat chapter of the users manual</a> for details. 2026a23d5eceSKris Buschelman 2027aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 2028aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 2029aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 2030aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 2031aa95bbe8SBarry Smith 203249a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 203349a6f317SBarry Smith 203449a6f317SBarry Smith 203569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2036a23d5eceSKris Buschelman @*/ 20377087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 203813f74950SBarry Smith { 20394ac538c5SBarry Smith PetscErrorCode ierr; 2040a23d5eceSKris Buschelman 2041a23d5eceSKris Buschelman PetscFunctionBegin; 20426ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 20436ba663aaSJed Brown PetscValidType(B,1); 20446ba663aaSJed Brown PetscValidLogicalCollectiveInt(B,bs,2); 20454ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr); 2046a23d5eceSKris Buschelman PetscFunctionReturn(0); 2047a23d5eceSKris Buschelman } 204849b5e25fSSatish Balay 20494a2ae208SSatish Balay #undef __FUNCT__ 205038f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR" 205138f409ebSLisandro Dalcin /*@C 205238f409ebSLisandro Dalcin MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format. 205338f409ebSLisandro Dalcin 205438f409ebSLisandro Dalcin Input Parameters: 205538f409ebSLisandro Dalcin + A - the matrix 205638f409ebSLisandro Dalcin . i - the indices into j for the start of each local row (starts with zero) 205738f409ebSLisandro Dalcin . j - the column indices for each local row (starts with zero) these must be sorted for each row 205838f409ebSLisandro Dalcin - v - optional values in the matrix 205938f409ebSLisandro Dalcin 206038f409ebSLisandro Dalcin Level: developer 206138f409ebSLisandro Dalcin 206238f409ebSLisandro Dalcin Notes: 206338f409ebSLisandro Dalcin The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED. For example, C programs 206438f409ebSLisandro Dalcin may want to use the default MAT_ROW_ORIENTED=PETSC_TRUE and use an array v[nnz][bs][bs] where the second index is 206538f409ebSLisandro Dalcin over rows within a block and the last index is over columns within a block row. Fortran programs will likely set 206638f409ebSLisandro Dalcin MAT_ROW_ORIENTED=PETSC_FALSE and use a Fortran array v(bs,bs,nnz) in which the first index is over rows within a 206738f409ebSLisandro Dalcin block column and the second index is over columns within a block. 206838f409ebSLisandro Dalcin 206938f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse 207038f409ebSLisandro Dalcin 207138f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ 207238f409ebSLisandro Dalcin @*/ 207338f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[]) 207438f409ebSLisandro Dalcin { 207538f409ebSLisandro Dalcin PetscErrorCode ierr; 207638f409ebSLisandro Dalcin 207738f409ebSLisandro Dalcin PetscFunctionBegin; 207838f409ebSLisandro Dalcin PetscValidHeaderSpecific(B,MAT_CLASSID,1); 207938f409ebSLisandro Dalcin PetscValidType(B,1); 208038f409ebSLisandro Dalcin PetscValidLogicalCollectiveInt(B,bs,2); 208138f409ebSLisandro Dalcin ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr); 208238f409ebSLisandro Dalcin PetscFunctionReturn(0); 208338f409ebSLisandro Dalcin } 208438f409ebSLisandro Dalcin 208538f409ebSLisandro Dalcin #undef __FUNCT__ 20864a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ" 2087c464158bSHong Zhang /*@C 2088c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 2089c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 2090c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 2091c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 2092c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 209349b5e25fSSatish Balay 2094c464158bSHong Zhang Collective on MPI_Comm 2095c464158bSHong Zhang 2096c464158bSHong Zhang Input Parameters: 2097c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2098c464158bSHong Zhang . bs - size of block 2099c464158bSHong Zhang . m - number of rows, or number of columns 2100c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 2101744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 21020298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2103c464158bSHong Zhang 2104c464158bSHong Zhang Output Parameter: 2105c464158bSHong Zhang . A - the symmetric matrix 2106c464158bSHong Zhang 2107c464158bSHong Zhang Options Database Keys: 2108c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 2109c464158bSHong Zhang block calculations (much slower) 2110c464158bSHong Zhang . -mat_block_size - size of the blocks to use 2111c464158bSHong Zhang 2112c464158bSHong Zhang Level: intermediate 2113c464158bSHong Zhang 2114175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2115ae1d86c5SBarry Smith MatXXXXSetPreallocation() paradgm instead of this routine directly. 2116175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 2117175b88e8SBarry Smith 2118c464158bSHong Zhang Notes: 21196d6d819aSHong Zhang The number of rows and columns must be divisible by blocksize. 21206d6d819aSHong Zhang This matrix type does not support complex Hermitian operation. 2121c464158bSHong Zhang 2122c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 21230298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 21240598bfebSBarry Smith allocation. See the <a href="../../docs/manual.pdf#nameddest=ch_mat">Mat chapter of the users manual</a> for details. 2125c464158bSHong Zhang 212649a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 212749a6f317SBarry Smith 212869b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2129c464158bSHong Zhang @*/ 21307087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 2131c464158bSHong Zhang { 2132dfbe8321SBarry Smith PetscErrorCode ierr; 2133c464158bSHong Zhang 2134c464158bSHong Zhang PetscFunctionBegin; 2135f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 2136f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2137c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 2138ab93d7beSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 213949b5e25fSSatish Balay PetscFunctionReturn(0); 214049b5e25fSSatish Balay } 214149b5e25fSSatish Balay 21424a2ae208SSatish Balay #undef __FUNCT__ 21434a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ" 2144dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 214549b5e25fSSatish Balay { 214649b5e25fSSatish Balay Mat C; 214749b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 21486849ba73SBarry Smith PetscErrorCode ierr; 2149b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 215049b5e25fSSatish Balay 215149b5e25fSSatish Balay PetscFunctionBegin; 2152e32f2f54SBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix"); 215349b5e25fSSatish Balay 215449b5e25fSSatish Balay *B = 0; 2155ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2156d0f46423SBarry Smith ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr); 21578e9a0fb8SHong Zhang ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr); 21581d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 2159692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 2160692f9cbeSHong Zhang 2161273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 2162d5f3da31SBarry Smith C->factortype = A->factortype; 216349b5e25fSSatish Balay c->row = 0; 216449b5e25fSSatish Balay c->icol = 0; 216549b5e25fSSatish Balay c->saved_values = 0; 2166a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 216749b5e25fSSatish Balay C->assembled = PETSC_TRUE; 216849b5e25fSSatish Balay 21691e1e43feSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 21701e1e43feSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 217149b5e25fSSatish Balay c->bs2 = a->bs2; 217249b5e25fSSatish Balay c->mbs = a->mbs; 217349b5e25fSSatish Balay c->nbs = a->nbs; 217449b5e25fSSatish Balay 2175c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2176c760cd28SBarry Smith c->imax = a->imax; 2177c760cd28SBarry Smith c->ilen = a->ilen; 2178c760cd28SBarry Smith c->free_imax_ilen = PETSC_FALSE; 2179c760cd28SBarry Smith } else { 2180dcca6d9dSJed Brown ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr); 21813bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 218249b5e25fSSatish Balay for (i=0; i<mbs; i++) { 218349b5e25fSSatish Balay c->imax[i] = a->imax[i]; 218449b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 218549b5e25fSSatish Balay } 2186c760cd28SBarry Smith c->free_imax_ilen = PETSC_TRUE; 2187c760cd28SBarry Smith } 218849b5e25fSSatish Balay 218949b5e25fSSatish Balay /* allocate the matrix space */ 21904da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2191785e854fSJed Brown ierr = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr); 21923bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 219344e1c64aSLisandro Dalcin c->i = a->i; 219444e1c64aSLisandro Dalcin c->j = a->j; 21954da8f245SBarry Smith c->singlemalloc = PETSC_FALSE; 219644e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 21974da8f245SBarry Smith c->free_ij = PETSC_FALSE; 21984da8f245SBarry Smith c->parent = A; 21994da8f245SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 22004da8f245SBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 22014da8f245SBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 22024da8f245SBarry Smith } else { 2203dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr); 220413f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 22053bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 22064da8f245SBarry Smith c->singlemalloc = PETSC_TRUE; 220744e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 22084da8f245SBarry Smith c->free_ij = PETSC_TRUE; 22094da8f245SBarry Smith } 221049b5e25fSSatish Balay if (mbs > 0) { 22114da8f245SBarry Smith if (cpvalues != MAT_SHARE_NONZERO_PATTERN) { 221213f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 22134da8f245SBarry Smith } 221449b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 221549b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 221649b5e25fSSatish Balay } else { 221749b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 221849b5e25fSSatish Balay } 2219a1c3900fSBarry Smith if (a->jshort) { 222044e1c64aSLisandro Dalcin /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */ 222144e1c64aSLisandro Dalcin /* if the parent matrix is reassembled, this child matrix will never notice */ 2222785e854fSJed Brown ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr); 22233bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr); 2224a1c3900fSBarry Smith ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr); 222526fbe8dcSKarl Rupp 22264da8f245SBarry Smith c->free_jshort = PETSC_TRUE; 22274da8f245SBarry Smith } 2228a1c3900fSBarry Smith } 222949b5e25fSSatish Balay 223049b5e25fSSatish Balay c->roworiented = a->roworiented; 223149b5e25fSSatish Balay c->nonew = a->nonew; 223249b5e25fSSatish Balay 223349b5e25fSSatish Balay if (a->diag) { 2234c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2235c760cd28SBarry Smith c->diag = a->diag; 2236c760cd28SBarry Smith c->free_diag = PETSC_FALSE; 2237c760cd28SBarry Smith } else { 2238785e854fSJed Brown ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr); 22393bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr); 224026fbe8dcSKarl Rupp for (i=0; i<mbs; i++) c->diag[i] = a->diag[i]; 2241c760cd28SBarry Smith c->free_diag = PETSC_TRUE; 2242c760cd28SBarry Smith } 224344e1c64aSLisandro Dalcin } 22446c6c5352SBarry Smith c->nz = a->nz; 2245f2cbd3d5SJed Brown c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 224649b5e25fSSatish Balay c->solve_work = 0; 224749b5e25fSSatish Balay c->mult_work = 0; 224826fbe8dcSKarl Rupp 224949b5e25fSSatish Balay *B = C; 2250140e18c1SBarry Smith ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 225149b5e25fSSatish Balay PetscFunctionReturn(0); 225249b5e25fSSatish Balay } 225349b5e25fSSatish Balay 22544a2ae208SSatish Balay #undef __FUNCT__ 22555bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ" 2256112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer) 22572f480046SShri Abhyankar { 22582f480046SShri Abhyankar Mat_SeqSBAIJ *a; 22592f480046SShri Abhyankar PetscErrorCode ierr; 22602f480046SShri Abhyankar int fd; 22612f480046SShri Abhyankar PetscMPIInt size; 22622f480046SShri Abhyankar PetscInt i,nz,header[4],*rowlengths=0,M,N,bs=1; 22632f480046SShri Abhyankar PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 22642f480046SShri Abhyankar PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols; 22652f480046SShri Abhyankar PetscInt *masked,nmask,tmp,bs2,ishift; 22662f480046SShri Abhyankar PetscScalar *aa; 2267ce94432eSBarry Smith MPI_Comm comm; 22682f480046SShri Abhyankar 22692f480046SShri Abhyankar PetscFunctionBegin; 2270ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 22710298fd71SBarry Smith ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr); 22722f480046SShri Abhyankar bs2 = bs*bs; 22732f480046SShri Abhyankar 22742f480046SShri Abhyankar ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 22752f480046SShri Abhyankar if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor"); 22762f480046SShri Abhyankar ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 22772f480046SShri Abhyankar ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 22782f480046SShri Abhyankar if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 22792f480046SShri Abhyankar M = header[1]; N = header[2]; nz = header[3]; 22802f480046SShri Abhyankar 22812f480046SShri Abhyankar if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 22822f480046SShri Abhyankar 22832f480046SShri Abhyankar if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices"); 22842f480046SShri Abhyankar 22852f480046SShri Abhyankar /* 22862f480046SShri Abhyankar This code adds extra rows to make sure the number of rows is 22872f480046SShri Abhyankar divisible by the blocksize 22882f480046SShri Abhyankar */ 22892f480046SShri Abhyankar mbs = M/bs; 22902f480046SShri Abhyankar extra_rows = bs - M + bs*(mbs); 22912f480046SShri Abhyankar if (extra_rows == bs) extra_rows = 0; 22922f480046SShri Abhyankar else mbs++; 22932f480046SShri Abhyankar if (extra_rows) { 22942f480046SShri Abhyankar ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr); 22952f480046SShri Abhyankar } 22962f480046SShri Abhyankar 22972f480046SShri Abhyankar /* Set global sizes if not already set */ 22982f480046SShri Abhyankar if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) { 22992f480046SShri Abhyankar ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr); 23002f480046SShri Abhyankar } else { /* Check if the matrix global sizes are correct */ 23012f480046SShri Abhyankar ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr); 23022f480046SShri Abhyankar if (M != rows || N != cols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix in file of different length (%d, %d) than the input matrix (%d, %d)",M,N,rows,cols); 23032f480046SShri Abhyankar } 23042f480046SShri Abhyankar 23052f480046SShri Abhyankar /* read in row lengths */ 2306785e854fSJed Brown ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr); 23072f480046SShri Abhyankar ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 23082f480046SShri Abhyankar for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 23092f480046SShri Abhyankar 23102f480046SShri Abhyankar /* read in column indices */ 2311785e854fSJed Brown ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr); 23122f480046SShri Abhyankar ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 23132f480046SShri Abhyankar for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 23142f480046SShri Abhyankar 23152f480046SShri Abhyankar /* loop over row lengths determining block row lengths */ 23161795a4d1SJed Brown ierr = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr); 2317dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr); 23182f480046SShri Abhyankar ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 23192f480046SShri Abhyankar rowcount = 0; 23202f480046SShri Abhyankar nzcount = 0; 23212f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23222f480046SShri Abhyankar nmask = 0; 23232f480046SShri Abhyankar for (j=0; j<bs; j++) { 23242f480046SShri Abhyankar kmax = rowlengths[rowcount]; 23252f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23262f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23272f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 23282f480046SShri Abhyankar } 23292f480046SShri Abhyankar rowcount++; 23302f480046SShri Abhyankar } 23312f480046SShri Abhyankar s_browlengths[i] += nmask; 23322f480046SShri Abhyankar 23332f480046SShri Abhyankar /* zero out the mask elements we set */ 23342f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23352f480046SShri Abhyankar } 23362f480046SShri Abhyankar 23372f480046SShri Abhyankar /* Do preallocation */ 23382f480046SShri Abhyankar ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr); 23392f480046SShri Abhyankar a = (Mat_SeqSBAIJ*)newmat->data; 23402f480046SShri Abhyankar 23412f480046SShri Abhyankar /* set matrix "i" values */ 23422f480046SShri Abhyankar a->i[0] = 0; 23432f480046SShri Abhyankar for (i=1; i<= mbs; i++) { 23442f480046SShri Abhyankar a->i[i] = a->i[i-1] + s_browlengths[i-1]; 23452f480046SShri Abhyankar a->ilen[i-1] = s_browlengths[i-1]; 23462f480046SShri Abhyankar } 23472f480046SShri Abhyankar a->nz = a->i[mbs]; 23482f480046SShri Abhyankar 23492f480046SShri Abhyankar /* read in nonzero values */ 2350785e854fSJed Brown ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr); 23512f480046SShri Abhyankar ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 23522f480046SShri Abhyankar for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 23532f480046SShri Abhyankar 23542f480046SShri Abhyankar /* set "a" and "j" values into matrix */ 23552f480046SShri Abhyankar nzcount = 0; jcount = 0; 23562f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23572f480046SShri Abhyankar nzcountb = nzcount; 23582f480046SShri Abhyankar nmask = 0; 23592f480046SShri Abhyankar for (j=0; j<bs; j++) { 23602f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23612f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23622f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23632f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 23642f480046SShri Abhyankar } 23652f480046SShri Abhyankar } 23662f480046SShri Abhyankar /* sort the masked values */ 23672f480046SShri Abhyankar ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 23682f480046SShri Abhyankar 23692f480046SShri Abhyankar /* set "j" values into matrix */ 23702f480046SShri Abhyankar maskcount = 1; 23712f480046SShri Abhyankar for (j=0; j<nmask; j++) { 23722f480046SShri Abhyankar a->j[jcount++] = masked[j]; 23732f480046SShri Abhyankar mask[masked[j]] = maskcount++; 23742f480046SShri Abhyankar } 23752f480046SShri Abhyankar 23762f480046SShri Abhyankar /* set "a" values into matrix */ 23772f480046SShri Abhyankar ishift = bs2*a->i[i]; 23782f480046SShri Abhyankar for (j=0; j<bs; j++) { 23792f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23802f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23812f480046SShri Abhyankar tmp = jj[nzcountb]/bs; /* block col. index */ 23822f480046SShri Abhyankar if (tmp >= i) { 23832f480046SShri Abhyankar block = mask[tmp] - 1; 23842f480046SShri Abhyankar point = jj[nzcountb] - bs*tmp; 23852f480046SShri Abhyankar idx = ishift + bs2*block + j + bs*point; 23862f480046SShri Abhyankar a->a[idx] = aa[nzcountb]; 23872f480046SShri Abhyankar } 23882f480046SShri Abhyankar nzcountb++; 23892f480046SShri Abhyankar } 23902f480046SShri Abhyankar } 23912f480046SShri Abhyankar /* zero out the mask elements we set */ 23922f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23932f480046SShri Abhyankar } 23942f480046SShri Abhyankar if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 23952f480046SShri Abhyankar 23962f480046SShri Abhyankar ierr = PetscFree(rowlengths);CHKERRQ(ierr); 23972f480046SShri Abhyankar ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 23982f480046SShri Abhyankar ierr = PetscFree(aa);CHKERRQ(ierr); 23992f480046SShri Abhyankar ierr = PetscFree(jj);CHKERRQ(ierr); 24002f480046SShri Abhyankar ierr = PetscFree2(mask,masked);CHKERRQ(ierr); 24012f480046SShri Abhyankar 24022f480046SShri Abhyankar ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 24032f480046SShri Abhyankar ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 24042f480046SShri Abhyankar PetscFunctionReturn(0); 24052f480046SShri Abhyankar } 24062f480046SShri Abhyankar 24072f480046SShri Abhyankar #undef __FUNCT__ 2408c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays" 2409c75a6043SHong Zhang /*@ 2410c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2411c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2412c75a6043SHong Zhang 2413c75a6043SHong Zhang Collective on MPI_Comm 2414c75a6043SHong Zhang 2415c75a6043SHong Zhang Input Parameters: 2416c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2417c75a6043SHong Zhang . bs - size of block 2418c75a6043SHong Zhang . m - number of rows 2419c75a6043SHong Zhang . n - number of columns 2420c75a6043SHong Zhang . i - row indices 2421c75a6043SHong Zhang . j - column indices 2422c75a6043SHong Zhang - a - matrix values 2423c75a6043SHong Zhang 2424c75a6043SHong Zhang Output Parameter: 2425c75a6043SHong Zhang . mat - the matrix 2426c75a6043SHong Zhang 2427dfb205c3SBarry Smith Level: advanced 2428c75a6043SHong Zhang 2429c75a6043SHong Zhang Notes: 2430c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2431c75a6043SHong Zhang once the matrix is destroyed 2432c75a6043SHong Zhang 2433c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2434c75a6043SHong Zhang 2435c75a6043SHong Zhang The i and j indices are 0 based 2436c75a6043SHong Zhang 2437dfb205c3SBarry Smith When block size is greater than 1 the matrix values must be stored using the SBAIJ storage format (see the SBAIJ code to determine this). For block size of 1 2438dfb205c3SBarry Smith it is the regular CSR format excluding the lower triangular elements. 2439dfb205c3SBarry Smith 244069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ() 2441c75a6043SHong Zhang 2442c75a6043SHong Zhang @*/ 24437087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat) 2444c75a6043SHong Zhang { 2445c75a6043SHong Zhang PetscErrorCode ierr; 2446c75a6043SHong Zhang PetscInt ii; 2447c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2448c75a6043SHong Zhang 2449c75a6043SHong Zhang PetscFunctionBegin; 2450e32f2f54SBarry Smith if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 2451e32f2f54SBarry Smith if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2452c75a6043SHong Zhang 2453c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2454c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2455c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2456c75a6043SHong Zhang ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 2457c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2458dcca6d9dSJed Brown ierr = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr); 24593bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr); 2460c75a6043SHong Zhang 2461c75a6043SHong Zhang sbaij->i = i; 2462c75a6043SHong Zhang sbaij->j = j; 2463c75a6043SHong Zhang sbaij->a = a; 246426fbe8dcSKarl Rupp 2465c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2466c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2467e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2468e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2469c75a6043SHong Zhang 2470c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2471c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2472c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2473e32f2f54SBarry Smith if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %d length = %d",ii,i[ii+1] - i[ii]); 2474c75a6043SHong Zhang #endif 2475c75a6043SHong Zhang } 2476c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2477c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2478e32f2f54SBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2479e32f2f54SBarry Smith if (j[ii] > n - 1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %d index = %d",ii,j[ii]); 2480c75a6043SHong Zhang } 2481c75a6043SHong Zhang #endif 2482c75a6043SHong Zhang 2483c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2484c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2485c75a6043SHong Zhang PetscFunctionReturn(0); 2486c75a6043SHong Zhang } 2487d06b337dSHong Zhang 2488d06b337dSHong Zhang 2489d06b337dSHong Zhang 249049b5e25fSSatish Balay 249149b5e25fSSatish Balay 2492