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 146214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 15cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat,MatType,MatReuse,Mat*); 166214f412SHong Zhang #endif 17d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK) 18d24d4204SJose E. Roman PETSC_INTERN PetscErrorCode MatConvert_SBAIJ_ScaLAPACK(Mat,MatType,MatReuse,Mat*); 19d24d4204SJose E. Roman #endif 2028d58a37SPierre Jolivet PETSC_INTERN PetscErrorCode MatConvert_MPISBAIJ_Basic(Mat,MatType,MatReuse,Mat*); 21b5b17502SBarry Smith 2249b5e25fSSatish Balay /* 2349b5e25fSSatish Balay Checks for missing diagonals 2449b5e25fSSatish Balay */ 25ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool *missing,PetscInt *dd) 2649b5e25fSSatish Balay { 27045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 286849ba73SBarry Smith PetscErrorCode ierr; 297734d3b5SMatthew G. Knepley PetscInt *diag,*ii = a->i,i; 3049b5e25fSSatish Balay 3149b5e25fSSatish Balay PetscFunctionBegin; 32045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 332af78befSBarry Smith *missing = PETSC_FALSE; 347734d3b5SMatthew G. Knepley if (A->rmap->n > 0 && !ii) { 35358d2f5dSShri Abhyankar *missing = PETSC_TRUE; 36358d2f5dSShri Abhyankar if (dd) *dd = 0; 37955c1f14SBarry Smith ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr); 38358d2f5dSShri Abhyankar } else { 39358d2f5dSShri Abhyankar diag = a->diag; 4049b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 417734d3b5SMatthew G. Knepley if (diag[i] >= ii[i+1]) { 422af78befSBarry Smith *missing = PETSC_TRUE; 432af78befSBarry Smith if (dd) *dd = i; 442af78befSBarry Smith break; 452af78befSBarry Smith } 4649b5e25fSSatish Balay } 47358d2f5dSShri Abhyankar } 4849b5e25fSSatish Balay PetscFunctionReturn(0); 4949b5e25fSSatish Balay } 5049b5e25fSSatish Balay 51dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 5249b5e25fSSatish Balay { 53045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 546849ba73SBarry Smith PetscErrorCode ierr; 5548dd3d27SHong Zhang PetscInt i,j; 5649b5e25fSSatish Balay 5749b5e25fSSatish Balay PetscFunctionBegin; 5809f38230SBarry Smith if (!a->diag) { 59785e854fSJed Brown ierr = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr); 603bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 61c760cd28SBarry Smith a->free_diag = PETSC_TRUE; 6209f38230SBarry Smith } 6348dd3d27SHong Zhang for (i=0; i<a->mbs; i++) { 6448dd3d27SHong Zhang a->diag[i] = a->i[i+1]; 6548dd3d27SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 6648dd3d27SHong Zhang if (a->j[j] == i) { 6748dd3d27SHong Zhang a->diag[i] = j; 6848dd3d27SHong Zhang break; 6948dd3d27SHong Zhang } 7048dd3d27SHong Zhang } 7148dd3d27SHong Zhang } 7249b5e25fSSatish Balay PetscFunctionReturn(0); 7349b5e25fSSatish Balay } 7449b5e25fSSatish Balay 751a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool *done) 7649b5e25fSSatish Balay { 77a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 788f7157efSSatish Balay PetscErrorCode ierr; 792462f5fdSStefano Zampini PetscInt i,j,n = a->mbs,nz = a->i[n],*tia,*tja,bs = A->rmap->bs,k,l,cnt; 802462f5fdSStefano Zampini PetscInt **ia = (PetscInt**)inia,**ja = (PetscInt**)inja; 8149b5e25fSSatish Balay 8249b5e25fSSatish Balay PetscFunctionBegin; 83d3e5a4abSHong Zhang *nn = n; 84a1373b80SHong Zhang if (!ia) PetscFunctionReturn(0); 852462f5fdSStefano Zampini if (symmetric) { 862462f5fdSStefano Zampini ierr = MatToSymmetricIJ_SeqAIJ(n,a->i,a->j,PETSC_FALSE,0,0,&tia,&tja);CHKERRQ(ierr); 872462f5fdSStefano Zampini nz = tia[n]; 882462f5fdSStefano Zampini } else { 892462f5fdSStefano Zampini tia = a->i; tja = a->j; 902462f5fdSStefano Zampini } 912462f5fdSStefano Zampini 922462f5fdSStefano Zampini if (!blockcompressed && bs > 1) { 932462f5fdSStefano Zampini (*nn) *= bs; 948f7157efSSatish Balay /* malloc & create the natural set of indices */ 952462f5fdSStefano Zampini ierr = PetscMalloc1((n+1)*bs,ia);CHKERRQ(ierr); 962462f5fdSStefano Zampini if (n) { 972462f5fdSStefano Zampini (*ia)[0] = oshift; 982462f5fdSStefano Zampini for (j=1; j<bs; j++) { 992462f5fdSStefano Zampini (*ia)[j] = (tia[1]-tia[0])*bs+(*ia)[j-1]; 1002462f5fdSStefano Zampini } 1012462f5fdSStefano Zampini } 1022462f5fdSStefano Zampini 1032462f5fdSStefano Zampini for (i=1; i<n; i++) { 1042462f5fdSStefano Zampini (*ia)[i*bs] = (tia[i]-tia[i-1])*bs + (*ia)[i*bs-1]; 1052462f5fdSStefano Zampini for (j=1; j<bs; j++) { 1062462f5fdSStefano Zampini (*ia)[i*bs+j] = (tia[i+1]-tia[i])*bs + (*ia)[i*bs+j-1]; 1072462f5fdSStefano Zampini } 1082462f5fdSStefano Zampini } 1092462f5fdSStefano Zampini if (n) { 1102462f5fdSStefano Zampini (*ia)[n*bs] = (tia[n]-tia[n-1])*bs + (*ia)[n*bs-1]; 1112462f5fdSStefano Zampini } 1122462f5fdSStefano Zampini 1132462f5fdSStefano Zampini if (inja) { 1142462f5fdSStefano Zampini ierr = PetscMalloc1(nz*bs*bs,ja);CHKERRQ(ierr); 1152462f5fdSStefano Zampini cnt = 0; 1162462f5fdSStefano Zampini for (i=0; i<n; i++) { 1178f7157efSSatish Balay for (j=0; j<bs; j++) { 1182462f5fdSStefano Zampini for (k=tia[i]; k<tia[i+1]; k++) { 1192462f5fdSStefano Zampini for (l=0; l<bs; l++) { 1202462f5fdSStefano Zampini (*ja)[cnt++] = bs*tja[k] + l; 1218f7157efSSatish Balay } 1228f7157efSSatish Balay } 1238f7157efSSatish Balay } 1248f7157efSSatish Balay } 1258f7157efSSatish Balay } 1262462f5fdSStefano Zampini 1272462f5fdSStefano Zampini if (symmetric) { /* deallocate memory allocated in MatToSymmetricIJ_SeqAIJ() */ 1282462f5fdSStefano Zampini ierr = PetscFree(tia);CHKERRQ(ierr); 1292462f5fdSStefano Zampini ierr = PetscFree(tja);CHKERRQ(ierr); 1302462f5fdSStefano Zampini } 1312462f5fdSStefano Zampini } else if (oshift == 1) { 1322462f5fdSStefano Zampini if (symmetric) { 1332462f5fdSStefano Zampini nz = tia[A->rmap->n/bs]; 1342462f5fdSStefano Zampini /* add 1 to i and j indices */ 1352462f5fdSStefano Zampini for (i=0; i<A->rmap->n/bs+1; i++) tia[i] = tia[i] + 1; 1362462f5fdSStefano Zampini *ia = tia; 1372462f5fdSStefano Zampini if (ja) { 1382462f5fdSStefano Zampini for (i=0; i<nz; i++) tja[i] = tja[i] + 1; 1392462f5fdSStefano Zampini *ja = tja; 1402462f5fdSStefano Zampini } 1412462f5fdSStefano Zampini } else { 1422462f5fdSStefano Zampini nz = a->i[A->rmap->n/bs]; 1432462f5fdSStefano Zampini /* malloc space and add 1 to i and j indices */ 1442462f5fdSStefano Zampini ierr = PetscMalloc1(A->rmap->n/bs+1,ia);CHKERRQ(ierr); 1452462f5fdSStefano Zampini for (i=0; i<A->rmap->n/bs+1; i++) (*ia)[i] = a->i[i] + 1; 1462462f5fdSStefano Zampini if (ja) { 1472462f5fdSStefano Zampini ierr = PetscMalloc1(nz,ja);CHKERRQ(ierr); 1482462f5fdSStefano Zampini for (i=0; i<nz; i++) (*ja)[i] = a->j[i] + 1; 1492462f5fdSStefano Zampini } 1502462f5fdSStefano Zampini } 1512462f5fdSStefano Zampini } else { 1522462f5fdSStefano Zampini *ia = tia; 1532462f5fdSStefano Zampini if (ja) *ja = tja; 154a6ece127SHong Zhang } 15549b5e25fSSatish Balay PetscFunctionReturn(0); 15649b5e25fSSatish Balay } 15749b5e25fSSatish Balay 1581a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 15949b5e25fSSatish Balay { 1608f7157efSSatish Balay PetscErrorCode ierr; 161a6ece127SHong Zhang 16249b5e25fSSatish Balay PetscFunctionBegin; 16349b5e25fSSatish Balay if (!ia) PetscFunctionReturn(0); 1642462f5fdSStefano Zampini if ((!blockcompressed && A->rmap->bs > 1) || (symmetric || oshift == 1)) { 1652462f5fdSStefano Zampini ierr = PetscFree(*ia);CHKERRQ(ierr); 1662462f5fdSStefano Zampini if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);} 167a6ece127SHong Zhang } 168a6ece127SHong Zhang PetscFunctionReturn(0); 16949b5e25fSSatish Balay } 17049b5e25fSSatish Balay 171dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A) 17249b5e25fSSatish Balay { 17349b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 174dfbe8321SBarry Smith PetscErrorCode ierr; 17549b5e25fSSatish Balay 17649b5e25fSSatish Balay PetscFunctionBegin; 177a9f03627SSatish Balay #if defined(PETSC_USE_LOG) 178d0f46423SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz); 179a9f03627SSatish Balay #endif 180e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 1817f53bb6cSHong Zhang if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);} 1826bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1836bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1846bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 185c31cb41cSBarry Smith ierr = PetscFree(a->idiag);CHKERRQ(ierr); 186c31cb41cSBarry Smith ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 187c760cd28SBarry Smith if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);} 18805b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 18941f059aeSBarry Smith ierr = PetscFree(a->sor_work);CHKERRQ(ierr); 19005b42c5fSBarry Smith ierr = PetscFree(a->solves_work);CHKERRQ(ierr); 19105b42c5fSBarry Smith ierr = PetscFree(a->mult_work);CHKERRQ(ierr); 19205b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 1934da8f245SBarry Smith if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);} 1941a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1956bf464f9SBarry Smith ierr = MatDestroy(&a->parent);CHKERRQ(ierr); 196bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 197901853e0SKris Buschelman 198f4259b30SLisandro Dalcin ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr); 199bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 200bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 201bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr); 202bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr); 203bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr); 204bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr); 20538f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr); 2066214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 2076214f412SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_elemental_C",NULL);CHKERRQ(ierr); 2086214f412SHong Zhang #endif 209d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK) 210d24d4204SJose E. Roman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_scalapack_C",NULL);CHKERRQ(ierr); 211d24d4204SJose E. Roman #endif 21249b5e25fSSatish Balay PetscFunctionReturn(0); 21349b5e25fSSatish Balay } 21449b5e25fSSatish Balay 215ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg) 21649b5e25fSSatish Balay { 217045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 218eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 219eb1ec7c1SStefano Zampini PetscInt bs; 220eb1ec7c1SStefano Zampini #endif 22163ba0a88SBarry Smith PetscErrorCode ierr; 22249b5e25fSSatish Balay 22349b5e25fSSatish Balay PetscFunctionBegin; 224eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 225eb1ec7c1SStefano Zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 226eb1ec7c1SStefano Zampini #endif 2274d9d31abSKris Buschelman switch (op) { 2284d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 2294e0d8c25SBarry Smith a->roworiented = flg; 2304d9d31abSKris Buschelman break; 231a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 232a9817697SBarry Smith a->keepnonzeropattern = flg; 2334d9d31abSKris Buschelman break; 234512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 235512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 2364d9d31abSKris Buschelman break; 2374d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 2384e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 2394d9d31abSKris Buschelman break; 2404d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 2414e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 2424d9d31abSKris Buschelman break; 24328b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 24428b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 24528b2fa4aSMatthew Knepley break; 2468c78258cSHong Zhang case MAT_FORCE_DIAGONAL_ENTRIES: 2474d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 2484d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 249071fcb05SBarry Smith case MAT_SORTED_FULL: 250290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 2514d9d31abSKris Buschelman break; 2529a4540c5SBarry Smith case MAT_HERMITIAN: 253eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 254eb1ec7c1SStefano Zampini if (flg) { /* disable transpose ops */ 255eb1ec7c1SStefano Zampini if (bs > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1"); 256eb1ec7c1SStefano Zampini A->ops->multtranspose = NULL; 257eb1ec7c1SStefano Zampini A->ops->multtransposeadd = NULL; 258eb1ec7c1SStefano Zampini A->symmetric = PETSC_FALSE; 259eb1ec7c1SStefano Zampini } 2600f2140c7SStefano Zampini #endif 261eeffb40dSHong Zhang break; 26277e54ba9SKris Buschelman case MAT_SYMMETRIC: 263eb1ec7c1SStefano Zampini case MAT_SPD: 264eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 265eb1ec7c1SStefano Zampini if (flg) { /* An hermitian and symmetric matrix has zero imaginary part (restore back transpose ops) */ 266eb1ec7c1SStefano Zampini A->ops->multtranspose = A->ops->mult; 267eb1ec7c1SStefano Zampini A->ops->multtransposeadd = A->ops->multadd; 268eb1ec7c1SStefano Zampini } 269eb1ec7c1SStefano Zampini #endif 270eb1ec7c1SStefano Zampini break; 271eb1ec7c1SStefano Zampini /* These options are handled directly by MatSetOption() */ 27277e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 2739a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 274672ba085SHong Zhang case MAT_STRUCTURE_ONLY: 2754dcd73b1SHong Zhang /* These options are handled directly by MatSetOption() */ 276290bbb0aSBarry Smith break; 277941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 2784e0d8c25SBarry Smith a->ignore_ltriangular = flg; 279941593c8SHong Zhang break; 280941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 2814e0d8c25SBarry Smith a->ignore_ltriangular = flg; 28277e54ba9SKris Buschelman break; 283f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 2844e0d8c25SBarry Smith a->getrow_utriangular = flg; 285f5edf698SHong Zhang break; 286c10200c1SHong Zhang case MAT_SUBMAT_SINGLEIS: 287c10200c1SHong Zhang break; 2884d9d31abSKris Buschelman default: 289e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 29049b5e25fSSatish Balay } 29149b5e25fSSatish Balay PetscFunctionReturn(0); 29249b5e25fSSatish Balay } 29349b5e25fSSatish Balay 29452768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 29549b5e25fSSatish Balay { 29649b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2976849ba73SBarry Smith PetscErrorCode ierr; 29849b5e25fSSatish Balay 29949b5e25fSSatish Balay PetscFunctionBegin; 300e32f2f54SBarry 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()"); 30152768537SHong Zhang 302f5edf698SHong Zhang /* Get the upper triangular part of the row */ 30352768537SHong Zhang ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr); 30449b5e25fSSatish Balay PetscFunctionReturn(0); 30549b5e25fSSatish Balay } 30649b5e25fSSatish Balay 30713f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 30849b5e25fSSatish Balay { 309dfbe8321SBarry Smith PetscErrorCode ierr; 31049b5e25fSSatish Balay 31149b5e25fSSatish Balay PetscFunctionBegin; 31205b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 31305b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 31449b5e25fSSatish Balay PetscFunctionReturn(0); 31549b5e25fSSatish Balay } 31649b5e25fSSatish Balay 317f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 318f5edf698SHong Zhang { 319f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 320f5edf698SHong Zhang 321f5edf698SHong Zhang PetscFunctionBegin; 322f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 323f5edf698SHong Zhang PetscFunctionReturn(0); 324f5edf698SHong Zhang } 325a323099bSStefano Zampini 326f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 327f5edf698SHong Zhang { 328f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 329f5edf698SHong Zhang 330f5edf698SHong Zhang PetscFunctionBegin; 331f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 332f5edf698SHong Zhang PetscFunctionReturn(0); 333f5edf698SHong Zhang } 334f5edf698SHong Zhang 335fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B) 33649b5e25fSSatish Balay { 337dfbe8321SBarry Smith PetscErrorCode ierr; 3385fd66863SKarl Rupp 33949b5e25fSSatish Balay PetscFunctionBegin; 340cf37664fSBarry Smith if (reuse == MAT_INITIAL_MATRIX) { 341999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 342cf37664fSBarry Smith } else if (reuse == MAT_REUSE_MATRIX) { 343cf37664fSBarry Smith ierr = MatCopy(A,*B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 344fc4dec0aSBarry Smith } 3458115998fSBarry Smith PetscFunctionReturn(0); 34649b5e25fSSatish Balay } 34749b5e25fSSatish Balay 3487da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 34949b5e25fSSatish Balay { 35049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 351dfbe8321SBarry Smith PetscErrorCode ierr; 352d0f46423SBarry Smith PetscInt i,j,bs = A->rmap->bs,k,l,bs2=a->bs2; 353f3ef73ceSBarry Smith PetscViewerFormat format; 354121deb67SSatish Balay PetscInt *diag; 35549b5e25fSSatish Balay 35649b5e25fSSatish Balay PetscFunctionBegin; 357b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 358456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 35977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 360fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 361d2507d54SMatthew Knepley Mat aij; 362ade3a672SBarry Smith const char *matname; 363ade3a672SBarry Smith 364d5f3da31SBarry Smith if (A->factortype && bs>1) { 36570d5e725SHong 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); 36670d5e725SHong Zhang PetscFunctionReturn(0); 36770d5e725SHong Zhang } 368c9f458caSMatthew Knepley ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr); 369ade3a672SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr); 370ade3a672SBarry Smith ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr); 371c9f458caSMatthew Knepley ierr = MatView(aij,viewer);CHKERRQ(ierr); 3726bf464f9SBarry Smith ierr = MatDestroy(&aij);CHKERRQ(ierr); 373fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 374d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 37549b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 37649b5e25fSSatish Balay for (j=0; j<bs; j++) { 37777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 37849b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 37949b5e25fSSatish Balay for (l=0; l<bs; l++) { 38049b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 38149b5e25fSSatish Balay 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 (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 38557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 38657622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38749b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 38857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38949b5e25fSSatish Balay } 39049b5e25fSSatish Balay #else 39149b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 39257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 39349b5e25fSSatish Balay } 39449b5e25fSSatish Balay #endif 39549b5e25fSSatish Balay } 39649b5e25fSSatish Balay } 397b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 39849b5e25fSSatish Balay } 39949b5e25fSSatish Balay } 400d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 401c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 402c1490034SHong Zhang PetscFunctionReturn(0); 40349b5e25fSSatish Balay } else { 404d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 4052c990fa1SHong Zhang if (A->factortype) { /* for factored matrix */ 4062c990fa1SHong Zhang if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet"); 4072c990fa1SHong Zhang 408121deb67SSatish Balay diag=a->diag; 409121deb67SSatish Balay for (i=0; i<a->mbs; i++) { /* for row block i */ 4102c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 4112c990fa1SHong Zhang /* diagonal entry */ 4122c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 4132c990fa1SHong Zhang if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) { 41457622a8eSBarry 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); 4152c990fa1SHong Zhang } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) { 41657622a8eSBarry 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); 4172c990fa1SHong Zhang } else { 41857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4192c990fa1SHong Zhang } 4202c990fa1SHong Zhang #else 4216712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4222c990fa1SHong Zhang #endif 4232c990fa1SHong Zhang /* off-diagonal entries */ 4242c990fa1SHong Zhang for (k=a->i[i]; k<a->i[i+1]-1; k++) { 4252c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 426ca0704adSBarry Smith if (PetscImaginaryPart(a->a[k]) > 0.0) { 42757622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 428ca0704adSBarry Smith } else if (PetscImaginaryPart(a->a[k]) < 0.0) { 42957622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 4302c990fa1SHong Zhang } else { 43157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr); 4322c990fa1SHong Zhang } 4332c990fa1SHong Zhang #else 43457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr); 4352c990fa1SHong Zhang #endif 4362c990fa1SHong Zhang } 4372c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 4382c990fa1SHong Zhang } 4392c990fa1SHong Zhang 4402c990fa1SHong Zhang } else { /* for non-factored matrix */ 4410c74a584SJed Brown for (i=0; i<a->mbs; i++) { /* for row block i */ 4420c74a584SJed Brown for (j=0; j<bs; j++) { /* for row bs*i + j */ 44377431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 4440c74a584SJed Brown for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */ 4450c74a584SJed Brown for (l=0; l<bs; l++) { /* for column */ 44649b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 44749b5e25fSSatish Balay 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 if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 45157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 45257622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 45349b5e25fSSatish Balay } else { 45457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 45549b5e25fSSatish Balay } 45649b5e25fSSatish Balay #else 45757622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 45849b5e25fSSatish Balay #endif 45949b5e25fSSatish Balay } 46049b5e25fSSatish Balay } 461b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 46249b5e25fSSatish Balay } 46349b5e25fSSatish Balay } 4642c990fa1SHong Zhang } 465d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 46649b5e25fSSatish Balay } 467b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 46849b5e25fSSatish Balay PetscFunctionReturn(0); 46949b5e25fSSatish Balay } 47049b5e25fSSatish Balay 4719804daf3SBarry Smith #include <petscdraw.h> 4726849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 47349b5e25fSSatish Balay { 47449b5e25fSSatish Balay Mat A = (Mat) Aa; 47549b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 4766849ba73SBarry Smith PetscErrorCode ierr; 477d0f46423SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2; 47849b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 47949b5e25fSSatish Balay MatScalar *aa; 480b0a32e0cSBarry Smith PetscViewer viewer; 48149b5e25fSSatish Balay 48249b5e25fSSatish Balay PetscFunctionBegin; 48349b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 484b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 48549b5e25fSSatish Balay 48649b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 487383922c3SLisandro Dalcin 488383922c3SLisandro Dalcin ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 489383922c3SLisandro Dalcin ierr = PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");CHKERRQ(ierr); 490383922c3SLisandro Dalcin /* Blue for negative, Cyan for zero and Red for positive */ 491b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 49249b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 49349b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 494d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 49549b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 49649b5e25fSSatish Balay aa = a->a + j*bs2; 49749b5e25fSSatish Balay for (k=0; k<bs; k++) { 49849b5e25fSSatish Balay for (l=0; l<bs; l++) { 49949b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 500b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 50149b5e25fSSatish Balay } 50249b5e25fSSatish Balay } 50349b5e25fSSatish Balay } 50449b5e25fSSatish Balay } 505b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 50649b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 50749b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 508d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 50949b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 51049b5e25fSSatish Balay aa = a->a + j*bs2; 51149b5e25fSSatish Balay for (k=0; k<bs; k++) { 51249b5e25fSSatish Balay for (l=0; l<bs; l++) { 51349b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 514b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 51549b5e25fSSatish Balay } 51649b5e25fSSatish Balay } 51749b5e25fSSatish Balay } 51849b5e25fSSatish Balay } 519b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 52049b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 52149b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 522d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 52349b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 52449b5e25fSSatish Balay aa = a->a + j*bs2; 52549b5e25fSSatish Balay for (k=0; k<bs; k++) { 52649b5e25fSSatish Balay for (l=0; l<bs; l++) { 52749b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 528b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 52949b5e25fSSatish Balay } 53049b5e25fSSatish Balay } 53149b5e25fSSatish Balay } 53249b5e25fSSatish Balay } 533383922c3SLisandro Dalcin ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 53449b5e25fSSatish Balay PetscFunctionReturn(0); 53549b5e25fSSatish Balay } 53649b5e25fSSatish Balay 5376849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 53849b5e25fSSatish Balay { 539dfbe8321SBarry Smith PetscErrorCode ierr; 54049b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 541b0a32e0cSBarry Smith PetscDraw draw; 542ace3abfcSBarry Smith PetscBool isnull; 54349b5e25fSSatish Balay 54449b5e25fSSatish Balay PetscFunctionBegin; 545b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 546383922c3SLisandro Dalcin ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 547383922c3SLisandro Dalcin if (isnull) PetscFunctionReturn(0); 54849b5e25fSSatish Balay 549d0f46423SBarry Smith xr = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0; 55049b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 551b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 552832b7cebSLisandro Dalcin ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 553b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 5540298fd71SBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 555832b7cebSLisandro Dalcin ierr = PetscDrawSave(draw);CHKERRQ(ierr); 55649b5e25fSSatish Balay PetscFunctionReturn(0); 55749b5e25fSSatish Balay } 55849b5e25fSSatish Balay 559618cc2edSLisandro Dalcin /* Used for both MPIBAIJ and MPISBAIJ matrices */ 560618cc2edSLisandro Dalcin #define MatView_SeqSBAIJ_Binary MatView_SeqBAIJ_Binary 561618cc2edSLisandro Dalcin 562dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 56349b5e25fSSatish Balay { 564dfbe8321SBarry Smith PetscErrorCode ierr; 565618cc2edSLisandro Dalcin PetscBool iascii,isbinary,isdraw; 56649b5e25fSSatish Balay 56749b5e25fSSatish Balay PetscFunctionBegin; 568251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 569618cc2edSLisandro Dalcin ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 570251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 57132077d6dSBarry Smith if (iascii) { 57249b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 573618cc2edSLisandro Dalcin } else if (isbinary) { 574618cc2edSLisandro Dalcin ierr = MatView_SeqSBAIJ_Binary(A,viewer);CHKERRQ(ierr); 57549b5e25fSSatish Balay } else if (isdraw) { 57649b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 57749b5e25fSSatish Balay } else { 578a5e6ed63SBarry Smith Mat B; 579ade3a672SBarry Smith const char *matname; 580ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 581ade3a672SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr); 582ade3a672SBarry Smith ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr); 583a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 5846bf464f9SBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 58549b5e25fSSatish Balay } 58649b5e25fSSatish Balay PetscFunctionReturn(0); 58749b5e25fSSatish Balay } 58849b5e25fSSatish Balay 58949b5e25fSSatish Balay 59013f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 59149b5e25fSSatish Balay { 592045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 59313f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 59413f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 595d0f46423SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2; 59697e567efSBarry Smith MatScalar *ap,*aa = a->a; 59749b5e25fSSatish Balay 59849b5e25fSSatish Balay PetscFunctionBegin; 59949b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 60049b5e25fSSatish Balay row = im[k]; brow = row/bs; 601e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 602e32f2f54SBarry 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); 60349b5e25fSSatish Balay rp = aj + ai[brow]; ap = aa + bs2*ai[brow]; 60449b5e25fSSatish Balay nrow = ailen[brow]; 60549b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 606e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 607e32f2f54SBarry 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); 60849b5e25fSSatish Balay col = in[l]; 60949b5e25fSSatish Balay bcol = col/bs; 61049b5e25fSSatish Balay cidx = col%bs; 61149b5e25fSSatish Balay ridx = row%bs; 61249b5e25fSSatish Balay high = nrow; 61349b5e25fSSatish Balay low = 0; /* assume unsorted */ 61449b5e25fSSatish Balay while (high-low > 5) { 61549b5e25fSSatish Balay t = (low+high)/2; 61649b5e25fSSatish Balay if (rp[t] > bcol) high = t; 61749b5e25fSSatish Balay else low = t; 61849b5e25fSSatish Balay } 61949b5e25fSSatish Balay for (i=low; i<high; i++) { 62049b5e25fSSatish Balay if (rp[i] > bcol) break; 62149b5e25fSSatish Balay if (rp[i] == bcol) { 62249b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 62349b5e25fSSatish Balay goto finished; 62449b5e25fSSatish Balay } 62549b5e25fSSatish Balay } 62697e567efSBarry Smith *v++ = 0.0; 62749b5e25fSSatish Balay finished:; 62849b5e25fSSatish Balay } 62949b5e25fSSatish Balay } 63049b5e25fSSatish Balay PetscFunctionReturn(0); 63149b5e25fSSatish Balay } 63249b5e25fSSatish Balay 633dc29a518SPierre Jolivet PetscErrorCode MatPermute_SeqSBAIJ(Mat A,IS rowp,IS colp,Mat *B) 634dc29a518SPierre Jolivet { 635dc29a518SPierre Jolivet Mat C; 636dc29a518SPierre Jolivet PetscErrorCode ierr; 637dc29a518SPierre Jolivet 638dc29a518SPierre Jolivet PetscFunctionBegin; 639dc29a518SPierre Jolivet ierr = MatConvert(A,MATSEQBAIJ,MAT_INITIAL_MATRIX,&C);CHKERRQ(ierr); 640dc29a518SPierre Jolivet ierr = MatPermute(C,rowp,colp,B);CHKERRQ(ierr); 641dc29a518SPierre Jolivet ierr = MatDestroy(&C);CHKERRQ(ierr); 642dc29a518SPierre Jolivet if (rowp == colp) { 643dc29a518SPierre Jolivet ierr = MatConvert(*B,MATSEQSBAIJ,MAT_INPLACE_MATRIX,B);CHKERRQ(ierr); 644dc29a518SPierre Jolivet } 645dc29a518SPierre Jolivet PetscFunctionReturn(0); 646dc29a518SPierre Jolivet } 64749b5e25fSSatish Balay 64813f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 64949b5e25fSSatish Balay { 6500880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 6516849ba73SBarry Smith PetscErrorCode ierr; 652e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 65313f74950SBarry Smith PetscInt *imax =a->imax,*ai=a->i,*ailen=a->ilen; 654d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval; 655ace3abfcSBarry Smith PetscBool roworiented=a->roworiented; 656dd6ea824SBarry Smith const PetscScalar *value = v; 657f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 6580880e062SHong Zhang 65949b5e25fSSatish Balay PetscFunctionBegin; 66026fbe8dcSKarl Rupp if (roworiented) stepval = (n-1)*bs; 66126fbe8dcSKarl Rupp else stepval = (m-1)*bs; 66226fbe8dcSKarl Rupp 6630880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 6640880e062SHong Zhang row = im[k]; 6650880e062SHong Zhang if (row < 0) continue; 666cf9c20a2SJed Brown if (PetscUnlikelyDebug(row >= a->mbs)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index row too large %D max %D",row,a->mbs-1); 6670880e062SHong Zhang rp = aj + ai[row]; 6680880e062SHong Zhang ap = aa + bs2*ai[row]; 6690880e062SHong Zhang rmax = imax[row]; 6700880e062SHong Zhang nrow = ailen[row]; 6710880e062SHong Zhang low = 0; 672818f2c47SBarry Smith high = nrow; 6730880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 6740880e062SHong Zhang if (in[l] < 0) continue; 6750880e062SHong Zhang col = in[l]; 676cf9c20a2SJed Brown if (PetscUnlikelyDebug(col >= a->nbs)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index column too large %D max %D",col,a->nbs-1); 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; 7302f7d4af7SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new block index nonzero block (%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 */ 734580bdb30SBarry Smith ierr = PetscArraymove(rp+i+1,rp+i,N-i+1);CHKERRQ(ierr); 735580bdb30SBarry Smith ierr = PetscArraymove(ap+bs2*(i+1),ap+bs2*i,bs2*(N-i+1));CHKERRQ(ierr); 736580bdb30SBarry Smith ierr = PetscArrayzero(ap+bs2*i,bs2);CHKERRQ(ierr); 7370880e062SHong Zhang rp[i] = col; 7380880e062SHong Zhang bap = ap + bs2*i; 7390880e062SHong Zhang if (roworiented) { 7400880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7410880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7420880e062SHong Zhang bap[jj] = *value++; 7430880e062SHong Zhang } 7440880e062SHong Zhang } 7450880e062SHong Zhang } else { 7460880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7470880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7480880e062SHong Zhang *bap++ = *value++; 7490880e062SHong Zhang } 7500880e062SHong Zhang } 7510880e062SHong Zhang } 7520880e062SHong Zhang noinsert2:; 7530880e062SHong Zhang low = i; 7540880e062SHong Zhang } 7550880e062SHong Zhang ailen[row] = nrow; 7560880e062SHong Zhang } 7570880e062SHong Zhang PetscFunctionReturn(0); 75849b5e25fSSatish Balay } 75949b5e25fSSatish Balay 76064831d72SBarry Smith /* 76164831d72SBarry Smith This is not yet used 76264831d72SBarry Smith */ 7634108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A) 7640def2e27SBarry Smith { 7650def2e27SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7660def2e27SBarry Smith PetscErrorCode ierr; 7670def2e27SBarry Smith const PetscInt *ai = a->i, *aj = a->j,*cols; 7680def2e27SBarry Smith PetscInt i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts; 769ace3abfcSBarry Smith PetscBool flag; 7700def2e27SBarry Smith 7710def2e27SBarry Smith PetscFunctionBegin; 772785e854fSJed Brown ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr); 7730def2e27SBarry Smith while (i < m) { 7740def2e27SBarry Smith nzx = ai[i+1] - ai[i]; /* Number of nonzeros */ 7750def2e27SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 7760def2e27SBarry Smith for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 7770def2e27SBarry Smith nzy = ai[j+1] - ai[j]; 7780def2e27SBarry Smith if (nzy != (nzx - j + i)) break; 779580bdb30SBarry Smith ierr = PetscArraycmp(aj + ai[i] + j - i,aj + ai[j],nzy,&flag);CHKERRQ(ierr); 7800def2e27SBarry Smith if (!flag) break; 7810def2e27SBarry Smith } 7820def2e27SBarry Smith ns[node_count++] = blk_size; 78326fbe8dcSKarl Rupp 7840def2e27SBarry Smith i = j; 7850def2e27SBarry Smith } 7860def2e27SBarry Smith if (!a->inode.size && m && node_count > .9*m) { 7870def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7880def2e27SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 7890def2e27SBarry Smith } else { 7900def2e27SBarry Smith a->inode.node_count = node_count; 79126fbe8dcSKarl Rupp 792785e854fSJed Brown ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr); 7933bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr); 794580bdb30SBarry Smith ierr = PetscArraycpy(a->inode.size,ns,node_count);CHKERRQ(ierr); 7950def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7960def2e27SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 7970def2e27SBarry Smith 7980def2e27SBarry Smith /* count collections of adjacent columns in each inode */ 7990def2e27SBarry Smith row = 0; 8000def2e27SBarry Smith cnt = 0; 8010def2e27SBarry Smith for (i=0; i<node_count; i++) { 8020def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 8030def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 8040def2e27SBarry Smith for (j=1; j<nz; j++) { 80526fbe8dcSKarl Rupp if (cols[j] != cols[j-1]+1) cnt++; 8060def2e27SBarry Smith } 8070def2e27SBarry Smith cnt++; 8080def2e27SBarry Smith row += a->inode.size[i]; 8090def2e27SBarry Smith } 810785e854fSJed Brown ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr); 8110def2e27SBarry Smith cnt = 0; 8120def2e27SBarry Smith row = 0; 8130def2e27SBarry Smith for (i=0; i<node_count; i++) { 8140def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 8150def2e27SBarry Smith counts[2*cnt] = cols[0]; 8160def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 8170def2e27SBarry Smith cnt2 = 1; 8180def2e27SBarry Smith for (j=1; j<nz; j++) { 8190def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 8200def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 8210def2e27SBarry Smith counts[2*cnt] = cols[j]; 8220def2e27SBarry Smith cnt2 = 1; 8230def2e27SBarry Smith } else cnt2++; 8240def2e27SBarry Smith } 8250def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 8260def2e27SBarry Smith row += a->inode.size[i]; 8270def2e27SBarry Smith } 828f4259b30SLisandro Dalcin ierr = PetscIntView(2*cnt,counts,NULL);CHKERRQ(ierr); 8290def2e27SBarry Smith } 83038702af4SBarry Smith PetscFunctionReturn(0); 83138702af4SBarry Smith } 83238702af4SBarry Smith 833dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 83449b5e25fSSatish Balay { 83549b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 8366849ba73SBarry Smith PetscErrorCode ierr; 8378f8f2f0dSBarry Smith PetscInt fshift = 0,i,*ai = a->i,*aj = a->j,*imax = a->imax; 838d0f46423SBarry Smith PetscInt m = A->rmap->N,*ip,N,*ailen = a->ilen; 83913f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 84049b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 84149b5e25fSSatish Balay 84249b5e25fSSatish Balay PetscFunctionBegin; 84349b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 84449b5e25fSSatish Balay 84549b5e25fSSatish Balay if (m) rmax = ailen[0]; 84649b5e25fSSatish Balay for (i=1; i<mbs; i++) { 84749b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 84849b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 84949b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 85049b5e25fSSatish Balay if (fshift) { 851580bdb30SBarry Smith ip = aj + ai[i]; 852580bdb30SBarry Smith ap = aa + bs2*ai[i]; 85349b5e25fSSatish Balay N = ailen[i]; 854580bdb30SBarry Smith ierr = PetscArraymove(ip-fshift,ip,N);CHKERRQ(ierr); 855580bdb30SBarry Smith ierr = PetscArraymove(ap-bs2*fshift,ap,bs2*N);CHKERRQ(ierr); 85649b5e25fSSatish Balay } 85749b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 85849b5e25fSSatish Balay } 85949b5e25fSSatish Balay if (mbs) { 86049b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 86149b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 86249b5e25fSSatish Balay } 86349b5e25fSSatish Balay /* reset ilen and imax for each row */ 86449b5e25fSSatish Balay for (i=0; i<mbs; i++) { 86549b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 86649b5e25fSSatish Balay } 8676c6c5352SBarry Smith a->nz = ai[mbs]; 86849b5e25fSSatish Balay 869b424e231SHong Zhang /* diagonals may have moved, reset it */ 870b424e231SHong Zhang if (a->diag) { 871580bdb30SBarry Smith ierr = PetscArraycpy(a->diag,ai,mbs);CHKERRQ(ierr); 87249b5e25fSSatish Balay } 87326fbe8dcSKarl 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); 87426fbe8dcSKarl Rupp 875d0f46423SBarry 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); 876ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 877ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 87826fbe8dcSKarl Rupp 8798e58a170SBarry Smith A->info.mallocs += a->reallocs; 88049b5e25fSSatish Balay a->reallocs = 0; 88149b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 882061b2667SBarry Smith a->idiagvalid = PETSC_FALSE; 8834dcd73b1SHong Zhang a->rmax = rmax; 88438702af4SBarry Smith 88538702af4SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 88644e1c64aSLisandro Dalcin if (a->jshort && a->free_jshort) { 88717803ae8SHong Zhang /* when matrix data structure is changed, previous jshort must be replaced */ 88817803ae8SHong Zhang ierr = PetscFree(a->jshort);CHKERRQ(ierr); 88917803ae8SHong Zhang } 890785e854fSJed Brown ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr); 8913bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr); 89238702af4SBarry Smith for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i]; 89338702af4SBarry Smith A->ops->mult = MatMult_SeqSBAIJ_1_ushort; 89441f059aeSBarry Smith A->ops->sor = MatSOR_SeqSBAIJ_ushort; 8954da8f245SBarry Smith a->free_jshort = PETSC_TRUE; 89638702af4SBarry Smith } 89749b5e25fSSatish Balay PetscFunctionReturn(0); 89849b5e25fSSatish Balay } 89949b5e25fSSatish Balay 90049b5e25fSSatish Balay /* 90149b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 90249b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 90349b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 90449b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 90549b5e25fSSatish Balay */ 90613f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 90749b5e25fSSatish Balay { 90813f74950SBarry Smith PetscInt i,j,k,row; 909ace3abfcSBarry Smith PetscBool flg; 91049b5e25fSSatish Balay 91149b5e25fSSatish Balay PetscFunctionBegin; 91249b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 91349b5e25fSSatish Balay row = idx[i]; 91449b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 91549b5e25fSSatish Balay sizes[j] = 1; 91649b5e25fSSatish Balay i++; 91749b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 91849b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 91949b5e25fSSatish Balay i++; 92049b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 92149b5e25fSSatish Balay flg = PETSC_TRUE; 92249b5e25fSSatish Balay for (k=1; k<bs; k++) { 92349b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 92449b5e25fSSatish Balay flg = PETSC_FALSE; 92549b5e25fSSatish Balay break; 92649b5e25fSSatish Balay } 92749b5e25fSSatish Balay } 928abc0a331SBarry Smith if (flg) { /* No break in the bs */ 92949b5e25fSSatish Balay sizes[j] = bs; 93049b5e25fSSatish Balay i += bs; 93149b5e25fSSatish Balay } else { 93249b5e25fSSatish Balay sizes[j] = 1; 93349b5e25fSSatish Balay i++; 93449b5e25fSSatish Balay } 93549b5e25fSSatish Balay } 93649b5e25fSSatish Balay } 93749b5e25fSSatish Balay *bs_max = j; 93849b5e25fSSatish Balay PetscFunctionReturn(0); 93949b5e25fSSatish Balay } 94049b5e25fSSatish Balay 94149b5e25fSSatish Balay 94249b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 94349b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 94449b5e25fSSatish Balay */ 94549b5e25fSSatish Balay 94613f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 94749b5e25fSSatish Balay { 94849b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 9496849ba73SBarry Smith PetscErrorCode ierr; 950e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 95113f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 952d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol; 95313f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 95449b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 95549b5e25fSSatish Balay 95649b5e25fSSatish Balay PetscFunctionBegin; 95749b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 95849b5e25fSSatish Balay row = im[k]; /* row number */ 95949b5e25fSSatish Balay brow = row/bs; /* block row number */ 96049b5e25fSSatish Balay if (row < 0) continue; 961cf9c20a2SJed Brown if (PetscUnlikelyDebug(row >= A->rmap->N)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1); 96249b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 96349b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 96449b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 96549b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 96649b5e25fSSatish Balay low = 0; 9678509e838SStefano Zampini high = nrow; 96849b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 96949b5e25fSSatish Balay if (in[l] < 0) continue; 970cf9c20a2SJed Brown if (PetscUnlikelyDebug(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); 97149b5e25fSSatish Balay col = in[l]; 97249b5e25fSSatish Balay bcol = col/bs; /* block col number */ 97349b5e25fSSatish Balay 974941593c8SHong Zhang if (brow > bcol) { 97526fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular values */ 97626fbe8dcSKarl 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)"); 977941593c8SHong Zhang } 978f4989cb3SHong Zhang 97949b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 9808549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)) { 98149b5e25fSSatish Balay /* element value a(k,l) */ 98226fbe8dcSKarl Rupp if (roworiented) value = v[l + k*n]; 98326fbe8dcSKarl Rupp else value = v[k + l*m]; 98449b5e25fSSatish Balay 98549b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 98626fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 9878509e838SStefano Zampini else high = nrow; 9888509e838SStefano Zampini 989e2ee6c50SBarry Smith lastcol = col; 99049b5e25fSSatish Balay while (high-low > 7) { 99149b5e25fSSatish Balay t = (low+high)/2; 99249b5e25fSSatish Balay if (rp[t] > bcol) high = t; 99349b5e25fSSatish Balay else low = t; 99449b5e25fSSatish Balay } 99549b5e25fSSatish Balay for (i=low; i<high; i++) { 99649b5e25fSSatish Balay if (rp[i] > bcol) break; 99749b5e25fSSatish Balay if (rp[i] == bcol) { 99849b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 99949b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 100049b5e25fSSatish Balay else *bap = value; 10018549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 10028549e402SHong Zhang if (brow == bcol && ridx < cidx) { 10038549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 10048549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 10058549e402SHong Zhang else *bap = value; 10068549e402SHong Zhang } 100749b5e25fSSatish Balay goto noinsert1; 100849b5e25fSSatish Balay } 100949b5e25fSSatish Balay } 101049b5e25fSSatish Balay 101149b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 1012e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 1013fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 101449b5e25fSSatish Balay 1015c03d1d03SSatish Balay N = nrow++ - 1; high++; 101649b5e25fSSatish Balay /* shift up all the later entries in this row */ 1017580bdb30SBarry Smith ierr = PetscArraymove(rp+i+1,rp+i,N-i+1);CHKERRQ(ierr); 1018580bdb30SBarry Smith ierr = PetscArraymove(ap+bs2*(i+1),ap+bs2*i,bs2*(N-i+1));CHKERRQ(ierr); 1019580bdb30SBarry Smith ierr = PetscArrayzero(ap+bs2*i,bs2);CHKERRQ(ierr); 102049b5e25fSSatish Balay rp[i] = bcol; 102149b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 10228509e838SStefano Zampini /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 10238509e838SStefano Zampini if (brow == bcol && ridx < cidx) { 10248509e838SStefano Zampini ap[bs2*i + bs*ridx + cidx] = value; 10258509e838SStefano Zampini } 1026e56f5c9eSBarry Smith A->nonzerostate++; 102749b5e25fSSatish Balay noinsert1:; 102849b5e25fSSatish Balay low = i; 10298549e402SHong Zhang } 103049b5e25fSSatish Balay } /* end of loop over added columns */ 103149b5e25fSSatish Balay ailen[brow] = nrow; 103249b5e25fSSatish Balay } /* end of loop over added rows */ 103349b5e25fSSatish Balay PetscFunctionReturn(0); 103449b5e25fSSatish Balay } 103549b5e25fSSatish Balay 10360481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info) 103749b5e25fSSatish Balay { 10384ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 103949b5e25fSSatish Balay Mat outA; 1040dfbe8321SBarry Smith PetscErrorCode ierr; 1041ace3abfcSBarry Smith PetscBool row_identity; 104249b5e25fSSatish Balay 104349b5e25fSSatish Balay PetscFunctionBegin; 1044e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 1045c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 1046e32f2f54SBarry Smith if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported"); 1047e32f2f54SBarry 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()! */ 1048c84f5b01SHong Zhang 104949b5e25fSSatish Balay outA = inA; 1050d5f3da31SBarry Smith inA->factortype = MAT_FACTOR_ICC; 1051f6224b95SHong Zhang ierr = PetscFree(inA->solvertype);CHKERRQ(ierr); 1052f6224b95SHong Zhang ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr); 105349b5e25fSSatish Balay 10541a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 1055d595f711SHong Zhang ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr); 105649b5e25fSSatish Balay 1057c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10586bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1059c84f5b01SHong Zhang a->row = row; 1060c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10616bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1062c84f5b01SHong Zhang a->col = row; 1063c84f5b01SHong Zhang 1064c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1065c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 10663bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr); 106749b5e25fSSatish Balay 106849b5e25fSSatish Balay if (!a->solve_work) { 1069854ce69bSBarry Smith ierr = PetscMalloc1(inA->rmap->N+inA->rmap->bs,&a->solve_work);CHKERRQ(ierr); 10703bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 107149b5e25fSSatish Balay } 107249b5e25fSSatish Balay 1073719d5645SBarry Smith ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr); 107449b5e25fSSatish Balay PetscFunctionReturn(0); 107549b5e25fSSatish Balay } 1076950f1e5bSHong Zhang 10777087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 107849b5e25fSSatish Balay { 1079045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ*)mat->data; 108013f74950SBarry Smith PetscInt i,nz,n; 10817827cd58SJed Brown PetscErrorCode ierr; 108249b5e25fSSatish Balay 108349b5e25fSSatish Balay PetscFunctionBegin; 10846c6c5352SBarry Smith nz = baij->maxnz; 1085d0f46423SBarry Smith n = mat->cmap->n; 108626fbe8dcSKarl Rupp for (i=0; i<nz; i++) baij->j[i] = indices[i]; 108726fbe8dcSKarl Rupp 10886c6c5352SBarry Smith baij->nz = nz; 108926fbe8dcSKarl Rupp for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i]; 109026fbe8dcSKarl Rupp 10917827cd58SJed Brown ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 109249b5e25fSSatish Balay PetscFunctionReturn(0); 109349b5e25fSSatish Balay } 109449b5e25fSSatish Balay 109549b5e25fSSatish Balay /*@ 109619585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 109749b5e25fSSatish Balay in the matrix. 109849b5e25fSSatish Balay 109949b5e25fSSatish Balay Input Parameters: 110019585528SSatish Balay + mat - the SeqSBAIJ matrix 110149b5e25fSSatish Balay - indices - the column indices 110249b5e25fSSatish Balay 110349b5e25fSSatish Balay Level: advanced 110449b5e25fSSatish Balay 110549b5e25fSSatish Balay Notes: 110649b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 110749b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 110849b5e25fSSatish Balay of the MatSetValues() operation. 110949b5e25fSSatish Balay 111049b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1111d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 111249b5e25fSSatish Balay 1113ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 111449b5e25fSSatish Balay 1115ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 111649b5e25fSSatish Balay @*/ 11177087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 111849b5e25fSSatish Balay { 11194ac538c5SBarry Smith PetscErrorCode ierr; 112049b5e25fSSatish Balay 112149b5e25fSSatish Balay PetscFunctionBegin; 11220700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 11234482741eSBarry Smith PetscValidPointer(indices,2); 11244ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr); 112549b5e25fSSatish Balay PetscFunctionReturn(0); 112649b5e25fSSatish Balay } 112749b5e25fSSatish Balay 11283c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 11293c896bc6SHong Zhang { 11303c896bc6SHong Zhang PetscErrorCode ierr; 11314c7a3774SStefano Zampini PetscBool isbaij; 11323c896bc6SHong Zhang 11333c896bc6SHong Zhang PetscFunctionBegin; 11344c7a3774SStefano Zampini ierr = PetscObjectTypeCompareAny((PetscObject)B,&isbaij,MATSEQSBAIJ,MATMPISBAIJ,"");CHKERRQ(ierr); 11354c7a3774SStefano Zampini if (!isbaij) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"Not for matrix type %s",((PetscObject)B)->type_name); 11364c7a3774SStefano Zampini /* If the two matrices have the same copy implementation and nonzero pattern, use fast copy. */ 11374c7a3774SStefano Zampini if (str == SAME_NONZERO_PATTERN && A->ops->copy == B->ops->copy) { 11383c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11393c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11403c896bc6SHong Zhang 11414c7a3774SStefano Zampini if (a->i[a->mbs] != b->i[b->mbs]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 11424c7a3774SStefano Zampini if (a->mbs != b->mbs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of rows in two matrices are different"); 11434c7a3774SStefano Zampini if (a->bs2 != b->bs2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Different block size"); 1144580bdb30SBarry Smith ierr = PetscArraycpy(b->a,a->a,a->bs2*a->i[a->mbs]);CHKERRQ(ierr); 1145cdc753b6SBarry Smith ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr); 11463c896bc6SHong Zhang } else { 1147f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11483c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1149f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11503c896bc6SHong Zhang } 11513c896bc6SHong Zhang PetscFunctionReturn(0); 11523c896bc6SHong Zhang } 11533c896bc6SHong Zhang 11544994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A) 1155273d9f13SBarry Smith { 1156dfbe8321SBarry Smith PetscErrorCode ierr; 1157273d9f13SBarry Smith 1158273d9f13SBarry Smith PetscFunctionBegin; 1159f4259b30SLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(A,A->rmap->bs,PETSC_DEFAULT,NULL);CHKERRQ(ierr); 1160273d9f13SBarry Smith PetscFunctionReturn(0); 1161273d9f13SBarry Smith } 1162273d9f13SBarry Smith 1163cda14afcSprj- static PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1164a6ece127SHong Zhang { 1165a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11665fd66863SKarl Rupp 1167a6ece127SHong Zhang PetscFunctionBegin; 1168a6ece127SHong Zhang *array = a->a; 1169a6ece127SHong Zhang PetscFunctionReturn(0); 1170a6ece127SHong Zhang } 1171a6ece127SHong Zhang 1172cda14afcSprj- static PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1173a6ece127SHong Zhang { 1174a6ece127SHong Zhang PetscFunctionBegin; 1175cda14afcSprj- *array = NULL; 1176a6ece127SHong Zhang PetscFunctionReturn(0); 1177a6ece127SHong Zhang } 1178a6ece127SHong Zhang 117952768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz) 118052768537SHong Zhang { 1181b264fe52SHong Zhang PetscInt bs = Y->rmap->bs,mbs = Y->rmap->N/bs; 118252768537SHong Zhang Mat_SeqSBAIJ *x = (Mat_SeqSBAIJ*)X->data; 118352768537SHong Zhang Mat_SeqSBAIJ *y = (Mat_SeqSBAIJ*)Y->data; 1184b264fe52SHong Zhang PetscErrorCode ierr; 118552768537SHong Zhang 118652768537SHong Zhang PetscFunctionBegin; 118752768537SHong Zhang /* Set the number of nonzeros in the new matrix */ 1188b264fe52SHong Zhang ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr); 118952768537SHong Zhang PetscFunctionReturn(0); 119052768537SHong Zhang } 119152768537SHong Zhang 1192f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 119342ee4b1aSHong Zhang { 119442ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data; 1195dfbe8321SBarry Smith PetscErrorCode ierr; 119631ce2d13SHong Zhang PetscInt bs=Y->rmap->bs,bs2=bs*bs; 1197e838b9e7SJed Brown PetscBLASInt one = 1; 119842ee4b1aSHong Zhang 119942ee4b1aSHong Zhang PetscFunctionBegin; 120042ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1201f4df32b1SMatthew Knepley PetscScalar alpha = a; 1202c5df96a5SBarry Smith PetscBLASInt bnz; 1203c5df96a5SBarry Smith ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr); 12048b83055fSJed Brown PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one)); 1205a3fa217bSJose E. Roman ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 1206ab784542SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1207ab784542SHong Zhang ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr); 1208ab784542SHong Zhang ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1209ab784542SHong Zhang ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr); 121042ee4b1aSHong Zhang } else { 121152768537SHong Zhang Mat B; 121252768537SHong Zhang PetscInt *nnz; 121352768537SHong Zhang if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size"); 1214f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 121552768537SHong Zhang ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr); 121652768537SHong Zhang ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr); 121752768537SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr); 121852768537SHong Zhang ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr); 121952768537SHong Zhang ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr); 122052768537SHong Zhang ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr); 12214c7a3774SStefano Zampini ierr = MatSetType(B,((PetscObject)Y)->type_name);CHKERRQ(ierr); 122252768537SHong Zhang ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr); 122352768537SHong Zhang ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 122452768537SHong Zhang 122552768537SHong Zhang ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr); 122652768537SHong Zhang 122728be2f97SBarry Smith ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr); 122852768537SHong Zhang ierr = PetscFree(nnz);CHKERRQ(ierr); 1229f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 123052768537SHong Zhang ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr); 123142ee4b1aSHong Zhang } 123242ee4b1aSHong Zhang PetscFunctionReturn(0); 123342ee4b1aSHong Zhang } 123442ee4b1aSHong Zhang 1235ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1236efcf0fc3SBarry Smith { 1237efcf0fc3SBarry Smith PetscFunctionBegin; 1238efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1239efcf0fc3SBarry Smith PetscFunctionReturn(0); 1240efcf0fc3SBarry Smith } 1241efcf0fc3SBarry Smith 1242ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool *flg) 1243efcf0fc3SBarry Smith { 1244efcf0fc3SBarry Smith PetscFunctionBegin; 1245efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1246efcf0fc3SBarry Smith PetscFunctionReturn(0); 1247efcf0fc3SBarry Smith } 1248efcf0fc3SBarry Smith 1249ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1250efcf0fc3SBarry Smith { 1251efcf0fc3SBarry Smith PetscFunctionBegin; 1252efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1253efcf0fc3SBarry Smith PetscFunctionReturn(0); 1254efcf0fc3SBarry Smith } 1255efcf0fc3SBarry Smith 12562726fb6dSPierre Jolivet PetscErrorCode MatConjugate_SeqSBAIJ(Mat A) 12572726fb6dSPierre Jolivet { 12582726fb6dSPierre Jolivet #if defined(PETSC_USE_COMPLEX) 12592726fb6dSPierre Jolivet Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 12602726fb6dSPierre Jolivet PetscInt i,nz = a->bs2*a->i[a->mbs]; 12612726fb6dSPierre Jolivet MatScalar *aa = a->a; 12622726fb6dSPierre Jolivet 12632726fb6dSPierre Jolivet PetscFunctionBegin; 12642726fb6dSPierre Jolivet for (i=0; i<nz; i++) aa[i] = PetscConj(aa[i]); 12652726fb6dSPierre Jolivet #else 12662726fb6dSPierre Jolivet PetscFunctionBegin; 12672726fb6dSPierre Jolivet #endif 12682726fb6dSPierre Jolivet PetscFunctionReturn(0); 12692726fb6dSPierre Jolivet } 12702726fb6dSPierre Jolivet 127199cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 127299cafbc1SBarry Smith { 127399cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 127499cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1275dd6ea824SBarry Smith MatScalar *aa = a->a; 127699cafbc1SBarry Smith 127799cafbc1SBarry Smith PetscFunctionBegin; 127899cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 127999cafbc1SBarry Smith PetscFunctionReturn(0); 128099cafbc1SBarry Smith } 128199cafbc1SBarry Smith 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 PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b) 12943bededecSBarry Smith { 12953bededecSBarry Smith Mat_SeqSBAIJ *baij=(Mat_SeqSBAIJ*)A->data; 12963bededecSBarry Smith PetscErrorCode ierr; 12973bededecSBarry Smith PetscInt i,j,k,count; 12983bededecSBarry Smith PetscInt bs =A->rmap->bs,bs2=baij->bs2,row,col; 12993bededecSBarry Smith PetscScalar zero = 0.0; 13003bededecSBarry Smith MatScalar *aa; 13013bededecSBarry Smith const PetscScalar *xx; 13023bededecSBarry Smith PetscScalar *bb; 130356777dd2SBarry Smith PetscBool *zeroed,vecs = PETSC_FALSE; 13043bededecSBarry Smith 13053bededecSBarry Smith PetscFunctionBegin; 13063bededecSBarry Smith /* fix right hand side if needed */ 13073bededecSBarry Smith if (x && b) { 13083bededecSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 13093bededecSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 131056777dd2SBarry Smith vecs = PETSC_TRUE; 13113bededecSBarry Smith } 13123bededecSBarry Smith 13133bededecSBarry Smith /* zero the columns */ 13141795a4d1SJed Brown ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr); 13153bededecSBarry Smith for (i=0; i<is_n; i++) { 13163bededecSBarry 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]); 13173bededecSBarry Smith zeroed[is_idx[i]] = PETSC_TRUE; 13183bededecSBarry Smith } 131956777dd2SBarry Smith if (vecs) { 132056777dd2SBarry Smith for (i=0; i<A->rmap->N; i++) { 132156777dd2SBarry Smith row = i/bs; 132256777dd2SBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 132356777dd2SBarry Smith for (k=0; k<bs; k++) { 132456777dd2SBarry Smith col = bs*baij->j[j] + k; 132556777dd2SBarry Smith if (col <= i) continue; 132656777dd2SBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 132726fbe8dcSKarl Rupp if (!zeroed[i] && zeroed[col]) bb[i] -= aa[0]*xx[col]; 132826fbe8dcSKarl Rupp if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i]; 132956777dd2SBarry Smith } 133056777dd2SBarry Smith } 133156777dd2SBarry Smith } 133226fbe8dcSKarl Rupp for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]]; 133356777dd2SBarry Smith } 133456777dd2SBarry Smith 13353bededecSBarry Smith for (i=0; i<A->rmap->N; i++) { 13363bededecSBarry Smith if (!zeroed[i]) { 13373bededecSBarry Smith row = i/bs; 13383bededecSBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 13393bededecSBarry Smith for (k=0; k<bs; k++) { 13403bededecSBarry Smith col = bs*baij->j[j] + k; 13413bededecSBarry Smith if (zeroed[col]) { 13423bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 13433bededecSBarry Smith aa[0] = 0.0; 13443bededecSBarry Smith } 13453bededecSBarry Smith } 13463bededecSBarry Smith } 13473bededecSBarry Smith } 13483bededecSBarry Smith } 13493bededecSBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 135056777dd2SBarry Smith if (vecs) { 135156777dd2SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 135256777dd2SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 135356777dd2SBarry Smith } 13543bededecSBarry Smith 13553bededecSBarry Smith /* zero the rows */ 13563bededecSBarry Smith for (i=0; i<is_n; i++) { 13573bededecSBarry Smith row = is_idx[i]; 13583bededecSBarry Smith count = (baij->i[row/bs +1] - baij->i[row/bs])*bs; 13593bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs); 13603bededecSBarry Smith for (k=0; k<count; k++) { 13613bededecSBarry Smith aa[0] = zero; 13623bededecSBarry Smith aa += bs; 13633bededecSBarry Smith } 13643bededecSBarry Smith if (diag != 0.0) { 13653bededecSBarry Smith ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr); 13663bededecSBarry Smith } 13673bededecSBarry Smith } 13683bededecSBarry Smith ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 13693bededecSBarry Smith PetscFunctionReturn(0); 13703bededecSBarry Smith } 13713bededecSBarry Smith 13727d68702bSBarry Smith PetscErrorCode MatShift_SeqSBAIJ(Mat Y,PetscScalar a) 13737d68702bSBarry Smith { 13747d68702bSBarry Smith PetscErrorCode ierr; 13757d68702bSBarry Smith Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)Y->data; 13767d68702bSBarry Smith 13777d68702bSBarry Smith PetscFunctionBegin; 13786f33a894SBarry Smith if (!Y->preallocated || !aij->nz) { 13797d68702bSBarry Smith ierr = MatSeqSBAIJSetPreallocation(Y,Y->rmap->bs,1,NULL);CHKERRQ(ierr); 13807d68702bSBarry Smith } 13817d68702bSBarry Smith ierr = MatShift_Basic(Y,a);CHKERRQ(ierr); 13827d68702bSBarry Smith PetscFunctionReturn(0); 13837d68702bSBarry Smith } 13847d68702bSBarry Smith 138549b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 13863964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 138749b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 138849b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 138949b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 139097304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1391431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1392e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 1393f4259b30SLisandro Dalcin NULL, 1394f4259b30SLisandro Dalcin NULL, 1395f4259b30SLisandro Dalcin NULL, 1396f4259b30SLisandro Dalcin /* 10*/ NULL, 1397f4259b30SLisandro Dalcin NULL, 1398c078aec8SLisandro Dalcin MatCholeskyFactor_SeqSBAIJ, 139941f059aeSBarry Smith MatSOR_SeqSBAIJ, 140049b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 140197304618SKris Buschelman /* 15*/ MatGetInfo_SeqSBAIJ, 140249b5e25fSSatish Balay MatEqual_SeqSBAIJ, 140349b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 140449b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 140549b5e25fSSatish Balay MatNorm_SeqSBAIJ, 1406f4259b30SLisandro Dalcin /* 20*/ NULL, 140749b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 140849b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 140949b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1410f4259b30SLisandro Dalcin /* 24*/ NULL, 1411f4259b30SLisandro Dalcin NULL, 1412f4259b30SLisandro Dalcin NULL, 1413f4259b30SLisandro Dalcin NULL, 1414f4259b30SLisandro Dalcin NULL, 14154994cf47SJed Brown /* 29*/ MatSetUp_SeqSBAIJ, 1416f4259b30SLisandro Dalcin NULL, 1417f4259b30SLisandro Dalcin NULL, 1418f4259b30SLisandro Dalcin NULL, 1419f4259b30SLisandro Dalcin NULL, 1420d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqSBAIJ, 1421f4259b30SLisandro Dalcin NULL, 1422f4259b30SLisandro Dalcin NULL, 1423f4259b30SLisandro Dalcin NULL, 1424c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 1425d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqSBAIJ, 14267dae84e0SHong Zhang MatCreateSubMatrices_SeqSBAIJ, 142749b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 142849b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 14293c896bc6SHong Zhang MatCopy_SeqSBAIJ, 1430f4259b30SLisandro Dalcin /* 44*/ NULL, 143149b5e25fSSatish Balay MatScale_SeqSBAIJ, 14327d68702bSBarry Smith MatShift_SeqSBAIJ, 1433f4259b30SLisandro Dalcin NULL, 14343bededecSBarry Smith MatZeroRowsColumns_SeqSBAIJ, 1435f4259b30SLisandro Dalcin /* 49*/ NULL, 143649b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 143749b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 1438f4259b30SLisandro Dalcin NULL, 1439f4259b30SLisandro Dalcin NULL, 1440f4259b30SLisandro Dalcin /* 54*/ NULL, 1441f4259b30SLisandro Dalcin NULL, 1442f4259b30SLisandro Dalcin NULL, 1443dc29a518SPierre Jolivet MatPermute_SeqSBAIJ, 144449b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 14457dae84e0SHong Zhang /* 59*/ MatCreateSubMatrix_SeqSBAIJ, 1446f4259b30SLisandro Dalcin NULL, 1447f4259b30SLisandro Dalcin NULL, 1448f4259b30SLisandro Dalcin NULL, 1449f4259b30SLisandro Dalcin NULL, 1450f4259b30SLisandro Dalcin /* 64*/ NULL, 1451f4259b30SLisandro Dalcin NULL, 1452f4259b30SLisandro Dalcin NULL, 1453f4259b30SLisandro Dalcin NULL, 1454f4259b30SLisandro Dalcin NULL, 1455d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqSBAIJ, 1456f4259b30SLisandro Dalcin NULL, 145728d58a37SPierre Jolivet MatConvert_MPISBAIJ_Basic, 1458f4259b30SLisandro Dalcin NULL, 1459f4259b30SLisandro Dalcin NULL, 1460f4259b30SLisandro Dalcin /* 74*/ NULL, 1461f4259b30SLisandro Dalcin NULL, 1462f4259b30SLisandro Dalcin NULL, 1463f4259b30SLisandro Dalcin NULL, 1464f4259b30SLisandro Dalcin NULL, 1465f4259b30SLisandro Dalcin /* 79*/ NULL, 1466f4259b30SLisandro Dalcin NULL, 1467f4259b30SLisandro Dalcin NULL, 146897304618SKris Buschelman MatGetInertia_SeqSBAIJ, 14695bba2384SShri Abhyankar MatLoad_SeqSBAIJ, 1470d519adbfSMatthew Knepley /* 84*/ MatIsSymmetric_SeqSBAIJ, 1471865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1472efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1473f4259b30SLisandro Dalcin NULL, 1474f4259b30SLisandro Dalcin NULL, 1475f4259b30SLisandro Dalcin /* 89*/ NULL, 1476f4259b30SLisandro Dalcin NULL, 1477f4259b30SLisandro Dalcin NULL, 1478f4259b30SLisandro Dalcin NULL, 1479f4259b30SLisandro Dalcin NULL, 1480f4259b30SLisandro Dalcin /* 94*/ NULL, 1481f4259b30SLisandro Dalcin NULL, 1482f4259b30SLisandro Dalcin NULL, 1483f4259b30SLisandro Dalcin NULL, 1484f4259b30SLisandro Dalcin NULL, 1485f4259b30SLisandro Dalcin /* 99*/ NULL, 1486f4259b30SLisandro Dalcin NULL, 1487f4259b30SLisandro Dalcin NULL, 14882726fb6dSPierre Jolivet MatConjugate_SeqSBAIJ, 1489f4259b30SLisandro Dalcin NULL, 1490f4259b30SLisandro Dalcin /*104*/ NULL, 149199cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1492f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1493f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 14942af78befSBarry Smith MatRestoreRowUpperTriangular_SeqSBAIJ, 1495f4259b30SLisandro Dalcin /*109*/ NULL, 1496f4259b30SLisandro Dalcin NULL, 1497f4259b30SLisandro Dalcin NULL, 1498f4259b30SLisandro Dalcin NULL, 1499547795f9SHong Zhang MatMissingDiagonal_SeqSBAIJ, 1500f4259b30SLisandro Dalcin /*114*/ NULL, 1501f4259b30SLisandro Dalcin NULL, 1502f4259b30SLisandro Dalcin NULL, 1503f4259b30SLisandro Dalcin NULL, 1504f4259b30SLisandro Dalcin NULL, 1505f4259b30SLisandro Dalcin /*119*/ NULL, 1506f4259b30SLisandro Dalcin NULL, 1507f4259b30SLisandro Dalcin NULL, 1508f4259b30SLisandro Dalcin NULL, 1509f4259b30SLisandro Dalcin NULL, 1510f4259b30SLisandro Dalcin /*124*/ NULL, 1511f4259b30SLisandro Dalcin NULL, 1512f4259b30SLisandro Dalcin NULL, 1513f4259b30SLisandro Dalcin NULL, 1514f4259b30SLisandro Dalcin NULL, 1515f4259b30SLisandro Dalcin /*129*/ NULL, 1516f4259b30SLisandro Dalcin NULL, 1517f4259b30SLisandro Dalcin NULL, 1518f4259b30SLisandro Dalcin NULL, 1519f4259b30SLisandro Dalcin NULL, 1520f4259b30SLisandro Dalcin /*134*/ NULL, 1521f4259b30SLisandro Dalcin NULL, 1522f4259b30SLisandro Dalcin NULL, 1523f4259b30SLisandro Dalcin NULL, 1524f4259b30SLisandro Dalcin NULL, 152546533700Sstefano_zampini /*139*/ MatSetBlockSizes_Default, 1526f4259b30SLisandro Dalcin NULL, 1527f4259b30SLisandro Dalcin NULL, 1528f4259b30SLisandro Dalcin NULL, 1529f4259b30SLisandro Dalcin NULL, 153059f5e6ceSHong Zhang /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ 153199cafbc1SBarry Smith }; 1532be1d678aSKris Buschelman 15337087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 153449b5e25fSSatish Balay { 15354afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 1536d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 1537dfbe8321SBarry Smith PetscErrorCode ierr; 153849b5e25fSSatish Balay 153949b5e25fSSatish Balay PetscFunctionBegin; 1540e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 154149b5e25fSSatish Balay 154249b5e25fSSatish Balay /* allocate space for values if not already there */ 154349b5e25fSSatish Balay if (!aij->saved_values) { 1544854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr); 154549b5e25fSSatish Balay } 154649b5e25fSSatish Balay 154749b5e25fSSatish Balay /* copy values over */ 1548580bdb30SBarry Smith ierr = PetscArraycpy(aij->saved_values,aij->a,nz);CHKERRQ(ierr); 154949b5e25fSSatish Balay PetscFunctionReturn(0); 155049b5e25fSSatish Balay } 155149b5e25fSSatish Balay 15527087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 155349b5e25fSSatish Balay { 15544afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 15556849ba73SBarry Smith PetscErrorCode ierr; 1556d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 155749b5e25fSSatish Balay 155849b5e25fSSatish Balay PetscFunctionBegin; 1559e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1560e7e72b3dSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 156149b5e25fSSatish Balay 156249b5e25fSSatish Balay /* copy values over */ 1563580bdb30SBarry Smith ierr = PetscArraycpy(aij->a,aij->saved_values,nz);CHKERRQ(ierr); 156449b5e25fSSatish Balay PetscFunctionReturn(0); 156549b5e25fSSatish Balay } 156649b5e25fSSatish Balay 1567367daffbSBarry Smith static PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 156849b5e25fSSatish Balay { 1569c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 15706849ba73SBarry Smith PetscErrorCode ierr; 15714dcd73b1SHong Zhang PetscInt i,mbs,nbs,bs2; 15722576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE; 157349b5e25fSSatish Balay 157449b5e25fSSatish Balay PetscFunctionBegin; 15752576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 1576db4efbfdSBarry Smith 157733d57670SJed Brown ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr); 157826283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 157926283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1580476417e5SBarry Smith if (B->rmap->N > B->cmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"SEQSBAIJ matrix cannot have more rows %D than columns %D",B->rmap->N,B->cmap->N); 1581e02043d6SBarry Smith ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 1582899cda47SBarry Smith 158321940c7eSstefano_zampini B->preallocated = PETSC_TRUE; 158421940c7eSstefano_zampini 1585d0f46423SBarry Smith mbs = B->rmap->N/bs; 15864dcd73b1SHong Zhang nbs = B->cmap->n/bs; 158749b5e25fSSatish Balay bs2 = bs*bs; 158849b5e25fSSatish Balay 15894dcd73b1SHong Zhang if (mbs*bs != B->rmap->N || nbs*bs!=B->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize"); 159049b5e25fSSatish Balay 1591ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1592ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1593ab93d7beSBarry Smith nz = 0; 1594ab93d7beSBarry Smith } 1595ab93d7beSBarry Smith 1596435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 1597e32f2f54SBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 159849b5e25fSSatish Balay if (nnz) { 159949b5e25fSSatish Balay for (i=0; i<mbs; i++) { 1600e32f2f54SBarry 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]); 1601de64b629SHong Zhang if (nnz[i] > nbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D block rowlength %D",i,nnz[i],nbs); 160249b5e25fSSatish Balay } 160349b5e25fSSatish Balay } 160449b5e25fSSatish Balay 1605db4efbfdSBarry Smith B->ops->mult = MatMult_SeqSBAIJ_N; 1606db4efbfdSBarry Smith B->ops->multadd = MatMultAdd_SeqSBAIJ_N; 1607db4efbfdSBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_N; 1608db4efbfdSBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N; 160926fbe8dcSKarl Rupp 1610c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 161149b5e25fSSatish Balay if (!flg) { 161249b5e25fSSatish Balay switch (bs) { 161349b5e25fSSatish Balay case 1: 161449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 161549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1616431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1617431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 161849b5e25fSSatish Balay break; 161949b5e25fSSatish Balay case 2: 162049b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 162149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1622431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1623431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 162449b5e25fSSatish Balay break; 162549b5e25fSSatish Balay case 3: 162649b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 162749b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1628431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1629431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 163049b5e25fSSatish Balay break; 163149b5e25fSSatish Balay case 4: 163249b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 163349b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1634431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1635431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 163649b5e25fSSatish Balay break; 163749b5e25fSSatish Balay case 5: 163849b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 163949b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1640431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1641431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 164249b5e25fSSatish Balay break; 164349b5e25fSSatish Balay case 6: 164449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 164549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1646431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1647431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 164849b5e25fSSatish Balay break; 164949b5e25fSSatish Balay case 7: 1650de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 165149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1652431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1653431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 165449b5e25fSSatish Balay break; 165549b5e25fSSatish Balay } 165649b5e25fSSatish Balay } 165749b5e25fSSatish Balay 165849b5e25fSSatish Balay b->mbs = mbs; 16594dcd73b1SHong Zhang b->nbs = nbs; 1660ab93d7beSBarry Smith if (!skipallocation) { 16612ee49352SLisandro Dalcin if (!b->imax) { 1662dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr); 166326fbe8dcSKarl Rupp 1664c760cd28SBarry Smith b->free_imax_ilen = PETSC_TRUE; 166526fbe8dcSKarl Rupp 16663bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr); 16672ee49352SLisandro Dalcin } 166849b5e25fSSatish Balay if (!nnz) { 1669435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 167049b5e25fSSatish Balay else if (nz <= 0) nz = 1; 16715d2a9ed1SStefano Zampini nz = PetscMin(nbs,nz); 167226fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->imax[i] = nz; 167361778c46SBarry Smith ierr = PetscIntMultError(nz,mbs,&nz);CHKERRQ(ierr); 167449b5e25fSSatish Balay } else { 1675c73702f5SBarry Smith PetscInt64 nz64 = 0; 1676c73702f5SBarry Smith for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz64 += nnz[i];} 1677c73702f5SBarry Smith ierr = PetscIntCast(nz64,&nz);CHKERRQ(ierr); 167849b5e25fSSatish Balay } 16792ee49352SLisandro Dalcin /* b->ilen will count nonzeros in each block row so far. */ 168026fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->ilen[i] = 0; 16816c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 168249b5e25fSSatish Balay 168349b5e25fSSatish Balay /* allocate the matrix space */ 16842ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 1685dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr); 16863bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 1687580bdb30SBarry Smith ierr = PetscArrayzero(b->a,nz*bs2);CHKERRQ(ierr); 1688580bdb30SBarry Smith ierr = PetscArrayzero(b->j,nz);CHKERRQ(ierr); 168926fbe8dcSKarl Rupp 169049b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 169149b5e25fSSatish Balay 169249b5e25fSSatish Balay /* pointer to beginning of each row */ 1693e60cf9a0SBarry Smith b->i[0] = 0; 169426fbe8dcSKarl Rupp for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1]; 169526fbe8dcSKarl Rupp 1696e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1697e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1698e811da20SHong Zhang } else { 1699e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1700e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1701ab93d7beSBarry Smith } 170249b5e25fSSatish Balay 170349b5e25fSSatish Balay b->bs2 = bs2; 17046c6c5352SBarry Smith b->nz = 0; 1705b32cb4a7SJed Brown b->maxnz = nz; 1706f4259b30SLisandro Dalcin b->inew = NULL; 1707f4259b30SLisandro Dalcin b->jnew = NULL; 1708f4259b30SLisandro Dalcin b->anew = NULL; 1709f4259b30SLisandro Dalcin b->a2anew = NULL; 17101a3463dfSHong Zhang b->permute = PETSC_FALSE; 1711cb7b82ddSBarry Smith 1712cb7b82ddSBarry Smith B->was_assembled = PETSC_FALSE; 1713cb7b82ddSBarry Smith B->assembled = PETSC_FALSE; 17142576faa2SJed Brown if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);} 1715c464158bSHong Zhang PetscFunctionReturn(0); 1716c464158bSHong Zhang } 1717153ea458SHong Zhang 171838f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[]) 171938f409ebSLisandro Dalcin { 17200cd7f59aSBarry Smith PetscInt i,j,m,nz,anz, nz_max=0,*nnz; 1721f4259b30SLisandro Dalcin PetscScalar *values=NULL; 172238f409ebSLisandro Dalcin PetscBool roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented; 172338f409ebSLisandro Dalcin PetscErrorCode ierr; 17240cd7f59aSBarry Smith 172538f409ebSLisandro Dalcin PetscFunctionBegin; 172638f409ebSLisandro Dalcin if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs); 172738f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr); 172838f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr); 172938f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 173038f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 173138f409ebSLisandro Dalcin ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 173238f409ebSLisandro Dalcin m = B->rmap->n/bs; 173338f409ebSLisandro Dalcin 173438f409ebSLisandro Dalcin if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]); 1735854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&nnz);CHKERRQ(ierr); 173638f409ebSLisandro Dalcin for (i=0; i<m; i++) { 173738f409ebSLisandro Dalcin nz = ii[i+1] - ii[i]; 173838f409ebSLisandro Dalcin if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz); 17390cd7f59aSBarry Smith anz = 0; 17400cd7f59aSBarry Smith for (j=0; j<nz; j++) { 17410cd7f59aSBarry Smith /* count only values on the diagonal or above */ 17420cd7f59aSBarry Smith if (jj[ii[i] + j] >= i) { 17430cd7f59aSBarry Smith anz = nz - j; 17440cd7f59aSBarry Smith break; 17450cd7f59aSBarry Smith } 17460cd7f59aSBarry Smith } 17470cd7f59aSBarry Smith nz_max = PetscMax(nz_max,anz); 17480cd7f59aSBarry Smith nnz[i] = anz; 174938f409ebSLisandro Dalcin } 175038f409ebSLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 175138f409ebSLisandro Dalcin ierr = PetscFree(nnz);CHKERRQ(ierr); 175238f409ebSLisandro Dalcin 175338f409ebSLisandro Dalcin values = (PetscScalar*)V; 175438f409ebSLisandro Dalcin if (!values) { 17551795a4d1SJed Brown ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr); 175638f409ebSLisandro Dalcin } 175738f409ebSLisandro Dalcin for (i=0; i<m; i++) { 175838f409ebSLisandro Dalcin PetscInt ncols = ii[i+1] - ii[i]; 175938f409ebSLisandro Dalcin const PetscInt *icols = jj + ii[i]; 176038f409ebSLisandro Dalcin if (!roworiented || bs == 1) { 176138f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0); 176238f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr); 176338f409ebSLisandro Dalcin } else { 176438f409ebSLisandro Dalcin for (j=0; j<ncols; j++) { 176538f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0); 176638f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr); 176738f409ebSLisandro Dalcin } 176838f409ebSLisandro Dalcin } 176938f409ebSLisandro Dalcin } 177038f409ebSLisandro Dalcin if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); } 177138f409ebSLisandro Dalcin ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 177238f409ebSLisandro Dalcin ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 177338f409ebSLisandro Dalcin ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 177438f409ebSLisandro Dalcin PetscFunctionReturn(0); 177538f409ebSLisandro Dalcin } 177638f409ebSLisandro Dalcin 1777db4efbfdSBarry Smith /* 1778db4efbfdSBarry Smith This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization 1779db4efbfdSBarry Smith */ 1780ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural) 1781db4efbfdSBarry Smith { 1782db4efbfdSBarry Smith PetscErrorCode ierr; 1783ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 1784db4efbfdSBarry Smith PetscInt bs = B->rmap->bs; 1785db4efbfdSBarry Smith 1786db4efbfdSBarry Smith PetscFunctionBegin; 1787c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 1788db4efbfdSBarry Smith if (flg) bs = 8; 1789db4efbfdSBarry Smith 1790db4efbfdSBarry Smith if (!natural) { 1791db4efbfdSBarry Smith switch (bs) { 1792db4efbfdSBarry Smith case 1: 1793d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace; 1794db4efbfdSBarry Smith break; 1795db4efbfdSBarry Smith case 2: 1796db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 1797db4efbfdSBarry Smith break; 1798db4efbfdSBarry Smith case 3: 1799db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 1800db4efbfdSBarry Smith break; 1801db4efbfdSBarry Smith case 4: 1802db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 1803db4efbfdSBarry Smith break; 1804db4efbfdSBarry Smith case 5: 1805db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 1806db4efbfdSBarry Smith break; 1807db4efbfdSBarry Smith case 6: 1808db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 1809db4efbfdSBarry Smith break; 1810db4efbfdSBarry Smith case 7: 1811db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 1812db4efbfdSBarry Smith break; 1813db4efbfdSBarry Smith default: 1814db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N; 1815db4efbfdSBarry Smith break; 1816db4efbfdSBarry Smith } 1817db4efbfdSBarry Smith } else { 1818db4efbfdSBarry Smith switch (bs) { 1819db4efbfdSBarry Smith case 1: 1820d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace; 1821db4efbfdSBarry Smith break; 1822db4efbfdSBarry Smith case 2: 1823db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 1824db4efbfdSBarry Smith break; 1825db4efbfdSBarry Smith case 3: 1826db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 1827db4efbfdSBarry Smith break; 1828db4efbfdSBarry Smith case 4: 1829db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 1830db4efbfdSBarry Smith break; 1831db4efbfdSBarry Smith case 5: 1832db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 1833db4efbfdSBarry Smith break; 1834db4efbfdSBarry Smith case 6: 1835db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 1836db4efbfdSBarry Smith break; 1837db4efbfdSBarry Smith case 7: 1838db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 1839db4efbfdSBarry Smith break; 1840db4efbfdSBarry Smith default: 1841db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1842db4efbfdSBarry Smith break; 1843db4efbfdSBarry Smith } 1844db4efbfdSBarry Smith } 1845db4efbfdSBarry Smith PetscFunctionReturn(0); 1846db4efbfdSBarry Smith } 1847db4efbfdSBarry Smith 1848cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat,MatType,MatReuse,Mat*); 1849cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat,MatType,MatReuse,Mat*); 1850d769727bSBarry Smith 1851cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B) 18525c9eb25fSBarry Smith { 1853d0f46423SBarry Smith PetscInt n = A->rmap->n; 18545c9eb25fSBarry Smith PetscErrorCode ierr; 18555c9eb25fSBarry Smith 18565c9eb25fSBarry Smith PetscFunctionBegin; 18570e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX) 1858eb1ec7c1SStefano Zampini if (A->hermitian && !A->symmetric && (ftype == MAT_FACTOR_CHOLESKY||ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian CHOLESKY or ICC Factor is not supported"); 18590e92d65fSHong Zhang #endif 1860eb1ec7c1SStefano Zampini 1861ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 18625c9eb25fSBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 18635c9eb25fSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 18645c9eb25fSBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 18650298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 186626fbe8dcSKarl Rupp 18677b056e98SHong Zhang (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ; 1868c6d0d4f0SHong Zhang (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqSBAIJ; 1869e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 187000c67f3bSHong Zhang 1871d5f3da31SBarry Smith (*B)->factortype = ftype; 1872*f73b0415SBarry Smith (*B)->canuseordering = PETSC_TRUE; 187300c67f3bSHong Zhang ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr); 187400c67f3bSHong Zhang ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr); 18755c9eb25fSBarry Smith PetscFunctionReturn(0); 18765c9eb25fSBarry Smith } 18775c9eb25fSBarry Smith 18788397e458SBarry Smith /*@C 18798397e458SBarry Smith MatSeqSBAIJGetArray - gives access to the array where the data for a MATSEQSBAIJ matrix is stored 18808397e458SBarry Smith 18818397e458SBarry Smith Not Collective 18828397e458SBarry Smith 18838397e458SBarry Smith Input Parameter: 18848397e458SBarry Smith . mat - a MATSEQSBAIJ matrix 18858397e458SBarry Smith 18868397e458SBarry Smith Output Parameter: 18878397e458SBarry Smith . array - pointer to the data 18888397e458SBarry Smith 18898397e458SBarry Smith Level: intermediate 18908397e458SBarry Smith 18918397e458SBarry Smith .seealso: MatSeqSBAIJRestoreArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray() 18928397e458SBarry Smith @*/ 18938397e458SBarry Smith PetscErrorCode MatSeqSBAIJGetArray(Mat A,PetscScalar **array) 18948397e458SBarry Smith { 18958397e458SBarry Smith PetscErrorCode ierr; 18968397e458SBarry Smith 18978397e458SBarry Smith PetscFunctionBegin; 18988397e458SBarry Smith ierr = PetscUseMethod(A,"MatSeqSBAIJGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 18998397e458SBarry Smith PetscFunctionReturn(0); 19008397e458SBarry Smith } 19018397e458SBarry Smith 19028397e458SBarry Smith /*@C 19038397e458SBarry Smith MatSeqSBAIJRestoreArray - returns access to the array where the data for a MATSEQSBAIJ matrix is stored obtained by MatSeqSBAIJGetArray() 19048397e458SBarry Smith 19058397e458SBarry Smith Not Collective 19068397e458SBarry Smith 19078397e458SBarry Smith Input Parameters: 1908a2b725a8SWilliam Gropp + mat - a MATSEQSBAIJ matrix 1909a2b725a8SWilliam Gropp - array - pointer to the data 19108397e458SBarry Smith 19118397e458SBarry Smith Level: intermediate 19128397e458SBarry Smith 19138397e458SBarry Smith .seealso: MatSeqSBAIJGetArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray() 19148397e458SBarry Smith @*/ 19158397e458SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray(Mat A,PetscScalar **array) 19168397e458SBarry Smith { 19178397e458SBarry Smith PetscErrorCode ierr; 19188397e458SBarry Smith 19198397e458SBarry Smith PetscFunctionBegin; 19208397e458SBarry Smith ierr = PetscUseMethod(A,"MatSeqSBAIJRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 19218397e458SBarry Smith PetscFunctionReturn(0); 19228397e458SBarry Smith } 19238397e458SBarry Smith 19240bad9183SKris Buschelman /*MC 1925fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 19260bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 19270bad9183SKris Buschelman 1928828413b8SBarry Smith For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you 1929eb1ec7c1SStefano Zampini can call MatSetOption(Mat, MAT_HERMITIAN). 1930828413b8SBarry Smith 19310bad9183SKris Buschelman Options Database Keys: 19320bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 19330bad9183SKris Buschelman 193495452b02SPatrick Sanan Notes: 193595452b02SPatrick Sanan By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not 193671dad5bbSBarry 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 193771dad5bbSBarry 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. 193871dad5bbSBarry Smith 1939476417e5SBarry Smith The number of rows in the matrix must be less than or equal to the number of columns 194071dad5bbSBarry Smith 19410bad9183SKris Buschelman Level: beginner 19420bad9183SKris Buschelman 1943476417e5SBarry Smith .seealso: MatCreateSeqSBAIJ(), MatType, MATMPISBAIJ 19440bad9183SKris Buschelman M*/ 19458cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1946a23d5eceSKris Buschelman { 1947a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1948dfbe8321SBarry Smith PetscErrorCode ierr; 194913f74950SBarry Smith PetscMPIInt size; 1950ace3abfcSBarry Smith PetscBool no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE; 1951a23d5eceSKris Buschelman 1952a23d5eceSKris Buschelman PetscFunctionBegin; 1953ffc4695bSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRMPI(ierr); 1954e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1955a23d5eceSKris Buschelman 1956b00a9115SJed Brown ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1957a23d5eceSKris Buschelman B->data = (void*)b; 1958a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 195926fbe8dcSKarl Rupp 1960a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1961a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1962f4259b30SLisandro Dalcin b->row = NULL; 1963f4259b30SLisandro Dalcin b->icol = NULL; 1964a23d5eceSKris Buschelman b->reallocs = 0; 1965f4259b30SLisandro Dalcin b->saved_values = NULL; 19660def2e27SBarry Smith b->inode.limit = 5; 19670def2e27SBarry Smith b->inode.max_limit = 5; 1968a23d5eceSKris Buschelman 1969a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1970a23d5eceSKris Buschelman b->nonew = 0; 1971f4259b30SLisandro Dalcin b->diag = NULL; 1972f4259b30SLisandro Dalcin b->solve_work = NULL; 1973f4259b30SLisandro Dalcin b->mult_work = NULL; 1974f4259b30SLisandro Dalcin B->spptr = NULL; 1975f2cbd3d5SJed Brown B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2; 1976a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 1977a23d5eceSKris Buschelman 1978f4259b30SLisandro Dalcin b->inew = NULL; 1979f4259b30SLisandro Dalcin b->jnew = NULL; 1980f4259b30SLisandro Dalcin b->anew = NULL; 1981f4259b30SLisandro Dalcin b->a2anew = NULL; 1982a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1983a23d5eceSKris Buschelman 198471dad5bbSBarry Smith b->ignore_ltriangular = PETSC_TRUE; 198526fbe8dcSKarl Rupp 1986c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr); 1987941593c8SHong Zhang 1988f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 198926fbe8dcSKarl Rupp 1990c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr); 1991f5edf698SHong Zhang 19928397e458SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJGetArray_C",MatSeqSBAIJGetArray_SeqSBAIJ);CHKERRQ(ierr); 19938397e458SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJRestoreArray_C",MatSeqSBAIJRestoreArray_SeqSBAIJ);CHKERRQ(ierr); 1994bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1995bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1996bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 1997bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1998bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1999bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 200038f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr); 20016214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 20026214f412SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_elemental_C",MatConvert_SeqSBAIJ_Elemental);CHKERRQ(ierr); 20036214f412SHong Zhang #endif 2004d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK) 2005d24d4204SJose E. Roman ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_scalapack_C",MatConvert_SBAIJ_ScaLAPACK);CHKERRQ(ierr); 2006d24d4204SJose E. Roman #endif 200723ce1328SBarry Smith 200823ce1328SBarry Smith B->symmetric = PETSC_TRUE; 200923ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 201023ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 201123ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 20129899f194SHong Zhang B->symmetric_eternal = PETSC_TRUE; 2013eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 201413647f61SHong Zhang B->hermitian = PETSC_FALSE; 201513647f61SHong Zhang B->hermitian_set = PETSC_FALSE; 2016eb1ec7c1SStefano Zampini #else 2017eb1ec7c1SStefano Zampini B->hermitian = PETSC_TRUE; 2018eb1ec7c1SStefano Zampini B->hermitian_set = PETSC_TRUE; 2019eb1ec7c1SStefano Zampini #endif 202013647f61SHong Zhang 202117667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 20220def2e27SBarry Smith 2023ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr); 20240298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr); 202526fbe8dcSKarl Rupp if (no_unroll) { 202626fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr); 202726fbe8dcSKarl Rupp } 20280298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr); 202926fbe8dcSKarl Rupp if (no_inode) { 203026fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr); 203126fbe8dcSKarl Rupp } 20320298fd71SBarry Smith ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr); 20330def2e27SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 2034ace3abfcSBarry Smith b->inode.use = (PetscBool)(!(no_unroll || no_inode)); 20350def2e27SBarry Smith if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit; 2036a23d5eceSKris Buschelman PetscFunctionReturn(0); 2037a23d5eceSKris Buschelman } 2038a23d5eceSKris Buschelman 2039a23d5eceSKris Buschelman /*@C 2040a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 2041a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 2042a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 2043a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 2044a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 2045a23d5eceSKris Buschelman 2046a23d5eceSKris Buschelman Collective on Mat 2047a23d5eceSKris Buschelman 2048a23d5eceSKris Buschelman Input Parameters: 20491c4f3114SJed Brown + B - the symmetric matrix 2050bb7ae925SBarry Smith . bs - size of block, the blocks are ALWAYS square. One can use MatSetBlockSizes() to set a different row and column blocksize but the row 2051bb7ae925SBarry Smith blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs() 2052a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 2053a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 20540298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2055a23d5eceSKris Buschelman 2056a23d5eceSKris Buschelman Options Database Keys: 2057a2b725a8SWilliam Gropp + -mat_no_unroll - uses code that does not unroll the loops in the 2058a23d5eceSKris Buschelman block calculations (much slower) 2059a2b725a8SWilliam Gropp - -mat_block_size - size of the blocks to use (only works if a negative bs is passed in 2060a23d5eceSKris Buschelman 2061a23d5eceSKris Buschelman Level: intermediate 2062a23d5eceSKris Buschelman 2063a23d5eceSKris Buschelman Notes: 2064a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 20650298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 2066a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 2067a23d5eceSKris Buschelman 2068aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 2069aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 2070aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 2071aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 2072aa95bbe8SBarry Smith 207349a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 207449a6f317SBarry Smith 207549a6f317SBarry Smith 207669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2077a23d5eceSKris Buschelman @*/ 20787087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 207913f74950SBarry Smith { 20804ac538c5SBarry Smith PetscErrorCode ierr; 2081a23d5eceSKris Buschelman 2082a23d5eceSKris Buschelman PetscFunctionBegin; 20836ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 20846ba663aaSJed Brown PetscValidType(B,1); 20856ba663aaSJed Brown PetscValidLogicalCollectiveInt(B,bs,2); 20864ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr); 2087a23d5eceSKris Buschelman PetscFunctionReturn(0); 2088a23d5eceSKris Buschelman } 208949b5e25fSSatish Balay 209038f409ebSLisandro Dalcin /*@C 2091664954b6SBarry Smith MatSeqSBAIJSetPreallocationCSR - Creates a sparse parallel matrix in SBAIJ format using the given nonzero structure and (optional) numerical values 209238f409ebSLisandro Dalcin 209338f409ebSLisandro Dalcin Input Parameters: 20941c4f3114SJed Brown + B - the matrix 2095eab78319SHong Zhang . bs - size of block, the blocks are ALWAYS square. 209638f409ebSLisandro Dalcin . i - the indices into j for the start of each local row (starts with zero) 209738f409ebSLisandro Dalcin . j - the column indices for each local row (starts with zero) these must be sorted for each row 209838f409ebSLisandro Dalcin - v - optional values in the matrix 209938f409ebSLisandro Dalcin 2100664954b6SBarry Smith Level: advanced 210138f409ebSLisandro Dalcin 210238f409ebSLisandro Dalcin Notes: 210338f409ebSLisandro Dalcin The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED. For example, C programs 210438f409ebSLisandro 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 210538f409ebSLisandro Dalcin over rows within a block and the last index is over columns within a block row. Fortran programs will likely set 210638f409ebSLisandro 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 210738f409ebSLisandro Dalcin block column and the second index is over columns within a block. 210838f409ebSLisandro Dalcin 210950c5228eSBarry Smith Any entries below the diagonal are ignored 21100cd7f59aSBarry Smith 21110cd7f59aSBarry Smith Though this routine has Preallocation() in the name it also sets the exact nonzero locations of the matrix entries 21120cd7f59aSBarry Smith and usually the numerical values as well 2113664954b6SBarry Smith 211438f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ 211538f409ebSLisandro Dalcin @*/ 211638f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[]) 211738f409ebSLisandro Dalcin { 211838f409ebSLisandro Dalcin PetscErrorCode ierr; 211938f409ebSLisandro Dalcin 212038f409ebSLisandro Dalcin PetscFunctionBegin; 212138f409ebSLisandro Dalcin PetscValidHeaderSpecific(B,MAT_CLASSID,1); 212238f409ebSLisandro Dalcin PetscValidType(B,1); 212338f409ebSLisandro Dalcin PetscValidLogicalCollectiveInt(B,bs,2); 212438f409ebSLisandro Dalcin ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr); 212538f409ebSLisandro Dalcin PetscFunctionReturn(0); 212638f409ebSLisandro Dalcin } 212738f409ebSLisandro Dalcin 2128c464158bSHong Zhang /*@C 2129c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 2130c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 2131c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 2132c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 2133c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 213449b5e25fSSatish Balay 2135d083f849SBarry Smith Collective 2136c464158bSHong Zhang 2137c464158bSHong Zhang Input Parameters: 2138c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2139bb7ae925SBarry Smith . bs - size of block, the blocks are ALWAYS square. One can use MatSetBlockSizes() to set a different row and column blocksize but the row 2140bb7ae925SBarry Smith blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs() 2141c464158bSHong Zhang . m - number of rows, or number of columns 2142c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 2143744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 21440298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2145c464158bSHong Zhang 2146c464158bSHong Zhang Output Parameter: 2147c464158bSHong Zhang . A - the symmetric matrix 2148c464158bSHong Zhang 2149c464158bSHong Zhang Options Database Keys: 2150a2b725a8SWilliam Gropp + -mat_no_unroll - uses code that does not unroll the loops in the 2151c464158bSHong Zhang block calculations (much slower) 2152a2b725a8SWilliam Gropp - -mat_block_size - size of the blocks to use 2153c464158bSHong Zhang 2154c464158bSHong Zhang Level: intermediate 2155c464158bSHong Zhang 2156175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2157f6f02116SRichard Tran Mills MatXXXXSetPreallocation() paradigm instead of this routine directly. 2158175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 2159175b88e8SBarry Smith 2160c464158bSHong Zhang Notes: 21616d6d819aSHong Zhang The number of rows and columns must be divisible by blocksize. 21626d6d819aSHong Zhang This matrix type does not support complex Hermitian operation. 2163c464158bSHong Zhang 2164c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 21650298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 2166a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 2167c464158bSHong Zhang 216849a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 216949a6f317SBarry Smith 217069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2171c464158bSHong Zhang @*/ 21727087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 2173c464158bSHong Zhang { 2174dfbe8321SBarry Smith PetscErrorCode ierr; 2175c464158bSHong Zhang 2176c464158bSHong Zhang PetscFunctionBegin; 2177f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 2178f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2179c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 2180367daffbSBarry Smith ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 218149b5e25fSSatish Balay PetscFunctionReturn(0); 218249b5e25fSSatish Balay } 218349b5e25fSSatish Balay 2184dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 218549b5e25fSSatish Balay { 218649b5e25fSSatish Balay Mat C; 218749b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 21886849ba73SBarry Smith PetscErrorCode ierr; 2189b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 219049b5e25fSSatish Balay 219149b5e25fSSatish Balay PetscFunctionBegin; 2192e32f2f54SBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix"); 219349b5e25fSSatish Balay 2194f4259b30SLisandro Dalcin *B = NULL; 2195ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2196d0f46423SBarry Smith ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr); 21974c7a3774SStefano Zampini ierr = MatSetBlockSizesFromMats(C,A,A);CHKERRQ(ierr); 21988e9a0fb8SHong Zhang ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr); 2199692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 2200692f9cbeSHong Zhang 2201273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 2202d5f3da31SBarry Smith C->factortype = A->factortype; 2203f4259b30SLisandro Dalcin c->row = NULL; 2204f4259b30SLisandro Dalcin c->icol = NULL; 2205f4259b30SLisandro Dalcin c->saved_values = NULL; 2206a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 220749b5e25fSSatish Balay C->assembled = PETSC_TRUE; 220849b5e25fSSatish Balay 22091e1e43feSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 22101e1e43feSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 221149b5e25fSSatish Balay c->bs2 = a->bs2; 221249b5e25fSSatish Balay c->mbs = a->mbs; 221349b5e25fSSatish Balay c->nbs = a->nbs; 221449b5e25fSSatish Balay 2215c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2216c760cd28SBarry Smith c->imax = a->imax; 2217c760cd28SBarry Smith c->ilen = a->ilen; 2218c760cd28SBarry Smith c->free_imax_ilen = PETSC_FALSE; 2219c760cd28SBarry Smith } else { 2220dcca6d9dSJed Brown ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr); 22213bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 222249b5e25fSSatish Balay for (i=0; i<mbs; i++) { 222349b5e25fSSatish Balay c->imax[i] = a->imax[i]; 222449b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 222549b5e25fSSatish Balay } 2226c760cd28SBarry Smith c->free_imax_ilen = PETSC_TRUE; 2227c760cd28SBarry Smith } 222849b5e25fSSatish Balay 222949b5e25fSSatish Balay /* allocate the matrix space */ 22304da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2231785e854fSJed Brown ierr = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr); 22323bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 223344e1c64aSLisandro Dalcin c->i = a->i; 223444e1c64aSLisandro Dalcin c->j = a->j; 22354da8f245SBarry Smith c->singlemalloc = PETSC_FALSE; 223644e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 22374da8f245SBarry Smith c->free_ij = PETSC_FALSE; 22384da8f245SBarry Smith c->parent = A; 22394da8f245SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 22404da8f245SBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 22414da8f245SBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 22424da8f245SBarry Smith } else { 2243dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr); 2244580bdb30SBarry Smith ierr = PetscArraycpy(c->i,a->i,mbs+1);CHKERRQ(ierr); 22453bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 22464da8f245SBarry Smith c->singlemalloc = PETSC_TRUE; 224744e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 22484da8f245SBarry Smith c->free_ij = PETSC_TRUE; 22494da8f245SBarry Smith } 225049b5e25fSSatish Balay if (mbs > 0) { 22514da8f245SBarry Smith if (cpvalues != MAT_SHARE_NONZERO_PATTERN) { 2252580bdb30SBarry Smith ierr = PetscArraycpy(c->j,a->j,nz);CHKERRQ(ierr); 22534da8f245SBarry Smith } 225449b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 2255580bdb30SBarry Smith ierr = PetscArraycpy(c->a,a->a,bs2*nz);CHKERRQ(ierr); 225649b5e25fSSatish Balay } else { 2257580bdb30SBarry Smith ierr = PetscArrayzero(c->a,bs2*nz);CHKERRQ(ierr); 225849b5e25fSSatish Balay } 2259a1c3900fSBarry Smith if (a->jshort) { 226044e1c64aSLisandro Dalcin /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */ 226144e1c64aSLisandro Dalcin /* if the parent matrix is reassembled, this child matrix will never notice */ 2262785e854fSJed Brown ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr); 22633bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr); 2264580bdb30SBarry Smith ierr = PetscArraycpy(c->jshort,a->jshort,nz);CHKERRQ(ierr); 226526fbe8dcSKarl Rupp 22664da8f245SBarry Smith c->free_jshort = PETSC_TRUE; 22674da8f245SBarry Smith } 2268a1c3900fSBarry Smith } 226949b5e25fSSatish Balay 227049b5e25fSSatish Balay c->roworiented = a->roworiented; 227149b5e25fSSatish Balay c->nonew = a->nonew; 227249b5e25fSSatish Balay 227349b5e25fSSatish Balay if (a->diag) { 2274c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2275c760cd28SBarry Smith c->diag = a->diag; 2276c760cd28SBarry Smith c->free_diag = PETSC_FALSE; 2277c760cd28SBarry Smith } else { 2278785e854fSJed Brown ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr); 22793bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr); 228026fbe8dcSKarl Rupp for (i=0; i<mbs; i++) c->diag[i] = a->diag[i]; 2281c760cd28SBarry Smith c->free_diag = PETSC_TRUE; 2282c760cd28SBarry Smith } 228344e1c64aSLisandro Dalcin } 22846c6c5352SBarry Smith c->nz = a->nz; 2285f2cbd3d5SJed Brown c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 2286f4259b30SLisandro Dalcin c->solve_work = NULL; 2287f4259b30SLisandro Dalcin c->mult_work = NULL; 228826fbe8dcSKarl Rupp 228949b5e25fSSatish Balay *B = C; 2290140e18c1SBarry Smith ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 229149b5e25fSSatish Balay PetscFunctionReturn(0); 229249b5e25fSSatish Balay } 229349b5e25fSSatish Balay 2294618cc2edSLisandro Dalcin /* Used for both SeqBAIJ and SeqSBAIJ matrices */ 2295618cc2edSLisandro Dalcin #define MatLoad_SeqSBAIJ_Binary MatLoad_SeqBAIJ_Binary 2296618cc2edSLisandro Dalcin 2297618cc2edSLisandro Dalcin PetscErrorCode MatLoad_SeqSBAIJ(Mat mat,PetscViewer viewer) 22982f480046SShri Abhyankar { 22992f480046SShri Abhyankar PetscErrorCode ierr; 23007f489da9SVaclav Hapla PetscBool isbinary; 23012f480046SShri Abhyankar 23022f480046SShri Abhyankar PetscFunctionBegin; 23037f489da9SVaclav Hapla ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 2304618cc2edSLisandro Dalcin if (!isbinary) SETERRQ2(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer type %s not yet supported for reading %s matrices",((PetscObject)viewer)->type_name,((PetscObject)mat)->type_name); 2305618cc2edSLisandro Dalcin ierr = MatLoad_SeqSBAIJ_Binary(mat,viewer);CHKERRQ(ierr); 23062f480046SShri Abhyankar PetscFunctionReturn(0); 23072f480046SShri Abhyankar } 23082f480046SShri Abhyankar 2309c75a6043SHong Zhang /*@ 2310c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2311c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2312c75a6043SHong Zhang 2313d083f849SBarry Smith Collective 2314c75a6043SHong Zhang 2315c75a6043SHong Zhang Input Parameters: 2316c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2317c75a6043SHong Zhang . bs - size of block 2318c75a6043SHong Zhang . m - number of rows 2319c75a6043SHong Zhang . n - number of columns 2320483a2f95SBarry Smith . i - row indices; that is i[0] = 0, i[row] = i[row-1] + number of block elements in that row block row of the matrix 2321c75a6043SHong Zhang . j - column indices 2322c75a6043SHong Zhang - a - matrix values 2323c75a6043SHong Zhang 2324c75a6043SHong Zhang Output Parameter: 2325c75a6043SHong Zhang . mat - the matrix 2326c75a6043SHong Zhang 2327dfb205c3SBarry Smith Level: advanced 2328c75a6043SHong Zhang 2329c75a6043SHong Zhang Notes: 2330c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2331c75a6043SHong Zhang once the matrix is destroyed 2332c75a6043SHong Zhang 2333c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2334c75a6043SHong Zhang 2335c75a6043SHong Zhang The i and j indices are 0 based 2336c75a6043SHong Zhang 2337dfb205c3SBarry 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 2338dfb205c3SBarry Smith it is the regular CSR format excluding the lower triangular elements. 2339dfb205c3SBarry Smith 234069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ() 2341c75a6043SHong Zhang 2342c75a6043SHong Zhang @*/ 2343c3c607ccSBarry Smith PetscErrorCode MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat) 2344c75a6043SHong Zhang { 2345c75a6043SHong Zhang PetscErrorCode ierr; 2346c75a6043SHong Zhang PetscInt ii; 2347c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2348c75a6043SHong Zhang 2349c75a6043SHong Zhang PetscFunctionBegin; 2350e32f2f54SBarry Smith if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 235141096f02SStefano Zampini if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2352c75a6043SHong Zhang 2353c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2354c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2355c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2356f4259b30SLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(*mat,bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 2357c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2358dcca6d9dSJed Brown ierr = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr); 23593bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr); 2360c75a6043SHong Zhang 2361c75a6043SHong Zhang sbaij->i = i; 2362c75a6043SHong Zhang sbaij->j = j; 2363c75a6043SHong Zhang sbaij->a = a; 236426fbe8dcSKarl Rupp 2365c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2366c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2367e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2368e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2369ddf7884eSMatthew Knepley sbaij->free_imax_ilen = PETSC_TRUE; 2370c75a6043SHong Zhang 2371c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2372c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2373cf9c20a2SJed Brown if (PetscUnlikelyDebug(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]); 2374c75a6043SHong Zhang } 237576bd3646SJed Brown if (PetscDefined(USE_DEBUG)) { 2376c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2377e32f2f54SBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2378e32f2f54SBarry 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]); 2379c75a6043SHong Zhang } 238076bd3646SJed Brown } 2381c75a6043SHong Zhang 2382c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2383c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2384c75a6043SHong Zhang PetscFunctionReturn(0); 2385c75a6043SHong Zhang } 2386d06b337dSHong Zhang 238759f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat) 238859f5e6ceSHong Zhang { 238959f5e6ceSHong Zhang PetscErrorCode ierr; 23908761c3d6SHong Zhang PetscMPIInt size; 239159f5e6ceSHong Zhang 239259f5e6ceSHong Zhang PetscFunctionBegin; 2393ffc4695bSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr); 23948761c3d6SHong Zhang if (size == 1 && scall == MAT_REUSE_MATRIX) { 23958761c3d6SHong Zhang ierr = MatCopy(inmat,*outmat,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 23968761c3d6SHong Zhang } else { 239759f5e6ceSHong Zhang ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr); 23988761c3d6SHong Zhang } 239959f5e6ceSHong Zhang PetscFunctionReturn(0); 240059f5e6ceSHong Zhang } 2401