149b5e25fSSatish Balay 249b5e25fSSatish Balay /* 3a1373b80SHong Zhang Defines the basic matrix operations for the SBAIJ (compressed row) 449b5e25fSSatish Balay matrix storage format. 549b5e25fSSatish Balay */ 6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h> /*I "petscmat.h" I*/ 7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h> 8c6db04a5SJed Brown #include <petscblaslapack.h> 949b5e25fSSatish Balay 10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h> 1170dcbbb9SBarry Smith #define USESHORT 12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h> 1370dcbbb9SBarry Smith 14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool); 15b5b17502SBarry Smith 1649b5e25fSSatish Balay /* 1749b5e25fSSatish Balay Checks for missing diagonals 1849b5e25fSSatish Balay */ 194a2ae208SSatish Balay #undef __FUNCT__ 204a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ" 21ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool *missing,PetscInt *dd) 2249b5e25fSSatish Balay { 23045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 246849ba73SBarry Smith PetscErrorCode ierr; 257734d3b5SMatthew G. Knepley PetscInt *diag,*ii = a->i,i; 2649b5e25fSSatish Balay 2749b5e25fSSatish Balay PetscFunctionBegin; 28045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 292af78befSBarry Smith *missing = PETSC_FALSE; 307734d3b5SMatthew G. Knepley if (A->rmap->n > 0 && !ii) { 31358d2f5dSShri Abhyankar *missing = PETSC_TRUE; 32358d2f5dSShri Abhyankar if (dd) *dd = 0; 33358d2f5dSShri Abhyankar PetscInfo(A,"Matrix has no entries therefore is missing diagonal"); 34358d2f5dSShri Abhyankar } else { 35358d2f5dSShri Abhyankar diag = a->diag; 3649b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 377734d3b5SMatthew G. Knepley if (diag[i] >= ii[i+1]) { 382af78befSBarry Smith *missing = PETSC_TRUE; 392af78befSBarry Smith if (dd) *dd = i; 402af78befSBarry Smith break; 412af78befSBarry Smith } 4249b5e25fSSatish Balay } 43358d2f5dSShri Abhyankar } 4449b5e25fSSatish Balay PetscFunctionReturn(0); 4549b5e25fSSatish Balay } 4649b5e25fSSatish Balay 474a2ae208SSatish Balay #undef __FUNCT__ 484a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ" 49dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 5049b5e25fSSatish Balay { 51045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 526849ba73SBarry Smith PetscErrorCode ierr; 5348dd3d27SHong Zhang PetscInt i,j; 5449b5e25fSSatish Balay 5549b5e25fSSatish Balay PetscFunctionBegin; 5609f38230SBarry Smith if (!a->diag) { 57785e854fSJed Brown ierr = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr); 583bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 59c760cd28SBarry Smith a->free_diag = PETSC_TRUE; 6009f38230SBarry Smith } 6148dd3d27SHong Zhang for (i=0; i<a->mbs; i++) { 6248dd3d27SHong Zhang a->diag[i] = a->i[i+1]; 6348dd3d27SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 6448dd3d27SHong Zhang if (a->j[j] == i) { 6548dd3d27SHong Zhang a->diag[i] = j; 6648dd3d27SHong Zhang break; 6748dd3d27SHong Zhang } 6848dd3d27SHong Zhang } 6948dd3d27SHong Zhang } 7049b5e25fSSatish Balay PetscFunctionReturn(0); 7149b5e25fSSatish Balay } 7249b5e25fSSatish Balay 734a2ae208SSatish Balay #undef __FUNCT__ 744a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ" 751a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool *done) 7649b5e25fSSatish Balay { 77a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 78d0f46423SBarry Smith PetscInt i,j,n = a->mbs,nz = a->i[n],bs = A->rmap->bs; 791a83f524SJed Brown PetscInt **ia = (PetscInt**)inia,**ja = (PetscInt**)inja; 808f7157efSSatish Balay PetscErrorCode ierr; 8149b5e25fSSatish Balay 8249b5e25fSSatish Balay PetscFunctionBegin; 83d3e5a4abSHong Zhang *nn = n; 84a1373b80SHong Zhang if (!ia) PetscFunctionReturn(0); 858f7157efSSatish Balay if (!blockcompressed) { 868f7157efSSatish Balay /* malloc & create the natural set of indices */ 87dcca6d9dSJed Brown ierr = PetscMalloc2((n+1)*bs,ia,nz*bs,ja);CHKERRQ(ierr); 888f7157efSSatish Balay for (i=0; i<n+1; i++) { 898f7157efSSatish Balay for (j=0; j<bs; j++) { 908f7157efSSatish Balay *ia[i*bs+j] = a->i[i]*bs+j+oshift; 918f7157efSSatish Balay } 928f7157efSSatish Balay } 938f7157efSSatish Balay for (i=0; i<nz; i++) { 948f7157efSSatish Balay for (j=0; j<bs; j++) { 958f7157efSSatish Balay *ja[i*bs+j] = a->j[i]*bs+j+oshift; 968f7157efSSatish Balay } 978f7157efSSatish Balay } 988f7157efSSatish Balay } else { /* blockcompressed */ 99a6ece127SHong Zhang if (oshift == 1) { 100a6ece127SHong Zhang /* temporarily add 1 to i and j indices */ 1016c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]++; 102a1373b80SHong Zhang for (i=0; i<n+1; i++) a->i[i]++; 1038f7157efSSatish Balay } 104a1373b80SHong Zhang *ia = a->i; *ja = a->j; 105a6ece127SHong Zhang } 10649b5e25fSSatish Balay PetscFunctionReturn(0); 10749b5e25fSSatish Balay } 10849b5e25fSSatish Balay 1094a2ae208SSatish Balay #undef __FUNCT__ 1104a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ" 1111a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 11249b5e25fSSatish Balay { 113b7aaefc3SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1148f7157efSSatish Balay PetscInt i,n = a->mbs,nz = a->i[n]; 1158f7157efSSatish Balay PetscErrorCode ierr; 116a6ece127SHong Zhang 11749b5e25fSSatish Balay PetscFunctionBegin; 11849b5e25fSSatish Balay if (!ia) PetscFunctionReturn(0); 119a6ece127SHong Zhang 1208f7157efSSatish Balay if (!blockcompressed) { 1218f7157efSSatish Balay ierr = PetscFree2(*ia,*ja);CHKERRQ(ierr); 1228f7157efSSatish Balay } else if (oshift == 1) { /* blockcompressed */ 1236c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]--; 124a6ece127SHong Zhang for (i=0; i<n+1; i++) a->i[i]--; 125a6ece127SHong Zhang } 126a6ece127SHong Zhang PetscFunctionReturn(0); 12749b5e25fSSatish Balay } 12849b5e25fSSatish Balay 1294a2ae208SSatish Balay #undef __FUNCT__ 1304a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ" 131dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A) 13249b5e25fSSatish Balay { 13349b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 134dfbe8321SBarry Smith PetscErrorCode ierr; 13549b5e25fSSatish Balay 13649b5e25fSSatish Balay PetscFunctionBegin; 137a9f03627SSatish Balay #if defined(PETSC_USE_LOG) 138d0f46423SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz); 139a9f03627SSatish Balay #endif 140e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 1417f53bb6cSHong Zhang if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);} 1426bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1436bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1446bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 145c31cb41cSBarry Smith ierr = PetscFree(a->idiag);CHKERRQ(ierr); 146c31cb41cSBarry Smith ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 147c760cd28SBarry Smith if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);} 14805b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 14941f059aeSBarry Smith ierr = PetscFree(a->sor_work);CHKERRQ(ierr); 15005b42c5fSBarry Smith ierr = PetscFree(a->solves_work);CHKERRQ(ierr); 15105b42c5fSBarry Smith ierr = PetscFree(a->mult_work);CHKERRQ(ierr); 15205b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 1534da8f245SBarry Smith if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);} 1541a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1556bf464f9SBarry Smith ierr = MatDestroy(&a->parent);CHKERRQ(ierr); 156bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 157901853e0SKris Buschelman 158dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 159bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 160bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 161bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr); 162bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr); 163bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr); 164bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr); 16538f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr); 166bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr); 16749b5e25fSSatish Balay PetscFunctionReturn(0); 16849b5e25fSSatish Balay } 16949b5e25fSSatish Balay 1704a2ae208SSatish Balay #undef __FUNCT__ 1714a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ" 172ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg) 17349b5e25fSSatish Balay { 174045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 17563ba0a88SBarry Smith PetscErrorCode ierr; 17649b5e25fSSatish Balay 17749b5e25fSSatish Balay PetscFunctionBegin; 1784d9d31abSKris Buschelman switch (op) { 1794d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 1804e0d8c25SBarry Smith a->roworiented = flg; 1814d9d31abSKris Buschelman break; 182a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 183a9817697SBarry Smith a->keepnonzeropattern = flg; 1844d9d31abSKris Buschelman break; 185512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 186512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 1874d9d31abSKris Buschelman break; 1884d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 1894e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 1904d9d31abSKris Buschelman break; 1914d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 1924e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 1934d9d31abSKris Buschelman break; 19428b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 19528b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 19628b2fa4aSMatthew Knepley break; 1974e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1984d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1994d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 200290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 2014d9d31abSKris Buschelman break; 2029a4540c5SBarry Smith case MAT_HERMITIAN: 203e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first"); 2040bc54ff2SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 205eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort; 2060bc54ff2SBarry Smith } else if (A->cmap->bs == 1) { 207eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian; 208e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1"); 209eeffb40dSHong Zhang break; 2103d472b54SHong Zhang case MAT_SPD: 2115021d80fSJed Brown /* These options are handled directly by MatSetOption() */ 2123d472b54SHong Zhang break; 21377e54ba9SKris Buschelman case MAT_SYMMETRIC: 21477e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 2159a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 2164dcd73b1SHong Zhang /* These options are handled directly by MatSetOption() */ 217290bbb0aSBarry Smith break; 218941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 2194e0d8c25SBarry Smith a->ignore_ltriangular = flg; 220941593c8SHong Zhang break; 221941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 2224e0d8c25SBarry Smith a->ignore_ltriangular = flg; 22377e54ba9SKris Buschelman break; 224f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 2254e0d8c25SBarry Smith a->getrow_utriangular = flg; 226f5edf698SHong Zhang break; 2274d9d31abSKris Buschelman default: 228e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 22949b5e25fSSatish Balay } 23049b5e25fSSatish Balay PetscFunctionReturn(0); 23149b5e25fSSatish Balay } 23249b5e25fSSatish Balay 2334a2ae208SSatish Balay #undef __FUNCT__ 2344a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ" 23552768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 23649b5e25fSSatish Balay { 23749b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2386849ba73SBarry Smith PetscErrorCode ierr; 23949b5e25fSSatish Balay 24049b5e25fSSatish Balay PetscFunctionBegin; 241e32f2f54SBarry 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()"); 24252768537SHong Zhang 243f5edf698SHong Zhang /* Get the upper triangular part of the row */ 24452768537SHong Zhang ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr); 24549b5e25fSSatish Balay PetscFunctionReturn(0); 24649b5e25fSSatish Balay } 24749b5e25fSSatish Balay 2484a2ae208SSatish Balay #undef __FUNCT__ 2494a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ" 25013f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 25149b5e25fSSatish Balay { 252dfbe8321SBarry Smith PetscErrorCode ierr; 25349b5e25fSSatish Balay 25449b5e25fSSatish Balay PetscFunctionBegin; 25505b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 25605b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 25749b5e25fSSatish Balay PetscFunctionReturn(0); 25849b5e25fSSatish Balay } 25949b5e25fSSatish Balay 2604a2ae208SSatish Balay #undef __FUNCT__ 261f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ" 262f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 263f5edf698SHong Zhang { 264f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 265f5edf698SHong Zhang 266f5edf698SHong Zhang PetscFunctionBegin; 267f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 268f5edf698SHong Zhang PetscFunctionReturn(0); 269f5edf698SHong Zhang } 270f5edf698SHong Zhang #undef __FUNCT__ 271f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ" 272f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 273f5edf698SHong Zhang { 274f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 275f5edf698SHong Zhang 276f5edf698SHong Zhang PetscFunctionBegin; 277f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 278f5edf698SHong Zhang PetscFunctionReturn(0); 279f5edf698SHong Zhang } 280f5edf698SHong Zhang 281f5edf698SHong Zhang #undef __FUNCT__ 2824a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ" 283fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B) 28449b5e25fSSatish Balay { 285dfbe8321SBarry Smith PetscErrorCode ierr; 2865fd66863SKarl Rupp 28749b5e25fSSatish Balay PetscFunctionBegin; 288815cbec1SBarry Smith if (reuse == MAT_INITIAL_MATRIX || *B != A) { 289999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 290fc4dec0aSBarry Smith } 2918115998fSBarry Smith PetscFunctionReturn(0); 29249b5e25fSSatish Balay } 29349b5e25fSSatish Balay 2944a2ae208SSatish Balay #undef __FUNCT__ 2954a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII" 2967da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 29749b5e25fSSatish Balay { 29849b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 299dfbe8321SBarry Smith PetscErrorCode ierr; 300d0f46423SBarry Smith PetscInt i,j,bs = A->rmap->bs,k,l,bs2=a->bs2; 301f3ef73ceSBarry Smith PetscViewerFormat format; 302121deb67SSatish Balay PetscInt *diag; 30349b5e25fSSatish Balay 30449b5e25fSSatish Balay PetscFunctionBegin; 305b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 306456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 30777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 308fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 309d2507d54SMatthew Knepley Mat aij; 310d5f3da31SBarry Smith if (A->factortype && bs>1) { 31170d5e725SHong 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); 31270d5e725SHong Zhang PetscFunctionReturn(0); 31370d5e725SHong Zhang } 314c9f458caSMatthew Knepley ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr); 315c9f458caSMatthew Knepley ierr = MatView(aij,viewer);CHKERRQ(ierr); 3166bf464f9SBarry Smith ierr = MatDestroy(&aij);CHKERRQ(ierr); 317fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 318d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 31949b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 32049b5e25fSSatish Balay for (j=0; j<bs; j++) { 32177431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 32249b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 32349b5e25fSSatish Balay for (l=0; l<bs; l++) { 32449b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 32549b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 32657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 32757622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 32849b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 32957622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 33057622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 33149b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 33257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 33349b5e25fSSatish Balay } 33449b5e25fSSatish Balay #else 33549b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 33657622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 33749b5e25fSSatish Balay } 33849b5e25fSSatish Balay #endif 33949b5e25fSSatish Balay } 34049b5e25fSSatish Balay } 341b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 34249b5e25fSSatish Balay } 34349b5e25fSSatish Balay } 344d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 345c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 346c1490034SHong Zhang PetscFunctionReturn(0); 34749b5e25fSSatish Balay } else { 348d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 3492c990fa1SHong Zhang if (A->factortype) { /* for factored matrix */ 3502c990fa1SHong Zhang if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet"); 3512c990fa1SHong Zhang 352121deb67SSatish Balay diag=a->diag; 353121deb67SSatish Balay for (i=0; i<a->mbs; i++) { /* for row block i */ 3542c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 3552c990fa1SHong Zhang /* diagonal entry */ 3562c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 3572c990fa1SHong Zhang if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) { 35857622a8eSBarry 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); 3592c990fa1SHong Zhang } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) { 36057622a8eSBarry 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); 3612c990fa1SHong Zhang } else { 36257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 3632c990fa1SHong Zhang } 3642c990fa1SHong Zhang #else 3656712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr); 3662c990fa1SHong Zhang #endif 3672c990fa1SHong Zhang /* off-diagonal entries */ 3682c990fa1SHong Zhang for (k=a->i[i]; k<a->i[i+1]-1; k++) { 3692c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 370ca0704adSBarry Smith if (PetscImaginaryPart(a->a[k]) > 0.0) { 37157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 372ca0704adSBarry Smith } else if (PetscImaginaryPart(a->a[k]) < 0.0) { 37357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 3742c990fa1SHong Zhang } else { 37557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr); 3762c990fa1SHong Zhang } 3772c990fa1SHong Zhang #else 37857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr); 3792c990fa1SHong Zhang #endif 3802c990fa1SHong Zhang } 3812c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 3822c990fa1SHong Zhang } 3832c990fa1SHong Zhang 3842c990fa1SHong Zhang } else { /* for non-factored matrix */ 3850c74a584SJed Brown for (i=0; i<a->mbs; i++) { /* for row block i */ 3860c74a584SJed Brown for (j=0; j<bs; j++) { /* for row bs*i + j */ 38777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 3880c74a584SJed Brown for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */ 3890c74a584SJed Brown for (l=0; l<bs; l++) { /* for column */ 39049b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 39149b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 39257622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 39357622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 39449b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 39557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 39657622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 39749b5e25fSSatish Balay } else { 39857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 39949b5e25fSSatish Balay } 40049b5e25fSSatish Balay #else 40157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 40249b5e25fSSatish Balay #endif 40349b5e25fSSatish Balay } 40449b5e25fSSatish Balay } 405b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 40649b5e25fSSatish Balay } 40749b5e25fSSatish Balay } 4082c990fa1SHong Zhang } 409d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 41049b5e25fSSatish Balay } 411b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 41249b5e25fSSatish Balay PetscFunctionReturn(0); 41349b5e25fSSatish Balay } 41449b5e25fSSatish Balay 4159804daf3SBarry Smith #include <petscdraw.h> 4164a2ae208SSatish Balay #undef __FUNCT__ 4174a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom" 4186849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 41949b5e25fSSatish Balay { 42049b5e25fSSatish Balay Mat A = (Mat) Aa; 42149b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 4226849ba73SBarry Smith PetscErrorCode ierr; 423d0f46423SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2; 42413f74950SBarry Smith PetscMPIInt rank; 42549b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 42649b5e25fSSatish Balay MatScalar *aa; 42749b5e25fSSatish Balay MPI_Comm comm; 428b0a32e0cSBarry Smith PetscViewer viewer; 42949b5e25fSSatish Balay 43049b5e25fSSatish Balay PetscFunctionBegin; 43149b5e25fSSatish Balay /* 43249b5e25fSSatish Balay This is nasty. If this is called from an originally parallel matrix 43349b5e25fSSatish Balay then all processes call this,but only the first has the matrix so the 43449b5e25fSSatish Balay rest should return immediately. 43549b5e25fSSatish Balay */ 43649b5e25fSSatish Balay ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr); 43749b5e25fSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 43849b5e25fSSatish Balay if (rank) PetscFunctionReturn(0); 43949b5e25fSSatish Balay 44049b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 44149b5e25fSSatish Balay 442b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 443b0a32e0cSBarry Smith PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric"); 44449b5e25fSSatish Balay 44549b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 446b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 44749b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 44849b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 449d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 45049b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 45149b5e25fSSatish Balay aa = a->a + j*bs2; 45249b5e25fSSatish Balay for (k=0; k<bs; k++) { 45349b5e25fSSatish Balay for (l=0; l<bs; l++) { 45449b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 455b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 45649b5e25fSSatish Balay } 45749b5e25fSSatish Balay } 45849b5e25fSSatish Balay } 45949b5e25fSSatish Balay } 460b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 46149b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 46249b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 463d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 46449b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 46549b5e25fSSatish Balay aa = a->a + j*bs2; 46649b5e25fSSatish Balay for (k=0; k<bs; k++) { 46749b5e25fSSatish Balay for (l=0; l<bs; l++) { 46849b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 469b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 47049b5e25fSSatish Balay } 47149b5e25fSSatish Balay } 47249b5e25fSSatish Balay } 47349b5e25fSSatish Balay } 47449b5e25fSSatish Balay 475b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 47649b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 47749b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 478d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 47949b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 48049b5e25fSSatish Balay aa = a->a + j*bs2; 48149b5e25fSSatish Balay for (k=0; k<bs; k++) { 48249b5e25fSSatish Balay for (l=0; l<bs; l++) { 48349b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 484b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 48549b5e25fSSatish Balay } 48649b5e25fSSatish Balay } 48749b5e25fSSatish Balay } 48849b5e25fSSatish Balay } 48949b5e25fSSatish Balay PetscFunctionReturn(0); 49049b5e25fSSatish Balay } 49149b5e25fSSatish Balay 4924a2ae208SSatish Balay #undef __FUNCT__ 4934a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw" 4946849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 49549b5e25fSSatish Balay { 496dfbe8321SBarry Smith PetscErrorCode ierr; 49749b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 498b0a32e0cSBarry Smith PetscDraw draw; 499ace3abfcSBarry Smith PetscBool isnull; 50049b5e25fSSatish Balay 50149b5e25fSSatish Balay PetscFunctionBegin; 502b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 503b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 50449b5e25fSSatish Balay 50549b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 506d0f46423SBarry Smith xr = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0; 50749b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 508b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 509b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 5100298fd71SBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 51149b5e25fSSatish Balay PetscFunctionReturn(0); 51249b5e25fSSatish Balay } 51349b5e25fSSatish Balay 5144a2ae208SSatish Balay #undef __FUNCT__ 5154a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ" 516dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 51749b5e25fSSatish Balay { 518dfbe8321SBarry Smith PetscErrorCode ierr; 519ace3abfcSBarry Smith PetscBool iascii,isdraw; 52008917f38SBarry Smith FILE *file = 0; 52149b5e25fSSatish Balay 52249b5e25fSSatish Balay PetscFunctionBegin; 523251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 524251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 52532077d6dSBarry Smith if (iascii) { 52649b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 52749b5e25fSSatish Balay } else if (isdraw) { 52849b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 52949b5e25fSSatish Balay } else { 530a5e6ed63SBarry Smith Mat B; 531ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 532a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 5336bf464f9SBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 53408917f38SBarry Smith ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr); 53508917f38SBarry Smith if (file) { 53608917f38SBarry Smith fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs); 53708917f38SBarry Smith } 53849b5e25fSSatish Balay } 53949b5e25fSSatish Balay PetscFunctionReturn(0); 54049b5e25fSSatish Balay } 54149b5e25fSSatish Balay 54249b5e25fSSatish Balay 5434a2ae208SSatish Balay #undef __FUNCT__ 5444a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ" 54513f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 54649b5e25fSSatish Balay { 547045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 54813f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 54913f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 550d0f46423SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2; 55197e567efSBarry Smith MatScalar *ap,*aa = a->a; 55249b5e25fSSatish Balay 55349b5e25fSSatish Balay PetscFunctionBegin; 55449b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 55549b5e25fSSatish Balay row = im[k]; brow = row/bs; 556e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 557e32f2f54SBarry 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); 55849b5e25fSSatish Balay rp = aj + ai[brow]; ap = aa + bs2*ai[brow]; 55949b5e25fSSatish Balay nrow = ailen[brow]; 56049b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 561e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 562e32f2f54SBarry 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); 56349b5e25fSSatish Balay col = in[l]; 56449b5e25fSSatish Balay bcol = col/bs; 56549b5e25fSSatish Balay cidx = col%bs; 56649b5e25fSSatish Balay ridx = row%bs; 56749b5e25fSSatish Balay high = nrow; 56849b5e25fSSatish Balay low = 0; /* assume unsorted */ 56949b5e25fSSatish Balay while (high-low > 5) { 57049b5e25fSSatish Balay t = (low+high)/2; 57149b5e25fSSatish Balay if (rp[t] > bcol) high = t; 57249b5e25fSSatish Balay else low = t; 57349b5e25fSSatish Balay } 57449b5e25fSSatish Balay for (i=low; i<high; i++) { 57549b5e25fSSatish Balay if (rp[i] > bcol) break; 57649b5e25fSSatish Balay if (rp[i] == bcol) { 57749b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 57849b5e25fSSatish Balay goto finished; 57949b5e25fSSatish Balay } 58049b5e25fSSatish Balay } 58197e567efSBarry Smith *v++ = 0.0; 58249b5e25fSSatish Balay finished:; 58349b5e25fSSatish Balay } 58449b5e25fSSatish Balay } 58549b5e25fSSatish Balay PetscFunctionReturn(0); 58649b5e25fSSatish Balay } 58749b5e25fSSatish Balay 58849b5e25fSSatish Balay 5894a2ae208SSatish Balay #undef __FUNCT__ 5904a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ" 59113f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 59249b5e25fSSatish Balay { 5930880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 5946849ba73SBarry Smith PetscErrorCode ierr; 595e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 59613f74950SBarry Smith PetscInt *imax =a->imax,*ai=a->i,*ailen=a->ilen; 597d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval; 598ace3abfcSBarry Smith PetscBool roworiented=a->roworiented; 599dd6ea824SBarry Smith const PetscScalar *value = v; 600f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 6010880e062SHong Zhang 60249b5e25fSSatish Balay PetscFunctionBegin; 60326fbe8dcSKarl Rupp if (roworiented) stepval = (n-1)*bs; 60426fbe8dcSKarl Rupp else stepval = (m-1)*bs; 60526fbe8dcSKarl Rupp 6060880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 6070880e062SHong Zhang row = im[k]; 6080880e062SHong Zhang if (row < 0) continue; 6092515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 610e32f2f54SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1); 6110880e062SHong Zhang #endif 6120880e062SHong Zhang rp = aj + ai[row]; 6130880e062SHong Zhang ap = aa + bs2*ai[row]; 6140880e062SHong Zhang rmax = imax[row]; 6150880e062SHong Zhang nrow = ailen[row]; 6160880e062SHong Zhang low = 0; 617818f2c47SBarry Smith high = nrow; 6180880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 6190880e062SHong Zhang if (in[l] < 0) continue; 6200880e062SHong Zhang col = in[l]; 6212515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 622e32f2f54SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1); 623b1823623SSatish Balay #endif 624b98bf0e1SJed Brown if (col < row) { 62526fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular block */ 62626fbe8dcSKarl 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)"); 627b98bf0e1SJed Brown } 62826fbe8dcSKarl Rupp if (roworiented) value = v + k*(stepval+bs)*bs + l*bs; 62926fbe8dcSKarl Rupp else value = v + l*(stepval+bs)*bs + k*bs; 63026fbe8dcSKarl Rupp 63126fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 63226fbe8dcSKarl Rupp else high = nrow; 63326fbe8dcSKarl Rupp 634e2ee6c50SBarry Smith lastcol = col; 6350880e062SHong Zhang while (high-low > 7) { 6360880e062SHong Zhang t = (low+high)/2; 6370880e062SHong Zhang if (rp[t] > col) high = t; 6380880e062SHong Zhang else low = t; 6390880e062SHong Zhang } 6400880e062SHong Zhang for (i=low; i<high; i++) { 6410880e062SHong Zhang if (rp[i] > col) break; 6420880e062SHong Zhang if (rp[i] == col) { 6430880e062SHong Zhang bap = ap + bs2*i; 6440880e062SHong Zhang if (roworiented) { 6450880e062SHong Zhang if (is == ADD_VALUES) { 6460880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6470880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6480880e062SHong Zhang bap[jj] += *value++; 6490880e062SHong Zhang } 6500880e062SHong Zhang } 6510880e062SHong Zhang } else { 6520880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6530880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6540880e062SHong Zhang bap[jj] = *value++; 6550880e062SHong Zhang } 6560880e062SHong Zhang } 6570880e062SHong Zhang } 6580880e062SHong Zhang } else { 6590880e062SHong Zhang if (is == ADD_VALUES) { 6600880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6610880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6620880e062SHong Zhang *bap++ += *value++; 6630880e062SHong Zhang } 6640880e062SHong Zhang } 6650880e062SHong Zhang } else { 6660880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6670880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6680880e062SHong Zhang *bap++ = *value++; 6690880e062SHong Zhang } 6700880e062SHong Zhang } 6710880e062SHong Zhang } 6720880e062SHong Zhang } 6730880e062SHong Zhang goto noinsert2; 6740880e062SHong Zhang } 6750880e062SHong Zhang } 6760880e062SHong Zhang if (nonew == 1) goto noinsert2; 677e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 678fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 679c03d1d03SSatish Balay N = nrow++ - 1; high++; 6800880e062SHong Zhang /* shift up all the later entries in this row */ 6810880e062SHong Zhang for (ii=N; ii>=i; ii--) { 6820880e062SHong Zhang rp[ii+1] = rp[ii]; 6830880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 6840880e062SHong Zhang } 6850880e062SHong Zhang if (N >= i) { 6860880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 6870880e062SHong Zhang } 6880880e062SHong Zhang rp[i] = col; 6890880e062SHong Zhang bap = ap + bs2*i; 6900880e062SHong Zhang if (roworiented) { 6910880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6920880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6930880e062SHong Zhang bap[jj] = *value++; 6940880e062SHong Zhang } 6950880e062SHong Zhang } 6960880e062SHong Zhang } else { 6970880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6980880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6990880e062SHong Zhang *bap++ = *value++; 7000880e062SHong Zhang } 7010880e062SHong Zhang } 7020880e062SHong Zhang } 7030880e062SHong Zhang noinsert2:; 7040880e062SHong Zhang low = i; 7050880e062SHong Zhang } 7060880e062SHong Zhang ailen[row] = nrow; 7070880e062SHong Zhang } 7080880e062SHong Zhang PetscFunctionReturn(0); 70949b5e25fSSatish Balay } 71049b5e25fSSatish Balay 71164831d72SBarry Smith /* 71264831d72SBarry Smith This is not yet used 71364831d72SBarry Smith */ 7144a2ae208SSatish Balay #undef __FUNCT__ 7154108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode" 7164108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A) 7170def2e27SBarry Smith { 7180def2e27SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7190def2e27SBarry Smith PetscErrorCode ierr; 7200def2e27SBarry Smith const PetscInt *ai = a->i, *aj = a->j,*cols; 7210def2e27SBarry Smith PetscInt i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts; 722ace3abfcSBarry Smith PetscBool flag; 7230def2e27SBarry Smith 7240def2e27SBarry Smith PetscFunctionBegin; 725785e854fSJed Brown ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr); 7260def2e27SBarry Smith while (i < m) { 7270def2e27SBarry Smith nzx = ai[i+1] - ai[i]; /* Number of nonzeros */ 7280def2e27SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 7290def2e27SBarry Smith for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 7300def2e27SBarry Smith nzy = ai[j+1] - ai[j]; 7310def2e27SBarry Smith if (nzy != (nzx - j + i)) break; 7320def2e27SBarry Smith ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr); 7330def2e27SBarry Smith if (!flag) break; 7340def2e27SBarry Smith } 7350def2e27SBarry Smith ns[node_count++] = blk_size; 73626fbe8dcSKarl Rupp 7370def2e27SBarry Smith i = j; 7380def2e27SBarry Smith } 7390def2e27SBarry Smith if (!a->inode.size && m && node_count > .9*m) { 7400def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7410def2e27SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 7420def2e27SBarry Smith } else { 7430def2e27SBarry Smith a->inode.node_count = node_count; 74426fbe8dcSKarl Rupp 745785e854fSJed Brown ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr); 7463bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr); 74722d28d08SBarry Smith ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr); 7480def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7490def2e27SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 7500def2e27SBarry Smith 7510def2e27SBarry Smith /* count collections of adjacent columns in each inode */ 7520def2e27SBarry Smith row = 0; 7530def2e27SBarry Smith cnt = 0; 7540def2e27SBarry Smith for (i=0; i<node_count; i++) { 7550def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 7560def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 7570def2e27SBarry Smith for (j=1; j<nz; j++) { 75826fbe8dcSKarl Rupp if (cols[j] != cols[j-1]+1) cnt++; 7590def2e27SBarry Smith } 7600def2e27SBarry Smith cnt++; 7610def2e27SBarry Smith row += a->inode.size[i]; 7620def2e27SBarry Smith } 763785e854fSJed Brown ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr); 7640def2e27SBarry Smith cnt = 0; 7650def2e27SBarry Smith row = 0; 7660def2e27SBarry Smith for (i=0; i<node_count; i++) { 7670def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 7680def2e27SBarry Smith counts[2*cnt] = cols[0]; 7690def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 7700def2e27SBarry Smith cnt2 = 1; 7710def2e27SBarry Smith for (j=1; j<nz; j++) { 7720def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 7730def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 7740def2e27SBarry Smith counts[2*cnt] = cols[j]; 7750def2e27SBarry Smith cnt2 = 1; 7760def2e27SBarry Smith } else cnt2++; 7770def2e27SBarry Smith } 7780def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 7790def2e27SBarry Smith row += a->inode.size[i]; 7800def2e27SBarry Smith } 78122d28d08SBarry Smith ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr); 7820def2e27SBarry Smith } 78338702af4SBarry Smith PetscFunctionReturn(0); 78438702af4SBarry Smith } 78538702af4SBarry Smith 78638702af4SBarry Smith #undef __FUNCT__ 7874a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ" 788dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 78949b5e25fSSatish Balay { 79049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7916849ba73SBarry Smith PetscErrorCode ierr; 79213f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 793d0f46423SBarry Smith PetscInt m = A->rmap->N,*ip,N,*ailen = a->ilen; 79413f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 79549b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 79649b5e25fSSatish Balay 79749b5e25fSSatish Balay PetscFunctionBegin; 79849b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 79949b5e25fSSatish Balay 80049b5e25fSSatish Balay if (m) rmax = ailen[0]; 80149b5e25fSSatish Balay for (i=1; i<mbs; i++) { 80249b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 80349b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 80449b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 80549b5e25fSSatish Balay if (fshift) { 80649b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 80749b5e25fSSatish Balay N = ailen[i]; 80849b5e25fSSatish Balay for (j=0; j<N; j++) { 80949b5e25fSSatish Balay ip[j-fshift] = ip[j]; 81049b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 81149b5e25fSSatish Balay } 81249b5e25fSSatish Balay } 81349b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 81449b5e25fSSatish Balay } 81549b5e25fSSatish Balay if (mbs) { 81649b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 81749b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 81849b5e25fSSatish Balay } 81949b5e25fSSatish Balay /* reset ilen and imax for each row */ 82049b5e25fSSatish Balay for (i=0; i<mbs; i++) { 82149b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 82249b5e25fSSatish Balay } 8236c6c5352SBarry Smith a->nz = ai[mbs]; 82449b5e25fSSatish Balay 825b424e231SHong Zhang /* diagonals may have moved, reset it */ 826b424e231SHong Zhang if (a->diag) { 8272ed38d0bSJed Brown ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr); 82849b5e25fSSatish Balay } 82926fbe8dcSKarl 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); 83026fbe8dcSKarl Rupp 831d0f46423SBarry 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); 832ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 833ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 83426fbe8dcSKarl Rupp 8358e58a170SBarry Smith A->info.mallocs += a->reallocs; 83649b5e25fSSatish Balay a->reallocs = 0; 83749b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 838061b2667SBarry Smith a->idiagvalid = PETSC_FALSE; 8394dcd73b1SHong Zhang a->rmax = rmax; 84038702af4SBarry Smith 84138702af4SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 84244e1c64aSLisandro Dalcin if (a->jshort && a->free_jshort) { 84317803ae8SHong Zhang /* when matrix data structure is changed, previous jshort must be replaced */ 84417803ae8SHong Zhang ierr = PetscFree(a->jshort);CHKERRQ(ierr); 84517803ae8SHong Zhang } 846785e854fSJed Brown ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr); 8473bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr); 84838702af4SBarry Smith for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i]; 84938702af4SBarry Smith A->ops->mult = MatMult_SeqSBAIJ_1_ushort; 85041f059aeSBarry Smith A->ops->sor = MatSOR_SeqSBAIJ_ushort; 8514da8f245SBarry Smith a->free_jshort = PETSC_TRUE; 85238702af4SBarry Smith } 85349b5e25fSSatish Balay PetscFunctionReturn(0); 85449b5e25fSSatish Balay } 85549b5e25fSSatish Balay 85649b5e25fSSatish Balay /* 85749b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 85849b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 85949b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 86049b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 86149b5e25fSSatish Balay */ 8624a2ae208SSatish Balay #undef __FUNCT__ 8634a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks" 86413f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 86549b5e25fSSatish Balay { 86613f74950SBarry Smith PetscInt i,j,k,row; 867ace3abfcSBarry Smith PetscBool flg; 86849b5e25fSSatish Balay 86949b5e25fSSatish Balay PetscFunctionBegin; 87049b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 87149b5e25fSSatish Balay row = idx[i]; 87249b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 87349b5e25fSSatish Balay sizes[j] = 1; 87449b5e25fSSatish Balay i++; 87549b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 87649b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 87749b5e25fSSatish Balay i++; 87849b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 87949b5e25fSSatish Balay flg = PETSC_TRUE; 88049b5e25fSSatish Balay for (k=1; k<bs; k++) { 88149b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 88249b5e25fSSatish Balay flg = PETSC_FALSE; 88349b5e25fSSatish Balay break; 88449b5e25fSSatish Balay } 88549b5e25fSSatish Balay } 886abc0a331SBarry Smith if (flg) { /* No break in the bs */ 88749b5e25fSSatish Balay sizes[j] = bs; 88849b5e25fSSatish Balay i += bs; 88949b5e25fSSatish Balay } else { 89049b5e25fSSatish Balay sizes[j] = 1; 89149b5e25fSSatish Balay i++; 89249b5e25fSSatish Balay } 89349b5e25fSSatish Balay } 89449b5e25fSSatish Balay } 89549b5e25fSSatish Balay *bs_max = j; 89649b5e25fSSatish Balay PetscFunctionReturn(0); 89749b5e25fSSatish Balay } 89849b5e25fSSatish Balay 89949b5e25fSSatish Balay 90049b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 90149b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 90249b5e25fSSatish Balay */ 90349b5e25fSSatish Balay 9044a2ae208SSatish Balay #undef __FUNCT__ 9054a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ" 90613f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 90749b5e25fSSatish Balay { 90849b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 9096849ba73SBarry Smith PetscErrorCode ierr; 910e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 91113f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 912d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol; 91313f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 91449b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 91549b5e25fSSatish Balay 91649b5e25fSSatish Balay PetscFunctionBegin; 91749b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 91849b5e25fSSatish Balay row = im[k]; /* row number */ 91949b5e25fSSatish Balay brow = row/bs; /* block row number */ 92049b5e25fSSatish Balay if (row < 0) continue; 9212515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 922e32f2f54SBarry 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); 92349b5e25fSSatish Balay #endif 92449b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 92549b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 92649b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 92749b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 92849b5e25fSSatish Balay low = 0; 92949b5e25fSSatish Balay 93049b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 93149b5e25fSSatish Balay if (in[l] < 0) continue; 9322515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 933e32f2f54SBarry 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); 93449b5e25fSSatish Balay #endif 93549b5e25fSSatish Balay col = in[l]; 93649b5e25fSSatish Balay bcol = col/bs; /* block col number */ 93749b5e25fSSatish Balay 938941593c8SHong Zhang if (brow > bcol) { 93926fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular values */ 94026fbe8dcSKarl 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)"); 941941593c8SHong Zhang } 942f4989cb3SHong Zhang 94349b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 9448549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)) { 94549b5e25fSSatish Balay /* element value a(k,l) */ 94626fbe8dcSKarl Rupp if (roworiented) value = v[l + k*n]; 94726fbe8dcSKarl Rupp else value = v[k + l*m]; 94849b5e25fSSatish Balay 94949b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 95026fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 95126fbe8dcSKarl Rupp high = nrow; 952e2ee6c50SBarry Smith lastcol = col; 95349b5e25fSSatish Balay while (high-low > 7) { 95449b5e25fSSatish Balay t = (low+high)/2; 95549b5e25fSSatish Balay if (rp[t] > bcol) high = t; 95649b5e25fSSatish Balay else low = t; 95749b5e25fSSatish Balay } 95849b5e25fSSatish Balay for (i=low; i<high; i++) { 95949b5e25fSSatish Balay if (rp[i] > bcol) break; 96049b5e25fSSatish Balay if (rp[i] == bcol) { 96149b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 96249b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 96349b5e25fSSatish Balay else *bap = value; 9648549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 9658549e402SHong Zhang if (brow == bcol && ridx < cidx) { 9668549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 9678549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 9688549e402SHong Zhang else *bap = value; 9698549e402SHong Zhang } 97049b5e25fSSatish Balay goto noinsert1; 97149b5e25fSSatish Balay } 97249b5e25fSSatish Balay } 97349b5e25fSSatish Balay 97449b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 975e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 976fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 97749b5e25fSSatish Balay 978c03d1d03SSatish Balay N = nrow++ - 1; high++; 97949b5e25fSSatish Balay /* shift up all the later entries in this row */ 98049b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 98149b5e25fSSatish Balay rp[ii+1] = rp[ii]; 98249b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 98349b5e25fSSatish Balay } 98449b5e25fSSatish Balay if (N>=i) { 98549b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 98649b5e25fSSatish Balay } 98749b5e25fSSatish Balay rp[i] = bcol; 98849b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 989e56f5c9eSBarry Smith A->nonzerostate++; 99049b5e25fSSatish Balay noinsert1:; 99149b5e25fSSatish Balay low = i; 9928549e402SHong Zhang } 99349b5e25fSSatish Balay } /* end of loop over added columns */ 99449b5e25fSSatish Balay ailen[brow] = nrow; 99549b5e25fSSatish Balay } /* end of loop over added rows */ 99649b5e25fSSatish Balay PetscFunctionReturn(0); 99749b5e25fSSatish Balay } 99849b5e25fSSatish Balay 9994a2ae208SSatish Balay #undef __FUNCT__ 10004d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ" 10010481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info) 100249b5e25fSSatish Balay { 10034ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 100449b5e25fSSatish Balay Mat outA; 1005dfbe8321SBarry Smith PetscErrorCode ierr; 1006ace3abfcSBarry Smith PetscBool row_identity; 100749b5e25fSSatish Balay 100849b5e25fSSatish Balay PetscFunctionBegin; 1009e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 1010c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 1011e32f2f54SBarry Smith if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported"); 1012e32f2f54SBarry 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()! */ 1013c84f5b01SHong Zhang 101449b5e25fSSatish Balay outA = inA; 1015d5f3da31SBarry Smith inA->factortype = MAT_FACTOR_ICC; 101649b5e25fSSatish Balay 10171a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 1018d595f711SHong Zhang ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr); 101949b5e25fSSatish Balay 1020c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10216bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1022c84f5b01SHong Zhang a->row = row; 1023c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10246bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1025c84f5b01SHong Zhang a->col = row; 1026c84f5b01SHong Zhang 1027c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1028c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 10293bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr); 103049b5e25fSSatish Balay 103149b5e25fSSatish Balay if (!a->solve_work) { 1032785e854fSJed Brown ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr); 10333bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 103449b5e25fSSatish Balay } 103549b5e25fSSatish Balay 1036719d5645SBarry Smith ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr); 103749b5e25fSSatish Balay PetscFunctionReturn(0); 103849b5e25fSSatish Balay } 1039950f1e5bSHong Zhang 10404a2ae208SSatish Balay #undef __FUNCT__ 10414a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ" 10427087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 104349b5e25fSSatish Balay { 1044045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ*)mat->data; 104513f74950SBarry Smith PetscInt i,nz,n; 10467827cd58SJed Brown PetscErrorCode ierr; 104749b5e25fSSatish Balay 104849b5e25fSSatish Balay PetscFunctionBegin; 10496c6c5352SBarry Smith nz = baij->maxnz; 1050d0f46423SBarry Smith n = mat->cmap->n; 105126fbe8dcSKarl Rupp for (i=0; i<nz; i++) baij->j[i] = indices[i]; 105226fbe8dcSKarl Rupp 10536c6c5352SBarry Smith baij->nz = nz; 105426fbe8dcSKarl Rupp for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i]; 105526fbe8dcSKarl Rupp 10567827cd58SJed Brown ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 105749b5e25fSSatish Balay PetscFunctionReturn(0); 105849b5e25fSSatish Balay } 105949b5e25fSSatish Balay 10604a2ae208SSatish Balay #undef __FUNCT__ 10614a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices" 106249b5e25fSSatish Balay /*@ 106319585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 106449b5e25fSSatish Balay in the matrix. 106549b5e25fSSatish Balay 106649b5e25fSSatish Balay Input Parameters: 106719585528SSatish Balay + mat - the SeqSBAIJ matrix 106849b5e25fSSatish Balay - indices - the column indices 106949b5e25fSSatish Balay 107049b5e25fSSatish Balay Level: advanced 107149b5e25fSSatish Balay 107249b5e25fSSatish Balay Notes: 107349b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 107449b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 107549b5e25fSSatish Balay of the MatSetValues() operation. 107649b5e25fSSatish Balay 107749b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1078d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 107949b5e25fSSatish Balay 1080ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 108149b5e25fSSatish Balay 1082ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 108349b5e25fSSatish Balay @*/ 10847087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 108549b5e25fSSatish Balay { 10864ac538c5SBarry Smith PetscErrorCode ierr; 108749b5e25fSSatish Balay 108849b5e25fSSatish Balay PetscFunctionBegin; 10890700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 10904482741eSBarry Smith PetscValidPointer(indices,2); 10914ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr); 109249b5e25fSSatish Balay PetscFunctionReturn(0); 109349b5e25fSSatish Balay } 109449b5e25fSSatish Balay 10954a2ae208SSatish Balay #undef __FUNCT__ 10963c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ" 10973c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 10983c896bc6SHong Zhang { 10993c896bc6SHong Zhang PetscErrorCode ierr; 11003c896bc6SHong Zhang 11013c896bc6SHong Zhang PetscFunctionBegin; 11023c896bc6SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 11033c896bc6SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 11043c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11053c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11063c896bc6SHong Zhang 1107e7e72b3dSBarry 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"); 1108d0f46423SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr); 11093c896bc6SHong Zhang } else { 1110f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11113c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1112f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11133c896bc6SHong Zhang } 11143c896bc6SHong Zhang PetscFunctionReturn(0); 11153c896bc6SHong Zhang } 11163c896bc6SHong Zhang 11173c896bc6SHong Zhang #undef __FUNCT__ 11184994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ" 11194994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A) 1120273d9f13SBarry Smith { 1121dfbe8321SBarry Smith PetscErrorCode ierr; 1122273d9f13SBarry Smith 1123273d9f13SBarry Smith PetscFunctionBegin; 1124535b19f3SBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr); 1125273d9f13SBarry Smith PetscFunctionReturn(0); 1126273d9f13SBarry Smith } 1127273d9f13SBarry Smith 1128a6ece127SHong Zhang #undef __FUNCT__ 11298c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ" 11308c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1131a6ece127SHong Zhang { 1132a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11335fd66863SKarl Rupp 1134a6ece127SHong Zhang PetscFunctionBegin; 1135a6ece127SHong Zhang *array = a->a; 1136a6ece127SHong Zhang PetscFunctionReturn(0); 1137a6ece127SHong Zhang } 1138a6ece127SHong Zhang 1139a6ece127SHong Zhang #undef __FUNCT__ 11408c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ" 11418c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1142a6ece127SHong Zhang { 1143a6ece127SHong Zhang PetscFunctionBegin; 1144a6ece127SHong Zhang PetscFunctionReturn(0); 1145a6ece127SHong Zhang } 1146a6ece127SHong Zhang 114742ee4b1aSHong Zhang #undef __FUNCT__ 114852768537SHong Zhang #define __FUNCT__ "MatAXPYGetPreallocation_SeqSBAIJ" 114952768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz) 115052768537SHong Zhang { 1151b264fe52SHong Zhang PetscInt bs = Y->rmap->bs,mbs = Y->rmap->N/bs; 115252768537SHong Zhang Mat_SeqSBAIJ *x = (Mat_SeqSBAIJ*)X->data; 115352768537SHong Zhang Mat_SeqSBAIJ *y = (Mat_SeqSBAIJ*)Y->data; 1154b264fe52SHong Zhang PetscErrorCode ierr; 115552768537SHong Zhang 115652768537SHong Zhang PetscFunctionBegin; 115752768537SHong Zhang /* Set the number of nonzeros in the new matrix */ 1158b264fe52SHong Zhang ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr); 115952768537SHong Zhang PetscFunctionReturn(0); 116052768537SHong Zhang } 116152768537SHong Zhang 116252768537SHong Zhang #undef __FUNCT__ 116342ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ" 1164f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 116542ee4b1aSHong Zhang { 116642ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data; 1167dfbe8321SBarry Smith PetscErrorCode ierr; 116831ce2d13SHong Zhang PetscInt bs=Y->rmap->bs,bs2=bs*bs; 1169e838b9e7SJed Brown PetscBLASInt one = 1; 117042ee4b1aSHong Zhang 117142ee4b1aSHong Zhang PetscFunctionBegin; 117242ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1173f4df32b1SMatthew Knepley PetscScalar alpha = a; 1174c5df96a5SBarry Smith PetscBLASInt bnz; 1175c5df96a5SBarry Smith ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr); 11768b83055fSJed Brown PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one)); 1177a3fa217bSJose E. Roman ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 1178*ab784542SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1179*ab784542SHong Zhang ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr); 1180*ab784542SHong Zhang ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1181*ab784542SHong Zhang ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr); 118242ee4b1aSHong Zhang } else { 118352768537SHong Zhang Mat B; 118452768537SHong Zhang PetscInt *nnz; 118552768537SHong Zhang if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size"); 1186f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 118752768537SHong Zhang ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr); 118852768537SHong Zhang ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr); 118952768537SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr); 119052768537SHong Zhang ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr); 119152768537SHong Zhang ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr); 119252768537SHong Zhang ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr); 119352768537SHong Zhang ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr); 119452768537SHong Zhang ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr); 119552768537SHong Zhang ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 119652768537SHong Zhang 119752768537SHong Zhang ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr); 119852768537SHong Zhang 119952768537SHong Zhang ierr = MatHeaderReplace(Y,B);CHKERRQ(ierr); 120052768537SHong Zhang ierr = PetscFree(nnz);CHKERRQ(ierr); 1201f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 120252768537SHong Zhang ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr); 120342ee4b1aSHong Zhang } 120442ee4b1aSHong Zhang PetscFunctionReturn(0); 120542ee4b1aSHong Zhang } 120642ee4b1aSHong Zhang 1207efcf0fc3SBarry Smith #undef __FUNCT__ 1208efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ" 1209ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1210efcf0fc3SBarry Smith { 1211efcf0fc3SBarry Smith PetscFunctionBegin; 1212efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1213efcf0fc3SBarry Smith PetscFunctionReturn(0); 1214efcf0fc3SBarry Smith } 1215efcf0fc3SBarry Smith 1216efcf0fc3SBarry Smith #undef __FUNCT__ 1217efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ" 1218ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool *flg) 1219efcf0fc3SBarry Smith { 1220efcf0fc3SBarry Smith PetscFunctionBegin; 1221efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1222efcf0fc3SBarry Smith PetscFunctionReturn(0); 1223efcf0fc3SBarry Smith } 1224efcf0fc3SBarry Smith 1225efcf0fc3SBarry Smith #undef __FUNCT__ 1226efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ" 1227ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1228efcf0fc3SBarry Smith { 1229efcf0fc3SBarry Smith PetscFunctionBegin; 1230efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1231efcf0fc3SBarry Smith PetscFunctionReturn(0); 1232efcf0fc3SBarry Smith } 1233efcf0fc3SBarry Smith 123499cafbc1SBarry Smith #undef __FUNCT__ 123599cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ" 123699cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 123799cafbc1SBarry Smith { 123899cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 123999cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1240dd6ea824SBarry Smith MatScalar *aa = a->a; 124199cafbc1SBarry Smith 124299cafbc1SBarry Smith PetscFunctionBegin; 124399cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 124499cafbc1SBarry Smith PetscFunctionReturn(0); 124599cafbc1SBarry Smith } 124699cafbc1SBarry Smith 124799cafbc1SBarry Smith #undef __FUNCT__ 124899cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ" 124999cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A) 125099cafbc1SBarry Smith { 125199cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 125299cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1253dd6ea824SBarry Smith MatScalar *aa = a->a; 125499cafbc1SBarry Smith 125599cafbc1SBarry Smith PetscFunctionBegin; 125699cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 125799cafbc1SBarry Smith PetscFunctionReturn(0); 125899cafbc1SBarry Smith } 125999cafbc1SBarry Smith 12603bededecSBarry Smith #undef __FUNCT__ 12613bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ" 12623bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b) 12633bededecSBarry Smith { 12643bededecSBarry Smith Mat_SeqSBAIJ *baij=(Mat_SeqSBAIJ*)A->data; 12653bededecSBarry Smith PetscErrorCode ierr; 12663bededecSBarry Smith PetscInt i,j,k,count; 12673bededecSBarry Smith PetscInt bs =A->rmap->bs,bs2=baij->bs2,row,col; 12683bededecSBarry Smith PetscScalar zero = 0.0; 12693bededecSBarry Smith MatScalar *aa; 12703bededecSBarry Smith const PetscScalar *xx; 12713bededecSBarry Smith PetscScalar *bb; 127256777dd2SBarry Smith PetscBool *zeroed,vecs = PETSC_FALSE; 12733bededecSBarry Smith 12743bededecSBarry Smith PetscFunctionBegin; 12753bededecSBarry Smith /* fix right hand side if needed */ 12763bededecSBarry Smith if (x && b) { 12773bededecSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 12783bededecSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 127956777dd2SBarry Smith vecs = PETSC_TRUE; 12803bededecSBarry Smith } 12813bededecSBarry Smith 12823bededecSBarry Smith /* zero the columns */ 12831795a4d1SJed Brown ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr); 12843bededecSBarry Smith for (i=0; i<is_n; i++) { 12853bededecSBarry 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]); 12863bededecSBarry Smith zeroed[is_idx[i]] = PETSC_TRUE; 12873bededecSBarry Smith } 128856777dd2SBarry Smith if (vecs) { 128956777dd2SBarry Smith for (i=0; i<A->rmap->N; i++) { 129056777dd2SBarry Smith row = i/bs; 129156777dd2SBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 129256777dd2SBarry Smith for (k=0; k<bs; k++) { 129356777dd2SBarry Smith col = bs*baij->j[j] + k; 129456777dd2SBarry Smith if (col <= i) continue; 129556777dd2SBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 129626fbe8dcSKarl Rupp if (!zeroed[i] && zeroed[col]) bb[i] -= aa[0]*xx[col]; 129726fbe8dcSKarl Rupp if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i]; 129856777dd2SBarry Smith } 129956777dd2SBarry Smith } 130056777dd2SBarry Smith } 130126fbe8dcSKarl Rupp for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]]; 130256777dd2SBarry Smith } 130356777dd2SBarry Smith 13043bededecSBarry Smith for (i=0; i<A->rmap->N; i++) { 13053bededecSBarry Smith if (!zeroed[i]) { 13063bededecSBarry Smith row = i/bs; 13073bededecSBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 13083bededecSBarry Smith for (k=0; k<bs; k++) { 13093bededecSBarry Smith col = bs*baij->j[j] + k; 13103bededecSBarry Smith if (zeroed[col]) { 13113bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 13123bededecSBarry Smith aa[0] = 0.0; 13133bededecSBarry Smith } 13143bededecSBarry Smith } 13153bededecSBarry Smith } 13163bededecSBarry Smith } 13173bededecSBarry Smith } 13183bededecSBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 131956777dd2SBarry Smith if (vecs) { 132056777dd2SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 132156777dd2SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 132256777dd2SBarry Smith } 13233bededecSBarry Smith 13243bededecSBarry Smith /* zero the rows */ 13253bededecSBarry Smith for (i=0; i<is_n; i++) { 13263bededecSBarry Smith row = is_idx[i]; 13273bededecSBarry Smith count = (baij->i[row/bs +1] - baij->i[row/bs])*bs; 13283bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs); 13293bededecSBarry Smith for (k=0; k<count; k++) { 13303bededecSBarry Smith aa[0] = zero; 13313bededecSBarry Smith aa += bs; 13323bededecSBarry Smith } 13333bededecSBarry Smith if (diag != 0.0) { 13343bededecSBarry Smith ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr); 13353bededecSBarry Smith } 13363bededecSBarry Smith } 13373bededecSBarry Smith ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 13383bededecSBarry Smith PetscFunctionReturn(0); 13393bededecSBarry Smith } 13403bededecSBarry Smith 134149b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 13423964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 134349b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 134449b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 134549b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 134697304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1347431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1348e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 1349db4efbfdSBarry Smith 0, 135049b5e25fSSatish Balay 0, 135149b5e25fSSatish Balay 0, 135297304618SKris Buschelman /* 10*/ 0, 135349b5e25fSSatish Balay 0, 1354c078aec8SLisandro Dalcin MatCholeskyFactor_SeqSBAIJ, 135541f059aeSBarry Smith MatSOR_SeqSBAIJ, 135649b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 135797304618SKris Buschelman /* 15*/ MatGetInfo_SeqSBAIJ, 135849b5e25fSSatish Balay MatEqual_SeqSBAIJ, 135949b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 136049b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 136149b5e25fSSatish Balay MatNorm_SeqSBAIJ, 136297304618SKris Buschelman /* 20*/ 0, 136349b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 136449b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 136549b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1366d519adbfSMatthew Knepley /* 24*/ 0, 136749b5e25fSSatish Balay 0, 136849b5e25fSSatish Balay 0, 1369db4efbfdSBarry Smith 0, 1370db4efbfdSBarry Smith 0, 13714994cf47SJed Brown /* 29*/ MatSetUp_SeqSBAIJ, 1372c464158bSHong Zhang 0, 1373db4efbfdSBarry Smith 0, 13748c778c55SBarry Smith 0, 13758c778c55SBarry Smith 0, 1376d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqSBAIJ, 1377719d5645SBarry Smith 0, 1378719d5645SBarry Smith 0, 137949b5e25fSSatish Balay 0, 1380c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 1381d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqSBAIJ, 138249b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 138349b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 138449b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 13853c896bc6SHong Zhang MatCopy_SeqSBAIJ, 1386d519adbfSMatthew Knepley /* 44*/ 0, 138749b5e25fSSatish Balay MatScale_SeqSBAIJ, 138849b5e25fSSatish Balay 0, 138949b5e25fSSatish Balay 0, 13903bededecSBarry Smith MatZeroRowsColumns_SeqSBAIJ, 1391f73d5cc4SBarry Smith /* 49*/ 0, 139249b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 139349b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 139449b5e25fSSatish Balay 0, 139549b5e25fSSatish Balay 0, 1396d519adbfSMatthew Knepley /* 54*/ 0, 139749b5e25fSSatish Balay 0, 139849b5e25fSSatish Balay 0, 139949b5e25fSSatish Balay 0, 140049b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 1401d519adbfSMatthew Knepley /* 59*/ MatGetSubMatrix_SeqSBAIJ, 140249b5e25fSSatish Balay 0, 140349b5e25fSSatish Balay 0, 1404357abbc8SBarry Smith 0, 1405d959ec07SHong Zhang 0, 1406d519adbfSMatthew Knepley /* 64*/ 0, 1407d959ec07SHong Zhang 0, 1408d959ec07SHong Zhang 0, 1409d959ec07SHong Zhang 0, 1410d959ec07SHong Zhang 0, 1411d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqSBAIJ, 14123e0d88b5SBarry Smith 0, 14133e0d88b5SBarry Smith 0, 14143e0d88b5SBarry Smith 0, 14153e0d88b5SBarry Smith 0, 1416d519adbfSMatthew Knepley /* 74*/ 0, 14173e0d88b5SBarry Smith 0, 14183e0d88b5SBarry Smith 0, 14193e0d88b5SBarry Smith 0, 14203e0d88b5SBarry Smith 0, 1421d519adbfSMatthew Knepley /* 79*/ 0, 14223e0d88b5SBarry Smith 0, 14233e0d88b5SBarry Smith 0, 142497304618SKris Buschelman MatGetInertia_SeqSBAIJ, 14255bba2384SShri Abhyankar MatLoad_SeqSBAIJ, 1426d519adbfSMatthew Knepley /* 84*/ MatIsSymmetric_SeqSBAIJ, 1427865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1428efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1429865e5f61SKris Buschelman 0, 1430865e5f61SKris Buschelman 0, 1431d519adbfSMatthew Knepley /* 89*/ 0, 1432865e5f61SKris Buschelman 0, 1433865e5f61SKris Buschelman 0, 1434865e5f61SKris Buschelman 0, 1435865e5f61SKris Buschelman 0, 1436d519adbfSMatthew Knepley /* 94*/ 0, 1437865e5f61SKris Buschelman 0, 1438865e5f61SKris Buschelman 0, 143999cafbc1SBarry Smith 0, 144099cafbc1SBarry Smith 0, 1441d519adbfSMatthew Knepley /* 99*/ 0, 144299cafbc1SBarry Smith 0, 144399cafbc1SBarry Smith 0, 144499cafbc1SBarry Smith 0, 144599cafbc1SBarry Smith 0, 1446d519adbfSMatthew Knepley /*104*/ 0, 144799cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1448f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1449f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 14502af78befSBarry Smith MatRestoreRowUpperTriangular_SeqSBAIJ, 1451d519adbfSMatthew Knepley /*109*/ 0, 14522af78befSBarry Smith 0, 14532af78befSBarry Smith 0, 14542af78befSBarry Smith 0, 1455547795f9SHong Zhang MatMissingDiagonal_SeqSBAIJ, 1456547795f9SHong Zhang /*114*/ 0, 1457547795f9SHong Zhang 0, 1458547795f9SHong Zhang 0, 1459547795f9SHong Zhang 0, 1460547795f9SHong Zhang 0, 1461547795f9SHong Zhang /*119*/ 0, 1462547795f9SHong Zhang 0, 14632f480046SShri Abhyankar 0, 14643964eb88SJed Brown 0, 14653964eb88SJed Brown 0, 14663964eb88SJed Brown /*124*/ 0, 14673964eb88SJed Brown 0, 14683964eb88SJed Brown 0, 14693964eb88SJed Brown 0, 14703964eb88SJed Brown 0, 14713964eb88SJed Brown /*129*/ 0, 14723964eb88SJed Brown 0, 14733964eb88SJed Brown 0, 14743964eb88SJed Brown 0, 14753964eb88SJed Brown 0, 14763964eb88SJed Brown /*134*/ 0, 14773964eb88SJed Brown 0, 14783964eb88SJed Brown 0, 14793964eb88SJed Brown 0, 14803964eb88SJed Brown 0, 14813964eb88SJed Brown /*139*/ 0, 1482f9426fe0SMark Adams 0, 148359f5e6ceSHong Zhang 0, 148459f5e6ceSHong Zhang 0, 148559f5e6ceSHong Zhang 0, 148659f5e6ceSHong Zhang /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ 148799cafbc1SBarry Smith }; 1488be1d678aSKris Buschelman 14894a2ae208SSatish Balay #undef __FUNCT__ 14904a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ" 14917087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 149249b5e25fSSatish Balay { 14934afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 1494d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 1495dfbe8321SBarry Smith PetscErrorCode ierr; 149649b5e25fSSatish Balay 149749b5e25fSSatish Balay PetscFunctionBegin; 1498e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 149949b5e25fSSatish Balay 150049b5e25fSSatish Balay /* allocate space for values if not already there */ 150149b5e25fSSatish Balay if (!aij->saved_values) { 1502785e854fSJed Brown ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr); 150349b5e25fSSatish Balay } 150449b5e25fSSatish Balay 150549b5e25fSSatish Balay /* copy values over */ 150687828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 150749b5e25fSSatish Balay PetscFunctionReturn(0); 150849b5e25fSSatish Balay } 150949b5e25fSSatish Balay 15104a2ae208SSatish Balay #undef __FUNCT__ 15114a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ" 15127087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 151349b5e25fSSatish Balay { 15144afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 15156849ba73SBarry Smith PetscErrorCode ierr; 1516d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 151749b5e25fSSatish Balay 151849b5e25fSSatish Balay PetscFunctionBegin; 1519e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1520e7e72b3dSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 152149b5e25fSSatish Balay 152249b5e25fSSatish Balay /* copy values over */ 152387828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 152449b5e25fSSatish Balay PetscFunctionReturn(0); 152549b5e25fSSatish Balay } 152649b5e25fSSatish Balay 15274a2ae208SSatish Balay #undef __FUNCT__ 1528a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ" 15297087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 153049b5e25fSSatish Balay { 1531c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 15326849ba73SBarry Smith PetscErrorCode ierr; 15334dcd73b1SHong Zhang PetscInt i,mbs,nbs,bs2; 15342576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE; 153549b5e25fSSatish Balay 153649b5e25fSSatish Balay PetscFunctionBegin; 15372576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 1538273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1539db4efbfdSBarry Smith 154033d57670SJed Brown ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr); 154126283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 154226283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1543e02043d6SBarry Smith ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 1544899cda47SBarry Smith 1545d0f46423SBarry Smith mbs = B->rmap->N/bs; 15464dcd73b1SHong Zhang nbs = B->cmap->n/bs; 154749b5e25fSSatish Balay bs2 = bs*bs; 154849b5e25fSSatish Balay 15494dcd73b1SHong 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"); 155049b5e25fSSatish Balay 1551ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1552ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1553ab93d7beSBarry Smith nz = 0; 1554ab93d7beSBarry Smith } 1555ab93d7beSBarry Smith 1556435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 1557e32f2f54SBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 155849b5e25fSSatish Balay if (nnz) { 155949b5e25fSSatish Balay for (i=0; i<mbs; i++) { 1560e32f2f54SBarry 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]); 1561de64b629SHong 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); 156249b5e25fSSatish Balay } 156349b5e25fSSatish Balay } 156449b5e25fSSatish Balay 1565db4efbfdSBarry Smith B->ops->mult = MatMult_SeqSBAIJ_N; 1566db4efbfdSBarry Smith B->ops->multadd = MatMultAdd_SeqSBAIJ_N; 1567db4efbfdSBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_N; 1568db4efbfdSBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N; 156926fbe8dcSKarl Rupp 15700298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 157149b5e25fSSatish Balay if (!flg) { 157249b5e25fSSatish Balay switch (bs) { 157349b5e25fSSatish Balay case 1: 157449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 157549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1576431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1577431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 157849b5e25fSSatish Balay break; 157949b5e25fSSatish Balay case 2: 158049b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 158149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1582431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1583431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 158449b5e25fSSatish Balay break; 158549b5e25fSSatish Balay case 3: 158649b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 158749b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1588431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1589431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 159049b5e25fSSatish Balay break; 159149b5e25fSSatish Balay case 4: 159249b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 159349b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1594431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1595431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 159649b5e25fSSatish Balay break; 159749b5e25fSSatish Balay case 5: 159849b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 159949b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1600431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1601431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 160249b5e25fSSatish Balay break; 160349b5e25fSSatish Balay case 6: 160449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 160549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1606431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1607431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 160849b5e25fSSatish Balay break; 160949b5e25fSSatish Balay case 7: 1610de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 161149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1612431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1613431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 161449b5e25fSSatish Balay break; 161549b5e25fSSatish Balay } 161649b5e25fSSatish Balay } 161749b5e25fSSatish Balay 161849b5e25fSSatish Balay b->mbs = mbs; 16194dcd73b1SHong Zhang b->nbs = nbs; 1620ab93d7beSBarry Smith if (!skipallocation) { 16212ee49352SLisandro Dalcin if (!b->imax) { 1622dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr); 162326fbe8dcSKarl Rupp 1624c760cd28SBarry Smith b->free_imax_ilen = PETSC_TRUE; 162526fbe8dcSKarl Rupp 16263bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr); 16272ee49352SLisandro Dalcin } 162849b5e25fSSatish Balay if (!nnz) { 1629435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 163049b5e25fSSatish Balay else if (nz <= 0) nz = 1; 163126fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->imax[i] = nz; 1632153ea458SHong Zhang nz = nz*mbs; /* total nz */ 163349b5e25fSSatish Balay } else { 163449b5e25fSSatish Balay nz = 0; 16358cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 163649b5e25fSSatish Balay } 16372ee49352SLisandro Dalcin /* b->ilen will count nonzeros in each block row so far. */ 163826fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->ilen[i] = 0; 16396c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 164049b5e25fSSatish Balay 164149b5e25fSSatish Balay /* allocate the matrix space */ 16422ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 1643dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr); 16443bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 16456c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 164613f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 164726fbe8dcSKarl Rupp 164849b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 164949b5e25fSSatish Balay 165049b5e25fSSatish Balay /* pointer to beginning of each row */ 1651e60cf9a0SBarry Smith b->i[0] = 0; 165226fbe8dcSKarl Rupp for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1]; 165326fbe8dcSKarl Rupp 1654e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1655e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1656e811da20SHong Zhang } else { 1657e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1658e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1659ab93d7beSBarry Smith } 166049b5e25fSSatish Balay 1661d0f46423SBarry Smith B->rmap->bs = bs; 166249b5e25fSSatish Balay b->bs2 = bs2; 16636c6c5352SBarry Smith b->nz = 0; 1664b32cb4a7SJed Brown b->maxnz = nz; 1665153ea458SHong Zhang 166616cdd363SHong Zhang b->inew = 0; 166716cdd363SHong Zhang b->jnew = 0; 166816cdd363SHong Zhang b->anew = 0; 166916cdd363SHong Zhang b->a2anew = 0; 16701a3463dfSHong Zhang b->permute = PETSC_FALSE; 16712576faa2SJed Brown if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);} 1672c464158bSHong Zhang PetscFunctionReturn(0); 1673c464158bSHong Zhang } 1674153ea458SHong Zhang 167538f409ebSLisandro Dalcin #undef __FUNCT__ 167638f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ" 167738f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[]) 167838f409ebSLisandro Dalcin { 167938f409ebSLisandro Dalcin PetscInt i,j,m,nz,nz_max=0,*nnz; 168038f409ebSLisandro Dalcin PetscScalar *values=0; 168138f409ebSLisandro Dalcin PetscBool roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented; 168238f409ebSLisandro Dalcin PetscErrorCode ierr; 168338f409ebSLisandro Dalcin PetscFunctionBegin; 168438f409ebSLisandro Dalcin if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs); 168538f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr); 168638f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr); 168738f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 168838f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 168938f409ebSLisandro Dalcin ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 169038f409ebSLisandro Dalcin m = B->rmap->n/bs; 169138f409ebSLisandro Dalcin 169238f409ebSLisandro Dalcin if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]); 1693785e854fSJed Brown ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr); 169438f409ebSLisandro Dalcin for (i=0; i<m; i++) { 169538f409ebSLisandro Dalcin nz = ii[i+1] - ii[i]; 169638f409ebSLisandro Dalcin if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz); 169738f409ebSLisandro Dalcin nz_max = PetscMax(nz_max,nz); 169838f409ebSLisandro Dalcin nnz[i] = nz; 169938f409ebSLisandro Dalcin } 170038f409ebSLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 170138f409ebSLisandro Dalcin ierr = PetscFree(nnz);CHKERRQ(ierr); 170238f409ebSLisandro Dalcin 170338f409ebSLisandro Dalcin values = (PetscScalar*)V; 170438f409ebSLisandro Dalcin if (!values) { 17051795a4d1SJed Brown ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr); 170638f409ebSLisandro Dalcin } 170738f409ebSLisandro Dalcin for (i=0; i<m; i++) { 170838f409ebSLisandro Dalcin PetscInt ncols = ii[i+1] - ii[i]; 170938f409ebSLisandro Dalcin const PetscInt *icols = jj + ii[i]; 171038f409ebSLisandro Dalcin if (!roworiented || bs == 1) { 171138f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0); 171238f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr); 171338f409ebSLisandro Dalcin } else { 171438f409ebSLisandro Dalcin for (j=0; j<ncols; j++) { 171538f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0); 171638f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr); 171738f409ebSLisandro Dalcin } 171838f409ebSLisandro Dalcin } 171938f409ebSLisandro Dalcin } 172038f409ebSLisandro Dalcin if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); } 172138f409ebSLisandro Dalcin ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 172238f409ebSLisandro Dalcin ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 172338f409ebSLisandro Dalcin ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 172438f409ebSLisandro Dalcin PetscFunctionReturn(0); 172538f409ebSLisandro Dalcin } 172638f409ebSLisandro Dalcin 1727db4efbfdSBarry Smith /* 1728db4efbfdSBarry Smith This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization 1729db4efbfdSBarry Smith */ 17308b1456e3SHong Zhang #undef __FUNCT__ 1731d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace" 1732ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural) 1733db4efbfdSBarry Smith { 1734db4efbfdSBarry Smith PetscErrorCode ierr; 1735ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 1736db4efbfdSBarry Smith PetscInt bs = B->rmap->bs; 1737db4efbfdSBarry Smith 1738db4efbfdSBarry Smith PetscFunctionBegin; 17390298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 1740db4efbfdSBarry Smith if (flg) bs = 8; 1741db4efbfdSBarry Smith 1742db4efbfdSBarry Smith if (!natural) { 1743db4efbfdSBarry Smith switch (bs) { 1744db4efbfdSBarry Smith case 1: 1745d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace; 1746db4efbfdSBarry Smith break; 1747db4efbfdSBarry Smith case 2: 1748db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 1749db4efbfdSBarry Smith break; 1750db4efbfdSBarry Smith case 3: 1751db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 1752db4efbfdSBarry Smith break; 1753db4efbfdSBarry Smith case 4: 1754db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 1755db4efbfdSBarry Smith break; 1756db4efbfdSBarry Smith case 5: 1757db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 1758db4efbfdSBarry Smith break; 1759db4efbfdSBarry Smith case 6: 1760db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 1761db4efbfdSBarry Smith break; 1762db4efbfdSBarry Smith case 7: 1763db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 1764db4efbfdSBarry Smith break; 1765db4efbfdSBarry Smith default: 1766db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N; 1767db4efbfdSBarry Smith break; 1768db4efbfdSBarry Smith } 1769db4efbfdSBarry Smith } else { 1770db4efbfdSBarry Smith switch (bs) { 1771db4efbfdSBarry Smith case 1: 1772d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace; 1773db4efbfdSBarry Smith break; 1774db4efbfdSBarry Smith case 2: 1775db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 1776db4efbfdSBarry Smith break; 1777db4efbfdSBarry Smith case 3: 1778db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 1779db4efbfdSBarry Smith break; 1780db4efbfdSBarry Smith case 4: 1781db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 1782db4efbfdSBarry Smith break; 1783db4efbfdSBarry Smith case 5: 1784db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 1785db4efbfdSBarry Smith break; 1786db4efbfdSBarry Smith case 6: 1787db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 1788db4efbfdSBarry Smith break; 1789db4efbfdSBarry Smith case 7: 1790db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 1791db4efbfdSBarry Smith break; 1792db4efbfdSBarry Smith default: 1793db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1794db4efbfdSBarry Smith break; 1795db4efbfdSBarry Smith } 1796db4efbfdSBarry Smith } 1797db4efbfdSBarry Smith PetscFunctionReturn(0); 1798db4efbfdSBarry Smith } 1799db4efbfdSBarry Smith 18008cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*); 18018cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*); 1802d769727bSBarry Smith 18035c9eb25fSBarry Smith #undef __FUNCT__ 18045c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc" 18058cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B) 18065c9eb25fSBarry Smith { 1807d0f46423SBarry Smith PetscInt n = A->rmap->n; 18085c9eb25fSBarry Smith PetscErrorCode ierr; 18095c9eb25fSBarry Smith 18105c9eb25fSBarry Smith PetscFunctionBegin; 18110e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX) 18120e92d65fSHong Zhang if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 18130e92d65fSHong Zhang #endif 1814ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 18155c9eb25fSBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 18165c9eb25fSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 18175c9eb25fSBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 18180298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 181926fbe8dcSKarl Rupp 18207b056e98SHong Zhang (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ; 1821c6d0d4f0SHong Zhang (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqSBAIJ; 1822e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 1823d5f3da31SBarry Smith (*B)->factortype = ftype; 18245c9eb25fSBarry Smith PetscFunctionReturn(0); 18255c9eb25fSBarry Smith } 18265c9eb25fSBarry Smith 18270bad9183SKris Buschelman /*MC 1828fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 18290bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 18300bad9183SKris Buschelman 1831828413b8SBarry Smith For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you 183271dad5bbSBarry Smith can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd() 1833828413b8SBarry Smith 18340bad9183SKris Buschelman Options Database Keys: 18350bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 18360bad9183SKris Buschelman 183771dad5bbSBarry Smith Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not 183871dad5bbSBarry 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 183971dad5bbSBarry 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. 184071dad5bbSBarry Smith 184171dad5bbSBarry Smith 18420bad9183SKris Buschelman Level: beginner 18430bad9183SKris Buschelman 18440bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 18450bad9183SKris Buschelman M*/ 18460bad9183SKris Buschelman 18478cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*); 1848aa5a9175SDahai Guo 1849a23d5eceSKris Buschelman #undef __FUNCT__ 1850a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ" 18518cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1852a23d5eceSKris Buschelman { 1853a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1854dfbe8321SBarry Smith PetscErrorCode ierr; 185513f74950SBarry Smith PetscMPIInt size; 1856ace3abfcSBarry Smith PetscBool no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE; 1857a23d5eceSKris Buschelman 1858a23d5eceSKris Buschelman PetscFunctionBegin; 1859ce94432eSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 1860e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1861a23d5eceSKris Buschelman 1862b00a9115SJed Brown ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1863a23d5eceSKris Buschelman B->data = (void*)b; 1864a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 186526fbe8dcSKarl Rupp 1866a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1867a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1868a23d5eceSKris Buschelman b->row = 0; 1869a23d5eceSKris Buschelman b->icol = 0; 1870a23d5eceSKris Buschelman b->reallocs = 0; 1871a23d5eceSKris Buschelman b->saved_values = 0; 18720def2e27SBarry Smith b->inode.limit = 5; 18730def2e27SBarry Smith b->inode.max_limit = 5; 1874a23d5eceSKris Buschelman 1875a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1876a23d5eceSKris Buschelman b->nonew = 0; 1877a23d5eceSKris Buschelman b->diag = 0; 1878a23d5eceSKris Buschelman b->solve_work = 0; 1879a23d5eceSKris Buschelman b->mult_work = 0; 1880a23d5eceSKris Buschelman B->spptr = 0; 1881f2cbd3d5SJed Brown B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2; 1882a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 1883a23d5eceSKris Buschelman 1884a23d5eceSKris Buschelman b->inew = 0; 1885a23d5eceSKris Buschelman b->jnew = 0; 1886a23d5eceSKris Buschelman b->anew = 0; 1887a23d5eceSKris Buschelman b->a2anew = 0; 1888a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1889a23d5eceSKris Buschelman 189071dad5bbSBarry Smith b->ignore_ltriangular = PETSC_TRUE; 189126fbe8dcSKarl Rupp 18920298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr); 1893941593c8SHong Zhang 1894f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 189526fbe8dcSKarl Rupp 18960298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr); 1897f5edf698SHong Zhang 1898bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1899bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1900bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 1901bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1902bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1903bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 190438f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr); 1905bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr); 190623ce1328SBarry Smith 190723ce1328SBarry Smith B->symmetric = PETSC_TRUE; 190823ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 190923ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 191023ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 191126fbe8dcSKarl Rupp 191217667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 19130def2e27SBarry Smith 1914ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr); 19150298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr); 191626fbe8dcSKarl Rupp if (no_unroll) { 191726fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr); 191826fbe8dcSKarl Rupp } 19190298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr); 192026fbe8dcSKarl Rupp if (no_inode) { 192126fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr); 192226fbe8dcSKarl Rupp } 19230298fd71SBarry Smith ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr); 19240def2e27SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 1925ace3abfcSBarry Smith b->inode.use = (PetscBool)(!(no_unroll || no_inode)); 19260def2e27SBarry Smith if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit; 1927a23d5eceSKris Buschelman PetscFunctionReturn(0); 1928a23d5eceSKris Buschelman } 1929a23d5eceSKris Buschelman 1930a23d5eceSKris Buschelman #undef __FUNCT__ 1931a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation" 1932a23d5eceSKris Buschelman /*@C 1933a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 1934a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 1935a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 1936a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 1937a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 1938a23d5eceSKris Buschelman 1939a23d5eceSKris Buschelman Collective on Mat 1940a23d5eceSKris Buschelman 1941a23d5eceSKris Buschelman Input Parameters: 19421c4f3114SJed Brown + B - the symmetric matrix 1943bb7ae925SBarry 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 1944bb7ae925SBarry Smith blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs() 1945a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 1946a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 19470298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 1948a23d5eceSKris Buschelman 1949a23d5eceSKris Buschelman Options Database Keys: 1950a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 1951a23d5eceSKris Buschelman block calculations (much slower) 1952db4efbfdSBarry Smith . -mat_block_size - size of the blocks to use (only works if a negative bs is passed in 1953a23d5eceSKris Buschelman 1954a23d5eceSKris Buschelman Level: intermediate 1955a23d5eceSKris Buschelman 1956a23d5eceSKris Buschelman Notes: 1957a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 19580298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 1959a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 1960a23d5eceSKris Buschelman 1961aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 1962aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 1963aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 1964aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 1965aa95bbe8SBarry Smith 196649a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 196749a6f317SBarry Smith 196849a6f317SBarry Smith 196969b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 1970a23d5eceSKris Buschelman @*/ 19717087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 197213f74950SBarry Smith { 19734ac538c5SBarry Smith PetscErrorCode ierr; 1974a23d5eceSKris Buschelman 1975a23d5eceSKris Buschelman PetscFunctionBegin; 19766ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 19776ba663aaSJed Brown PetscValidType(B,1); 19786ba663aaSJed Brown PetscValidLogicalCollectiveInt(B,bs,2); 19794ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr); 1980a23d5eceSKris Buschelman PetscFunctionReturn(0); 1981a23d5eceSKris Buschelman } 198249b5e25fSSatish Balay 19834a2ae208SSatish Balay #undef __FUNCT__ 198438f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR" 198538f409ebSLisandro Dalcin /*@C 198638f409ebSLisandro Dalcin MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format. 198738f409ebSLisandro Dalcin 198838f409ebSLisandro Dalcin Input Parameters: 19891c4f3114SJed Brown + B - the matrix 199038f409ebSLisandro Dalcin . i - the indices into j for the start of each local row (starts with zero) 199138f409ebSLisandro Dalcin . j - the column indices for each local row (starts with zero) these must be sorted for each row 199238f409ebSLisandro Dalcin - v - optional values in the matrix 199338f409ebSLisandro Dalcin 199438f409ebSLisandro Dalcin Level: developer 199538f409ebSLisandro Dalcin 199638f409ebSLisandro Dalcin Notes: 199738f409ebSLisandro Dalcin The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED. For example, C programs 199838f409ebSLisandro 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 199938f409ebSLisandro Dalcin over rows within a block and the last index is over columns within a block row. Fortran programs will likely set 200038f409ebSLisandro 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 200138f409ebSLisandro Dalcin block column and the second index is over columns within a block. 200238f409ebSLisandro Dalcin 200338f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse 200438f409ebSLisandro Dalcin 200538f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ 200638f409ebSLisandro Dalcin @*/ 200738f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[]) 200838f409ebSLisandro Dalcin { 200938f409ebSLisandro Dalcin PetscErrorCode ierr; 201038f409ebSLisandro Dalcin 201138f409ebSLisandro Dalcin PetscFunctionBegin; 201238f409ebSLisandro Dalcin PetscValidHeaderSpecific(B,MAT_CLASSID,1); 201338f409ebSLisandro Dalcin PetscValidType(B,1); 201438f409ebSLisandro Dalcin PetscValidLogicalCollectiveInt(B,bs,2); 201538f409ebSLisandro Dalcin ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr); 201638f409ebSLisandro Dalcin PetscFunctionReturn(0); 201738f409ebSLisandro Dalcin } 201838f409ebSLisandro Dalcin 201938f409ebSLisandro Dalcin #undef __FUNCT__ 20204a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ" 2021c464158bSHong Zhang /*@C 2022c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 2023c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 2024c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 2025c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 2026c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 202749b5e25fSSatish Balay 2028c464158bSHong Zhang Collective on MPI_Comm 2029c464158bSHong Zhang 2030c464158bSHong Zhang Input Parameters: 2031c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2032bb7ae925SBarry 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 2033bb7ae925SBarry Smith blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs() 2034c464158bSHong Zhang . m - number of rows, or number of columns 2035c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 2036744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 20370298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2038c464158bSHong Zhang 2039c464158bSHong Zhang Output Parameter: 2040c464158bSHong Zhang . A - the symmetric matrix 2041c464158bSHong Zhang 2042c464158bSHong Zhang Options Database Keys: 2043c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 2044c464158bSHong Zhang block calculations (much slower) 2045c464158bSHong Zhang . -mat_block_size - size of the blocks to use 2046c464158bSHong Zhang 2047c464158bSHong Zhang Level: intermediate 2048c464158bSHong Zhang 2049175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2050ae1d86c5SBarry Smith MatXXXXSetPreallocation() paradgm instead of this routine directly. 2051175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 2052175b88e8SBarry Smith 2053c464158bSHong Zhang Notes: 20546d6d819aSHong Zhang The number of rows and columns must be divisible by blocksize. 20556d6d819aSHong Zhang This matrix type does not support complex Hermitian operation. 2056c464158bSHong Zhang 2057c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 20580298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 2059a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 2060c464158bSHong Zhang 206149a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 206249a6f317SBarry Smith 206369b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2064c464158bSHong Zhang @*/ 20657087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 2066c464158bSHong Zhang { 2067dfbe8321SBarry Smith PetscErrorCode ierr; 2068c464158bSHong Zhang 2069c464158bSHong Zhang PetscFunctionBegin; 2070f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 2071f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2072c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 2073ab93d7beSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 207449b5e25fSSatish Balay PetscFunctionReturn(0); 207549b5e25fSSatish Balay } 207649b5e25fSSatish Balay 20774a2ae208SSatish Balay #undef __FUNCT__ 20784a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ" 2079dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 208049b5e25fSSatish Balay { 208149b5e25fSSatish Balay Mat C; 208249b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 20836849ba73SBarry Smith PetscErrorCode ierr; 2084b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 208549b5e25fSSatish Balay 208649b5e25fSSatish Balay PetscFunctionBegin; 2087e32f2f54SBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix"); 208849b5e25fSSatish Balay 208949b5e25fSSatish Balay *B = 0; 2090ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2091d0f46423SBarry Smith ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr); 20928e9a0fb8SHong Zhang ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr); 20931d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 2094692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 2095692f9cbeSHong Zhang 2096273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 2097d5f3da31SBarry Smith C->factortype = A->factortype; 209849b5e25fSSatish Balay c->row = 0; 209949b5e25fSSatish Balay c->icol = 0; 210049b5e25fSSatish Balay c->saved_values = 0; 2101a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 210249b5e25fSSatish Balay C->assembled = PETSC_TRUE; 210349b5e25fSSatish Balay 21041e1e43feSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 21051e1e43feSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 210649b5e25fSSatish Balay c->bs2 = a->bs2; 210749b5e25fSSatish Balay c->mbs = a->mbs; 210849b5e25fSSatish Balay c->nbs = a->nbs; 210949b5e25fSSatish Balay 2110c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2111c760cd28SBarry Smith c->imax = a->imax; 2112c760cd28SBarry Smith c->ilen = a->ilen; 2113c760cd28SBarry Smith c->free_imax_ilen = PETSC_FALSE; 2114c760cd28SBarry Smith } else { 2115dcca6d9dSJed Brown ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr); 21163bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 211749b5e25fSSatish Balay for (i=0; i<mbs; i++) { 211849b5e25fSSatish Balay c->imax[i] = a->imax[i]; 211949b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 212049b5e25fSSatish Balay } 2121c760cd28SBarry Smith c->free_imax_ilen = PETSC_TRUE; 2122c760cd28SBarry Smith } 212349b5e25fSSatish Balay 212449b5e25fSSatish Balay /* allocate the matrix space */ 21254da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2126785e854fSJed Brown ierr = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr); 21273bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 212844e1c64aSLisandro Dalcin c->i = a->i; 212944e1c64aSLisandro Dalcin c->j = a->j; 21304da8f245SBarry Smith c->singlemalloc = PETSC_FALSE; 213144e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 21324da8f245SBarry Smith c->free_ij = PETSC_FALSE; 21334da8f245SBarry Smith c->parent = A; 21344da8f245SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 21354da8f245SBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21364da8f245SBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21374da8f245SBarry Smith } else { 2138dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr); 213913f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 21403bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 21414da8f245SBarry Smith c->singlemalloc = PETSC_TRUE; 214244e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 21434da8f245SBarry Smith c->free_ij = PETSC_TRUE; 21444da8f245SBarry Smith } 214549b5e25fSSatish Balay if (mbs > 0) { 21464da8f245SBarry Smith if (cpvalues != MAT_SHARE_NONZERO_PATTERN) { 214713f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 21484da8f245SBarry Smith } 214949b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 215049b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 215149b5e25fSSatish Balay } else { 215249b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 215349b5e25fSSatish Balay } 2154a1c3900fSBarry Smith if (a->jshort) { 215544e1c64aSLisandro Dalcin /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */ 215644e1c64aSLisandro Dalcin /* if the parent matrix is reassembled, this child matrix will never notice */ 2157785e854fSJed Brown ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr); 21583bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr); 2159a1c3900fSBarry Smith ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr); 216026fbe8dcSKarl Rupp 21614da8f245SBarry Smith c->free_jshort = PETSC_TRUE; 21624da8f245SBarry Smith } 2163a1c3900fSBarry Smith } 216449b5e25fSSatish Balay 216549b5e25fSSatish Balay c->roworiented = a->roworiented; 216649b5e25fSSatish Balay c->nonew = a->nonew; 216749b5e25fSSatish Balay 216849b5e25fSSatish Balay if (a->diag) { 2169c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2170c760cd28SBarry Smith c->diag = a->diag; 2171c760cd28SBarry Smith c->free_diag = PETSC_FALSE; 2172c760cd28SBarry Smith } else { 2173785e854fSJed Brown ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr); 21743bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr); 217526fbe8dcSKarl Rupp for (i=0; i<mbs; i++) c->diag[i] = a->diag[i]; 2176c760cd28SBarry Smith c->free_diag = PETSC_TRUE; 2177c760cd28SBarry Smith } 217844e1c64aSLisandro Dalcin } 21796c6c5352SBarry Smith c->nz = a->nz; 2180f2cbd3d5SJed Brown c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 218149b5e25fSSatish Balay c->solve_work = 0; 218249b5e25fSSatish Balay c->mult_work = 0; 218326fbe8dcSKarl Rupp 218449b5e25fSSatish Balay *B = C; 2185140e18c1SBarry Smith ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 218649b5e25fSSatish Balay PetscFunctionReturn(0); 218749b5e25fSSatish Balay } 218849b5e25fSSatish Balay 21894a2ae208SSatish Balay #undef __FUNCT__ 21905bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ" 2191112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer) 21922f480046SShri Abhyankar { 21932f480046SShri Abhyankar Mat_SeqSBAIJ *a; 21942f480046SShri Abhyankar PetscErrorCode ierr; 21952f480046SShri Abhyankar int fd; 21962f480046SShri Abhyankar PetscMPIInt size; 21972f480046SShri Abhyankar PetscInt i,nz,header[4],*rowlengths=0,M,N,bs=1; 21982f480046SShri Abhyankar PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 21992f480046SShri Abhyankar PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols; 22002f480046SShri Abhyankar PetscInt *masked,nmask,tmp,bs2,ishift; 22012f480046SShri Abhyankar PetscScalar *aa; 2202ce94432eSBarry Smith MPI_Comm comm; 22032f480046SShri Abhyankar 22042f480046SShri Abhyankar PetscFunctionBegin; 2205ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 22060298fd71SBarry Smith ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr); 22072f480046SShri Abhyankar bs2 = bs*bs; 22082f480046SShri Abhyankar 22092f480046SShri Abhyankar ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 22102f480046SShri Abhyankar if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor"); 22112f480046SShri Abhyankar ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 22122f480046SShri Abhyankar ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 22132f480046SShri Abhyankar if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 22142f480046SShri Abhyankar M = header[1]; N = header[2]; nz = header[3]; 22152f480046SShri Abhyankar 22162f480046SShri Abhyankar if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 22172f480046SShri Abhyankar 22182f480046SShri Abhyankar if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices"); 22192f480046SShri Abhyankar 22202f480046SShri Abhyankar /* 22212f480046SShri Abhyankar This code adds extra rows to make sure the number of rows is 22222f480046SShri Abhyankar divisible by the blocksize 22232f480046SShri Abhyankar */ 22242f480046SShri Abhyankar mbs = M/bs; 22252f480046SShri Abhyankar extra_rows = bs - M + bs*(mbs); 22262f480046SShri Abhyankar if (extra_rows == bs) extra_rows = 0; 22272f480046SShri Abhyankar else mbs++; 22282f480046SShri Abhyankar if (extra_rows) { 22292f480046SShri Abhyankar ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr); 22302f480046SShri Abhyankar } 22312f480046SShri Abhyankar 22322f480046SShri Abhyankar /* Set global sizes if not already set */ 22332f480046SShri Abhyankar if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) { 22342f480046SShri Abhyankar ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr); 22352f480046SShri Abhyankar } else { /* Check if the matrix global sizes are correct */ 22362f480046SShri Abhyankar ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr); 22372f480046SShri 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); 22382f480046SShri Abhyankar } 22392f480046SShri Abhyankar 22402f480046SShri Abhyankar /* read in row lengths */ 2241785e854fSJed Brown ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr); 22422f480046SShri Abhyankar ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 22432f480046SShri Abhyankar for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 22442f480046SShri Abhyankar 22452f480046SShri Abhyankar /* read in column indices */ 2246785e854fSJed Brown ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr); 22472f480046SShri Abhyankar ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 22482f480046SShri Abhyankar for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 22492f480046SShri Abhyankar 22502f480046SShri Abhyankar /* loop over row lengths determining block row lengths */ 22511795a4d1SJed Brown ierr = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr); 2252dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr); 22532f480046SShri Abhyankar ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 22542f480046SShri Abhyankar rowcount = 0; 22552f480046SShri Abhyankar nzcount = 0; 22562f480046SShri Abhyankar for (i=0; i<mbs; i++) { 22572f480046SShri Abhyankar nmask = 0; 22582f480046SShri Abhyankar for (j=0; j<bs; j++) { 22592f480046SShri Abhyankar kmax = rowlengths[rowcount]; 22602f480046SShri Abhyankar for (k=0; k<kmax; k++) { 22612f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 22622f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 22632f480046SShri Abhyankar } 22642f480046SShri Abhyankar rowcount++; 22652f480046SShri Abhyankar } 22662f480046SShri Abhyankar s_browlengths[i] += nmask; 22672f480046SShri Abhyankar 22682f480046SShri Abhyankar /* zero out the mask elements we set */ 22692f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 22702f480046SShri Abhyankar } 22712f480046SShri Abhyankar 22722f480046SShri Abhyankar /* Do preallocation */ 22732f480046SShri Abhyankar ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr); 22742f480046SShri Abhyankar a = (Mat_SeqSBAIJ*)newmat->data; 22752f480046SShri Abhyankar 22762f480046SShri Abhyankar /* set matrix "i" values */ 22772f480046SShri Abhyankar a->i[0] = 0; 22782f480046SShri Abhyankar for (i=1; i<= mbs; i++) { 22792f480046SShri Abhyankar a->i[i] = a->i[i-1] + s_browlengths[i-1]; 22802f480046SShri Abhyankar a->ilen[i-1] = s_browlengths[i-1]; 22812f480046SShri Abhyankar } 22822f480046SShri Abhyankar a->nz = a->i[mbs]; 22832f480046SShri Abhyankar 22842f480046SShri Abhyankar /* read in nonzero values */ 2285785e854fSJed Brown ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr); 22862f480046SShri Abhyankar ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 22872f480046SShri Abhyankar for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 22882f480046SShri Abhyankar 22892f480046SShri Abhyankar /* set "a" and "j" values into matrix */ 22902f480046SShri Abhyankar nzcount = 0; jcount = 0; 22912f480046SShri Abhyankar for (i=0; i<mbs; i++) { 22922f480046SShri Abhyankar nzcountb = nzcount; 22932f480046SShri Abhyankar nmask = 0; 22942f480046SShri Abhyankar for (j=0; j<bs; j++) { 22952f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 22962f480046SShri Abhyankar for (k=0; k<kmax; k++) { 22972f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 22982f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 22992f480046SShri Abhyankar } 23002f480046SShri Abhyankar } 23012f480046SShri Abhyankar /* sort the masked values */ 23022f480046SShri Abhyankar ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 23032f480046SShri Abhyankar 23042f480046SShri Abhyankar /* set "j" values into matrix */ 23052f480046SShri Abhyankar maskcount = 1; 23062f480046SShri Abhyankar for (j=0; j<nmask; j++) { 23072f480046SShri Abhyankar a->j[jcount++] = masked[j]; 23082f480046SShri Abhyankar mask[masked[j]] = maskcount++; 23092f480046SShri Abhyankar } 23102f480046SShri Abhyankar 23112f480046SShri Abhyankar /* set "a" values into matrix */ 23122f480046SShri Abhyankar ishift = bs2*a->i[i]; 23132f480046SShri Abhyankar for (j=0; j<bs; j++) { 23142f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23152f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23162f480046SShri Abhyankar tmp = jj[nzcountb]/bs; /* block col. index */ 23172f480046SShri Abhyankar if (tmp >= i) { 23182f480046SShri Abhyankar block = mask[tmp] - 1; 23192f480046SShri Abhyankar point = jj[nzcountb] - bs*tmp; 23202f480046SShri Abhyankar idx = ishift + bs2*block + j + bs*point; 23212f480046SShri Abhyankar a->a[idx] = aa[nzcountb]; 23222f480046SShri Abhyankar } 23232f480046SShri Abhyankar nzcountb++; 23242f480046SShri Abhyankar } 23252f480046SShri Abhyankar } 23262f480046SShri Abhyankar /* zero out the mask elements we set */ 23272f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23282f480046SShri Abhyankar } 23292f480046SShri Abhyankar if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 23302f480046SShri Abhyankar 23312f480046SShri Abhyankar ierr = PetscFree(rowlengths);CHKERRQ(ierr); 23322f480046SShri Abhyankar ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 23332f480046SShri Abhyankar ierr = PetscFree(aa);CHKERRQ(ierr); 23342f480046SShri Abhyankar ierr = PetscFree(jj);CHKERRQ(ierr); 23352f480046SShri Abhyankar ierr = PetscFree2(mask,masked);CHKERRQ(ierr); 23362f480046SShri Abhyankar 23372f480046SShri Abhyankar ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23382f480046SShri Abhyankar ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23392f480046SShri Abhyankar PetscFunctionReturn(0); 23402f480046SShri Abhyankar } 23412f480046SShri Abhyankar 23422f480046SShri Abhyankar #undef __FUNCT__ 2343c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays" 2344c75a6043SHong Zhang /*@ 2345c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2346c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2347c75a6043SHong Zhang 2348c75a6043SHong Zhang Collective on MPI_Comm 2349c75a6043SHong Zhang 2350c75a6043SHong Zhang Input Parameters: 2351c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2352c75a6043SHong Zhang . bs - size of block 2353c75a6043SHong Zhang . m - number of rows 2354c75a6043SHong Zhang . n - number of columns 2355c75a6043SHong Zhang . i - row indices 2356c75a6043SHong Zhang . j - column indices 2357c75a6043SHong Zhang - a - matrix values 2358c75a6043SHong Zhang 2359c75a6043SHong Zhang Output Parameter: 2360c75a6043SHong Zhang . mat - the matrix 2361c75a6043SHong Zhang 2362dfb205c3SBarry Smith Level: advanced 2363c75a6043SHong Zhang 2364c75a6043SHong Zhang Notes: 2365c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2366c75a6043SHong Zhang once the matrix is destroyed 2367c75a6043SHong Zhang 2368c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2369c75a6043SHong Zhang 2370c75a6043SHong Zhang The i and j indices are 0 based 2371c75a6043SHong Zhang 2372dfb205c3SBarry 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 2373dfb205c3SBarry Smith it is the regular CSR format excluding the lower triangular elements. 2374dfb205c3SBarry Smith 237569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ() 2376c75a6043SHong Zhang 2377c75a6043SHong Zhang @*/ 23787087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat) 2379c75a6043SHong Zhang { 2380c75a6043SHong Zhang PetscErrorCode ierr; 2381c75a6043SHong Zhang PetscInt ii; 2382c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2383c75a6043SHong Zhang 2384c75a6043SHong Zhang PetscFunctionBegin; 2385e32f2f54SBarry Smith if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 2386e32f2f54SBarry Smith if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2387c75a6043SHong Zhang 2388c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2389c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2390c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2391c75a6043SHong Zhang ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 2392c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2393dcca6d9dSJed Brown ierr = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr); 23943bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr); 2395c75a6043SHong Zhang 2396c75a6043SHong Zhang sbaij->i = i; 2397c75a6043SHong Zhang sbaij->j = j; 2398c75a6043SHong Zhang sbaij->a = a; 239926fbe8dcSKarl Rupp 2400c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2401c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2402e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2403e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2404c75a6043SHong Zhang 2405c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2406c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2407c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2408e32f2f54SBarry 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]); 2409c75a6043SHong Zhang #endif 2410c75a6043SHong Zhang } 2411c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2412c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2413e32f2f54SBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2414e32f2f54SBarry 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]); 2415c75a6043SHong Zhang } 2416c75a6043SHong Zhang #endif 2417c75a6043SHong Zhang 2418c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2419c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2420c75a6043SHong Zhang PetscFunctionReturn(0); 2421c75a6043SHong Zhang } 2422d06b337dSHong Zhang 242359f5e6ceSHong Zhang #undef __FUNCT__ 242459f5e6ceSHong Zhang #define __FUNCT__ "MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ" 242559f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat) 242659f5e6ceSHong Zhang { 242759f5e6ceSHong Zhang PetscErrorCode ierr; 242859f5e6ceSHong Zhang 242959f5e6ceSHong Zhang PetscFunctionBegin; 243059f5e6ceSHong Zhang ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr); 243159f5e6ceSHong Zhang PetscFunctionReturn(0); 243259f5e6ceSHong Zhang } 2433d06b337dSHong Zhang 2434d06b337dSHong Zhang 243549b5e25fSSatish Balay 243649b5e25fSSatish Balay 2437