149b5e25fSSatish Balay 249b5e25fSSatish Balay /* 3a1373b80SHong Zhang Defines the basic matrix operations for the SBAIJ (compressed row) 449b5e25fSSatish Balay matrix storage format. 549b5e25fSSatish Balay */ 6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h> /*I "petscmat.h" I*/ 7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h> 8c6db04a5SJed Brown #include <petscblaslapack.h> 949b5e25fSSatish Balay 10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h> 1170dcbbb9SBarry Smith #define USESHORT 12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h> 1370dcbbb9SBarry Smith 14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool); 156214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 16cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat,MatType,MatReuse,Mat*); 176214f412SHong Zhang #endif 18b5b17502SBarry Smith 1949b5e25fSSatish Balay /* 2049b5e25fSSatish Balay Checks for missing diagonals 2149b5e25fSSatish Balay */ 22ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool *missing,PetscInt *dd) 2349b5e25fSSatish Balay { 24045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 256849ba73SBarry Smith PetscErrorCode ierr; 267734d3b5SMatthew G. Knepley PetscInt *diag,*ii = a->i,i; 2749b5e25fSSatish Balay 2849b5e25fSSatish Balay PetscFunctionBegin; 29045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 302af78befSBarry Smith *missing = PETSC_FALSE; 317734d3b5SMatthew G. Knepley if (A->rmap->n > 0 && !ii) { 32358d2f5dSShri Abhyankar *missing = PETSC_TRUE; 33358d2f5dSShri Abhyankar if (dd) *dd = 0; 34955c1f14SBarry Smith ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr); 35358d2f5dSShri Abhyankar } else { 36358d2f5dSShri Abhyankar diag = a->diag; 3749b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 387734d3b5SMatthew G. Knepley if (diag[i] >= ii[i+1]) { 392af78befSBarry Smith *missing = PETSC_TRUE; 402af78befSBarry Smith if (dd) *dd = i; 412af78befSBarry Smith break; 422af78befSBarry Smith } 4349b5e25fSSatish Balay } 44358d2f5dSShri Abhyankar } 4549b5e25fSSatish Balay PetscFunctionReturn(0); 4649b5e25fSSatish Balay } 4749b5e25fSSatish Balay 48dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 4949b5e25fSSatish Balay { 50045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 516849ba73SBarry Smith PetscErrorCode ierr; 5248dd3d27SHong Zhang PetscInt i,j; 5349b5e25fSSatish Balay 5449b5e25fSSatish Balay PetscFunctionBegin; 5509f38230SBarry Smith if (!a->diag) { 56785e854fSJed Brown ierr = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr); 573bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 58c760cd28SBarry Smith a->free_diag = PETSC_TRUE; 5909f38230SBarry Smith } 6048dd3d27SHong Zhang for (i=0; i<a->mbs; i++) { 6148dd3d27SHong Zhang a->diag[i] = a->i[i+1]; 6248dd3d27SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 6348dd3d27SHong Zhang if (a->j[j] == i) { 6448dd3d27SHong Zhang a->diag[i] = j; 6548dd3d27SHong Zhang break; 6648dd3d27SHong Zhang } 6748dd3d27SHong Zhang } 6848dd3d27SHong Zhang } 6949b5e25fSSatish Balay PetscFunctionReturn(0); 7049b5e25fSSatish Balay } 7149b5e25fSSatish Balay 721a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool *done) 7349b5e25fSSatish Balay { 74a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 758f7157efSSatish Balay PetscErrorCode ierr; 762462f5fdSStefano Zampini PetscInt i,j,n = a->mbs,nz = a->i[n],*tia,*tja,bs = A->rmap->bs,k,l,cnt; 772462f5fdSStefano Zampini PetscInt **ia = (PetscInt**)inia,**ja = (PetscInt**)inja; 7849b5e25fSSatish Balay 7949b5e25fSSatish Balay PetscFunctionBegin; 80d3e5a4abSHong Zhang *nn = n; 81a1373b80SHong Zhang if (!ia) PetscFunctionReturn(0); 822462f5fdSStefano Zampini if (symmetric) { 832462f5fdSStefano Zampini ierr = MatToSymmetricIJ_SeqAIJ(n,a->i,a->j,PETSC_FALSE,0,0,&tia,&tja);CHKERRQ(ierr); 842462f5fdSStefano Zampini nz = tia[n]; 852462f5fdSStefano Zampini } else { 862462f5fdSStefano Zampini tia = a->i; tja = a->j; 872462f5fdSStefano Zampini } 882462f5fdSStefano Zampini 892462f5fdSStefano Zampini if (!blockcompressed && bs > 1) { 902462f5fdSStefano Zampini (*nn) *= bs; 918f7157efSSatish Balay /* malloc & create the natural set of indices */ 922462f5fdSStefano Zampini ierr = PetscMalloc1((n+1)*bs,ia);CHKERRQ(ierr); 932462f5fdSStefano Zampini if (n) { 942462f5fdSStefano Zampini (*ia)[0] = oshift; 952462f5fdSStefano Zampini for (j=1; j<bs; j++) { 962462f5fdSStefano Zampini (*ia)[j] = (tia[1]-tia[0])*bs+(*ia)[j-1]; 972462f5fdSStefano Zampini } 982462f5fdSStefano Zampini } 992462f5fdSStefano Zampini 1002462f5fdSStefano Zampini for (i=1; i<n; i++) { 1012462f5fdSStefano Zampini (*ia)[i*bs] = (tia[i]-tia[i-1])*bs + (*ia)[i*bs-1]; 1022462f5fdSStefano Zampini for (j=1; j<bs; j++) { 1032462f5fdSStefano Zampini (*ia)[i*bs+j] = (tia[i+1]-tia[i])*bs + (*ia)[i*bs+j-1]; 1042462f5fdSStefano Zampini } 1052462f5fdSStefano Zampini } 1062462f5fdSStefano Zampini if (n) { 1072462f5fdSStefano Zampini (*ia)[n*bs] = (tia[n]-tia[n-1])*bs + (*ia)[n*bs-1]; 1082462f5fdSStefano Zampini } 1092462f5fdSStefano Zampini 1102462f5fdSStefano Zampini if (inja) { 1112462f5fdSStefano Zampini ierr = PetscMalloc1(nz*bs*bs,ja);CHKERRQ(ierr); 1122462f5fdSStefano Zampini cnt = 0; 1132462f5fdSStefano Zampini for (i=0; i<n; i++) { 1148f7157efSSatish Balay for (j=0; j<bs; j++) { 1152462f5fdSStefano Zampini for (k=tia[i]; k<tia[i+1]; k++) { 1162462f5fdSStefano Zampini for (l=0; l<bs; l++) { 1172462f5fdSStefano Zampini (*ja)[cnt++] = bs*tja[k] + l; 1188f7157efSSatish Balay } 1198f7157efSSatish Balay } 1208f7157efSSatish Balay } 1218f7157efSSatish Balay } 1228f7157efSSatish Balay } 1232462f5fdSStefano Zampini 1242462f5fdSStefano Zampini if (symmetric) { /* deallocate memory allocated in MatToSymmetricIJ_SeqAIJ() */ 1252462f5fdSStefano Zampini ierr = PetscFree(tia);CHKERRQ(ierr); 1262462f5fdSStefano Zampini ierr = PetscFree(tja);CHKERRQ(ierr); 1272462f5fdSStefano Zampini } 1282462f5fdSStefano Zampini } else if (oshift == 1) { 1292462f5fdSStefano Zampini if (symmetric) { 1302462f5fdSStefano Zampini nz = tia[A->rmap->n/bs]; 1312462f5fdSStefano Zampini /* add 1 to i and j indices */ 1322462f5fdSStefano Zampini for (i=0; i<A->rmap->n/bs+1; i++) tia[i] = tia[i] + 1; 1332462f5fdSStefano Zampini *ia = tia; 1342462f5fdSStefano Zampini if (ja) { 1352462f5fdSStefano Zampini for (i=0; i<nz; i++) tja[i] = tja[i] + 1; 1362462f5fdSStefano Zampini *ja = tja; 1372462f5fdSStefano Zampini } 1382462f5fdSStefano Zampini } else { 1392462f5fdSStefano Zampini nz = a->i[A->rmap->n/bs]; 1402462f5fdSStefano Zampini /* malloc space and add 1 to i and j indices */ 1412462f5fdSStefano Zampini ierr = PetscMalloc1(A->rmap->n/bs+1,ia);CHKERRQ(ierr); 1422462f5fdSStefano Zampini for (i=0; i<A->rmap->n/bs+1; i++) (*ia)[i] = a->i[i] + 1; 1432462f5fdSStefano Zampini if (ja) { 1442462f5fdSStefano Zampini ierr = PetscMalloc1(nz,ja);CHKERRQ(ierr); 1452462f5fdSStefano Zampini for (i=0; i<nz; i++) (*ja)[i] = a->j[i] + 1; 1462462f5fdSStefano Zampini } 1472462f5fdSStefano Zampini } 1482462f5fdSStefano Zampini } else { 1492462f5fdSStefano Zampini *ia = tia; 1502462f5fdSStefano Zampini if (ja) *ja = tja; 151a6ece127SHong Zhang } 15249b5e25fSSatish Balay PetscFunctionReturn(0); 15349b5e25fSSatish Balay } 15449b5e25fSSatish Balay 1551a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 15649b5e25fSSatish Balay { 1578f7157efSSatish Balay PetscErrorCode ierr; 158a6ece127SHong Zhang 15949b5e25fSSatish Balay PetscFunctionBegin; 16049b5e25fSSatish Balay if (!ia) PetscFunctionReturn(0); 1612462f5fdSStefano Zampini if ((!blockcompressed && A->rmap->bs > 1) || (symmetric || oshift == 1)) { 1622462f5fdSStefano Zampini ierr = PetscFree(*ia);CHKERRQ(ierr); 1632462f5fdSStefano Zampini if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);} 164a6ece127SHong Zhang } 165a6ece127SHong Zhang PetscFunctionReturn(0); 16649b5e25fSSatish Balay } 16749b5e25fSSatish Balay 168dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A) 16949b5e25fSSatish Balay { 17049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 171dfbe8321SBarry Smith PetscErrorCode ierr; 17249b5e25fSSatish Balay 17349b5e25fSSatish Balay PetscFunctionBegin; 174a9f03627SSatish Balay #if defined(PETSC_USE_LOG) 175d0f46423SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz); 176a9f03627SSatish Balay #endif 177e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 1787f53bb6cSHong Zhang if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);} 1796bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1806bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1816bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 182c31cb41cSBarry Smith ierr = PetscFree(a->idiag);CHKERRQ(ierr); 183c31cb41cSBarry Smith ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 184c760cd28SBarry Smith if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);} 18505b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 18641f059aeSBarry Smith ierr = PetscFree(a->sor_work);CHKERRQ(ierr); 18705b42c5fSBarry Smith ierr = PetscFree(a->solves_work);CHKERRQ(ierr); 18805b42c5fSBarry Smith ierr = PetscFree(a->mult_work);CHKERRQ(ierr); 18905b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 1904da8f245SBarry Smith if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);} 1911a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1926bf464f9SBarry Smith ierr = MatDestroy(&a->parent);CHKERRQ(ierr); 193bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 194901853e0SKris Buschelman 195dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 196bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 197bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 198bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr); 199bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr); 200bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr); 201bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr); 20238f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr); 203bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr); 2046214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 2056214f412SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_elemental_C",NULL);CHKERRQ(ierr); 2066214f412SHong Zhang #endif 20749b5e25fSSatish Balay PetscFunctionReturn(0); 20849b5e25fSSatish Balay } 20949b5e25fSSatish Balay 210ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg) 21149b5e25fSSatish Balay { 212045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 21363ba0a88SBarry Smith PetscErrorCode ierr; 21449b5e25fSSatish Balay 21549b5e25fSSatish Balay PetscFunctionBegin; 2164d9d31abSKris Buschelman switch (op) { 2174d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 2184e0d8c25SBarry Smith a->roworiented = flg; 2194d9d31abSKris Buschelman break; 220a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 221a9817697SBarry Smith a->keepnonzeropattern = flg; 2224d9d31abSKris Buschelman break; 223512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 224512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 2254d9d31abSKris Buschelman break; 2264d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 2274e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 2284d9d31abSKris Buschelman break; 2294d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 2304e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 2314d9d31abSKris Buschelman break; 23228b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 23328b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 23428b2fa4aSMatthew Knepley break; 2354e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 2364d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 2374d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 238290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 2394d9d31abSKris Buschelman break; 2409a4540c5SBarry Smith case MAT_HERMITIAN: 241e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first"); 2420bc54ff2SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 243eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort; 2440bc54ff2SBarry Smith } else if (A->cmap->bs == 1) { 245eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian; 246e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1"); 247eeffb40dSHong Zhang break; 2483d472b54SHong Zhang case MAT_SPD: 2495021d80fSJed Brown /* These options are handled directly by MatSetOption() */ 2503d472b54SHong Zhang break; 25177e54ba9SKris Buschelman case MAT_SYMMETRIC: 25277e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 2539a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 2544dcd73b1SHong Zhang /* These options are handled directly by MatSetOption() */ 255290bbb0aSBarry Smith break; 256941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 2574e0d8c25SBarry Smith a->ignore_ltriangular = flg; 258941593c8SHong Zhang break; 259941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 2604e0d8c25SBarry Smith a->ignore_ltriangular = flg; 26177e54ba9SKris Buschelman break; 262f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 2634e0d8c25SBarry Smith a->getrow_utriangular = flg; 264f5edf698SHong Zhang break; 265c10200c1SHong Zhang case MAT_SUBMAT_SINGLEIS: 266c10200c1SHong Zhang break; 2674d9d31abSKris Buschelman default: 268e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 26949b5e25fSSatish Balay } 27049b5e25fSSatish Balay PetscFunctionReturn(0); 27149b5e25fSSatish Balay } 27249b5e25fSSatish Balay 27352768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 27449b5e25fSSatish Balay { 27549b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2766849ba73SBarry Smith PetscErrorCode ierr; 27749b5e25fSSatish Balay 27849b5e25fSSatish Balay PetscFunctionBegin; 279e32f2f54SBarry 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()"); 28052768537SHong Zhang 281f5edf698SHong Zhang /* Get the upper triangular part of the row */ 28252768537SHong Zhang ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr); 28349b5e25fSSatish Balay PetscFunctionReturn(0); 28449b5e25fSSatish Balay } 28549b5e25fSSatish Balay 28613f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 28749b5e25fSSatish Balay { 288dfbe8321SBarry Smith PetscErrorCode ierr; 28949b5e25fSSatish Balay 29049b5e25fSSatish Balay PetscFunctionBegin; 29105b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 29205b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 29349b5e25fSSatish Balay PetscFunctionReturn(0); 29449b5e25fSSatish Balay } 29549b5e25fSSatish Balay 296f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 297f5edf698SHong Zhang { 298f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 299f5edf698SHong Zhang 300f5edf698SHong Zhang PetscFunctionBegin; 301f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 302f5edf698SHong Zhang PetscFunctionReturn(0); 303f5edf698SHong Zhang } 304f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 305f5edf698SHong Zhang { 306f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 307f5edf698SHong Zhang 308f5edf698SHong Zhang PetscFunctionBegin; 309f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 310f5edf698SHong Zhang PetscFunctionReturn(0); 311f5edf698SHong Zhang } 312f5edf698SHong Zhang 313fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B) 31449b5e25fSSatish Balay { 315dfbe8321SBarry Smith PetscErrorCode ierr; 3165fd66863SKarl Rupp 31749b5e25fSSatish Balay PetscFunctionBegin; 318cf37664fSBarry Smith if (reuse == MAT_INITIAL_MATRIX) { 319999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 320cf37664fSBarry Smith } else if (reuse == MAT_REUSE_MATRIX) { 321cf37664fSBarry Smith ierr = MatCopy(A,*B,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 322fc4dec0aSBarry Smith } 3238115998fSBarry Smith PetscFunctionReturn(0); 32449b5e25fSSatish Balay } 32549b5e25fSSatish Balay 3267da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 32749b5e25fSSatish Balay { 32849b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 329dfbe8321SBarry Smith PetscErrorCode ierr; 330d0f46423SBarry Smith PetscInt i,j,bs = A->rmap->bs,k,l,bs2=a->bs2; 331f3ef73ceSBarry Smith PetscViewerFormat format; 332121deb67SSatish Balay PetscInt *diag; 33349b5e25fSSatish Balay 33449b5e25fSSatish Balay PetscFunctionBegin; 335b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 336456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 33777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 338fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 339d2507d54SMatthew Knepley Mat aij; 340ade3a672SBarry Smith const char *matname; 341ade3a672SBarry Smith 342d5f3da31SBarry Smith if (A->factortype && bs>1) { 34370d5e725SHong 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); 34470d5e725SHong Zhang PetscFunctionReturn(0); 34570d5e725SHong Zhang } 346c9f458caSMatthew Knepley ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr); 347ade3a672SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr); 348ade3a672SBarry Smith ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr); 349c9f458caSMatthew Knepley ierr = MatView(aij,viewer);CHKERRQ(ierr); 3506bf464f9SBarry Smith ierr = MatDestroy(&aij);CHKERRQ(ierr); 351fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 352d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 35349b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 35449b5e25fSSatish Balay for (j=0; j<bs; j++) { 35577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 35649b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 35749b5e25fSSatish Balay for (l=0; l<bs; l++) { 35849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 35949b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 36057622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 36157622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 36249b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 36357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 36457622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 36549b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 36657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 36749b5e25fSSatish Balay } 36849b5e25fSSatish Balay #else 36949b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 37057622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 37149b5e25fSSatish Balay } 37249b5e25fSSatish Balay #endif 37349b5e25fSSatish Balay } 37449b5e25fSSatish Balay } 375b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 37649b5e25fSSatish Balay } 37749b5e25fSSatish Balay } 378d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 379c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 380c1490034SHong Zhang PetscFunctionReturn(0); 38149b5e25fSSatish Balay } else { 382d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 3832c990fa1SHong Zhang if (A->factortype) { /* for factored matrix */ 3842c990fa1SHong Zhang if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet"); 3852c990fa1SHong Zhang 386121deb67SSatish Balay diag=a->diag; 387121deb67SSatish Balay for (i=0; i<a->mbs; i++) { /* for row block i */ 3882c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 3892c990fa1SHong Zhang /* diagonal entry */ 3902c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 3912c990fa1SHong Zhang if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) { 39257622a8eSBarry 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); 3932c990fa1SHong Zhang } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) { 39457622a8eSBarry 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); 3952c990fa1SHong Zhang } else { 39657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 3972c990fa1SHong Zhang } 3982c990fa1SHong Zhang #else 3996712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4002c990fa1SHong Zhang #endif 4012c990fa1SHong Zhang /* off-diagonal entries */ 4022c990fa1SHong Zhang for (k=a->i[i]; k<a->i[i+1]-1; k++) { 4032c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 404ca0704adSBarry Smith if (PetscImaginaryPart(a->a[k]) > 0.0) { 40557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 406ca0704adSBarry Smith } else if (PetscImaginaryPart(a->a[k]) < 0.0) { 40757622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 4082c990fa1SHong Zhang } else { 40957622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr); 4102c990fa1SHong Zhang } 4112c990fa1SHong Zhang #else 41257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr); 4132c990fa1SHong Zhang #endif 4142c990fa1SHong Zhang } 4152c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 4162c990fa1SHong Zhang } 4172c990fa1SHong Zhang 4182c990fa1SHong Zhang } else { /* for non-factored matrix */ 4190c74a584SJed Brown for (i=0; i<a->mbs; i++) { /* for row block i */ 4200c74a584SJed Brown for (j=0; j<bs; j++) { /* for row bs*i + j */ 42177431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 4220c74a584SJed Brown for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */ 4230c74a584SJed Brown for (l=0; l<bs; l++) { /* for column */ 42449b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 42549b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 42657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 42757622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 42849b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 42957622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 43057622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 43149b5e25fSSatish Balay } else { 43257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 43349b5e25fSSatish Balay } 43449b5e25fSSatish Balay #else 43557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 43649b5e25fSSatish Balay #endif 43749b5e25fSSatish Balay } 43849b5e25fSSatish Balay } 439b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 44049b5e25fSSatish Balay } 44149b5e25fSSatish Balay } 4422c990fa1SHong Zhang } 443d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 44449b5e25fSSatish Balay } 445b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 44649b5e25fSSatish Balay PetscFunctionReturn(0); 44749b5e25fSSatish Balay } 44849b5e25fSSatish Balay 4499804daf3SBarry Smith #include <petscdraw.h> 4506849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 45149b5e25fSSatish Balay { 45249b5e25fSSatish Balay Mat A = (Mat) Aa; 45349b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 4546849ba73SBarry Smith PetscErrorCode ierr; 455d0f46423SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2; 45649b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 45749b5e25fSSatish Balay MatScalar *aa; 458b0a32e0cSBarry Smith PetscViewer viewer; 45949b5e25fSSatish Balay 46049b5e25fSSatish Balay PetscFunctionBegin; 46149b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 462b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 46349b5e25fSSatish Balay 46449b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 465383922c3SLisandro Dalcin 466383922c3SLisandro Dalcin ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 467383922c3SLisandro Dalcin ierr = PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");CHKERRQ(ierr); 468383922c3SLisandro Dalcin /* Blue for negative, Cyan for zero and Red for positive */ 469b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 47049b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 47149b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 472d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 47349b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 47449b5e25fSSatish Balay aa = a->a + j*bs2; 47549b5e25fSSatish Balay for (k=0; k<bs; k++) { 47649b5e25fSSatish Balay for (l=0; l<bs; l++) { 47749b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 478b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 47949b5e25fSSatish Balay } 48049b5e25fSSatish Balay } 48149b5e25fSSatish Balay } 48249b5e25fSSatish Balay } 483b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 48449b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 48549b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 486d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 48749b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 48849b5e25fSSatish Balay aa = a->a + j*bs2; 48949b5e25fSSatish Balay for (k=0; k<bs; k++) { 49049b5e25fSSatish Balay for (l=0; l<bs; l++) { 49149b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 492b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 49349b5e25fSSatish Balay } 49449b5e25fSSatish Balay } 49549b5e25fSSatish Balay } 49649b5e25fSSatish Balay } 497b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 49849b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 49949b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 500d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 50149b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 50249b5e25fSSatish Balay aa = a->a + j*bs2; 50349b5e25fSSatish Balay for (k=0; k<bs; k++) { 50449b5e25fSSatish Balay for (l=0; l<bs; l++) { 50549b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 506b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 50749b5e25fSSatish Balay } 50849b5e25fSSatish Balay } 50949b5e25fSSatish Balay } 51049b5e25fSSatish Balay } 511383922c3SLisandro Dalcin ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 51249b5e25fSSatish Balay PetscFunctionReturn(0); 51349b5e25fSSatish Balay } 51449b5e25fSSatish Balay 5156849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 51649b5e25fSSatish Balay { 517dfbe8321SBarry Smith PetscErrorCode ierr; 51849b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 519b0a32e0cSBarry Smith PetscDraw draw; 520ace3abfcSBarry Smith PetscBool isnull; 52149b5e25fSSatish Balay 52249b5e25fSSatish Balay PetscFunctionBegin; 523b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 524383922c3SLisandro Dalcin ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 525383922c3SLisandro Dalcin if (isnull) PetscFunctionReturn(0); 52649b5e25fSSatish Balay 527d0f46423SBarry Smith xr = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0; 52849b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 529b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 530832b7cebSLisandro Dalcin ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 531b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 5320298fd71SBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 533832b7cebSLisandro Dalcin ierr = PetscDrawSave(draw);CHKERRQ(ierr); 53449b5e25fSSatish Balay PetscFunctionReturn(0); 53549b5e25fSSatish Balay } 53649b5e25fSSatish Balay 537dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 53849b5e25fSSatish Balay { 539dfbe8321SBarry Smith PetscErrorCode ierr; 540ace3abfcSBarry Smith PetscBool iascii,isdraw; 54108917f38SBarry Smith FILE *file = 0; 54249b5e25fSSatish Balay 54349b5e25fSSatish Balay PetscFunctionBegin; 544251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 545251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 54632077d6dSBarry Smith if (iascii) { 54749b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 54849b5e25fSSatish Balay } else if (isdraw) { 54949b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 55049b5e25fSSatish Balay } else { 551a5e6ed63SBarry Smith Mat B; 552ade3a672SBarry Smith const char *matname; 553ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 554ade3a672SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr); 555ade3a672SBarry Smith ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr); 556a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 5576bf464f9SBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 55808917f38SBarry Smith ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr); 55908917f38SBarry Smith if (file) { 56008917f38SBarry Smith fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs); 56108917f38SBarry Smith } 56249b5e25fSSatish Balay } 56349b5e25fSSatish Balay PetscFunctionReturn(0); 56449b5e25fSSatish Balay } 56549b5e25fSSatish Balay 56649b5e25fSSatish Balay 56713f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 56849b5e25fSSatish Balay { 569045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 57013f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 57113f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 572d0f46423SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2; 57397e567efSBarry Smith MatScalar *ap,*aa = a->a; 57449b5e25fSSatish Balay 57549b5e25fSSatish Balay PetscFunctionBegin; 57649b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 57749b5e25fSSatish Balay row = im[k]; brow = row/bs; 578e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 579e32f2f54SBarry 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); 58049b5e25fSSatish Balay rp = aj + ai[brow]; ap = aa + bs2*ai[brow]; 58149b5e25fSSatish Balay nrow = ailen[brow]; 58249b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 583e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 584e32f2f54SBarry 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); 58549b5e25fSSatish Balay col = in[l]; 58649b5e25fSSatish Balay bcol = col/bs; 58749b5e25fSSatish Balay cidx = col%bs; 58849b5e25fSSatish Balay ridx = row%bs; 58949b5e25fSSatish Balay high = nrow; 59049b5e25fSSatish Balay low = 0; /* assume unsorted */ 59149b5e25fSSatish Balay while (high-low > 5) { 59249b5e25fSSatish Balay t = (low+high)/2; 59349b5e25fSSatish Balay if (rp[t] > bcol) high = t; 59449b5e25fSSatish Balay else low = t; 59549b5e25fSSatish Balay } 59649b5e25fSSatish Balay for (i=low; i<high; i++) { 59749b5e25fSSatish Balay if (rp[i] > bcol) break; 59849b5e25fSSatish Balay if (rp[i] == bcol) { 59949b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 60049b5e25fSSatish Balay goto finished; 60149b5e25fSSatish Balay } 60249b5e25fSSatish Balay } 60397e567efSBarry Smith *v++ = 0.0; 60449b5e25fSSatish Balay finished:; 60549b5e25fSSatish Balay } 60649b5e25fSSatish Balay } 60749b5e25fSSatish Balay PetscFunctionReturn(0); 60849b5e25fSSatish Balay } 60949b5e25fSSatish Balay 61049b5e25fSSatish Balay 61113f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 61249b5e25fSSatish Balay { 6130880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 6146849ba73SBarry Smith PetscErrorCode ierr; 615e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 61613f74950SBarry Smith PetscInt *imax =a->imax,*ai=a->i,*ailen=a->ilen; 617d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval; 618ace3abfcSBarry Smith PetscBool roworiented=a->roworiented; 619dd6ea824SBarry Smith const PetscScalar *value = v; 620f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 6210880e062SHong Zhang 62249b5e25fSSatish Balay PetscFunctionBegin; 62326fbe8dcSKarl Rupp if (roworiented) stepval = (n-1)*bs; 62426fbe8dcSKarl Rupp else stepval = (m-1)*bs; 62526fbe8dcSKarl Rupp 6260880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 6270880e062SHong Zhang row = im[k]; 6280880e062SHong Zhang if (row < 0) continue; 6292515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 6302f7d4af7SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index row too large %D max %D",row,a->mbs-1); 6310880e062SHong Zhang #endif 6320880e062SHong Zhang rp = aj + ai[row]; 6330880e062SHong Zhang ap = aa + bs2*ai[row]; 6340880e062SHong Zhang rmax = imax[row]; 6350880e062SHong Zhang nrow = ailen[row]; 6360880e062SHong Zhang low = 0; 637818f2c47SBarry Smith high = nrow; 6380880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 6390880e062SHong Zhang if (in[l] < 0) continue; 6400880e062SHong Zhang col = in[l]; 6412515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 6422f7d4af7SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index column too large %D max %D",col,a->nbs-1); 643b1823623SSatish Balay #endif 644b98bf0e1SJed Brown if (col < row) { 64526fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular block */ 64626fbe8dcSKarl 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)"); 647b98bf0e1SJed Brown } 64826fbe8dcSKarl Rupp if (roworiented) value = v + k*(stepval+bs)*bs + l*bs; 64926fbe8dcSKarl Rupp else value = v + l*(stepval+bs)*bs + k*bs; 65026fbe8dcSKarl Rupp 65126fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 65226fbe8dcSKarl Rupp else high = nrow; 65326fbe8dcSKarl Rupp 654e2ee6c50SBarry Smith lastcol = col; 6550880e062SHong Zhang while (high-low > 7) { 6560880e062SHong Zhang t = (low+high)/2; 6570880e062SHong Zhang if (rp[t] > col) high = t; 6580880e062SHong Zhang else low = t; 6590880e062SHong Zhang } 6600880e062SHong Zhang for (i=low; i<high; i++) { 6610880e062SHong Zhang if (rp[i] > col) break; 6620880e062SHong Zhang if (rp[i] == col) { 6630880e062SHong Zhang bap = ap + bs2*i; 6640880e062SHong Zhang if (roworiented) { 6650880e062SHong Zhang if (is == ADD_VALUES) { 6660880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6670880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6680880e062SHong Zhang bap[jj] += *value++; 6690880e062SHong Zhang } 6700880e062SHong Zhang } 6710880e062SHong Zhang } else { 6720880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6730880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6740880e062SHong Zhang bap[jj] = *value++; 6750880e062SHong Zhang } 6760880e062SHong Zhang } 6770880e062SHong Zhang } 6780880e062SHong Zhang } else { 6790880e062SHong Zhang if (is == ADD_VALUES) { 6800880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6810880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6820880e062SHong Zhang *bap++ += *value++; 6830880e062SHong Zhang } 6840880e062SHong Zhang } 6850880e062SHong Zhang } else { 6860880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6870880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6880880e062SHong Zhang *bap++ = *value++; 6890880e062SHong Zhang } 6900880e062SHong Zhang } 6910880e062SHong Zhang } 6920880e062SHong Zhang } 6930880e062SHong Zhang goto noinsert2; 6940880e062SHong Zhang } 6950880e062SHong Zhang } 6960880e062SHong Zhang if (nonew == 1) goto noinsert2; 6972f7d4af7SBarry 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); 698fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 699c03d1d03SSatish Balay N = nrow++ - 1; high++; 7000880e062SHong Zhang /* shift up all the later entries in this row */ 7010880e062SHong Zhang for (ii=N; ii>=i; ii--) { 7020880e062SHong Zhang rp[ii+1] = rp[ii]; 7030880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 7040880e062SHong Zhang } 7050880e062SHong Zhang if (N >= i) { 7060880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 7070880e062SHong Zhang } 7080880e062SHong Zhang rp[i] = col; 7090880e062SHong Zhang bap = ap + bs2*i; 7100880e062SHong Zhang if (roworiented) { 7110880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7120880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7130880e062SHong Zhang bap[jj] = *value++; 7140880e062SHong Zhang } 7150880e062SHong Zhang } 7160880e062SHong Zhang } else { 7170880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7180880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7190880e062SHong Zhang *bap++ = *value++; 7200880e062SHong Zhang } 7210880e062SHong Zhang } 7220880e062SHong Zhang } 7230880e062SHong Zhang noinsert2:; 7240880e062SHong Zhang low = i; 7250880e062SHong Zhang } 7260880e062SHong Zhang ailen[row] = nrow; 7270880e062SHong Zhang } 7280880e062SHong Zhang PetscFunctionReturn(0); 72949b5e25fSSatish Balay } 73049b5e25fSSatish Balay 73164831d72SBarry Smith /* 73264831d72SBarry Smith This is not yet used 73364831d72SBarry Smith */ 7344108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A) 7350def2e27SBarry Smith { 7360def2e27SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7370def2e27SBarry Smith PetscErrorCode ierr; 7380def2e27SBarry Smith const PetscInt *ai = a->i, *aj = a->j,*cols; 7390def2e27SBarry Smith PetscInt i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts; 740ace3abfcSBarry Smith PetscBool flag; 7410def2e27SBarry Smith 7420def2e27SBarry Smith PetscFunctionBegin; 743785e854fSJed Brown ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr); 7440def2e27SBarry Smith while (i < m) { 7450def2e27SBarry Smith nzx = ai[i+1] - ai[i]; /* Number of nonzeros */ 7460def2e27SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 7470def2e27SBarry Smith for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 7480def2e27SBarry Smith nzy = ai[j+1] - ai[j]; 7490def2e27SBarry Smith if (nzy != (nzx - j + i)) break; 7500def2e27SBarry Smith ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr); 7510def2e27SBarry Smith if (!flag) break; 7520def2e27SBarry Smith } 7530def2e27SBarry Smith ns[node_count++] = blk_size; 75426fbe8dcSKarl Rupp 7550def2e27SBarry Smith i = j; 7560def2e27SBarry Smith } 7570def2e27SBarry Smith if (!a->inode.size && m && node_count > .9*m) { 7580def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7590def2e27SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 7600def2e27SBarry Smith } else { 7610def2e27SBarry Smith a->inode.node_count = node_count; 76226fbe8dcSKarl Rupp 763785e854fSJed Brown ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr); 7643bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr); 76522d28d08SBarry Smith ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr); 7660def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7670def2e27SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 7680def2e27SBarry Smith 7690def2e27SBarry Smith /* count collections of adjacent columns in each inode */ 7700def2e27SBarry Smith row = 0; 7710def2e27SBarry Smith cnt = 0; 7720def2e27SBarry Smith for (i=0; i<node_count; i++) { 7730def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 7740def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 7750def2e27SBarry Smith for (j=1; j<nz; j++) { 77626fbe8dcSKarl Rupp if (cols[j] != cols[j-1]+1) cnt++; 7770def2e27SBarry Smith } 7780def2e27SBarry Smith cnt++; 7790def2e27SBarry Smith row += a->inode.size[i]; 7800def2e27SBarry Smith } 781785e854fSJed Brown ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr); 7820def2e27SBarry Smith cnt = 0; 7830def2e27SBarry Smith row = 0; 7840def2e27SBarry Smith for (i=0; i<node_count; i++) { 7850def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 7860def2e27SBarry Smith counts[2*cnt] = cols[0]; 7870def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 7880def2e27SBarry Smith cnt2 = 1; 7890def2e27SBarry Smith for (j=1; j<nz; j++) { 7900def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 7910def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 7920def2e27SBarry Smith counts[2*cnt] = cols[j]; 7930def2e27SBarry Smith cnt2 = 1; 7940def2e27SBarry Smith } else cnt2++; 7950def2e27SBarry Smith } 7960def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 7970def2e27SBarry Smith row += a->inode.size[i]; 7980def2e27SBarry Smith } 79922d28d08SBarry Smith ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr); 8000def2e27SBarry Smith } 80138702af4SBarry Smith PetscFunctionReturn(0); 80238702af4SBarry Smith } 80338702af4SBarry Smith 804dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 80549b5e25fSSatish Balay { 80649b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 8076849ba73SBarry Smith PetscErrorCode ierr; 80813f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 809d0f46423SBarry Smith PetscInt m = A->rmap->N,*ip,N,*ailen = a->ilen; 81013f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 81149b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 81249b5e25fSSatish Balay 81349b5e25fSSatish Balay PetscFunctionBegin; 81449b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 81549b5e25fSSatish Balay 81649b5e25fSSatish Balay if (m) rmax = ailen[0]; 81749b5e25fSSatish Balay for (i=1; i<mbs; i++) { 81849b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 81949b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 82049b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 82149b5e25fSSatish Balay if (fshift) { 82249b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 82349b5e25fSSatish Balay N = ailen[i]; 82449b5e25fSSatish Balay for (j=0; j<N; j++) { 82549b5e25fSSatish Balay ip[j-fshift] = ip[j]; 82649b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 82749b5e25fSSatish Balay } 82849b5e25fSSatish Balay } 82949b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 83049b5e25fSSatish Balay } 83149b5e25fSSatish Balay if (mbs) { 83249b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 83349b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 83449b5e25fSSatish Balay } 83549b5e25fSSatish Balay /* reset ilen and imax for each row */ 83649b5e25fSSatish Balay for (i=0; i<mbs; i++) { 83749b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 83849b5e25fSSatish Balay } 8396c6c5352SBarry Smith a->nz = ai[mbs]; 84049b5e25fSSatish Balay 841b424e231SHong Zhang /* diagonals may have moved, reset it */ 842b424e231SHong Zhang if (a->diag) { 8432ed38d0bSJed Brown ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr); 84449b5e25fSSatish Balay } 84526fbe8dcSKarl 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); 84626fbe8dcSKarl Rupp 847d0f46423SBarry 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); 848ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 849ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 85026fbe8dcSKarl Rupp 8518e58a170SBarry Smith A->info.mallocs += a->reallocs; 85249b5e25fSSatish Balay a->reallocs = 0; 85349b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 854061b2667SBarry Smith a->idiagvalid = PETSC_FALSE; 8554dcd73b1SHong Zhang a->rmax = rmax; 85638702af4SBarry Smith 85738702af4SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 85844e1c64aSLisandro Dalcin if (a->jshort && a->free_jshort) { 85917803ae8SHong Zhang /* when matrix data structure is changed, previous jshort must be replaced */ 86017803ae8SHong Zhang ierr = PetscFree(a->jshort);CHKERRQ(ierr); 86117803ae8SHong Zhang } 862785e854fSJed Brown ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr); 8633bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr); 86438702af4SBarry Smith for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i]; 86538702af4SBarry Smith A->ops->mult = MatMult_SeqSBAIJ_1_ushort; 86641f059aeSBarry Smith A->ops->sor = MatSOR_SeqSBAIJ_ushort; 8674da8f245SBarry Smith a->free_jshort = PETSC_TRUE; 86838702af4SBarry Smith } 86949b5e25fSSatish Balay PetscFunctionReturn(0); 87049b5e25fSSatish Balay } 87149b5e25fSSatish Balay 87249b5e25fSSatish Balay /* 87349b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 87449b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 87549b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 87649b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 87749b5e25fSSatish Balay */ 87813f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 87949b5e25fSSatish Balay { 88013f74950SBarry Smith PetscInt i,j,k,row; 881ace3abfcSBarry Smith PetscBool flg; 88249b5e25fSSatish Balay 88349b5e25fSSatish Balay PetscFunctionBegin; 88449b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 88549b5e25fSSatish Balay row = idx[i]; 88649b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 88749b5e25fSSatish Balay sizes[j] = 1; 88849b5e25fSSatish Balay i++; 88949b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 89049b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 89149b5e25fSSatish Balay i++; 89249b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 89349b5e25fSSatish Balay flg = PETSC_TRUE; 89449b5e25fSSatish Balay for (k=1; k<bs; k++) { 89549b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 89649b5e25fSSatish Balay flg = PETSC_FALSE; 89749b5e25fSSatish Balay break; 89849b5e25fSSatish Balay } 89949b5e25fSSatish Balay } 900abc0a331SBarry Smith if (flg) { /* No break in the bs */ 90149b5e25fSSatish Balay sizes[j] = bs; 90249b5e25fSSatish Balay i += bs; 90349b5e25fSSatish Balay } else { 90449b5e25fSSatish Balay sizes[j] = 1; 90549b5e25fSSatish Balay i++; 90649b5e25fSSatish Balay } 90749b5e25fSSatish Balay } 90849b5e25fSSatish Balay } 90949b5e25fSSatish Balay *bs_max = j; 91049b5e25fSSatish Balay PetscFunctionReturn(0); 91149b5e25fSSatish Balay } 91249b5e25fSSatish Balay 91349b5e25fSSatish Balay 91449b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 91549b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 91649b5e25fSSatish Balay */ 91749b5e25fSSatish Balay 91813f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 91949b5e25fSSatish Balay { 92049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 9216849ba73SBarry Smith PetscErrorCode ierr; 922e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 92313f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 924d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol; 92513f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 92649b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 92749b5e25fSSatish Balay 92849b5e25fSSatish Balay PetscFunctionBegin; 92949b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 93049b5e25fSSatish Balay row = im[k]; /* row number */ 93149b5e25fSSatish Balay brow = row/bs; /* block row number */ 93249b5e25fSSatish Balay if (row < 0) continue; 9332515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 934e32f2f54SBarry 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); 93549b5e25fSSatish Balay #endif 93649b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 93749b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 93849b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 93949b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 94049b5e25fSSatish Balay low = 0; 94149b5e25fSSatish Balay 94249b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 94349b5e25fSSatish Balay if (in[l] < 0) continue; 9442515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 945e32f2f54SBarry Smith if (in[l] >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->rmap->N-1); 94649b5e25fSSatish Balay #endif 94749b5e25fSSatish Balay col = in[l]; 94849b5e25fSSatish Balay bcol = col/bs; /* block col number */ 94949b5e25fSSatish Balay 950941593c8SHong Zhang if (brow > bcol) { 95126fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular values */ 95226fbe8dcSKarl 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)"); 953941593c8SHong Zhang } 954f4989cb3SHong Zhang 95549b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 9568549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)) { 95749b5e25fSSatish Balay /* element value a(k,l) */ 95826fbe8dcSKarl Rupp if (roworiented) value = v[l + k*n]; 95926fbe8dcSKarl Rupp else value = v[k + l*m]; 96049b5e25fSSatish Balay 96149b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 96226fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 96326fbe8dcSKarl Rupp high = nrow; 964e2ee6c50SBarry Smith lastcol = col; 96549b5e25fSSatish Balay while (high-low > 7) { 96649b5e25fSSatish Balay t = (low+high)/2; 96749b5e25fSSatish Balay if (rp[t] > bcol) high = t; 96849b5e25fSSatish Balay else low = t; 96949b5e25fSSatish Balay } 97049b5e25fSSatish Balay for (i=low; i<high; i++) { 97149b5e25fSSatish Balay if (rp[i] > bcol) break; 97249b5e25fSSatish Balay if (rp[i] == bcol) { 97349b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 97449b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 97549b5e25fSSatish Balay else *bap = value; 9768549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 9778549e402SHong Zhang if (brow == bcol && ridx < cidx) { 9788549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 9798549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 9808549e402SHong Zhang else *bap = value; 9818549e402SHong Zhang } 98249b5e25fSSatish Balay goto noinsert1; 98349b5e25fSSatish Balay } 98449b5e25fSSatish Balay } 98549b5e25fSSatish Balay 98649b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 987e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 988fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 98949b5e25fSSatish Balay 990c03d1d03SSatish Balay N = nrow++ - 1; high++; 99149b5e25fSSatish Balay /* shift up all the later entries in this row */ 99249b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 99349b5e25fSSatish Balay rp[ii+1] = rp[ii]; 99449b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 99549b5e25fSSatish Balay } 99649b5e25fSSatish Balay if (N>=i) { 99749b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 99849b5e25fSSatish Balay } 99949b5e25fSSatish Balay rp[i] = bcol; 100049b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 1001e56f5c9eSBarry Smith A->nonzerostate++; 100249b5e25fSSatish Balay noinsert1:; 100349b5e25fSSatish Balay low = i; 10048549e402SHong Zhang } 100549b5e25fSSatish Balay } /* end of loop over added columns */ 100649b5e25fSSatish Balay ailen[brow] = nrow; 100749b5e25fSSatish Balay } /* end of loop over added rows */ 100849b5e25fSSatish Balay PetscFunctionReturn(0); 100949b5e25fSSatish Balay } 101049b5e25fSSatish Balay 10110481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info) 101249b5e25fSSatish Balay { 10134ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 101449b5e25fSSatish Balay Mat outA; 1015dfbe8321SBarry Smith PetscErrorCode ierr; 1016ace3abfcSBarry Smith PetscBool row_identity; 101749b5e25fSSatish Balay 101849b5e25fSSatish Balay PetscFunctionBegin; 1019e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 1020c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 1021e32f2f54SBarry Smith if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported"); 1022e32f2f54SBarry 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()! */ 1023c84f5b01SHong Zhang 102449b5e25fSSatish Balay outA = inA; 1025d5f3da31SBarry Smith inA->factortype = MAT_FACTOR_ICC; 1026f6224b95SHong Zhang ierr = PetscFree(inA->solvertype);CHKERRQ(ierr); 1027f6224b95SHong Zhang ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr); 102849b5e25fSSatish Balay 10291a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 1030d595f711SHong Zhang ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr); 103149b5e25fSSatish Balay 1032c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10336bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1034c84f5b01SHong Zhang a->row = row; 1035c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10366bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1037c84f5b01SHong Zhang a->col = row; 1038c84f5b01SHong Zhang 1039c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1040c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 10413bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr); 104249b5e25fSSatish Balay 104349b5e25fSSatish Balay if (!a->solve_work) { 1044854ce69bSBarry Smith ierr = PetscMalloc1(inA->rmap->N+inA->rmap->bs,&a->solve_work);CHKERRQ(ierr); 10453bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 104649b5e25fSSatish Balay } 104749b5e25fSSatish Balay 1048719d5645SBarry Smith ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr); 104949b5e25fSSatish Balay PetscFunctionReturn(0); 105049b5e25fSSatish Balay } 1051950f1e5bSHong Zhang 10527087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 105349b5e25fSSatish Balay { 1054045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ*)mat->data; 105513f74950SBarry Smith PetscInt i,nz,n; 10567827cd58SJed Brown PetscErrorCode ierr; 105749b5e25fSSatish Balay 105849b5e25fSSatish Balay PetscFunctionBegin; 10596c6c5352SBarry Smith nz = baij->maxnz; 1060d0f46423SBarry Smith n = mat->cmap->n; 106126fbe8dcSKarl Rupp for (i=0; i<nz; i++) baij->j[i] = indices[i]; 106226fbe8dcSKarl Rupp 10636c6c5352SBarry Smith baij->nz = nz; 106426fbe8dcSKarl Rupp for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i]; 106526fbe8dcSKarl Rupp 10667827cd58SJed Brown ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 106749b5e25fSSatish Balay PetscFunctionReturn(0); 106849b5e25fSSatish Balay } 106949b5e25fSSatish Balay 107049b5e25fSSatish Balay /*@ 107119585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 107249b5e25fSSatish Balay in the matrix. 107349b5e25fSSatish Balay 107449b5e25fSSatish Balay Input Parameters: 107519585528SSatish Balay + mat - the SeqSBAIJ matrix 107649b5e25fSSatish Balay - indices - the column indices 107749b5e25fSSatish Balay 107849b5e25fSSatish Balay Level: advanced 107949b5e25fSSatish Balay 108049b5e25fSSatish Balay Notes: 108149b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 108249b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 108349b5e25fSSatish Balay of the MatSetValues() operation. 108449b5e25fSSatish Balay 108549b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1086d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 108749b5e25fSSatish Balay 1088ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 108949b5e25fSSatish Balay 1090ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 109149b5e25fSSatish Balay @*/ 10927087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 109349b5e25fSSatish Balay { 10944ac538c5SBarry Smith PetscErrorCode ierr; 109549b5e25fSSatish Balay 109649b5e25fSSatish Balay PetscFunctionBegin; 10970700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 10984482741eSBarry Smith PetscValidPointer(indices,2); 10994ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr); 110049b5e25fSSatish Balay PetscFunctionReturn(0); 110149b5e25fSSatish Balay } 110249b5e25fSSatish Balay 11033c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 11043c896bc6SHong Zhang { 11053c896bc6SHong Zhang PetscErrorCode ierr; 11063c896bc6SHong Zhang 11073c896bc6SHong Zhang PetscFunctionBegin; 11083c896bc6SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 11093c896bc6SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 11103c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11113c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11123c896bc6SHong Zhang 1113e7e72b3dSBarry Smith if (a->i[A->rmap->N] != b->i[B->rmap->N]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 1114d0f46423SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr); 11153c896bc6SHong Zhang } else { 1116f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11173c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1118f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11193c896bc6SHong Zhang } 11203c896bc6SHong Zhang PetscFunctionReturn(0); 11213c896bc6SHong Zhang } 11223c896bc6SHong Zhang 11234994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A) 1124273d9f13SBarry Smith { 1125dfbe8321SBarry Smith PetscErrorCode ierr; 1126273d9f13SBarry Smith 1127273d9f13SBarry Smith PetscFunctionBegin; 1128367daffbSBarry Smith ierr = MatSeqSBAIJSetPreallocation(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr); 1129273d9f13SBarry Smith PetscFunctionReturn(0); 1130273d9f13SBarry Smith } 1131273d9f13SBarry Smith 11328c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1133a6ece127SHong Zhang { 1134a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11355fd66863SKarl Rupp 1136a6ece127SHong Zhang PetscFunctionBegin; 1137a6ece127SHong Zhang *array = a->a; 1138a6ece127SHong Zhang PetscFunctionReturn(0); 1139a6ece127SHong Zhang } 1140a6ece127SHong Zhang 11418c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1142a6ece127SHong Zhang { 1143a6ece127SHong Zhang PetscFunctionBegin; 1144a6ece127SHong Zhang PetscFunctionReturn(0); 1145a6ece127SHong Zhang } 1146a6ece127SHong Zhang 114752768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz) 114852768537SHong Zhang { 1149b264fe52SHong Zhang PetscInt bs = Y->rmap->bs,mbs = Y->rmap->N/bs; 115052768537SHong Zhang Mat_SeqSBAIJ *x = (Mat_SeqSBAIJ*)X->data; 115152768537SHong Zhang Mat_SeqSBAIJ *y = (Mat_SeqSBAIJ*)Y->data; 1152b264fe52SHong Zhang PetscErrorCode ierr; 115352768537SHong Zhang 115452768537SHong Zhang PetscFunctionBegin; 115552768537SHong Zhang /* Set the number of nonzeros in the new matrix */ 1156b264fe52SHong Zhang ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr); 115752768537SHong Zhang PetscFunctionReturn(0); 115852768537SHong Zhang } 115952768537SHong Zhang 1160f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 116142ee4b1aSHong Zhang { 116242ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data; 1163dfbe8321SBarry Smith PetscErrorCode ierr; 116431ce2d13SHong Zhang PetscInt bs=Y->rmap->bs,bs2=bs*bs; 1165e838b9e7SJed Brown PetscBLASInt one = 1; 116642ee4b1aSHong Zhang 116742ee4b1aSHong Zhang PetscFunctionBegin; 116842ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1169f4df32b1SMatthew Knepley PetscScalar alpha = a; 1170c5df96a5SBarry Smith PetscBLASInt bnz; 1171c5df96a5SBarry Smith ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr); 11728b83055fSJed Brown PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one)); 1173a3fa217bSJose E. Roman ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 1174ab784542SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1175ab784542SHong Zhang ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr); 1176ab784542SHong Zhang ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1177ab784542SHong Zhang ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr); 117842ee4b1aSHong Zhang } else { 117952768537SHong Zhang Mat B; 118052768537SHong Zhang PetscInt *nnz; 118152768537SHong Zhang if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size"); 1182f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 118352768537SHong Zhang ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr); 118452768537SHong Zhang ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr); 118552768537SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr); 118652768537SHong Zhang ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr); 118752768537SHong Zhang ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr); 118852768537SHong Zhang ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr); 118952768537SHong Zhang ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr); 119052768537SHong Zhang ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr); 119152768537SHong Zhang ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 119252768537SHong Zhang 119352768537SHong Zhang ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr); 119452768537SHong Zhang 119528be2f97SBarry Smith ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr); 119652768537SHong Zhang ierr = PetscFree(nnz);CHKERRQ(ierr); 1197f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 119852768537SHong Zhang ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr); 119942ee4b1aSHong Zhang } 120042ee4b1aSHong Zhang PetscFunctionReturn(0); 120142ee4b1aSHong Zhang } 120242ee4b1aSHong Zhang 1203ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1204efcf0fc3SBarry Smith { 1205efcf0fc3SBarry Smith PetscFunctionBegin; 1206efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1207efcf0fc3SBarry Smith PetscFunctionReturn(0); 1208efcf0fc3SBarry Smith } 1209efcf0fc3SBarry Smith 1210ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool *flg) 1211efcf0fc3SBarry Smith { 1212efcf0fc3SBarry Smith PetscFunctionBegin; 1213efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1214efcf0fc3SBarry Smith PetscFunctionReturn(0); 1215efcf0fc3SBarry Smith } 1216efcf0fc3SBarry Smith 1217ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1218efcf0fc3SBarry Smith { 1219efcf0fc3SBarry Smith PetscFunctionBegin; 1220efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1221efcf0fc3SBarry Smith PetscFunctionReturn(0); 1222efcf0fc3SBarry Smith } 1223efcf0fc3SBarry Smith 122499cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 122599cafbc1SBarry Smith { 122699cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 122799cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1228dd6ea824SBarry Smith MatScalar *aa = a->a; 122999cafbc1SBarry Smith 123099cafbc1SBarry Smith PetscFunctionBegin; 123199cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 123299cafbc1SBarry Smith PetscFunctionReturn(0); 123399cafbc1SBarry Smith } 123499cafbc1SBarry Smith 123599cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A) 123699cafbc1SBarry Smith { 123799cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 123899cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1239dd6ea824SBarry Smith MatScalar *aa = a->a; 124099cafbc1SBarry Smith 124199cafbc1SBarry Smith PetscFunctionBegin; 124299cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 124399cafbc1SBarry Smith PetscFunctionReturn(0); 124499cafbc1SBarry Smith } 124599cafbc1SBarry Smith 12463bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b) 12473bededecSBarry Smith { 12483bededecSBarry Smith Mat_SeqSBAIJ *baij=(Mat_SeqSBAIJ*)A->data; 12493bededecSBarry Smith PetscErrorCode ierr; 12503bededecSBarry Smith PetscInt i,j,k,count; 12513bededecSBarry Smith PetscInt bs =A->rmap->bs,bs2=baij->bs2,row,col; 12523bededecSBarry Smith PetscScalar zero = 0.0; 12533bededecSBarry Smith MatScalar *aa; 12543bededecSBarry Smith const PetscScalar *xx; 12553bededecSBarry Smith PetscScalar *bb; 125656777dd2SBarry Smith PetscBool *zeroed,vecs = PETSC_FALSE; 12573bededecSBarry Smith 12583bededecSBarry Smith PetscFunctionBegin; 12593bededecSBarry Smith /* fix right hand side if needed */ 12603bededecSBarry Smith if (x && b) { 12613bededecSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 12623bededecSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 126356777dd2SBarry Smith vecs = PETSC_TRUE; 12643bededecSBarry Smith } 12653bededecSBarry Smith 12663bededecSBarry Smith /* zero the columns */ 12671795a4d1SJed Brown ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr); 12683bededecSBarry Smith for (i=0; i<is_n; i++) { 12693bededecSBarry 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]); 12703bededecSBarry Smith zeroed[is_idx[i]] = PETSC_TRUE; 12713bededecSBarry Smith } 127256777dd2SBarry Smith if (vecs) { 127356777dd2SBarry Smith for (i=0; i<A->rmap->N; i++) { 127456777dd2SBarry Smith row = i/bs; 127556777dd2SBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 127656777dd2SBarry Smith for (k=0; k<bs; k++) { 127756777dd2SBarry Smith col = bs*baij->j[j] + k; 127856777dd2SBarry Smith if (col <= i) continue; 127956777dd2SBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 128026fbe8dcSKarl Rupp if (!zeroed[i] && zeroed[col]) bb[i] -= aa[0]*xx[col]; 128126fbe8dcSKarl Rupp if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i]; 128256777dd2SBarry Smith } 128356777dd2SBarry Smith } 128456777dd2SBarry Smith } 128526fbe8dcSKarl Rupp for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]]; 128656777dd2SBarry Smith } 128756777dd2SBarry Smith 12883bededecSBarry Smith for (i=0; i<A->rmap->N; i++) { 12893bededecSBarry Smith if (!zeroed[i]) { 12903bededecSBarry Smith row = i/bs; 12913bededecSBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 12923bededecSBarry Smith for (k=0; k<bs; k++) { 12933bededecSBarry Smith col = bs*baij->j[j] + k; 12943bededecSBarry Smith if (zeroed[col]) { 12953bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 12963bededecSBarry Smith aa[0] = 0.0; 12973bededecSBarry Smith } 12983bededecSBarry Smith } 12993bededecSBarry Smith } 13003bededecSBarry Smith } 13013bededecSBarry Smith } 13023bededecSBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 130356777dd2SBarry Smith if (vecs) { 130456777dd2SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 130556777dd2SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 130656777dd2SBarry Smith } 13073bededecSBarry Smith 13083bededecSBarry Smith /* zero the rows */ 13093bededecSBarry Smith for (i=0; i<is_n; i++) { 13103bededecSBarry Smith row = is_idx[i]; 13113bededecSBarry Smith count = (baij->i[row/bs +1] - baij->i[row/bs])*bs; 13123bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs); 13133bededecSBarry Smith for (k=0; k<count; k++) { 13143bededecSBarry Smith aa[0] = zero; 13153bededecSBarry Smith aa += bs; 13163bededecSBarry Smith } 13173bededecSBarry Smith if (diag != 0.0) { 13183bededecSBarry Smith ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr); 13193bededecSBarry Smith } 13203bededecSBarry Smith } 13213bededecSBarry Smith ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 13223bededecSBarry Smith PetscFunctionReturn(0); 13233bededecSBarry Smith } 13243bededecSBarry Smith 13257d68702bSBarry Smith PetscErrorCode MatShift_SeqSBAIJ(Mat Y,PetscScalar a) 13267d68702bSBarry Smith { 13277d68702bSBarry Smith PetscErrorCode ierr; 13287d68702bSBarry Smith Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)Y->data; 13297d68702bSBarry Smith 13307d68702bSBarry Smith PetscFunctionBegin; 13316f33a894SBarry Smith if (!Y->preallocated || !aij->nz) { 13327d68702bSBarry Smith ierr = MatSeqSBAIJSetPreallocation(Y,Y->rmap->bs,1,NULL);CHKERRQ(ierr); 13337d68702bSBarry Smith } 13347d68702bSBarry Smith ierr = MatShift_Basic(Y,a);CHKERRQ(ierr); 13357d68702bSBarry Smith PetscFunctionReturn(0); 13367d68702bSBarry Smith } 13377d68702bSBarry Smith 133849b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 13393964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 134049b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 134149b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 134249b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 134397304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1344431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1345e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 1346db4efbfdSBarry Smith 0, 134749b5e25fSSatish Balay 0, 134849b5e25fSSatish Balay 0, 134997304618SKris Buschelman /* 10*/ 0, 135049b5e25fSSatish Balay 0, 1351c078aec8SLisandro Dalcin MatCholeskyFactor_SeqSBAIJ, 135241f059aeSBarry Smith MatSOR_SeqSBAIJ, 135349b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 135497304618SKris Buschelman /* 15*/ MatGetInfo_SeqSBAIJ, 135549b5e25fSSatish Balay MatEqual_SeqSBAIJ, 135649b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 135749b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 135849b5e25fSSatish Balay MatNorm_SeqSBAIJ, 135997304618SKris Buschelman /* 20*/ 0, 136049b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 136149b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 136249b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1363d519adbfSMatthew Knepley /* 24*/ 0, 136449b5e25fSSatish Balay 0, 136549b5e25fSSatish Balay 0, 1366db4efbfdSBarry Smith 0, 1367db4efbfdSBarry Smith 0, 13684994cf47SJed Brown /* 29*/ MatSetUp_SeqSBAIJ, 1369c464158bSHong Zhang 0, 1370db4efbfdSBarry Smith 0, 13718c778c55SBarry Smith 0, 13728c778c55SBarry Smith 0, 1373d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqSBAIJ, 1374719d5645SBarry Smith 0, 1375719d5645SBarry Smith 0, 137649b5e25fSSatish Balay 0, 1377c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 1378d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqSBAIJ, 137949b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 138049b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 138149b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 13823c896bc6SHong Zhang MatCopy_SeqSBAIJ, 1383d519adbfSMatthew Knepley /* 44*/ 0, 138449b5e25fSSatish Balay MatScale_SeqSBAIJ, 13857d68702bSBarry Smith MatShift_SeqSBAIJ, 138649b5e25fSSatish Balay 0, 13873bededecSBarry Smith MatZeroRowsColumns_SeqSBAIJ, 1388f73d5cc4SBarry Smith /* 49*/ 0, 138949b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 139049b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 139149b5e25fSSatish Balay 0, 139249b5e25fSSatish Balay 0, 1393d519adbfSMatthew Knepley /* 54*/ 0, 139449b5e25fSSatish Balay 0, 139549b5e25fSSatish Balay 0, 139649b5e25fSSatish Balay 0, 139749b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 1398d519adbfSMatthew Knepley /* 59*/ MatGetSubMatrix_SeqSBAIJ, 139949b5e25fSSatish Balay 0, 140049b5e25fSSatish Balay 0, 1401357abbc8SBarry Smith 0, 1402d959ec07SHong Zhang 0, 1403d519adbfSMatthew Knepley /* 64*/ 0, 1404d959ec07SHong Zhang 0, 1405d959ec07SHong Zhang 0, 1406d959ec07SHong Zhang 0, 1407d959ec07SHong Zhang 0, 1408d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqSBAIJ, 14093e0d88b5SBarry Smith 0, 14103e0d88b5SBarry Smith 0, 14113e0d88b5SBarry Smith 0, 14123e0d88b5SBarry Smith 0, 1413d519adbfSMatthew Knepley /* 74*/ 0, 14143e0d88b5SBarry Smith 0, 14153e0d88b5SBarry Smith 0, 14163e0d88b5SBarry Smith 0, 14173e0d88b5SBarry Smith 0, 1418d519adbfSMatthew Knepley /* 79*/ 0, 14193e0d88b5SBarry Smith 0, 14203e0d88b5SBarry Smith 0, 142197304618SKris Buschelman MatGetInertia_SeqSBAIJ, 14225bba2384SShri Abhyankar MatLoad_SeqSBAIJ, 1423d519adbfSMatthew Knepley /* 84*/ MatIsSymmetric_SeqSBAIJ, 1424865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1425efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1426865e5f61SKris Buschelman 0, 1427865e5f61SKris Buschelman 0, 1428d519adbfSMatthew Knepley /* 89*/ 0, 1429865e5f61SKris Buschelman 0, 1430865e5f61SKris Buschelman 0, 1431865e5f61SKris Buschelman 0, 1432865e5f61SKris Buschelman 0, 1433d519adbfSMatthew Knepley /* 94*/ 0, 1434865e5f61SKris Buschelman 0, 1435865e5f61SKris Buschelman 0, 143699cafbc1SBarry Smith 0, 143799cafbc1SBarry Smith 0, 1438d519adbfSMatthew Knepley /* 99*/ 0, 143999cafbc1SBarry Smith 0, 144099cafbc1SBarry Smith 0, 144199cafbc1SBarry Smith 0, 144299cafbc1SBarry Smith 0, 1443d519adbfSMatthew Knepley /*104*/ 0, 144499cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1445f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1446f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 14472af78befSBarry Smith MatRestoreRowUpperTriangular_SeqSBAIJ, 1448d519adbfSMatthew Knepley /*109*/ 0, 14492af78befSBarry Smith 0, 14502af78befSBarry Smith 0, 14512af78befSBarry Smith 0, 1452547795f9SHong Zhang MatMissingDiagonal_SeqSBAIJ, 1453547795f9SHong Zhang /*114*/ 0, 1454547795f9SHong Zhang 0, 1455547795f9SHong Zhang 0, 1456547795f9SHong Zhang 0, 1457547795f9SHong Zhang 0, 1458547795f9SHong Zhang /*119*/ 0, 1459547795f9SHong Zhang 0, 14602f480046SShri Abhyankar 0, 14613964eb88SJed Brown 0, 14623964eb88SJed Brown 0, 14633964eb88SJed Brown /*124*/ 0, 14643964eb88SJed Brown 0, 14653964eb88SJed Brown 0, 14663964eb88SJed Brown 0, 14673964eb88SJed Brown 0, 14683964eb88SJed Brown /*129*/ 0, 14693964eb88SJed Brown 0, 14703964eb88SJed Brown 0, 14713964eb88SJed Brown 0, 14723964eb88SJed Brown 0, 14733964eb88SJed Brown /*134*/ 0, 14743964eb88SJed Brown 0, 14753964eb88SJed Brown 0, 14763964eb88SJed Brown 0, 14773964eb88SJed Brown 0, 147846533700Sstefano_zampini /*139*/ MatSetBlockSizes_Default, 1479f9426fe0SMark Adams 0, 148059f5e6ceSHong Zhang 0, 148159f5e6ceSHong Zhang 0, 148259f5e6ceSHong Zhang 0, 148359f5e6ceSHong Zhang /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ 148499cafbc1SBarry Smith }; 1485be1d678aSKris Buschelman 14867087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 148749b5e25fSSatish Balay { 14884afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 1489d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 1490dfbe8321SBarry Smith PetscErrorCode ierr; 149149b5e25fSSatish Balay 149249b5e25fSSatish Balay PetscFunctionBegin; 1493e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 149449b5e25fSSatish Balay 149549b5e25fSSatish Balay /* allocate space for values if not already there */ 149649b5e25fSSatish Balay if (!aij->saved_values) { 1497854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr); 149849b5e25fSSatish Balay } 149949b5e25fSSatish Balay 150049b5e25fSSatish Balay /* copy values over */ 150187828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 150249b5e25fSSatish Balay PetscFunctionReturn(0); 150349b5e25fSSatish Balay } 150449b5e25fSSatish Balay 15057087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 150649b5e25fSSatish Balay { 15074afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 15086849ba73SBarry Smith PetscErrorCode ierr; 1509d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 151049b5e25fSSatish Balay 151149b5e25fSSatish Balay PetscFunctionBegin; 1512e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1513e7e72b3dSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 151449b5e25fSSatish Balay 151549b5e25fSSatish Balay /* copy values over */ 151687828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 151749b5e25fSSatish Balay PetscFunctionReturn(0); 151849b5e25fSSatish Balay } 151949b5e25fSSatish Balay 1520367daffbSBarry Smith static PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 152149b5e25fSSatish Balay { 1522c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 15236849ba73SBarry Smith PetscErrorCode ierr; 15244dcd73b1SHong Zhang PetscInt i,mbs,nbs,bs2; 15252576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE; 152649b5e25fSSatish Balay 152749b5e25fSSatish Balay PetscFunctionBegin; 15282576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 1529db4efbfdSBarry Smith 153033d57670SJed Brown ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr); 153126283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 153226283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1533e02043d6SBarry Smith ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 1534899cda47SBarry Smith 153521940c7eSstefano_zampini B->preallocated = PETSC_TRUE; 153621940c7eSstefano_zampini 1537d0f46423SBarry Smith mbs = B->rmap->N/bs; 15384dcd73b1SHong Zhang nbs = B->cmap->n/bs; 153949b5e25fSSatish Balay bs2 = bs*bs; 154049b5e25fSSatish Balay 15414dcd73b1SHong 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"); 154249b5e25fSSatish Balay 1543ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1544ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1545ab93d7beSBarry Smith nz = 0; 1546ab93d7beSBarry Smith } 1547ab93d7beSBarry Smith 1548435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 1549e32f2f54SBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 155049b5e25fSSatish Balay if (nnz) { 155149b5e25fSSatish Balay for (i=0; i<mbs; i++) { 1552e32f2f54SBarry 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]); 1553de64b629SHong 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); 155449b5e25fSSatish Balay } 155549b5e25fSSatish Balay } 155649b5e25fSSatish Balay 1557db4efbfdSBarry Smith B->ops->mult = MatMult_SeqSBAIJ_N; 1558db4efbfdSBarry Smith B->ops->multadd = MatMultAdd_SeqSBAIJ_N; 1559db4efbfdSBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_N; 1560db4efbfdSBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N; 156126fbe8dcSKarl Rupp 1562c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 156349b5e25fSSatish Balay if (!flg) { 156449b5e25fSSatish Balay switch (bs) { 156549b5e25fSSatish Balay case 1: 156649b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 156749b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1568431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1569431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 157049b5e25fSSatish Balay break; 157149b5e25fSSatish Balay case 2: 157249b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 157349b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1574431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1575431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 157649b5e25fSSatish Balay break; 157749b5e25fSSatish Balay case 3: 157849b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 157949b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1580431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1581431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 158249b5e25fSSatish Balay break; 158349b5e25fSSatish Balay case 4: 158449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 158549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1586431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1587431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 158849b5e25fSSatish Balay break; 158949b5e25fSSatish Balay case 5: 159049b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 159149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1592431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1593431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 159449b5e25fSSatish Balay break; 159549b5e25fSSatish Balay case 6: 159649b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 159749b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1598431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1599431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 160049b5e25fSSatish Balay break; 160149b5e25fSSatish Balay case 7: 1602de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 160349b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1604431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1605431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 160649b5e25fSSatish Balay break; 160749b5e25fSSatish Balay } 160849b5e25fSSatish Balay } 160949b5e25fSSatish Balay 161049b5e25fSSatish Balay b->mbs = mbs; 16114dcd73b1SHong Zhang b->nbs = nbs; 1612ab93d7beSBarry Smith if (!skipallocation) { 16132ee49352SLisandro Dalcin if (!b->imax) { 1614dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr); 161526fbe8dcSKarl Rupp 1616c760cd28SBarry Smith b->free_imax_ilen = PETSC_TRUE; 161726fbe8dcSKarl Rupp 16183bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr); 16192ee49352SLisandro Dalcin } 162049b5e25fSSatish Balay if (!nnz) { 1621435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 162249b5e25fSSatish Balay else if (nz <= 0) nz = 1; 162326fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->imax[i] = nz; 1624153ea458SHong Zhang nz = nz*mbs; /* total nz */ 162549b5e25fSSatish Balay } else { 162649b5e25fSSatish Balay nz = 0; 16278cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 162849b5e25fSSatish Balay } 16292ee49352SLisandro Dalcin /* b->ilen will count nonzeros in each block row so far. */ 163026fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->ilen[i] = 0; 16316c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 163249b5e25fSSatish Balay 163349b5e25fSSatish Balay /* allocate the matrix space */ 16342ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 1635dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr); 16363bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 16376c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 163813f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 163926fbe8dcSKarl Rupp 164049b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 164149b5e25fSSatish Balay 164249b5e25fSSatish Balay /* pointer to beginning of each row */ 1643e60cf9a0SBarry Smith b->i[0] = 0; 164426fbe8dcSKarl Rupp for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1]; 164526fbe8dcSKarl Rupp 1646e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1647e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1648e811da20SHong Zhang } else { 1649e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1650e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1651ab93d7beSBarry Smith } 165249b5e25fSSatish Balay 165349b5e25fSSatish Balay b->bs2 = bs2; 16546c6c5352SBarry Smith b->nz = 0; 1655b32cb4a7SJed Brown b->maxnz = nz; 165616cdd363SHong Zhang b->inew = 0; 165716cdd363SHong Zhang b->jnew = 0; 165816cdd363SHong Zhang b->anew = 0; 165916cdd363SHong Zhang b->a2anew = 0; 16601a3463dfSHong Zhang b->permute = PETSC_FALSE; 1661cb7b82ddSBarry Smith 1662cb7b82ddSBarry Smith B->was_assembled = PETSC_FALSE; 1663cb7b82ddSBarry Smith B->assembled = PETSC_FALSE; 16642576faa2SJed Brown if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);} 1665c464158bSHong Zhang PetscFunctionReturn(0); 1666c464158bSHong Zhang } 1667153ea458SHong Zhang 166838f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[]) 166938f409ebSLisandro Dalcin { 167038f409ebSLisandro Dalcin PetscInt i,j,m,nz,nz_max=0,*nnz; 167138f409ebSLisandro Dalcin PetscScalar *values=0; 167238f409ebSLisandro Dalcin PetscBool roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented; 167338f409ebSLisandro Dalcin PetscErrorCode ierr; 167438f409ebSLisandro Dalcin PetscFunctionBegin; 167538f409ebSLisandro Dalcin if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs); 167638f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr); 167738f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr); 167838f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 167938f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 168038f409ebSLisandro Dalcin ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 168138f409ebSLisandro Dalcin m = B->rmap->n/bs; 168238f409ebSLisandro Dalcin 168338f409ebSLisandro Dalcin if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]); 1684854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&nnz);CHKERRQ(ierr); 168538f409ebSLisandro Dalcin for (i=0; i<m; i++) { 168638f409ebSLisandro Dalcin nz = ii[i+1] - ii[i]; 168738f409ebSLisandro Dalcin if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz); 168838f409ebSLisandro Dalcin nz_max = PetscMax(nz_max,nz); 168938f409ebSLisandro Dalcin nnz[i] = nz; 169038f409ebSLisandro Dalcin } 169138f409ebSLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 169238f409ebSLisandro Dalcin ierr = PetscFree(nnz);CHKERRQ(ierr); 169338f409ebSLisandro Dalcin 169438f409ebSLisandro Dalcin values = (PetscScalar*)V; 169538f409ebSLisandro Dalcin if (!values) { 16961795a4d1SJed Brown ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr); 169738f409ebSLisandro Dalcin } 169838f409ebSLisandro Dalcin for (i=0; i<m; i++) { 169938f409ebSLisandro Dalcin PetscInt ncols = ii[i+1] - ii[i]; 170038f409ebSLisandro Dalcin const PetscInt *icols = jj + ii[i]; 170138f409ebSLisandro Dalcin if (!roworiented || bs == 1) { 170238f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0); 170338f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr); 170438f409ebSLisandro Dalcin } else { 170538f409ebSLisandro Dalcin for (j=0; j<ncols; j++) { 170638f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0); 170738f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr); 170838f409ebSLisandro Dalcin } 170938f409ebSLisandro Dalcin } 171038f409ebSLisandro Dalcin } 171138f409ebSLisandro Dalcin if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); } 171238f409ebSLisandro Dalcin ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 171338f409ebSLisandro Dalcin ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 171438f409ebSLisandro Dalcin ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 171538f409ebSLisandro Dalcin PetscFunctionReturn(0); 171638f409ebSLisandro Dalcin } 171738f409ebSLisandro Dalcin 1718db4efbfdSBarry Smith /* 1719db4efbfdSBarry Smith This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization 1720db4efbfdSBarry Smith */ 1721ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural) 1722db4efbfdSBarry Smith { 1723db4efbfdSBarry Smith PetscErrorCode ierr; 1724ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 1725db4efbfdSBarry Smith PetscInt bs = B->rmap->bs; 1726db4efbfdSBarry Smith 1727db4efbfdSBarry Smith PetscFunctionBegin; 1728c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 1729db4efbfdSBarry Smith if (flg) bs = 8; 1730db4efbfdSBarry Smith 1731db4efbfdSBarry Smith if (!natural) { 1732db4efbfdSBarry Smith switch (bs) { 1733db4efbfdSBarry Smith case 1: 1734d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace; 1735db4efbfdSBarry Smith break; 1736db4efbfdSBarry Smith case 2: 1737db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 1738db4efbfdSBarry Smith break; 1739db4efbfdSBarry Smith case 3: 1740db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 1741db4efbfdSBarry Smith break; 1742db4efbfdSBarry Smith case 4: 1743db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 1744db4efbfdSBarry Smith break; 1745db4efbfdSBarry Smith case 5: 1746db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 1747db4efbfdSBarry Smith break; 1748db4efbfdSBarry Smith case 6: 1749db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 1750db4efbfdSBarry Smith break; 1751db4efbfdSBarry Smith case 7: 1752db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 1753db4efbfdSBarry Smith break; 1754db4efbfdSBarry Smith default: 1755db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N; 1756db4efbfdSBarry Smith break; 1757db4efbfdSBarry Smith } 1758db4efbfdSBarry Smith } else { 1759db4efbfdSBarry Smith switch (bs) { 1760db4efbfdSBarry Smith case 1: 1761d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace; 1762db4efbfdSBarry Smith break; 1763db4efbfdSBarry Smith case 2: 1764db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 1765db4efbfdSBarry Smith break; 1766db4efbfdSBarry Smith case 3: 1767db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 1768db4efbfdSBarry Smith break; 1769db4efbfdSBarry Smith case 4: 1770db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 1771db4efbfdSBarry Smith break; 1772db4efbfdSBarry Smith case 5: 1773db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 1774db4efbfdSBarry Smith break; 1775db4efbfdSBarry Smith case 6: 1776db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 1777db4efbfdSBarry Smith break; 1778db4efbfdSBarry Smith case 7: 1779db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 1780db4efbfdSBarry Smith break; 1781db4efbfdSBarry Smith default: 1782db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1783db4efbfdSBarry Smith break; 1784db4efbfdSBarry Smith } 1785db4efbfdSBarry Smith } 1786db4efbfdSBarry Smith PetscFunctionReturn(0); 1787db4efbfdSBarry Smith } 1788db4efbfdSBarry Smith 1789cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*); 1790cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*); 1791d769727bSBarry Smith 1792cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B) 17935c9eb25fSBarry Smith { 1794d0f46423SBarry Smith PetscInt n = A->rmap->n; 17955c9eb25fSBarry Smith PetscErrorCode ierr; 17965c9eb25fSBarry Smith 17975c9eb25fSBarry Smith PetscFunctionBegin; 17980e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX) 17990e92d65fSHong Zhang if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 18000e92d65fSHong Zhang #endif 1801ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 18025c9eb25fSBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 18035c9eb25fSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 18045c9eb25fSBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 18050298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 180626fbe8dcSKarl Rupp 18077b056e98SHong Zhang (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ; 1808c6d0d4f0SHong Zhang (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqSBAIJ; 1809e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 181000c67f3bSHong Zhang 1811d5f3da31SBarry Smith (*B)->factortype = ftype; 181200c67f3bSHong Zhang ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr); 181300c67f3bSHong Zhang ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr); 18145c9eb25fSBarry Smith PetscFunctionReturn(0); 18155c9eb25fSBarry Smith } 18165c9eb25fSBarry Smith 18178397e458SBarry Smith /*@C 18188397e458SBarry Smith MatSeqSBAIJGetArray - gives access to the array where the data for a MATSEQSBAIJ matrix is stored 18198397e458SBarry Smith 18208397e458SBarry Smith Not Collective 18218397e458SBarry Smith 18228397e458SBarry Smith Input Parameter: 18238397e458SBarry Smith . mat - a MATSEQSBAIJ matrix 18248397e458SBarry Smith 18258397e458SBarry Smith Output Parameter: 18268397e458SBarry Smith . array - pointer to the data 18278397e458SBarry Smith 18288397e458SBarry Smith Level: intermediate 18298397e458SBarry Smith 18308397e458SBarry Smith .seealso: MatSeqSBAIJRestoreArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray() 18318397e458SBarry Smith @*/ 18328397e458SBarry Smith PetscErrorCode MatSeqSBAIJGetArray(Mat A,PetscScalar **array) 18338397e458SBarry Smith { 18348397e458SBarry Smith PetscErrorCode ierr; 18358397e458SBarry Smith 18368397e458SBarry Smith PetscFunctionBegin; 18378397e458SBarry Smith ierr = PetscUseMethod(A,"MatSeqSBAIJGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 18388397e458SBarry Smith PetscFunctionReturn(0); 18398397e458SBarry Smith } 18408397e458SBarry Smith 18418397e458SBarry Smith /*@C 18428397e458SBarry Smith MatSeqSBAIJRestoreArray - returns access to the array where the data for a MATSEQSBAIJ matrix is stored obtained by MatSeqSBAIJGetArray() 18438397e458SBarry Smith 18448397e458SBarry Smith Not Collective 18458397e458SBarry Smith 18468397e458SBarry Smith Input Parameters: 18478397e458SBarry Smith . mat - a MATSEQSBAIJ matrix 18488397e458SBarry Smith . array - pointer to the data 18498397e458SBarry Smith 18508397e458SBarry Smith Level: intermediate 18518397e458SBarry Smith 18528397e458SBarry Smith .seealso: MatSeqSBAIJGetArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray() 18538397e458SBarry Smith @*/ 18548397e458SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray(Mat A,PetscScalar **array) 18558397e458SBarry Smith { 18568397e458SBarry Smith PetscErrorCode ierr; 18578397e458SBarry Smith 18588397e458SBarry Smith PetscFunctionBegin; 18598397e458SBarry Smith ierr = PetscUseMethod(A,"MatSeqSBAIJRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 18608397e458SBarry Smith PetscFunctionReturn(0); 18618397e458SBarry Smith } 18628397e458SBarry Smith 18630bad9183SKris Buschelman /*MC 1864fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 18650bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 18660bad9183SKris Buschelman 1867828413b8SBarry Smith For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you 186871dad5bbSBarry Smith can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd() 1869828413b8SBarry Smith 18700bad9183SKris Buschelman Options Database Keys: 18710bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 18720bad9183SKris Buschelman 187371dad5bbSBarry Smith Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not 187471dad5bbSBarry 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 187571dad5bbSBarry 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. 187671dad5bbSBarry Smith 187771dad5bbSBarry Smith 18780bad9183SKris Buschelman Level: beginner 18790bad9183SKris Buschelman 18800bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 18810bad9183SKris Buschelman M*/ 18820bad9183SKris Buschelman 1883cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*); 1884aa5a9175SDahai Guo 18858cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1886a23d5eceSKris Buschelman { 1887a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1888dfbe8321SBarry Smith PetscErrorCode ierr; 188913f74950SBarry Smith PetscMPIInt size; 1890ace3abfcSBarry Smith PetscBool no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE; 1891a23d5eceSKris Buschelman 1892a23d5eceSKris Buschelman PetscFunctionBegin; 1893ce94432eSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 1894e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1895a23d5eceSKris Buschelman 1896b00a9115SJed Brown ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1897a23d5eceSKris Buschelman B->data = (void*)b; 1898a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 189926fbe8dcSKarl Rupp 1900a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1901a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1902a23d5eceSKris Buschelman b->row = 0; 1903a23d5eceSKris Buschelman b->icol = 0; 1904a23d5eceSKris Buschelman b->reallocs = 0; 1905a23d5eceSKris Buschelman b->saved_values = 0; 19060def2e27SBarry Smith b->inode.limit = 5; 19070def2e27SBarry Smith b->inode.max_limit = 5; 1908a23d5eceSKris Buschelman 1909a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1910a23d5eceSKris Buschelman b->nonew = 0; 1911a23d5eceSKris Buschelman b->diag = 0; 1912a23d5eceSKris Buschelman b->solve_work = 0; 1913a23d5eceSKris Buschelman b->mult_work = 0; 1914a23d5eceSKris Buschelman B->spptr = 0; 1915f2cbd3d5SJed Brown B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2; 1916a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 1917a23d5eceSKris Buschelman 1918a23d5eceSKris Buschelman b->inew = 0; 1919a23d5eceSKris Buschelman b->jnew = 0; 1920a23d5eceSKris Buschelman b->anew = 0; 1921a23d5eceSKris Buschelman b->a2anew = 0; 1922a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1923a23d5eceSKris Buschelman 192471dad5bbSBarry Smith b->ignore_ltriangular = PETSC_TRUE; 192526fbe8dcSKarl Rupp 1926c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr); 1927941593c8SHong Zhang 1928f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 192926fbe8dcSKarl Rupp 1930c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr); 1931f5edf698SHong Zhang 19328397e458SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJGetArray_C",MatSeqSBAIJGetArray_SeqSBAIJ);CHKERRQ(ierr); 19338397e458SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJRestoreArray_C",MatSeqSBAIJRestoreArray_SeqSBAIJ);CHKERRQ(ierr); 1934bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1935bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1936bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 1937bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1938bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1939bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 194038f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr); 19416214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 19426214f412SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_elemental_C",MatConvert_SeqSBAIJ_Elemental);CHKERRQ(ierr); 19436214f412SHong Zhang #endif 194423ce1328SBarry Smith 194523ce1328SBarry Smith B->symmetric = PETSC_TRUE; 194623ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 194723ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 194823ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 194926fbe8dcSKarl Rupp 195017667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 19510def2e27SBarry Smith 1952ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr); 19530298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr); 195426fbe8dcSKarl Rupp if (no_unroll) { 195526fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr); 195626fbe8dcSKarl Rupp } 19570298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr); 195826fbe8dcSKarl Rupp if (no_inode) { 195926fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr); 196026fbe8dcSKarl Rupp } 19610298fd71SBarry Smith ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr); 19620def2e27SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 1963ace3abfcSBarry Smith b->inode.use = (PetscBool)(!(no_unroll || no_inode)); 19640def2e27SBarry Smith if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit; 1965a23d5eceSKris Buschelman PetscFunctionReturn(0); 1966a23d5eceSKris Buschelman } 1967a23d5eceSKris Buschelman 1968a23d5eceSKris Buschelman /*@C 1969a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 1970a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 1971a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 1972a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 1973a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 1974a23d5eceSKris Buschelman 1975a23d5eceSKris Buschelman Collective on Mat 1976a23d5eceSKris Buschelman 1977a23d5eceSKris Buschelman Input Parameters: 19781c4f3114SJed Brown + B - the symmetric matrix 1979bb7ae925SBarry 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 1980bb7ae925SBarry Smith blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs() 1981a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 1982a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 19830298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 1984a23d5eceSKris Buschelman 1985a23d5eceSKris Buschelman Options Database Keys: 1986a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 1987a23d5eceSKris Buschelman block calculations (much slower) 1988db4efbfdSBarry Smith . -mat_block_size - size of the blocks to use (only works if a negative bs is passed in 1989a23d5eceSKris Buschelman 1990a23d5eceSKris Buschelman Level: intermediate 1991a23d5eceSKris Buschelman 1992a23d5eceSKris Buschelman Notes: 1993a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 19940298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 1995a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 1996a23d5eceSKris Buschelman 1997aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 1998aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 1999aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 2000aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 2001aa95bbe8SBarry Smith 200249a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 200349a6f317SBarry Smith 200449a6f317SBarry Smith 200569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2006a23d5eceSKris Buschelman @*/ 20077087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 200813f74950SBarry Smith { 20094ac538c5SBarry Smith PetscErrorCode ierr; 2010a23d5eceSKris Buschelman 2011a23d5eceSKris Buschelman PetscFunctionBegin; 20126ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 20136ba663aaSJed Brown PetscValidType(B,1); 20146ba663aaSJed Brown PetscValidLogicalCollectiveInt(B,bs,2); 20154ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr); 2016a23d5eceSKris Buschelman PetscFunctionReturn(0); 2017a23d5eceSKris Buschelman } 201849b5e25fSSatish Balay 201938f409ebSLisandro Dalcin /*@C 202038f409ebSLisandro Dalcin MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format. 202138f409ebSLisandro Dalcin 202238f409ebSLisandro Dalcin Input Parameters: 20231c4f3114SJed Brown + B - the matrix 2024eab78319SHong Zhang . bs - size of block, the blocks are ALWAYS square. 202538f409ebSLisandro Dalcin . i - the indices into j for the start of each local row (starts with zero) 202638f409ebSLisandro Dalcin . j - the column indices for each local row (starts with zero) these must be sorted for each row 202738f409ebSLisandro Dalcin - v - optional values in the matrix 202838f409ebSLisandro Dalcin 202938f409ebSLisandro Dalcin Level: developer 203038f409ebSLisandro Dalcin 203138f409ebSLisandro Dalcin Notes: 203238f409ebSLisandro Dalcin The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED. For example, C programs 203338f409ebSLisandro 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 203438f409ebSLisandro Dalcin over rows within a block and the last index is over columns within a block row. Fortran programs will likely set 203538f409ebSLisandro 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 203638f409ebSLisandro Dalcin block column and the second index is over columns within a block. 203738f409ebSLisandro Dalcin 203838f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse 203938f409ebSLisandro Dalcin 204038f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ 204138f409ebSLisandro Dalcin @*/ 204238f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[]) 204338f409ebSLisandro Dalcin { 204438f409ebSLisandro Dalcin PetscErrorCode ierr; 204538f409ebSLisandro Dalcin 204638f409ebSLisandro Dalcin PetscFunctionBegin; 204738f409ebSLisandro Dalcin PetscValidHeaderSpecific(B,MAT_CLASSID,1); 204838f409ebSLisandro Dalcin PetscValidType(B,1); 204938f409ebSLisandro Dalcin PetscValidLogicalCollectiveInt(B,bs,2); 205038f409ebSLisandro Dalcin ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr); 205138f409ebSLisandro Dalcin PetscFunctionReturn(0); 205238f409ebSLisandro Dalcin } 205338f409ebSLisandro Dalcin 2054c464158bSHong Zhang /*@C 2055c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 2056c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 2057c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 2058c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 2059c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 206049b5e25fSSatish Balay 2061c464158bSHong Zhang Collective on MPI_Comm 2062c464158bSHong Zhang 2063c464158bSHong Zhang Input Parameters: 2064c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2065bb7ae925SBarry 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 2066bb7ae925SBarry Smith blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs() 2067c464158bSHong Zhang . m - number of rows, or number of columns 2068c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 2069744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 20700298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2071c464158bSHong Zhang 2072c464158bSHong Zhang Output Parameter: 2073c464158bSHong Zhang . A - the symmetric matrix 2074c464158bSHong Zhang 2075c464158bSHong Zhang Options Database Keys: 2076c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 2077c464158bSHong Zhang block calculations (much slower) 2078c464158bSHong Zhang . -mat_block_size - size of the blocks to use 2079c464158bSHong Zhang 2080c464158bSHong Zhang Level: intermediate 2081c464158bSHong Zhang 2082175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2083ae1d86c5SBarry Smith MatXXXXSetPreallocation() paradgm instead of this routine directly. 2084175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 2085175b88e8SBarry Smith 2086c464158bSHong Zhang Notes: 20876d6d819aSHong Zhang The number of rows and columns must be divisible by blocksize. 20886d6d819aSHong Zhang This matrix type does not support complex Hermitian operation. 2089c464158bSHong Zhang 2090c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 20910298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 2092a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 2093c464158bSHong Zhang 209449a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 209549a6f317SBarry Smith 209669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2097c464158bSHong Zhang @*/ 20987087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 2099c464158bSHong Zhang { 2100dfbe8321SBarry Smith PetscErrorCode ierr; 2101c464158bSHong Zhang 2102c464158bSHong Zhang PetscFunctionBegin; 2103f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 2104f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2105c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 2106367daffbSBarry Smith ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 210749b5e25fSSatish Balay PetscFunctionReturn(0); 210849b5e25fSSatish Balay } 210949b5e25fSSatish Balay 2110dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 211149b5e25fSSatish Balay { 211249b5e25fSSatish Balay Mat C; 211349b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 21146849ba73SBarry Smith PetscErrorCode ierr; 2115b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 211649b5e25fSSatish Balay 211749b5e25fSSatish Balay PetscFunctionBegin; 2118e32f2f54SBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix"); 211949b5e25fSSatish Balay 212049b5e25fSSatish Balay *B = 0; 2121ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2122d0f46423SBarry Smith ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr); 21238e9a0fb8SHong Zhang ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr); 21241d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 2125692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 2126692f9cbeSHong Zhang 2127273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 2128d5f3da31SBarry Smith C->factortype = A->factortype; 212949b5e25fSSatish Balay c->row = 0; 213049b5e25fSSatish Balay c->icol = 0; 213149b5e25fSSatish Balay c->saved_values = 0; 2132a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 213349b5e25fSSatish Balay C->assembled = PETSC_TRUE; 213449b5e25fSSatish Balay 21351e1e43feSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 21361e1e43feSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 213749b5e25fSSatish Balay c->bs2 = a->bs2; 213849b5e25fSSatish Balay c->mbs = a->mbs; 213949b5e25fSSatish Balay c->nbs = a->nbs; 214049b5e25fSSatish Balay 2141c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2142c760cd28SBarry Smith c->imax = a->imax; 2143c760cd28SBarry Smith c->ilen = a->ilen; 2144c760cd28SBarry Smith c->free_imax_ilen = PETSC_FALSE; 2145c760cd28SBarry Smith } else { 2146dcca6d9dSJed Brown ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr); 21473bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 214849b5e25fSSatish Balay for (i=0; i<mbs; i++) { 214949b5e25fSSatish Balay c->imax[i] = a->imax[i]; 215049b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 215149b5e25fSSatish Balay } 2152c760cd28SBarry Smith c->free_imax_ilen = PETSC_TRUE; 2153c760cd28SBarry Smith } 215449b5e25fSSatish Balay 215549b5e25fSSatish Balay /* allocate the matrix space */ 21564da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2157785e854fSJed Brown ierr = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr); 21583bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 215944e1c64aSLisandro Dalcin c->i = a->i; 216044e1c64aSLisandro Dalcin c->j = a->j; 21614da8f245SBarry Smith c->singlemalloc = PETSC_FALSE; 216244e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 21634da8f245SBarry Smith c->free_ij = PETSC_FALSE; 21644da8f245SBarry Smith c->parent = A; 21654da8f245SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 21664da8f245SBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21674da8f245SBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21684da8f245SBarry Smith } else { 2169dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr); 217013f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 21713bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 21724da8f245SBarry Smith c->singlemalloc = PETSC_TRUE; 217344e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 21744da8f245SBarry Smith c->free_ij = PETSC_TRUE; 21754da8f245SBarry Smith } 217649b5e25fSSatish Balay if (mbs > 0) { 21774da8f245SBarry Smith if (cpvalues != MAT_SHARE_NONZERO_PATTERN) { 217813f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 21794da8f245SBarry Smith } 218049b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 218149b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 218249b5e25fSSatish Balay } else { 218349b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 218449b5e25fSSatish Balay } 2185a1c3900fSBarry Smith if (a->jshort) { 218644e1c64aSLisandro Dalcin /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */ 218744e1c64aSLisandro Dalcin /* if the parent matrix is reassembled, this child matrix will never notice */ 2188785e854fSJed Brown ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr); 21893bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr); 2190a1c3900fSBarry Smith ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr); 219126fbe8dcSKarl Rupp 21924da8f245SBarry Smith c->free_jshort = PETSC_TRUE; 21934da8f245SBarry Smith } 2194a1c3900fSBarry Smith } 219549b5e25fSSatish Balay 219649b5e25fSSatish Balay c->roworiented = a->roworiented; 219749b5e25fSSatish Balay c->nonew = a->nonew; 219849b5e25fSSatish Balay 219949b5e25fSSatish Balay if (a->diag) { 2200c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2201c760cd28SBarry Smith c->diag = a->diag; 2202c760cd28SBarry Smith c->free_diag = PETSC_FALSE; 2203c760cd28SBarry Smith } else { 2204785e854fSJed Brown ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr); 22053bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr); 220626fbe8dcSKarl Rupp for (i=0; i<mbs; i++) c->diag[i] = a->diag[i]; 2207c760cd28SBarry Smith c->free_diag = PETSC_TRUE; 2208c760cd28SBarry Smith } 220944e1c64aSLisandro Dalcin } 22106c6c5352SBarry Smith c->nz = a->nz; 2211f2cbd3d5SJed Brown c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 221249b5e25fSSatish Balay c->solve_work = 0; 221349b5e25fSSatish Balay c->mult_work = 0; 221426fbe8dcSKarl Rupp 221549b5e25fSSatish Balay *B = C; 2216140e18c1SBarry Smith ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 221749b5e25fSSatish Balay PetscFunctionReturn(0); 221849b5e25fSSatish Balay } 221949b5e25fSSatish Balay 2220112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer) 22212f480046SShri Abhyankar { 22222f480046SShri Abhyankar Mat_SeqSBAIJ *a; 22232f480046SShri Abhyankar PetscErrorCode ierr; 22242f480046SShri Abhyankar int fd; 22252f480046SShri Abhyankar PetscMPIInt size; 22263059b6faSBarry Smith PetscInt i,nz,header[4],*rowlengths=0,M,N,bs = newmat->rmap->bs; 22272f480046SShri Abhyankar PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 22282f480046SShri Abhyankar PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols; 22292f480046SShri Abhyankar PetscInt *masked,nmask,tmp,bs2,ishift; 22302f480046SShri Abhyankar PetscScalar *aa; 2231ce94432eSBarry Smith MPI_Comm comm; 22322f480046SShri Abhyankar 22332f480046SShri Abhyankar PetscFunctionBegin; 2234c98fd787SBarry Smith /* force binary viewer to load .info file if it has not yet done so */ 2235c98fd787SBarry Smith ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr); 2236ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 2237c5929fdfSBarry Smith ierr = PetscOptionsGetInt(((PetscObject)newmat)->options,((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr); 22383059b6faSBarry Smith if (bs < 0) bs = 1; 22392f480046SShri Abhyankar bs2 = bs*bs; 22402f480046SShri Abhyankar 22412f480046SShri Abhyankar ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 22422f480046SShri Abhyankar if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor"); 22432f480046SShri Abhyankar ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 22442f480046SShri Abhyankar ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 22452f480046SShri Abhyankar if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 22462f480046SShri Abhyankar M = header[1]; N = header[2]; nz = header[3]; 22472f480046SShri Abhyankar 22482f480046SShri Abhyankar if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 22492f480046SShri Abhyankar 22502f480046SShri Abhyankar if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices"); 22512f480046SShri Abhyankar 22522f480046SShri Abhyankar /* 22532f480046SShri Abhyankar This code adds extra rows to make sure the number of rows is 22542f480046SShri Abhyankar divisible by the blocksize 22552f480046SShri Abhyankar */ 22562f480046SShri Abhyankar mbs = M/bs; 22572f480046SShri Abhyankar extra_rows = bs - M + bs*(mbs); 22582f480046SShri Abhyankar if (extra_rows == bs) extra_rows = 0; 22592f480046SShri Abhyankar else mbs++; 22602f480046SShri Abhyankar if (extra_rows) { 22612f480046SShri Abhyankar ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr); 22622f480046SShri Abhyankar } 22632f480046SShri Abhyankar 22642f480046SShri Abhyankar /* Set global sizes if not already set */ 22652f480046SShri Abhyankar if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) { 22662f480046SShri Abhyankar ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr); 22672f480046SShri Abhyankar } else { /* Check if the matrix global sizes are correct */ 22682f480046SShri Abhyankar ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr); 22692f480046SShri Abhyankar if (M != rows || N != cols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix in file of different length (%d, %d) than the input matrix (%d, %d)",M,N,rows,cols); 22702f480046SShri Abhyankar } 22712f480046SShri Abhyankar 22722f480046SShri Abhyankar /* read in row lengths */ 2273854ce69bSBarry Smith ierr = PetscMalloc1(M+extra_rows,&rowlengths);CHKERRQ(ierr); 22742f480046SShri Abhyankar ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 22752f480046SShri Abhyankar for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 22762f480046SShri Abhyankar 22772f480046SShri Abhyankar /* read in column indices */ 2278854ce69bSBarry Smith ierr = PetscMalloc1(nz+extra_rows,&jj);CHKERRQ(ierr); 22792f480046SShri Abhyankar ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 22802f480046SShri Abhyankar for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 22812f480046SShri Abhyankar 22822f480046SShri Abhyankar /* loop over row lengths determining block row lengths */ 22831795a4d1SJed Brown ierr = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr); 2284dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr); 22852f480046SShri Abhyankar ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 22862f480046SShri Abhyankar rowcount = 0; 22872f480046SShri Abhyankar nzcount = 0; 22882f480046SShri Abhyankar for (i=0; i<mbs; i++) { 22892f480046SShri Abhyankar nmask = 0; 22902f480046SShri Abhyankar for (j=0; j<bs; j++) { 22912f480046SShri Abhyankar kmax = rowlengths[rowcount]; 22922f480046SShri Abhyankar for (k=0; k<kmax; k++) { 22932f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 22942f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 22952f480046SShri Abhyankar } 22962f480046SShri Abhyankar rowcount++; 22972f480046SShri Abhyankar } 22982f480046SShri Abhyankar s_browlengths[i] += nmask; 22992f480046SShri Abhyankar 23002f480046SShri Abhyankar /* zero out the mask elements we set */ 23012f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23022f480046SShri Abhyankar } 23032f480046SShri Abhyankar 23042f480046SShri Abhyankar /* Do preallocation */ 2305367daffbSBarry Smith ierr = MatSeqSBAIJSetPreallocation(newmat,bs,0,s_browlengths);CHKERRQ(ierr); 23062f480046SShri Abhyankar a = (Mat_SeqSBAIJ*)newmat->data; 23072f480046SShri Abhyankar 23082f480046SShri Abhyankar /* set matrix "i" values */ 23092f480046SShri Abhyankar a->i[0] = 0; 23102f480046SShri Abhyankar for (i=1; i<= mbs; i++) { 23112f480046SShri Abhyankar a->i[i] = a->i[i-1] + s_browlengths[i-1]; 23122f480046SShri Abhyankar a->ilen[i-1] = s_browlengths[i-1]; 23132f480046SShri Abhyankar } 23142f480046SShri Abhyankar a->nz = a->i[mbs]; 23152f480046SShri Abhyankar 23162f480046SShri Abhyankar /* read in nonzero values */ 2317854ce69bSBarry Smith ierr = PetscMalloc1(nz+extra_rows,&aa);CHKERRQ(ierr); 23182f480046SShri Abhyankar ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 23192f480046SShri Abhyankar for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 23202f480046SShri Abhyankar 23212f480046SShri Abhyankar /* set "a" and "j" values into matrix */ 23222f480046SShri Abhyankar nzcount = 0; jcount = 0; 23232f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23242f480046SShri Abhyankar nzcountb = nzcount; 23252f480046SShri Abhyankar nmask = 0; 23262f480046SShri Abhyankar for (j=0; j<bs; j++) { 23272f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23282f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23292f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23302f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 23312f480046SShri Abhyankar } 23322f480046SShri Abhyankar } 23332f480046SShri Abhyankar /* sort the masked values */ 23342f480046SShri Abhyankar ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 23352f480046SShri Abhyankar 23362f480046SShri Abhyankar /* set "j" values into matrix */ 23372f480046SShri Abhyankar maskcount = 1; 23382f480046SShri Abhyankar for (j=0; j<nmask; j++) { 23392f480046SShri Abhyankar a->j[jcount++] = masked[j]; 23402f480046SShri Abhyankar mask[masked[j]] = maskcount++; 23412f480046SShri Abhyankar } 23422f480046SShri Abhyankar 23432f480046SShri Abhyankar /* set "a" values into matrix */ 23442f480046SShri Abhyankar ishift = bs2*a->i[i]; 23452f480046SShri Abhyankar for (j=0; j<bs; j++) { 23462f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23472f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23482f480046SShri Abhyankar tmp = jj[nzcountb]/bs; /* block col. index */ 23492f480046SShri Abhyankar if (tmp >= i) { 23502f480046SShri Abhyankar block = mask[tmp] - 1; 23512f480046SShri Abhyankar point = jj[nzcountb] - bs*tmp; 23522f480046SShri Abhyankar idx = ishift + bs2*block + j + bs*point; 23532f480046SShri Abhyankar a->a[idx] = aa[nzcountb]; 23542f480046SShri Abhyankar } 23552f480046SShri Abhyankar nzcountb++; 23562f480046SShri Abhyankar } 23572f480046SShri Abhyankar } 23582f480046SShri Abhyankar /* zero out the mask elements we set */ 23592f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23602f480046SShri Abhyankar } 23612f480046SShri Abhyankar if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 23622f480046SShri Abhyankar 23632f480046SShri Abhyankar ierr = PetscFree(rowlengths);CHKERRQ(ierr); 23642f480046SShri Abhyankar ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 23652f480046SShri Abhyankar ierr = PetscFree(aa);CHKERRQ(ierr); 23662f480046SShri Abhyankar ierr = PetscFree(jj);CHKERRQ(ierr); 23672f480046SShri Abhyankar ierr = PetscFree2(mask,masked);CHKERRQ(ierr); 23682f480046SShri Abhyankar 23692f480046SShri Abhyankar ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23702f480046SShri Abhyankar ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23712f480046SShri Abhyankar PetscFunctionReturn(0); 23722f480046SShri Abhyankar } 23732f480046SShri Abhyankar 2374c75a6043SHong Zhang /*@ 2375c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2376c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2377c75a6043SHong Zhang 2378c75a6043SHong Zhang Collective on MPI_Comm 2379c75a6043SHong Zhang 2380c75a6043SHong Zhang Input Parameters: 2381c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2382c75a6043SHong Zhang . bs - size of block 2383c75a6043SHong Zhang . m - number of rows 2384c75a6043SHong Zhang . n - number of columns 2385c75a6043SHong Zhang . i - row indices 2386c75a6043SHong Zhang . j - column indices 2387c75a6043SHong Zhang - a - matrix values 2388c75a6043SHong Zhang 2389c75a6043SHong Zhang Output Parameter: 2390c75a6043SHong Zhang . mat - the matrix 2391c75a6043SHong Zhang 2392dfb205c3SBarry Smith Level: advanced 2393c75a6043SHong Zhang 2394c75a6043SHong Zhang Notes: 2395c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2396c75a6043SHong Zhang once the matrix is destroyed 2397c75a6043SHong Zhang 2398c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2399c75a6043SHong Zhang 2400c75a6043SHong Zhang The i and j indices are 0 based 2401c75a6043SHong Zhang 2402dfb205c3SBarry 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 2403dfb205c3SBarry Smith it is the regular CSR format excluding the lower triangular elements. 2404dfb205c3SBarry Smith 240569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ() 2406c75a6043SHong Zhang 2407c75a6043SHong Zhang @*/ 2408*c3c607ccSBarry Smith PetscErrorCode MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat) 2409c75a6043SHong Zhang { 2410c75a6043SHong Zhang PetscErrorCode ierr; 2411c75a6043SHong Zhang PetscInt ii; 2412c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2413c75a6043SHong Zhang 2414c75a6043SHong Zhang PetscFunctionBegin; 2415e32f2f54SBarry Smith if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 241641096f02SStefano Zampini if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2417c75a6043SHong Zhang 2418c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2419c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2420c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2421367daffbSBarry Smith ierr = MatSeqSBAIJSetPreallocation(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 2422c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2423dcca6d9dSJed Brown ierr = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr); 24243bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr); 2425c75a6043SHong Zhang 2426c75a6043SHong Zhang sbaij->i = i; 2427c75a6043SHong Zhang sbaij->j = j; 2428c75a6043SHong Zhang sbaij->a = a; 242926fbe8dcSKarl Rupp 2430c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2431c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2432e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2433e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2434c75a6043SHong Zhang 2435c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2436c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2437c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2438e32f2f54SBarry Smith if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %d length = %d",ii,i[ii+1] - i[ii]); 2439c75a6043SHong Zhang #endif 2440c75a6043SHong Zhang } 2441c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2442c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2443e32f2f54SBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2444e32f2f54SBarry 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]); 2445c75a6043SHong Zhang } 2446c75a6043SHong Zhang #endif 2447c75a6043SHong Zhang 2448c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2449c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2450c75a6043SHong Zhang PetscFunctionReturn(0); 2451c75a6043SHong Zhang } 2452d06b337dSHong Zhang 245359f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat) 245459f5e6ceSHong Zhang { 245559f5e6ceSHong Zhang PetscErrorCode ierr; 245659f5e6ceSHong Zhang 245759f5e6ceSHong Zhang PetscFunctionBegin; 245859f5e6ceSHong Zhang ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr); 245959f5e6ceSHong Zhang PetscFunctionReturn(0); 246059f5e6ceSHong Zhang } 2461d06b337dSHong Zhang 2462d06b337dSHong Zhang 246349b5e25fSSatish Balay 246449b5e25fSSatish Balay 2465