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); 15305b42c5fSBarry Smith ierr = PetscFree(a->xtoy);CHKERRQ(ierr); 1544da8f245SBarry Smith if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);} 1551a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1566bf464f9SBarry Smith ierr = MatDestroy(&a->parent);CHKERRQ(ierr); 157bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 158901853e0SKris Buschelman 159dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 160bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 161bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 162bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr); 163bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr); 164bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr); 165bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr); 16638f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr); 167bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr); 16849b5e25fSSatish Balay PetscFunctionReturn(0); 16949b5e25fSSatish Balay } 17049b5e25fSSatish Balay 1714a2ae208SSatish Balay #undef __FUNCT__ 1724a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ" 173ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg) 17449b5e25fSSatish Balay { 175045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 17663ba0a88SBarry Smith PetscErrorCode ierr; 17749b5e25fSSatish Balay 17849b5e25fSSatish Balay PetscFunctionBegin; 1794d9d31abSKris Buschelman switch (op) { 1804d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 1814e0d8c25SBarry Smith a->roworiented = flg; 1824d9d31abSKris Buschelman break; 183a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 184a9817697SBarry Smith a->keepnonzeropattern = flg; 1854d9d31abSKris Buschelman break; 186512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 187512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 1884d9d31abSKris Buschelman break; 1894d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 1904e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 1914d9d31abSKris Buschelman break; 1924d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 1934e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 1944d9d31abSKris Buschelman break; 19528b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 19628b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 19728b2fa4aSMatthew Knepley break; 1984e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1994d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 2004d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 201290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 2024d9d31abSKris Buschelman break; 2039a4540c5SBarry Smith case MAT_HERMITIAN: 204e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first"); 2050bc54ff2SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 206eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort; 2070bc54ff2SBarry Smith } else if (A->cmap->bs == 1) { 208eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian; 209e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1"); 210eeffb40dSHong Zhang break; 2113d472b54SHong Zhang case MAT_SPD: 2125021d80fSJed Brown /* These options are handled directly by MatSetOption() */ 2133d472b54SHong Zhang break; 21477e54ba9SKris Buschelman case MAT_SYMMETRIC: 21577e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 2169a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 217e32f2f54SBarry Smith if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix must be symmetric"); 218290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr); 219290bbb0aSBarry Smith break; 220941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 2214e0d8c25SBarry Smith a->ignore_ltriangular = flg; 222941593c8SHong Zhang break; 223941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 2244e0d8c25SBarry Smith a->ignore_ltriangular = flg; 22577e54ba9SKris Buschelman break; 226f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 2274e0d8c25SBarry Smith a->getrow_utriangular = flg; 228f5edf698SHong Zhang break; 2294d9d31abSKris Buschelman default: 230e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 23149b5e25fSSatish Balay } 23249b5e25fSSatish Balay PetscFunctionReturn(0); 23349b5e25fSSatish Balay } 23449b5e25fSSatish Balay 2354a2ae208SSatish Balay #undef __FUNCT__ 2364a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ" 237*52768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 23849b5e25fSSatish Balay { 23949b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2406849ba73SBarry Smith PetscErrorCode ierr; 24149b5e25fSSatish Balay 24249b5e25fSSatish Balay PetscFunctionBegin; 243e32f2f54SBarry 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()"); 244*52768537SHong Zhang 245f5edf698SHong Zhang /* Get the upper triangular part of the row */ 246*52768537SHong Zhang ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr); 24749b5e25fSSatish Balay PetscFunctionReturn(0); 24849b5e25fSSatish Balay } 24949b5e25fSSatish Balay 2504a2ae208SSatish Balay #undef __FUNCT__ 2514a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ" 25213f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 25349b5e25fSSatish Balay { 254dfbe8321SBarry Smith PetscErrorCode ierr; 25549b5e25fSSatish Balay 25649b5e25fSSatish Balay PetscFunctionBegin; 25705b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 25805b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 25949b5e25fSSatish Balay PetscFunctionReturn(0); 26049b5e25fSSatish Balay } 26149b5e25fSSatish Balay 2624a2ae208SSatish Balay #undef __FUNCT__ 263f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ" 264f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 265f5edf698SHong Zhang { 266f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 267f5edf698SHong Zhang 268f5edf698SHong Zhang PetscFunctionBegin; 269f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 270f5edf698SHong Zhang PetscFunctionReturn(0); 271f5edf698SHong Zhang } 272f5edf698SHong Zhang #undef __FUNCT__ 273f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ" 274f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 275f5edf698SHong Zhang { 276f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 277f5edf698SHong Zhang 278f5edf698SHong Zhang PetscFunctionBegin; 279f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 280f5edf698SHong Zhang PetscFunctionReturn(0); 281f5edf698SHong Zhang } 282f5edf698SHong Zhang 283f5edf698SHong Zhang #undef __FUNCT__ 2844a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ" 285fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B) 28649b5e25fSSatish Balay { 287dfbe8321SBarry Smith PetscErrorCode ierr; 2885fd66863SKarl Rupp 28949b5e25fSSatish Balay PetscFunctionBegin; 290815cbec1SBarry Smith if (reuse == MAT_INITIAL_MATRIX || *B != A) { 291999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 292fc4dec0aSBarry Smith } 2938115998fSBarry Smith PetscFunctionReturn(0); 29449b5e25fSSatish Balay } 29549b5e25fSSatish Balay 2964a2ae208SSatish Balay #undef __FUNCT__ 2974a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII" 2987da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 29949b5e25fSSatish Balay { 30049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 301dfbe8321SBarry Smith PetscErrorCode ierr; 302d0f46423SBarry Smith PetscInt i,j,bs = A->rmap->bs,k,l,bs2=a->bs2; 303f3ef73ceSBarry Smith PetscViewerFormat format; 304121deb67SSatish Balay PetscInt *diag; 30549b5e25fSSatish Balay 30649b5e25fSSatish Balay PetscFunctionBegin; 307b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 308456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 30977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 310fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 311d2507d54SMatthew Knepley Mat aij; 312d5f3da31SBarry Smith if (A->factortype && bs>1) { 31370d5e725SHong 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); 31470d5e725SHong Zhang PetscFunctionReturn(0); 31570d5e725SHong Zhang } 316c9f458caSMatthew Knepley ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr); 317c9f458caSMatthew Knepley ierr = MatView(aij,viewer);CHKERRQ(ierr); 3186bf464f9SBarry Smith ierr = MatDestroy(&aij);CHKERRQ(ierr); 319fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 320d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 32149b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 32249b5e25fSSatish Balay for (j=0; j<bs; j++) { 32377431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 32449b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 32549b5e25fSSatish Balay for (l=0; l<bs; l++) { 32649b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 32749b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 32857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 32957622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 33049b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 33157622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 33257622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 33349b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 33457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 33549b5e25fSSatish Balay } 33649b5e25fSSatish Balay #else 33749b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 33857622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 33949b5e25fSSatish Balay } 34049b5e25fSSatish Balay #endif 34149b5e25fSSatish Balay } 34249b5e25fSSatish Balay } 343b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 34449b5e25fSSatish Balay } 34549b5e25fSSatish Balay } 346d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 347c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 348c1490034SHong Zhang PetscFunctionReturn(0); 34949b5e25fSSatish Balay } else { 350d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 3512c990fa1SHong Zhang if (A->factortype) { /* for factored matrix */ 3522c990fa1SHong Zhang if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet"); 3532c990fa1SHong Zhang 354121deb67SSatish Balay diag=a->diag; 355121deb67SSatish Balay for (i=0; i<a->mbs; i++) { /* for row block i */ 3562c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 3572c990fa1SHong Zhang /* diagonal entry */ 3582c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 3592c990fa1SHong Zhang 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 if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) { 36257622a8eSBarry 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); 3632c990fa1SHong Zhang } else { 36457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 3652c990fa1SHong Zhang } 3662c990fa1SHong Zhang #else 3676712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr); 3682c990fa1SHong Zhang #endif 3692c990fa1SHong Zhang /* off-diagonal entries */ 3702c990fa1SHong Zhang for (k=a->i[i]; k<a->i[i+1]-1; k++) { 3712c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 372ca0704adSBarry Smith 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); 374ca0704adSBarry Smith } else if (PetscImaginaryPart(a->a[k]) < 0.0) { 37557622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 3762c990fa1SHong Zhang } else { 37757622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr); 3782c990fa1SHong Zhang } 3792c990fa1SHong Zhang #else 38057622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr); 3812c990fa1SHong Zhang #endif 3822c990fa1SHong Zhang } 3832c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 3842c990fa1SHong Zhang } 3852c990fa1SHong Zhang 3862c990fa1SHong Zhang } else { /* for non-factored matrix */ 3870c74a584SJed Brown for (i=0; i<a->mbs; i++) { /* for row block i */ 3880c74a584SJed Brown for (j=0; j<bs; j++) { /* for row bs*i + j */ 38977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 3900c74a584SJed Brown for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */ 3910c74a584SJed Brown for (l=0; l<bs; l++) { /* for column */ 39249b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 39349b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 39457622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 39557622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 39649b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 39757622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 39857622a8eSBarry Smith (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 39949b5e25fSSatish Balay } else { 40057622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 40149b5e25fSSatish Balay } 40249b5e25fSSatish Balay #else 40357622a8eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 40449b5e25fSSatish Balay #endif 40549b5e25fSSatish Balay } 40649b5e25fSSatish Balay } 407b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 40849b5e25fSSatish Balay } 40949b5e25fSSatish Balay } 4102c990fa1SHong Zhang } 411d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 41249b5e25fSSatish Balay } 413b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 41449b5e25fSSatish Balay PetscFunctionReturn(0); 41549b5e25fSSatish Balay } 41649b5e25fSSatish Balay 4179804daf3SBarry Smith #include <petscdraw.h> 4184a2ae208SSatish Balay #undef __FUNCT__ 4194a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom" 4206849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 42149b5e25fSSatish Balay { 42249b5e25fSSatish Balay Mat A = (Mat) Aa; 42349b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 4246849ba73SBarry Smith PetscErrorCode ierr; 425d0f46423SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2; 42613f74950SBarry Smith PetscMPIInt rank; 42749b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 42849b5e25fSSatish Balay MatScalar *aa; 42949b5e25fSSatish Balay MPI_Comm comm; 430b0a32e0cSBarry Smith PetscViewer viewer; 43149b5e25fSSatish Balay 43249b5e25fSSatish Balay PetscFunctionBegin; 43349b5e25fSSatish Balay /* 43449b5e25fSSatish Balay This is nasty. If this is called from an originally parallel matrix 43549b5e25fSSatish Balay then all processes call this,but only the first has the matrix so the 43649b5e25fSSatish Balay rest should return immediately. 43749b5e25fSSatish Balay */ 43849b5e25fSSatish Balay ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr); 43949b5e25fSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 44049b5e25fSSatish Balay if (rank) PetscFunctionReturn(0); 44149b5e25fSSatish Balay 44249b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 44349b5e25fSSatish Balay 444b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 445b0a32e0cSBarry Smith PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric"); 44649b5e25fSSatish Balay 44749b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 448b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 44949b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 45049b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 451d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 45249b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 45349b5e25fSSatish Balay aa = a->a + j*bs2; 45449b5e25fSSatish Balay for (k=0; k<bs; k++) { 45549b5e25fSSatish Balay for (l=0; l<bs; l++) { 45649b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 457b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 45849b5e25fSSatish Balay } 45949b5e25fSSatish Balay } 46049b5e25fSSatish Balay } 46149b5e25fSSatish Balay } 462b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 46349b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 46449b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 465d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 46649b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 46749b5e25fSSatish Balay aa = a->a + j*bs2; 46849b5e25fSSatish Balay for (k=0; k<bs; k++) { 46949b5e25fSSatish Balay for (l=0; l<bs; l++) { 47049b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 471b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 47249b5e25fSSatish Balay } 47349b5e25fSSatish Balay } 47449b5e25fSSatish Balay } 47549b5e25fSSatish Balay } 47649b5e25fSSatish Balay 477b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 47849b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 47949b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 480d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 48149b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 48249b5e25fSSatish Balay aa = a->a + j*bs2; 48349b5e25fSSatish Balay for (k=0; k<bs; k++) { 48449b5e25fSSatish Balay for (l=0; l<bs; l++) { 48549b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 486b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 48749b5e25fSSatish Balay } 48849b5e25fSSatish Balay } 48949b5e25fSSatish Balay } 49049b5e25fSSatish Balay } 49149b5e25fSSatish Balay PetscFunctionReturn(0); 49249b5e25fSSatish Balay } 49349b5e25fSSatish Balay 4944a2ae208SSatish Balay #undef __FUNCT__ 4954a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw" 4966849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 49749b5e25fSSatish Balay { 498dfbe8321SBarry Smith PetscErrorCode ierr; 49949b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 500b0a32e0cSBarry Smith PetscDraw draw; 501ace3abfcSBarry Smith PetscBool isnull; 50249b5e25fSSatish Balay 50349b5e25fSSatish Balay PetscFunctionBegin; 504b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 505b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 50649b5e25fSSatish Balay 50749b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 508d0f46423SBarry Smith xr = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0; 50949b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 510b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 511b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 5120298fd71SBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 51349b5e25fSSatish Balay PetscFunctionReturn(0); 51449b5e25fSSatish Balay } 51549b5e25fSSatish Balay 5164a2ae208SSatish Balay #undef __FUNCT__ 5174a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ" 518dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 51949b5e25fSSatish Balay { 520dfbe8321SBarry Smith PetscErrorCode ierr; 521ace3abfcSBarry Smith PetscBool iascii,isdraw; 52208917f38SBarry Smith FILE *file = 0; 52349b5e25fSSatish Balay 52449b5e25fSSatish Balay PetscFunctionBegin; 525251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 526251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 52732077d6dSBarry Smith if (iascii) { 52849b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 52949b5e25fSSatish Balay } else if (isdraw) { 53049b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 53149b5e25fSSatish Balay } else { 532a5e6ed63SBarry Smith Mat B; 533ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 534a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 5356bf464f9SBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 53608917f38SBarry Smith ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr); 53708917f38SBarry Smith if (file) { 53808917f38SBarry Smith fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs); 53908917f38SBarry Smith } 54049b5e25fSSatish Balay } 54149b5e25fSSatish Balay PetscFunctionReturn(0); 54249b5e25fSSatish Balay } 54349b5e25fSSatish Balay 54449b5e25fSSatish Balay 5454a2ae208SSatish Balay #undef __FUNCT__ 5464a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ" 54713f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 54849b5e25fSSatish Balay { 549045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 55013f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 55113f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 552d0f46423SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2; 55397e567efSBarry Smith MatScalar *ap,*aa = a->a; 55449b5e25fSSatish Balay 55549b5e25fSSatish Balay PetscFunctionBegin; 55649b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 55749b5e25fSSatish Balay row = im[k]; brow = row/bs; 558e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 559e32f2f54SBarry 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); 56049b5e25fSSatish Balay rp = aj + ai[brow]; ap = aa + bs2*ai[brow]; 56149b5e25fSSatish Balay nrow = ailen[brow]; 56249b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 563e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 564e32f2f54SBarry 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); 56549b5e25fSSatish Balay col = in[l]; 56649b5e25fSSatish Balay bcol = col/bs; 56749b5e25fSSatish Balay cidx = col%bs; 56849b5e25fSSatish Balay ridx = row%bs; 56949b5e25fSSatish Balay high = nrow; 57049b5e25fSSatish Balay low = 0; /* assume unsorted */ 57149b5e25fSSatish Balay while (high-low > 5) { 57249b5e25fSSatish Balay t = (low+high)/2; 57349b5e25fSSatish Balay if (rp[t] > bcol) high = t; 57449b5e25fSSatish Balay else low = t; 57549b5e25fSSatish Balay } 57649b5e25fSSatish Balay for (i=low; i<high; i++) { 57749b5e25fSSatish Balay if (rp[i] > bcol) break; 57849b5e25fSSatish Balay if (rp[i] == bcol) { 57949b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 58049b5e25fSSatish Balay goto finished; 58149b5e25fSSatish Balay } 58249b5e25fSSatish Balay } 58397e567efSBarry Smith *v++ = 0.0; 58449b5e25fSSatish Balay finished:; 58549b5e25fSSatish Balay } 58649b5e25fSSatish Balay } 58749b5e25fSSatish Balay PetscFunctionReturn(0); 58849b5e25fSSatish Balay } 58949b5e25fSSatish Balay 59049b5e25fSSatish Balay 5914a2ae208SSatish Balay #undef __FUNCT__ 5924a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ" 59313f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 59449b5e25fSSatish Balay { 5950880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 5966849ba73SBarry Smith PetscErrorCode ierr; 597e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 59813f74950SBarry Smith PetscInt *imax =a->imax,*ai=a->i,*ailen=a->ilen; 599d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval; 600ace3abfcSBarry Smith PetscBool roworiented=a->roworiented; 601dd6ea824SBarry Smith const PetscScalar *value = v; 602f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 6030880e062SHong Zhang 60449b5e25fSSatish Balay PetscFunctionBegin; 60526fbe8dcSKarl Rupp if (roworiented) stepval = (n-1)*bs; 60626fbe8dcSKarl Rupp else stepval = (m-1)*bs; 60726fbe8dcSKarl Rupp 6080880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 6090880e062SHong Zhang row = im[k]; 6100880e062SHong Zhang if (row < 0) continue; 6112515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 612e32f2f54SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1); 6130880e062SHong Zhang #endif 6140880e062SHong Zhang rp = aj + ai[row]; 6150880e062SHong Zhang ap = aa + bs2*ai[row]; 6160880e062SHong Zhang rmax = imax[row]; 6170880e062SHong Zhang nrow = ailen[row]; 6180880e062SHong Zhang low = 0; 619818f2c47SBarry Smith high = nrow; 6200880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 6210880e062SHong Zhang if (in[l] < 0) continue; 6220880e062SHong Zhang col = in[l]; 6232515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 624e32f2f54SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1); 625b1823623SSatish Balay #endif 626b98bf0e1SJed Brown if (col < row) { 62726fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular block */ 62826fbe8dcSKarl 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)"); 629b98bf0e1SJed Brown } 63026fbe8dcSKarl Rupp if (roworiented) value = v + k*(stepval+bs)*bs + l*bs; 63126fbe8dcSKarl Rupp else value = v + l*(stepval+bs)*bs + k*bs; 63226fbe8dcSKarl Rupp 63326fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 63426fbe8dcSKarl Rupp else high = nrow; 63526fbe8dcSKarl Rupp 636e2ee6c50SBarry Smith lastcol = col; 6370880e062SHong Zhang while (high-low > 7) { 6380880e062SHong Zhang t = (low+high)/2; 6390880e062SHong Zhang if (rp[t] > col) high = t; 6400880e062SHong Zhang else low = t; 6410880e062SHong Zhang } 6420880e062SHong Zhang for (i=low; i<high; i++) { 6430880e062SHong Zhang if (rp[i] > col) break; 6440880e062SHong Zhang if (rp[i] == col) { 6450880e062SHong Zhang bap = ap + bs2*i; 6460880e062SHong Zhang if (roworiented) { 6470880e062SHong Zhang if (is == ADD_VALUES) { 6480880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6490880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6500880e062SHong Zhang bap[jj] += *value++; 6510880e062SHong Zhang } 6520880e062SHong Zhang } 6530880e062SHong Zhang } else { 6540880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6550880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6560880e062SHong Zhang bap[jj] = *value++; 6570880e062SHong Zhang } 6580880e062SHong Zhang } 6590880e062SHong Zhang } 6600880e062SHong Zhang } else { 6610880e062SHong Zhang if (is == ADD_VALUES) { 6620880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6630880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6640880e062SHong Zhang *bap++ += *value++; 6650880e062SHong Zhang } 6660880e062SHong Zhang } 6670880e062SHong Zhang } else { 6680880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6690880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6700880e062SHong Zhang *bap++ = *value++; 6710880e062SHong Zhang } 6720880e062SHong Zhang } 6730880e062SHong Zhang } 6740880e062SHong Zhang } 6750880e062SHong Zhang goto noinsert2; 6760880e062SHong Zhang } 6770880e062SHong Zhang } 6780880e062SHong Zhang if (nonew == 1) goto noinsert2; 679e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 680fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 681c03d1d03SSatish Balay N = nrow++ - 1; high++; 6820880e062SHong Zhang /* shift up all the later entries in this row */ 6830880e062SHong Zhang for (ii=N; ii>=i; ii--) { 6840880e062SHong Zhang rp[ii+1] = rp[ii]; 6850880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 6860880e062SHong Zhang } 6870880e062SHong Zhang if (N >= i) { 6880880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 6890880e062SHong Zhang } 6900880e062SHong Zhang rp[i] = col; 6910880e062SHong Zhang bap = ap + bs2*i; 6920880e062SHong Zhang if (roworiented) { 6930880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6940880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6950880e062SHong Zhang bap[jj] = *value++; 6960880e062SHong Zhang } 6970880e062SHong Zhang } 6980880e062SHong Zhang } else { 6990880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7000880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7010880e062SHong Zhang *bap++ = *value++; 7020880e062SHong Zhang } 7030880e062SHong Zhang } 7040880e062SHong Zhang } 7050880e062SHong Zhang noinsert2:; 7060880e062SHong Zhang low = i; 7070880e062SHong Zhang } 7080880e062SHong Zhang ailen[row] = nrow; 7090880e062SHong Zhang } 7100880e062SHong Zhang PetscFunctionReturn(0); 71149b5e25fSSatish Balay } 71249b5e25fSSatish Balay 71364831d72SBarry Smith /* 71464831d72SBarry Smith This is not yet used 71564831d72SBarry Smith */ 7164a2ae208SSatish Balay #undef __FUNCT__ 7174108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode" 7184108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A) 7190def2e27SBarry Smith { 7200def2e27SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7210def2e27SBarry Smith PetscErrorCode ierr; 7220def2e27SBarry Smith const PetscInt *ai = a->i, *aj = a->j,*cols; 7230def2e27SBarry Smith PetscInt i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts; 724ace3abfcSBarry Smith PetscBool flag; 7250def2e27SBarry Smith 7260def2e27SBarry Smith PetscFunctionBegin; 727785e854fSJed Brown ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr); 7280def2e27SBarry Smith while (i < m) { 7290def2e27SBarry Smith nzx = ai[i+1] - ai[i]; /* Number of nonzeros */ 7300def2e27SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 7310def2e27SBarry Smith for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 7320def2e27SBarry Smith nzy = ai[j+1] - ai[j]; 7330def2e27SBarry Smith if (nzy != (nzx - j + i)) break; 7340def2e27SBarry Smith ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr); 7350def2e27SBarry Smith if (!flag) break; 7360def2e27SBarry Smith } 7370def2e27SBarry Smith ns[node_count++] = blk_size; 73826fbe8dcSKarl Rupp 7390def2e27SBarry Smith i = j; 7400def2e27SBarry Smith } 7410def2e27SBarry Smith if (!a->inode.size && m && node_count > .9*m) { 7420def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7430def2e27SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 7440def2e27SBarry Smith } else { 7450def2e27SBarry Smith a->inode.node_count = node_count; 74626fbe8dcSKarl Rupp 747785e854fSJed Brown ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr); 7483bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr); 74922d28d08SBarry Smith ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr); 7500def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7510def2e27SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 7520def2e27SBarry Smith 7530def2e27SBarry Smith /* count collections of adjacent columns in each inode */ 7540def2e27SBarry Smith row = 0; 7550def2e27SBarry Smith cnt = 0; 7560def2e27SBarry Smith for (i=0; i<node_count; i++) { 7570def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 7580def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 7590def2e27SBarry Smith for (j=1; j<nz; j++) { 76026fbe8dcSKarl Rupp if (cols[j] != cols[j-1]+1) cnt++; 7610def2e27SBarry Smith } 7620def2e27SBarry Smith cnt++; 7630def2e27SBarry Smith row += a->inode.size[i]; 7640def2e27SBarry Smith } 765785e854fSJed Brown ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr); 7660def2e27SBarry Smith cnt = 0; 7670def2e27SBarry Smith row = 0; 7680def2e27SBarry Smith for (i=0; i<node_count; i++) { 7690def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 7700def2e27SBarry Smith counts[2*cnt] = cols[0]; 7710def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 7720def2e27SBarry Smith cnt2 = 1; 7730def2e27SBarry Smith for (j=1; j<nz; j++) { 7740def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 7750def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 7760def2e27SBarry Smith counts[2*cnt] = cols[j]; 7770def2e27SBarry Smith cnt2 = 1; 7780def2e27SBarry Smith } else cnt2++; 7790def2e27SBarry Smith } 7800def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 7810def2e27SBarry Smith row += a->inode.size[i]; 7820def2e27SBarry Smith } 78322d28d08SBarry Smith ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr); 7840def2e27SBarry Smith } 78538702af4SBarry Smith PetscFunctionReturn(0); 78638702af4SBarry Smith } 78738702af4SBarry Smith 78838702af4SBarry Smith #undef __FUNCT__ 7894a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ" 790dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 79149b5e25fSSatish Balay { 79249b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7936849ba73SBarry Smith PetscErrorCode ierr; 79413f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 795d0f46423SBarry Smith PetscInt m = A->rmap->N,*ip,N,*ailen = a->ilen; 79613f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 79749b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 79849b5e25fSSatish Balay 79949b5e25fSSatish Balay PetscFunctionBegin; 80049b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 80149b5e25fSSatish Balay 80249b5e25fSSatish Balay if (m) rmax = ailen[0]; 80349b5e25fSSatish Balay for (i=1; i<mbs; i++) { 80449b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 80549b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 80649b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 80749b5e25fSSatish Balay if (fshift) { 80849b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 80949b5e25fSSatish Balay N = ailen[i]; 81049b5e25fSSatish Balay for (j=0; j<N; j++) { 81149b5e25fSSatish Balay ip[j-fshift] = ip[j]; 81249b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 81349b5e25fSSatish Balay } 81449b5e25fSSatish Balay } 81549b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 81649b5e25fSSatish Balay } 81749b5e25fSSatish Balay if (mbs) { 81849b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 81949b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 82049b5e25fSSatish Balay } 82149b5e25fSSatish Balay /* reset ilen and imax for each row */ 82249b5e25fSSatish Balay for (i=0; i<mbs; i++) { 82349b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 82449b5e25fSSatish Balay } 8256c6c5352SBarry Smith a->nz = ai[mbs]; 82649b5e25fSSatish Balay 827b424e231SHong Zhang /* diagonals may have moved, reset it */ 828b424e231SHong Zhang if (a->diag) { 8292ed38d0bSJed Brown ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr); 83049b5e25fSSatish Balay } 83126fbe8dcSKarl 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); 83226fbe8dcSKarl Rupp 833d0f46423SBarry 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); 834ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 835ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 83626fbe8dcSKarl Rupp 8378e58a170SBarry Smith A->info.mallocs += a->reallocs; 83849b5e25fSSatish Balay a->reallocs = 0; 83949b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 840061b2667SBarry Smith a->idiagvalid = PETSC_FALSE; 84138702af4SBarry Smith 84238702af4SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 84344e1c64aSLisandro Dalcin if (a->jshort && a->free_jshort) { 84417803ae8SHong Zhang /* when matrix data structure is changed, previous jshort must be replaced */ 84517803ae8SHong Zhang ierr = PetscFree(a->jshort);CHKERRQ(ierr); 84617803ae8SHong Zhang } 847785e854fSJed Brown ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr); 8483bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr); 84938702af4SBarry Smith for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i]; 85038702af4SBarry Smith A->ops->mult = MatMult_SeqSBAIJ_1_ushort; 85141f059aeSBarry Smith A->ops->sor = MatSOR_SeqSBAIJ_ushort; 8524da8f245SBarry Smith a->free_jshort = PETSC_TRUE; 85338702af4SBarry Smith } 85449b5e25fSSatish Balay PetscFunctionReturn(0); 85549b5e25fSSatish Balay } 85649b5e25fSSatish Balay 85749b5e25fSSatish Balay /* 85849b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 85949b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 86049b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 86149b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 86249b5e25fSSatish Balay */ 8634a2ae208SSatish Balay #undef __FUNCT__ 8644a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks" 86513f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 86649b5e25fSSatish Balay { 86713f74950SBarry Smith PetscInt i,j,k,row; 868ace3abfcSBarry Smith PetscBool flg; 86949b5e25fSSatish Balay 87049b5e25fSSatish Balay PetscFunctionBegin; 87149b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 87249b5e25fSSatish Balay row = idx[i]; 87349b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 87449b5e25fSSatish Balay sizes[j] = 1; 87549b5e25fSSatish Balay i++; 87649b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 87749b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 87849b5e25fSSatish Balay i++; 87949b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 88049b5e25fSSatish Balay flg = PETSC_TRUE; 88149b5e25fSSatish Balay for (k=1; k<bs; k++) { 88249b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 88349b5e25fSSatish Balay flg = PETSC_FALSE; 88449b5e25fSSatish Balay break; 88549b5e25fSSatish Balay } 88649b5e25fSSatish Balay } 887abc0a331SBarry Smith if (flg) { /* No break in the bs */ 88849b5e25fSSatish Balay sizes[j] = bs; 88949b5e25fSSatish Balay i += bs; 89049b5e25fSSatish Balay } else { 89149b5e25fSSatish Balay sizes[j] = 1; 89249b5e25fSSatish Balay i++; 89349b5e25fSSatish Balay } 89449b5e25fSSatish Balay } 89549b5e25fSSatish Balay } 89649b5e25fSSatish Balay *bs_max = j; 89749b5e25fSSatish Balay PetscFunctionReturn(0); 89849b5e25fSSatish Balay } 89949b5e25fSSatish Balay 90049b5e25fSSatish Balay 90149b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 90249b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 90349b5e25fSSatish Balay */ 90449b5e25fSSatish Balay 9054a2ae208SSatish Balay #undef __FUNCT__ 9064a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ" 90713f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 90849b5e25fSSatish Balay { 90949b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 9106849ba73SBarry Smith PetscErrorCode ierr; 911e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 91213f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 913d0f46423SBarry Smith PetscInt *aj =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol; 91413f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 91549b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 91649b5e25fSSatish Balay 91749b5e25fSSatish Balay PetscFunctionBegin; 91849b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 91949b5e25fSSatish Balay row = im[k]; /* row number */ 92049b5e25fSSatish Balay brow = row/bs; /* block row number */ 92149b5e25fSSatish Balay if (row < 0) continue; 9222515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 923e32f2f54SBarry 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); 92449b5e25fSSatish Balay #endif 92549b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 92649b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 92749b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 92849b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 92949b5e25fSSatish Balay low = 0; 93049b5e25fSSatish Balay 93149b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 93249b5e25fSSatish Balay if (in[l] < 0) continue; 9332515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 934e32f2f54SBarry 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); 93549b5e25fSSatish Balay #endif 93649b5e25fSSatish Balay col = in[l]; 93749b5e25fSSatish Balay bcol = col/bs; /* block col number */ 93849b5e25fSSatish Balay 939941593c8SHong Zhang if (brow > bcol) { 94026fbe8dcSKarl Rupp if (a->ignore_ltriangular) continue; /* ignore lower triangular values */ 94126fbe8dcSKarl 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)"); 942941593c8SHong Zhang } 943f4989cb3SHong Zhang 94449b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 9458549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)) { 94649b5e25fSSatish Balay /* element value a(k,l) */ 94726fbe8dcSKarl Rupp if (roworiented) value = v[l + k*n]; 94826fbe8dcSKarl Rupp else value = v[k + l*m]; 94949b5e25fSSatish Balay 95049b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 95126fbe8dcSKarl Rupp if (col <= lastcol) low = 0; 95226fbe8dcSKarl Rupp high = nrow; 953e2ee6c50SBarry Smith lastcol = col; 95449b5e25fSSatish Balay while (high-low > 7) { 95549b5e25fSSatish Balay t = (low+high)/2; 95649b5e25fSSatish Balay if (rp[t] > bcol) high = t; 95749b5e25fSSatish Balay else low = t; 95849b5e25fSSatish Balay } 95949b5e25fSSatish Balay for (i=low; i<high; i++) { 96049b5e25fSSatish Balay if (rp[i] > bcol) break; 96149b5e25fSSatish Balay if (rp[i] == bcol) { 96249b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 96349b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 96449b5e25fSSatish Balay else *bap = value; 9658549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 9668549e402SHong Zhang if (brow == bcol && ridx < cidx) { 9678549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 9688549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 9698549e402SHong Zhang else *bap = value; 9708549e402SHong Zhang } 97149b5e25fSSatish Balay goto noinsert1; 97249b5e25fSSatish Balay } 97349b5e25fSSatish Balay } 97449b5e25fSSatish Balay 97549b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 976e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 977fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 97849b5e25fSSatish Balay 979c03d1d03SSatish Balay N = nrow++ - 1; high++; 98049b5e25fSSatish Balay /* shift up all the later entries in this row */ 98149b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 98249b5e25fSSatish Balay rp[ii+1] = rp[ii]; 98349b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 98449b5e25fSSatish Balay } 98549b5e25fSSatish Balay if (N>=i) { 98649b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 98749b5e25fSSatish Balay } 98849b5e25fSSatish Balay rp[i] = bcol; 98949b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 990e56f5c9eSBarry Smith A->nonzerostate++; 99149b5e25fSSatish Balay noinsert1:; 99249b5e25fSSatish Balay low = i; 9938549e402SHong Zhang } 99449b5e25fSSatish Balay } /* end of loop over added columns */ 99549b5e25fSSatish Balay ailen[brow] = nrow; 99649b5e25fSSatish Balay } /* end of loop over added rows */ 99749b5e25fSSatish Balay PetscFunctionReturn(0); 99849b5e25fSSatish Balay } 99949b5e25fSSatish Balay 10004a2ae208SSatish Balay #undef __FUNCT__ 10014d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ" 10020481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info) 100349b5e25fSSatish Balay { 10044ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 100549b5e25fSSatish Balay Mat outA; 1006dfbe8321SBarry Smith PetscErrorCode ierr; 1007ace3abfcSBarry Smith PetscBool row_identity; 100849b5e25fSSatish Balay 100949b5e25fSSatish Balay PetscFunctionBegin; 1010e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 1011c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 1012e32f2f54SBarry Smith if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported"); 1013e32f2f54SBarry 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()! */ 1014c84f5b01SHong Zhang 101549b5e25fSSatish Balay outA = inA; 1016d5f3da31SBarry Smith inA->factortype = MAT_FACTOR_ICC; 101749b5e25fSSatish Balay 10181a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 1019d595f711SHong Zhang ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr); 102049b5e25fSSatish Balay 1021c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10226bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1023c84f5b01SHong Zhang a->row = row; 1024c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10256bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1026c84f5b01SHong Zhang a->col = row; 1027c84f5b01SHong Zhang 1028c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1029c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 10303bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr); 103149b5e25fSSatish Balay 103249b5e25fSSatish Balay if (!a->solve_work) { 1033785e854fSJed Brown ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr); 10343bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 103549b5e25fSSatish Balay } 103649b5e25fSSatish Balay 1037719d5645SBarry Smith ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr); 103849b5e25fSSatish Balay PetscFunctionReturn(0); 103949b5e25fSSatish Balay } 1040950f1e5bSHong Zhang 10414a2ae208SSatish Balay #undef __FUNCT__ 10424a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ" 10437087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 104449b5e25fSSatish Balay { 1045045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ*)mat->data; 104613f74950SBarry Smith PetscInt i,nz,n; 10477827cd58SJed Brown PetscErrorCode ierr; 104849b5e25fSSatish Balay 104949b5e25fSSatish Balay PetscFunctionBegin; 10506c6c5352SBarry Smith nz = baij->maxnz; 1051d0f46423SBarry Smith n = mat->cmap->n; 105226fbe8dcSKarl Rupp for (i=0; i<nz; i++) baij->j[i] = indices[i]; 105326fbe8dcSKarl Rupp 10546c6c5352SBarry Smith baij->nz = nz; 105526fbe8dcSKarl Rupp for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i]; 105626fbe8dcSKarl Rupp 10577827cd58SJed Brown ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 105849b5e25fSSatish Balay PetscFunctionReturn(0); 105949b5e25fSSatish Balay } 106049b5e25fSSatish Balay 10614a2ae208SSatish Balay #undef __FUNCT__ 10624a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices" 106349b5e25fSSatish Balay /*@ 106419585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 106549b5e25fSSatish Balay in the matrix. 106649b5e25fSSatish Balay 106749b5e25fSSatish Balay Input Parameters: 106819585528SSatish Balay + mat - the SeqSBAIJ matrix 106949b5e25fSSatish Balay - indices - the column indices 107049b5e25fSSatish Balay 107149b5e25fSSatish Balay Level: advanced 107249b5e25fSSatish Balay 107349b5e25fSSatish Balay Notes: 107449b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 107549b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 107649b5e25fSSatish Balay of the MatSetValues() operation. 107749b5e25fSSatish Balay 107849b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1079d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 108049b5e25fSSatish Balay 1081ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 108249b5e25fSSatish Balay 1083ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 108449b5e25fSSatish Balay @*/ 10857087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 108649b5e25fSSatish Balay { 10874ac538c5SBarry Smith PetscErrorCode ierr; 108849b5e25fSSatish Balay 108949b5e25fSSatish Balay PetscFunctionBegin; 10900700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 10914482741eSBarry Smith PetscValidPointer(indices,2); 10924ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr); 109349b5e25fSSatish Balay PetscFunctionReturn(0); 109449b5e25fSSatish Balay } 109549b5e25fSSatish Balay 10964a2ae208SSatish Balay #undef __FUNCT__ 10973c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ" 10983c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 10993c896bc6SHong Zhang { 11003c896bc6SHong Zhang PetscErrorCode ierr; 11013c896bc6SHong Zhang 11023c896bc6SHong Zhang PetscFunctionBegin; 11033c896bc6SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 11043c896bc6SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 11053c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11063c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11073c896bc6SHong Zhang 1108e7e72b3dSBarry 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"); 1109d0f46423SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr); 11103c896bc6SHong Zhang } else { 1111f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11123c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1113f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11143c896bc6SHong Zhang } 11153c896bc6SHong Zhang PetscFunctionReturn(0); 11163c896bc6SHong Zhang } 11173c896bc6SHong Zhang 11183c896bc6SHong Zhang #undef __FUNCT__ 11194994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ" 11204994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A) 1121273d9f13SBarry Smith { 1122dfbe8321SBarry Smith PetscErrorCode ierr; 1123273d9f13SBarry Smith 1124273d9f13SBarry Smith PetscFunctionBegin; 1125535b19f3SBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr); 1126273d9f13SBarry Smith PetscFunctionReturn(0); 1127273d9f13SBarry Smith } 1128273d9f13SBarry Smith 1129a6ece127SHong Zhang #undef __FUNCT__ 11308c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ" 11318c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1132a6ece127SHong Zhang { 1133a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11345fd66863SKarl Rupp 1135a6ece127SHong Zhang PetscFunctionBegin; 1136a6ece127SHong Zhang *array = a->a; 1137a6ece127SHong Zhang PetscFunctionReturn(0); 1138a6ece127SHong Zhang } 1139a6ece127SHong Zhang 1140a6ece127SHong Zhang #undef __FUNCT__ 11418c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ" 11428c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1143a6ece127SHong Zhang { 1144a6ece127SHong Zhang PetscFunctionBegin; 1145a6ece127SHong Zhang PetscFunctionReturn(0); 1146a6ece127SHong Zhang } 1147a6ece127SHong Zhang 114842ee4b1aSHong Zhang #undef __FUNCT__ 1149*52768537SHong Zhang #define __FUNCT__ "MatAXPYGetPreallocation_SeqSBAIJ" 1150*52768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz) 1151*52768537SHong Zhang { 1152*52768537SHong Zhang PetscInt i,bs=Y->rmap->bs,mbs=Y->rmap->N/bs; 1153*52768537SHong Zhang Mat_SeqSBAIJ *x = (Mat_SeqSBAIJ*)X->data; 1154*52768537SHong Zhang Mat_SeqSBAIJ *y = (Mat_SeqSBAIJ*)Y->data; 1155*52768537SHong Zhang const PetscInt *xi = x->i,*yi = y->i; 1156*52768537SHong Zhang 1157*52768537SHong Zhang PetscFunctionBegin; 1158*52768537SHong Zhang /* Set the number of nonzeros in the new matrix */ 1159*52768537SHong Zhang printf("Y: mbs %d, m %d\n",mbs, Y->rmap->N); 1160*52768537SHong Zhang for (i=0; i<mbs; i++) { 1161*52768537SHong Zhang PetscInt j,k,nzx = xi[i+1] - xi[i],nzy = yi[i+1] - yi[i]; 1162*52768537SHong Zhang const PetscInt *xj = x->j+xi[i],*yj = y->j+yi[i]; 1163*52768537SHong Zhang nnz[i] = 0; 1164*52768537SHong Zhang for (j=0,k=0; j<nzx; j++) { /* Point in X */ 1165*52768537SHong Zhang for (; k<nzy && yj[k]<xj[j]; k++) nnz[i]++; /* Catch up to X */ 1166*52768537SHong Zhang if (k<nzy && yj[k]==xj[j]) k++; /* Skip duplicate */ 1167*52768537SHong Zhang nnz[i]++; 1168*52768537SHong Zhang } 1169*52768537SHong Zhang for (; k<nzy; k++) nnz[i]++; 1170*52768537SHong Zhang } 1171*52768537SHong Zhang PetscFunctionReturn(0); 1172*52768537SHong Zhang } 1173*52768537SHong Zhang 1174*52768537SHong Zhang #undef __FUNCT__ 117542ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ" 1176f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 117742ee4b1aSHong Zhang { 117842ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data; 1179dfbe8321SBarry Smith PetscErrorCode ierr; 1180e838b9e7SJed Brown PetscInt i,bs=Y->rmap->bs,bs2=bs*bs,j; 1181e838b9e7SJed Brown PetscBLASInt one = 1; 118242ee4b1aSHong Zhang 118342ee4b1aSHong Zhang PetscFunctionBegin; 118442ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1185f4df32b1SMatthew Knepley PetscScalar alpha = a; 1186c5df96a5SBarry Smith PetscBLASInt bnz; 1187c5df96a5SBarry Smith ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr); 11888b83055fSJed Brown PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one)); 1189a3fa217bSJose E. Roman ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 1190c537a176SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1191c4319e64SHong Zhang if (y->xtoy && y->XtoY != X) { 1192c4319e64SHong Zhang ierr = PetscFree(y->xtoy);CHKERRQ(ierr); 11936bf464f9SBarry Smith ierr = MatDestroy(&y->XtoY);CHKERRQ(ierr); 1194c537a176SHong Zhang } 1195c4319e64SHong Zhang if (!y->xtoy) { /* get xtoy */ 11960298fd71SBarry Smith ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,NULL, y->i,y->j,NULL, &y->xtoy);CHKERRQ(ierr); 1197c4319e64SHong Zhang y->XtoY = X; 1198c537a176SHong Zhang } 11996c6c5352SBarry Smith for (i=0; i<x->nz; i++) { 1200c4319e64SHong Zhang j = 0; 1201c4319e64SHong Zhang while (j < bs2) { 1202f4df32b1SMatthew Knepley y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]); 1203c4319e64SHong Zhang j++; 1204c537a176SHong Zhang } 1205c4319e64SHong Zhang } 1206a3fa217bSJose E. Roman ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 12076712e2f1SBarry Smith ierr = PetscInfo3(Y,"ratio of nnz_s(X)/nnz_s(Y): %D/%D = %g\n",bs2*x->nz,bs2*y->nz,(double)((PetscReal)(bs2*x->nz)/(PetscReal)(bs2*y->nz)));CHKERRQ(ierr); 120842ee4b1aSHong Zhang } else { 1209*52768537SHong Zhang Mat B; 1210*52768537SHong Zhang PetscInt *nnz; 1211*52768537SHong Zhang if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size"); 1212f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 1213*52768537SHong Zhang ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr); 1214*52768537SHong Zhang ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr); 1215*52768537SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr); 1216*52768537SHong Zhang ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr); 1217*52768537SHong Zhang ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr); 1218*52768537SHong Zhang ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr); 1219*52768537SHong Zhang ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr); 1220*52768537SHong Zhang ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr); 1221*52768537SHong Zhang ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 1222*52768537SHong Zhang 1223*52768537SHong Zhang ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr); 1224*52768537SHong Zhang 1225*52768537SHong Zhang ierr = MatHeaderReplace(Y,B);CHKERRQ(ierr); 1226*52768537SHong Zhang ierr = PetscFree(nnz);CHKERRQ(ierr); 1227*52768537SHong Zhang //ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1228f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 1229*52768537SHong Zhang ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr); 123042ee4b1aSHong Zhang } 123142ee4b1aSHong Zhang PetscFunctionReturn(0); 123242ee4b1aSHong Zhang } 123342ee4b1aSHong Zhang 1234efcf0fc3SBarry Smith #undef __FUNCT__ 1235efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ" 1236ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1237efcf0fc3SBarry Smith { 1238efcf0fc3SBarry Smith PetscFunctionBegin; 1239efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1240efcf0fc3SBarry Smith PetscFunctionReturn(0); 1241efcf0fc3SBarry Smith } 1242efcf0fc3SBarry Smith 1243efcf0fc3SBarry Smith #undef __FUNCT__ 1244efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ" 1245ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool *flg) 1246efcf0fc3SBarry Smith { 1247efcf0fc3SBarry Smith PetscFunctionBegin; 1248efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1249efcf0fc3SBarry Smith PetscFunctionReturn(0); 1250efcf0fc3SBarry Smith } 1251efcf0fc3SBarry Smith 1252efcf0fc3SBarry Smith #undef __FUNCT__ 1253efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ" 1254ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1255efcf0fc3SBarry Smith { 1256efcf0fc3SBarry Smith PetscFunctionBegin; 1257efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1258efcf0fc3SBarry Smith PetscFunctionReturn(0); 1259efcf0fc3SBarry Smith } 1260efcf0fc3SBarry Smith 126199cafbc1SBarry Smith #undef __FUNCT__ 126299cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ" 126399cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 126499cafbc1SBarry Smith { 126599cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 126699cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1267dd6ea824SBarry Smith MatScalar *aa = a->a; 126899cafbc1SBarry Smith 126999cafbc1SBarry Smith PetscFunctionBegin; 127099cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 127199cafbc1SBarry Smith PetscFunctionReturn(0); 127299cafbc1SBarry Smith } 127399cafbc1SBarry Smith 127499cafbc1SBarry Smith #undef __FUNCT__ 127599cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ" 127699cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A) 127799cafbc1SBarry Smith { 127899cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 127999cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1280dd6ea824SBarry Smith MatScalar *aa = a->a; 128199cafbc1SBarry Smith 128299cafbc1SBarry Smith PetscFunctionBegin; 128399cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 128499cafbc1SBarry Smith PetscFunctionReturn(0); 128599cafbc1SBarry Smith } 128699cafbc1SBarry Smith 12873bededecSBarry Smith #undef __FUNCT__ 12883bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ" 12893bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b) 12903bededecSBarry Smith { 12913bededecSBarry Smith Mat_SeqSBAIJ *baij=(Mat_SeqSBAIJ*)A->data; 12923bededecSBarry Smith PetscErrorCode ierr; 12933bededecSBarry Smith PetscInt i,j,k,count; 12943bededecSBarry Smith PetscInt bs =A->rmap->bs,bs2=baij->bs2,row,col; 12953bededecSBarry Smith PetscScalar zero = 0.0; 12963bededecSBarry Smith MatScalar *aa; 12973bededecSBarry Smith const PetscScalar *xx; 12983bededecSBarry Smith PetscScalar *bb; 129956777dd2SBarry Smith PetscBool *zeroed,vecs = PETSC_FALSE; 13003bededecSBarry Smith 13013bededecSBarry Smith PetscFunctionBegin; 13023bededecSBarry Smith /* fix right hand side if needed */ 13033bededecSBarry Smith if (x && b) { 13043bededecSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 13053bededecSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 130656777dd2SBarry Smith vecs = PETSC_TRUE; 13073bededecSBarry Smith } 13083bededecSBarry Smith 13093bededecSBarry Smith /* zero the columns */ 13101795a4d1SJed Brown ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr); 13113bededecSBarry Smith for (i=0; i<is_n; i++) { 13123bededecSBarry 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]); 13133bededecSBarry Smith zeroed[is_idx[i]] = PETSC_TRUE; 13143bededecSBarry Smith } 131556777dd2SBarry Smith if (vecs) { 131656777dd2SBarry Smith for (i=0; i<A->rmap->N; i++) { 131756777dd2SBarry Smith row = i/bs; 131856777dd2SBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 131956777dd2SBarry Smith for (k=0; k<bs; k++) { 132056777dd2SBarry Smith col = bs*baij->j[j] + k; 132156777dd2SBarry Smith if (col <= i) continue; 132256777dd2SBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 132326fbe8dcSKarl Rupp if (!zeroed[i] && zeroed[col]) bb[i] -= aa[0]*xx[col]; 132426fbe8dcSKarl Rupp if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i]; 132556777dd2SBarry Smith } 132656777dd2SBarry Smith } 132756777dd2SBarry Smith } 132826fbe8dcSKarl Rupp for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]]; 132956777dd2SBarry Smith } 133056777dd2SBarry Smith 13313bededecSBarry Smith for (i=0; i<A->rmap->N; i++) { 13323bededecSBarry Smith if (!zeroed[i]) { 13333bededecSBarry Smith row = i/bs; 13343bededecSBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 13353bededecSBarry Smith for (k=0; k<bs; k++) { 13363bededecSBarry Smith col = bs*baij->j[j] + k; 13373bededecSBarry Smith if (zeroed[col]) { 13383bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 13393bededecSBarry Smith aa[0] = 0.0; 13403bededecSBarry Smith } 13413bededecSBarry Smith } 13423bededecSBarry Smith } 13433bededecSBarry Smith } 13443bededecSBarry Smith } 13453bededecSBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 134656777dd2SBarry Smith if (vecs) { 134756777dd2SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 134856777dd2SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 134956777dd2SBarry Smith } 13503bededecSBarry Smith 13513bededecSBarry Smith /* zero the rows */ 13523bededecSBarry Smith for (i=0; i<is_n; i++) { 13533bededecSBarry Smith row = is_idx[i]; 13543bededecSBarry Smith count = (baij->i[row/bs +1] - baij->i[row/bs])*bs; 13553bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs); 13563bededecSBarry Smith for (k=0; k<count; k++) { 13573bededecSBarry Smith aa[0] = zero; 13583bededecSBarry Smith aa += bs; 13593bededecSBarry Smith } 13603bededecSBarry Smith if (diag != 0.0) { 13613bededecSBarry Smith ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr); 13623bededecSBarry Smith } 13633bededecSBarry Smith } 13643bededecSBarry Smith ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 13653bededecSBarry Smith PetscFunctionReturn(0); 13663bededecSBarry Smith } 13673bededecSBarry Smith 136849b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 13693964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 137049b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 137149b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 137249b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 137397304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1374431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1375e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 1376db4efbfdSBarry Smith 0, 137749b5e25fSSatish Balay 0, 137849b5e25fSSatish Balay 0, 137997304618SKris Buschelman /* 10*/ 0, 138049b5e25fSSatish Balay 0, 1381c078aec8SLisandro Dalcin MatCholeskyFactor_SeqSBAIJ, 138241f059aeSBarry Smith MatSOR_SeqSBAIJ, 138349b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 138497304618SKris Buschelman /* 15*/ MatGetInfo_SeqSBAIJ, 138549b5e25fSSatish Balay MatEqual_SeqSBAIJ, 138649b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 138749b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 138849b5e25fSSatish Balay MatNorm_SeqSBAIJ, 138997304618SKris Buschelman /* 20*/ 0, 139049b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 139149b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 139249b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1393d519adbfSMatthew Knepley /* 24*/ 0, 139449b5e25fSSatish Balay 0, 139549b5e25fSSatish Balay 0, 1396db4efbfdSBarry Smith 0, 1397db4efbfdSBarry Smith 0, 13984994cf47SJed Brown /* 29*/ MatSetUp_SeqSBAIJ, 1399c464158bSHong Zhang 0, 1400db4efbfdSBarry Smith 0, 14018c778c55SBarry Smith 0, 14028c778c55SBarry Smith 0, 1403d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqSBAIJ, 1404719d5645SBarry Smith 0, 1405719d5645SBarry Smith 0, 140649b5e25fSSatish Balay 0, 1407c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 1408d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqSBAIJ, 140949b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 141049b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 141149b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 14123c896bc6SHong Zhang MatCopy_SeqSBAIJ, 1413d519adbfSMatthew Knepley /* 44*/ 0, 141449b5e25fSSatish Balay MatScale_SeqSBAIJ, 141549b5e25fSSatish Balay 0, 141649b5e25fSSatish Balay 0, 14173bededecSBarry Smith MatZeroRowsColumns_SeqSBAIJ, 1418f73d5cc4SBarry Smith /* 49*/ 0, 141949b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 142049b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 142149b5e25fSSatish Balay 0, 142249b5e25fSSatish Balay 0, 1423d519adbfSMatthew Knepley /* 54*/ 0, 142449b5e25fSSatish Balay 0, 142549b5e25fSSatish Balay 0, 142649b5e25fSSatish Balay 0, 142749b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 1428d519adbfSMatthew Knepley /* 59*/ MatGetSubMatrix_SeqSBAIJ, 142949b5e25fSSatish Balay 0, 143049b5e25fSSatish Balay 0, 1431357abbc8SBarry Smith 0, 1432d959ec07SHong Zhang 0, 1433d519adbfSMatthew Knepley /* 64*/ 0, 1434d959ec07SHong Zhang 0, 1435d959ec07SHong Zhang 0, 1436d959ec07SHong Zhang 0, 1437d959ec07SHong Zhang 0, 1438d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqSBAIJ, 14393e0d88b5SBarry Smith 0, 14403e0d88b5SBarry Smith 0, 14413e0d88b5SBarry Smith 0, 14423e0d88b5SBarry Smith 0, 1443d519adbfSMatthew Knepley /* 74*/ 0, 14443e0d88b5SBarry Smith 0, 14453e0d88b5SBarry Smith 0, 14463e0d88b5SBarry Smith 0, 14473e0d88b5SBarry Smith 0, 1448d519adbfSMatthew Knepley /* 79*/ 0, 14493e0d88b5SBarry Smith 0, 14503e0d88b5SBarry Smith 0, 145197304618SKris Buschelman MatGetInertia_SeqSBAIJ, 14525bba2384SShri Abhyankar MatLoad_SeqSBAIJ, 1453d519adbfSMatthew Knepley /* 84*/ MatIsSymmetric_SeqSBAIJ, 1454865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1455efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1456865e5f61SKris Buschelman 0, 1457865e5f61SKris Buschelman 0, 1458d519adbfSMatthew Knepley /* 89*/ 0, 1459865e5f61SKris Buschelman 0, 1460865e5f61SKris Buschelman 0, 1461865e5f61SKris Buschelman 0, 1462865e5f61SKris Buschelman 0, 1463d519adbfSMatthew Knepley /* 94*/ 0, 1464865e5f61SKris Buschelman 0, 1465865e5f61SKris Buschelman 0, 146699cafbc1SBarry Smith 0, 146799cafbc1SBarry Smith 0, 1468d519adbfSMatthew Knepley /* 99*/ 0, 146999cafbc1SBarry Smith 0, 147099cafbc1SBarry Smith 0, 147199cafbc1SBarry Smith 0, 147299cafbc1SBarry Smith 0, 1473d519adbfSMatthew Knepley /*104*/ 0, 147499cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1475f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1476f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 14772af78befSBarry Smith MatRestoreRowUpperTriangular_SeqSBAIJ, 1478d519adbfSMatthew Knepley /*109*/ 0, 14792af78befSBarry Smith 0, 14802af78befSBarry Smith 0, 14812af78befSBarry Smith 0, 1482547795f9SHong Zhang MatMissingDiagonal_SeqSBAIJ, 1483547795f9SHong Zhang /*114*/ 0, 1484547795f9SHong Zhang 0, 1485547795f9SHong Zhang 0, 1486547795f9SHong Zhang 0, 1487547795f9SHong Zhang 0, 1488547795f9SHong Zhang /*119*/ 0, 1489547795f9SHong Zhang 0, 14902f480046SShri Abhyankar 0, 14913964eb88SJed Brown 0, 14923964eb88SJed Brown 0, 14933964eb88SJed Brown /*124*/ 0, 14943964eb88SJed Brown 0, 14953964eb88SJed Brown 0, 14963964eb88SJed Brown 0, 14973964eb88SJed Brown 0, 14983964eb88SJed Brown /*129*/ 0, 14993964eb88SJed Brown 0, 15003964eb88SJed Brown 0, 15013964eb88SJed Brown 0, 15023964eb88SJed Brown 0, 15033964eb88SJed Brown /*134*/ 0, 15043964eb88SJed Brown 0, 15053964eb88SJed Brown 0, 15063964eb88SJed Brown 0, 15073964eb88SJed Brown 0, 15083964eb88SJed Brown /*139*/ 0, 1509f9426fe0SMark Adams 0, 15105bba2384SShri Abhyankar 0 151199cafbc1SBarry Smith }; 1512be1d678aSKris Buschelman 15134a2ae208SSatish Balay #undef __FUNCT__ 15144a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ" 15157087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 151649b5e25fSSatish Balay { 15174afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 1518d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 1519dfbe8321SBarry Smith PetscErrorCode ierr; 152049b5e25fSSatish Balay 152149b5e25fSSatish Balay PetscFunctionBegin; 1522e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 152349b5e25fSSatish Balay 152449b5e25fSSatish Balay /* allocate space for values if not already there */ 152549b5e25fSSatish Balay if (!aij->saved_values) { 1526785e854fSJed Brown ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr); 152749b5e25fSSatish Balay } 152849b5e25fSSatish Balay 152949b5e25fSSatish Balay /* copy values over */ 153087828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 153149b5e25fSSatish Balay PetscFunctionReturn(0); 153249b5e25fSSatish Balay } 153349b5e25fSSatish Balay 15344a2ae208SSatish Balay #undef __FUNCT__ 15354a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ" 15367087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 153749b5e25fSSatish Balay { 15384afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ*)mat->data; 15396849ba73SBarry Smith PetscErrorCode ierr; 1540d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 154149b5e25fSSatish Balay 154249b5e25fSSatish Balay PetscFunctionBegin; 1543e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1544e7e72b3dSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 154549b5e25fSSatish Balay 154649b5e25fSSatish Balay /* copy values over */ 154787828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 154849b5e25fSSatish Balay PetscFunctionReturn(0); 154949b5e25fSSatish Balay } 155049b5e25fSSatish Balay 15514a2ae208SSatish Balay #undef __FUNCT__ 1552a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ" 15537087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 155449b5e25fSSatish Balay { 1555c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 15566849ba73SBarry Smith PetscErrorCode ierr; 1557535b19f3SBarry Smith PetscInt i,mbs,bs2; 15582576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE; 155949b5e25fSSatish Balay 156049b5e25fSSatish Balay PetscFunctionBegin; 15612576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 1562273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1563db4efbfdSBarry Smith 156433d57670SJed Brown ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr); 156526283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 156626283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1567e02043d6SBarry Smith ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 1568899cda47SBarry Smith 1569d0f46423SBarry Smith mbs = B->rmap->N/bs; 157049b5e25fSSatish Balay bs2 = bs*bs; 157149b5e25fSSatish Balay 1572e7e72b3dSBarry Smith if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize"); 157349b5e25fSSatish Balay 1574ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1575ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1576ab93d7beSBarry Smith nz = 0; 1577ab93d7beSBarry Smith } 1578ab93d7beSBarry Smith 1579435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 1580e32f2f54SBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 158149b5e25fSSatish Balay if (nnz) { 158249b5e25fSSatish Balay for (i=0; i<mbs; i++) { 1583e32f2f54SBarry 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]); 1584e32f2f54SBarry Smith if (nnz[i] > mbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs); 158549b5e25fSSatish Balay } 158649b5e25fSSatish Balay } 158749b5e25fSSatish Balay 1588db4efbfdSBarry Smith B->ops->mult = MatMult_SeqSBAIJ_N; 1589db4efbfdSBarry Smith B->ops->multadd = MatMultAdd_SeqSBAIJ_N; 1590db4efbfdSBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_N; 1591db4efbfdSBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N; 159226fbe8dcSKarl Rupp 15930298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 159449b5e25fSSatish Balay if (!flg) { 159549b5e25fSSatish Balay switch (bs) { 159649b5e25fSSatish Balay case 1: 159749b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 159849b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1599431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1600431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 160149b5e25fSSatish Balay break; 160249b5e25fSSatish Balay case 2: 160349b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 160449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1605431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1606431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 160749b5e25fSSatish Balay break; 160849b5e25fSSatish Balay case 3: 160949b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 161049b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1611431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1612431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 161349b5e25fSSatish Balay break; 161449b5e25fSSatish Balay case 4: 161549b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 161649b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1617431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1618431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 161949b5e25fSSatish Balay break; 162049b5e25fSSatish Balay case 5: 162149b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 162249b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1623431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1624431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 162549b5e25fSSatish Balay break; 162649b5e25fSSatish Balay case 6: 162749b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 162849b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1629431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1630431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 163149b5e25fSSatish Balay break; 163249b5e25fSSatish Balay case 7: 1633de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 163449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1635431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1636431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 163749b5e25fSSatish Balay break; 163849b5e25fSSatish Balay } 163949b5e25fSSatish Balay } 164049b5e25fSSatish Balay 164149b5e25fSSatish Balay b->mbs = mbs; 16424afc71dfSHong Zhang b->nbs = mbs; 1643ab93d7beSBarry Smith if (!skipallocation) { 16442ee49352SLisandro Dalcin if (!b->imax) { 1645dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr); 164626fbe8dcSKarl Rupp 1647c760cd28SBarry Smith b->free_imax_ilen = PETSC_TRUE; 164826fbe8dcSKarl Rupp 16493bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr); 16502ee49352SLisandro Dalcin } 165149b5e25fSSatish Balay if (!nnz) { 1652435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 165349b5e25fSSatish Balay else if (nz <= 0) nz = 1; 165426fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->imax[i] = nz; 1655153ea458SHong Zhang nz = nz*mbs; /* total nz */ 165649b5e25fSSatish Balay } else { 165749b5e25fSSatish Balay nz = 0; 16588cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 165949b5e25fSSatish Balay } 16602ee49352SLisandro Dalcin /* b->ilen will count nonzeros in each block row so far. */ 166126fbe8dcSKarl Rupp for (i=0; i<mbs; i++) b->ilen[i] = 0; 16626c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 166349b5e25fSSatish Balay 166449b5e25fSSatish Balay /* allocate the matrix space */ 16652ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 1666dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr); 16673bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 16686c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 166913f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 167026fbe8dcSKarl Rupp 167149b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 167249b5e25fSSatish Balay 167349b5e25fSSatish Balay /* pointer to beginning of each row */ 1674e60cf9a0SBarry Smith b->i[0] = 0; 167526fbe8dcSKarl Rupp for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1]; 167626fbe8dcSKarl Rupp 1677e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1678e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1679e811da20SHong Zhang } else { 1680e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1681e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1682ab93d7beSBarry Smith } 168349b5e25fSSatish Balay 1684d0f46423SBarry Smith B->rmap->bs = bs; 168549b5e25fSSatish Balay b->bs2 = bs2; 16866c6c5352SBarry Smith b->nz = 0; 1687b32cb4a7SJed Brown b->maxnz = nz; 1688153ea458SHong Zhang 168916cdd363SHong Zhang b->inew = 0; 169016cdd363SHong Zhang b->jnew = 0; 169116cdd363SHong Zhang b->anew = 0; 169216cdd363SHong Zhang b->a2anew = 0; 16931a3463dfSHong Zhang b->permute = PETSC_FALSE; 16942576faa2SJed Brown if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);} 1695c464158bSHong Zhang PetscFunctionReturn(0); 1696c464158bSHong Zhang } 1697153ea458SHong Zhang 169838f409ebSLisandro Dalcin #undef __FUNCT__ 169938f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ" 170038f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[]) 170138f409ebSLisandro Dalcin { 170238f409ebSLisandro Dalcin PetscInt i,j,m,nz,nz_max=0,*nnz; 170338f409ebSLisandro Dalcin PetscScalar *values=0; 170438f409ebSLisandro Dalcin PetscBool roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented; 170538f409ebSLisandro Dalcin PetscErrorCode ierr; 170638f409ebSLisandro Dalcin PetscFunctionBegin; 170738f409ebSLisandro Dalcin if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs); 170838f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr); 170938f409ebSLisandro Dalcin ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr); 171038f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 171138f409ebSLisandro Dalcin ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 171238f409ebSLisandro Dalcin ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr); 171338f409ebSLisandro Dalcin m = B->rmap->n/bs; 171438f409ebSLisandro Dalcin 171538f409ebSLisandro Dalcin if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]); 1716785e854fSJed Brown ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr); 171738f409ebSLisandro Dalcin for (i=0; i<m; i++) { 171838f409ebSLisandro Dalcin nz = ii[i+1] - ii[i]; 171938f409ebSLisandro Dalcin if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz); 172038f409ebSLisandro Dalcin nz_max = PetscMax(nz_max,nz); 172138f409ebSLisandro Dalcin nnz[i] = nz; 172238f409ebSLisandro Dalcin } 172338f409ebSLisandro Dalcin ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr); 172438f409ebSLisandro Dalcin ierr = PetscFree(nnz);CHKERRQ(ierr); 172538f409ebSLisandro Dalcin 172638f409ebSLisandro Dalcin values = (PetscScalar*)V; 172738f409ebSLisandro Dalcin if (!values) { 17281795a4d1SJed Brown ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr); 172938f409ebSLisandro Dalcin } 173038f409ebSLisandro Dalcin for (i=0; i<m; i++) { 173138f409ebSLisandro Dalcin PetscInt ncols = ii[i+1] - ii[i]; 173238f409ebSLisandro Dalcin const PetscInt *icols = jj + ii[i]; 173338f409ebSLisandro Dalcin if (!roworiented || bs == 1) { 173438f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0); 173538f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr); 173638f409ebSLisandro Dalcin } else { 173738f409ebSLisandro Dalcin for (j=0; j<ncols; j++) { 173838f409ebSLisandro Dalcin const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0); 173938f409ebSLisandro Dalcin ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr); 174038f409ebSLisandro Dalcin } 174138f409ebSLisandro Dalcin } 174238f409ebSLisandro Dalcin } 174338f409ebSLisandro Dalcin if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); } 174438f409ebSLisandro Dalcin ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 174538f409ebSLisandro Dalcin ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 174638f409ebSLisandro Dalcin ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 174738f409ebSLisandro Dalcin PetscFunctionReturn(0); 174838f409ebSLisandro Dalcin } 174938f409ebSLisandro Dalcin 1750db4efbfdSBarry Smith /* 1751db4efbfdSBarry Smith This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization 1752db4efbfdSBarry Smith */ 17538b1456e3SHong Zhang #undef __FUNCT__ 1754d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace" 1755ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural) 1756db4efbfdSBarry Smith { 1757db4efbfdSBarry Smith PetscErrorCode ierr; 1758ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 1759db4efbfdSBarry Smith PetscInt bs = B->rmap->bs; 1760db4efbfdSBarry Smith 1761db4efbfdSBarry Smith PetscFunctionBegin; 17620298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr); 1763db4efbfdSBarry Smith if (flg) bs = 8; 1764db4efbfdSBarry Smith 1765db4efbfdSBarry Smith if (!natural) { 1766db4efbfdSBarry Smith switch (bs) { 1767db4efbfdSBarry Smith case 1: 1768d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace; 1769db4efbfdSBarry Smith break; 1770db4efbfdSBarry Smith case 2: 1771db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 1772db4efbfdSBarry Smith break; 1773db4efbfdSBarry Smith case 3: 1774db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 1775db4efbfdSBarry Smith break; 1776db4efbfdSBarry Smith case 4: 1777db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 1778db4efbfdSBarry Smith break; 1779db4efbfdSBarry Smith case 5: 1780db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 1781db4efbfdSBarry Smith break; 1782db4efbfdSBarry Smith case 6: 1783db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 1784db4efbfdSBarry Smith break; 1785db4efbfdSBarry Smith case 7: 1786db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 1787db4efbfdSBarry Smith break; 1788db4efbfdSBarry Smith default: 1789db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N; 1790db4efbfdSBarry Smith break; 1791db4efbfdSBarry Smith } 1792db4efbfdSBarry Smith } else { 1793db4efbfdSBarry Smith switch (bs) { 1794db4efbfdSBarry Smith case 1: 1795d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace; 1796db4efbfdSBarry Smith break; 1797db4efbfdSBarry Smith case 2: 1798db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 1799db4efbfdSBarry Smith break; 1800db4efbfdSBarry Smith case 3: 1801db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 1802db4efbfdSBarry Smith break; 1803db4efbfdSBarry Smith case 4: 1804db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 1805db4efbfdSBarry Smith break; 1806db4efbfdSBarry Smith case 5: 1807db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 1808db4efbfdSBarry Smith break; 1809db4efbfdSBarry Smith case 6: 1810db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 1811db4efbfdSBarry Smith break; 1812db4efbfdSBarry Smith case 7: 1813db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 1814db4efbfdSBarry Smith break; 1815db4efbfdSBarry Smith default: 1816db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1817db4efbfdSBarry Smith break; 1818db4efbfdSBarry Smith } 1819db4efbfdSBarry Smith } 1820db4efbfdSBarry Smith PetscFunctionReturn(0); 1821db4efbfdSBarry Smith } 1822db4efbfdSBarry Smith 18238cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*); 18248cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*); 1825d769727bSBarry Smith 18265c9eb25fSBarry Smith #undef __FUNCT__ 18275c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc" 18288cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B) 18295c9eb25fSBarry Smith { 1830d0f46423SBarry Smith PetscInt n = A->rmap->n; 18315c9eb25fSBarry Smith PetscErrorCode ierr; 18325c9eb25fSBarry Smith 18335c9eb25fSBarry Smith PetscFunctionBegin; 18340e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX) 18350e92d65fSHong Zhang if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 18360e92d65fSHong Zhang #endif 1837ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 18385c9eb25fSBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 18395c9eb25fSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 18405c9eb25fSBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 18410298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 184226fbe8dcSKarl Rupp 18437b056e98SHong Zhang (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ; 1844c6d0d4f0SHong Zhang (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqSBAIJ; 1845e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 1846d5f3da31SBarry Smith (*B)->factortype = ftype; 18475c9eb25fSBarry Smith PetscFunctionReturn(0); 18485c9eb25fSBarry Smith } 18495c9eb25fSBarry Smith 1850db4efbfdSBarry Smith #undef __FUNCT__ 1851db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc" 1852ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool *flg) 1853db4efbfdSBarry Smith { 1854db4efbfdSBarry Smith PetscFunctionBegin; 1855db4efbfdSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 1856db4efbfdSBarry Smith *flg = PETSC_TRUE; 1857db4efbfdSBarry Smith } else { 1858db4efbfdSBarry Smith *flg = PETSC_FALSE; 1859db4efbfdSBarry Smith } 1860db4efbfdSBarry Smith PetscFunctionReturn(0); 1861db4efbfdSBarry Smith } 1862db4efbfdSBarry Smith 1863611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS) 18648cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*); 1865611f576cSBarry Smith #endif 1866b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX) 18678cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*); 1868b5e56a35SBarry Smith #endif 1869913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE) 18708cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*); 187120db9a53SJed Brown #endif 18728cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*); 18735c9eb25fSBarry Smith 18740bad9183SKris Buschelman /*MC 1875fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 18760bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 18770bad9183SKris Buschelman 1878828413b8SBarry Smith For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you 187971dad5bbSBarry Smith can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd() 1880828413b8SBarry Smith 18810bad9183SKris Buschelman Options Database Keys: 18820bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 18830bad9183SKris Buschelman 188471dad5bbSBarry Smith Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not 188571dad5bbSBarry 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 188671dad5bbSBarry 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. 188771dad5bbSBarry Smith 188871dad5bbSBarry Smith 18890bad9183SKris Buschelman Level: beginner 18900bad9183SKris Buschelman 18910bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 18920bad9183SKris Buschelman M*/ 18930bad9183SKris Buschelman 18948cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*); 1895aa5a9175SDahai Guo 1896a23d5eceSKris Buschelman #undef __FUNCT__ 1897a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ" 18988cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1899a23d5eceSKris Buschelman { 1900a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1901dfbe8321SBarry Smith PetscErrorCode ierr; 190213f74950SBarry Smith PetscMPIInt size; 1903ace3abfcSBarry Smith PetscBool no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE; 1904a23d5eceSKris Buschelman 1905a23d5eceSKris Buschelman PetscFunctionBegin; 1906ce94432eSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 1907e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1908a23d5eceSKris Buschelman 1909b00a9115SJed Brown ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1910a23d5eceSKris Buschelman B->data = (void*)b; 1911a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 191226fbe8dcSKarl Rupp 1913a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1914a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1915a23d5eceSKris Buschelman b->row = 0; 1916a23d5eceSKris Buschelman b->icol = 0; 1917a23d5eceSKris Buschelman b->reallocs = 0; 1918a23d5eceSKris Buschelman b->saved_values = 0; 19190def2e27SBarry Smith b->inode.limit = 5; 19200def2e27SBarry Smith b->inode.max_limit = 5; 1921a23d5eceSKris Buschelman 1922a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1923a23d5eceSKris Buschelman b->nonew = 0; 1924a23d5eceSKris Buschelman b->diag = 0; 1925a23d5eceSKris Buschelman b->solve_work = 0; 1926a23d5eceSKris Buschelman b->mult_work = 0; 1927a23d5eceSKris Buschelman B->spptr = 0; 1928f2cbd3d5SJed Brown B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2; 1929a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 1930a23d5eceSKris Buschelman b->xtoy = 0; 1931a23d5eceSKris Buschelman b->XtoY = 0; 1932a23d5eceSKris Buschelman 1933a23d5eceSKris Buschelman b->inew = 0; 1934a23d5eceSKris Buschelman b->jnew = 0; 1935a23d5eceSKris Buschelman b->anew = 0; 1936a23d5eceSKris Buschelman b->a2anew = 0; 1937a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1938a23d5eceSKris Buschelman 193971dad5bbSBarry Smith b->ignore_ltriangular = PETSC_TRUE; 194026fbe8dcSKarl Rupp 19410298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr); 1942941593c8SHong Zhang 1943f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 194426fbe8dcSKarl Rupp 19450298fd71SBarry Smith ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr); 1946f5edf698SHong Zhang 1947b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX) 1948bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_pastix_C",MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr); 1949b5e56a35SBarry Smith #endif 1950611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS) 1951bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_mumps_C",MatGetFactor_sbaij_mumps);CHKERRQ(ierr); 1952611f576cSBarry Smith #endif 1953913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE) 1954bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_cholmod_C",MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr); 195520db9a53SJed Brown #endif 1956bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactorAvailable_petsc_C",MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr); 1957bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr); 1958bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_sbstrm_C",MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr); 1959bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1960bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1961bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 1962bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1963bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1964bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 196538f409ebSLisandro Dalcin ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr); 1966bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr); 196723ce1328SBarry Smith 196823ce1328SBarry Smith B->symmetric = PETSC_TRUE; 196923ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 197023ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 197123ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 197226fbe8dcSKarl Rupp 197317667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 19740def2e27SBarry Smith 1975ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr); 19760298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr); 197726fbe8dcSKarl Rupp if (no_unroll) { 197826fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr); 197926fbe8dcSKarl Rupp } 19800298fd71SBarry Smith ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr); 198126fbe8dcSKarl Rupp if (no_inode) { 198226fbe8dcSKarl Rupp ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr); 198326fbe8dcSKarl Rupp } 19840298fd71SBarry Smith ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr); 19850def2e27SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 1986ace3abfcSBarry Smith b->inode.use = (PetscBool)(!(no_unroll || no_inode)); 19870def2e27SBarry Smith if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit; 1988a23d5eceSKris Buschelman PetscFunctionReturn(0); 1989a23d5eceSKris Buschelman } 1990a23d5eceSKris Buschelman 1991a23d5eceSKris Buschelman #undef __FUNCT__ 1992a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation" 1993a23d5eceSKris Buschelman /*@C 1994a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 1995a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 1996a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 1997a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 1998a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 1999a23d5eceSKris Buschelman 2000a23d5eceSKris Buschelman Collective on Mat 2001a23d5eceSKris Buschelman 2002a23d5eceSKris Buschelman Input Parameters: 20031c4f3114SJed Brown + B - the symmetric matrix 2004a23d5eceSKris Buschelman . bs - size of block 2005a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 2006a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 20070298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2008a23d5eceSKris Buschelman 2009a23d5eceSKris Buschelman Options Database Keys: 2010a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 2011a23d5eceSKris Buschelman block calculations (much slower) 2012db4efbfdSBarry Smith . -mat_block_size - size of the blocks to use (only works if a negative bs is passed in 2013a23d5eceSKris Buschelman 2014a23d5eceSKris Buschelman Level: intermediate 2015a23d5eceSKris Buschelman 2016a23d5eceSKris Buschelman Notes: 2017a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 20180298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 2019a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 2020a23d5eceSKris Buschelman 2021aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 2022aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 2023aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 2024aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 2025aa95bbe8SBarry Smith 202649a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 202749a6f317SBarry Smith 202849a6f317SBarry Smith 202969b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2030a23d5eceSKris Buschelman @*/ 20317087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 203213f74950SBarry Smith { 20334ac538c5SBarry Smith PetscErrorCode ierr; 2034a23d5eceSKris Buschelman 2035a23d5eceSKris Buschelman PetscFunctionBegin; 20366ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 20376ba663aaSJed Brown PetscValidType(B,1); 20386ba663aaSJed Brown PetscValidLogicalCollectiveInt(B,bs,2); 20394ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr); 2040a23d5eceSKris Buschelman PetscFunctionReturn(0); 2041a23d5eceSKris Buschelman } 204249b5e25fSSatish Balay 20434a2ae208SSatish Balay #undef __FUNCT__ 204438f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR" 204538f409ebSLisandro Dalcin /*@C 204638f409ebSLisandro Dalcin MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format. 204738f409ebSLisandro Dalcin 204838f409ebSLisandro Dalcin Input Parameters: 20491c4f3114SJed Brown + B - the matrix 205038f409ebSLisandro Dalcin . i - the indices into j for the start of each local row (starts with zero) 205138f409ebSLisandro Dalcin . j - the column indices for each local row (starts with zero) these must be sorted for each row 205238f409ebSLisandro Dalcin - v - optional values in the matrix 205338f409ebSLisandro Dalcin 205438f409ebSLisandro Dalcin Level: developer 205538f409ebSLisandro Dalcin 205638f409ebSLisandro Dalcin Notes: 205738f409ebSLisandro Dalcin The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED. For example, C programs 205838f409ebSLisandro 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 205938f409ebSLisandro Dalcin over rows within a block and the last index is over columns within a block row. Fortran programs will likely set 206038f409ebSLisandro 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 206138f409ebSLisandro Dalcin block column and the second index is over columns within a block. 206238f409ebSLisandro Dalcin 206338f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse 206438f409ebSLisandro Dalcin 206538f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ 206638f409ebSLisandro Dalcin @*/ 206738f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[]) 206838f409ebSLisandro Dalcin { 206938f409ebSLisandro Dalcin PetscErrorCode ierr; 207038f409ebSLisandro Dalcin 207138f409ebSLisandro Dalcin PetscFunctionBegin; 207238f409ebSLisandro Dalcin PetscValidHeaderSpecific(B,MAT_CLASSID,1); 207338f409ebSLisandro Dalcin PetscValidType(B,1); 207438f409ebSLisandro Dalcin PetscValidLogicalCollectiveInt(B,bs,2); 207538f409ebSLisandro Dalcin ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr); 207638f409ebSLisandro Dalcin PetscFunctionReturn(0); 207738f409ebSLisandro Dalcin } 207838f409ebSLisandro Dalcin 207938f409ebSLisandro Dalcin #undef __FUNCT__ 20804a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ" 2081c464158bSHong Zhang /*@C 2082c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 2083c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 2084c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 2085c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 2086c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 208749b5e25fSSatish Balay 2088c464158bSHong Zhang Collective on MPI_Comm 2089c464158bSHong Zhang 2090c464158bSHong Zhang Input Parameters: 2091c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2092c464158bSHong Zhang . bs - size of block 2093c464158bSHong Zhang . m - number of rows, or number of columns 2094c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 2095744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 20960298fd71SBarry Smith diagonal portion of each block (possibly different for each block row) or NULL 2097c464158bSHong Zhang 2098c464158bSHong Zhang Output Parameter: 2099c464158bSHong Zhang . A - the symmetric matrix 2100c464158bSHong Zhang 2101c464158bSHong Zhang Options Database Keys: 2102c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 2103c464158bSHong Zhang block calculations (much slower) 2104c464158bSHong Zhang . -mat_block_size - size of the blocks to use 2105c464158bSHong Zhang 2106c464158bSHong Zhang Level: intermediate 2107c464158bSHong Zhang 2108175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2109ae1d86c5SBarry Smith MatXXXXSetPreallocation() paradgm instead of this routine directly. 2110175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 2111175b88e8SBarry Smith 2112c464158bSHong Zhang Notes: 21136d6d819aSHong Zhang The number of rows and columns must be divisible by blocksize. 21146d6d819aSHong Zhang This matrix type does not support complex Hermitian operation. 2115c464158bSHong Zhang 2116c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 21170298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 2118a7f22e61SSatish Balay allocation. See Users-Manual: ch_mat for details. 2119c464158bSHong Zhang 212049a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 212149a6f317SBarry Smith 212269b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2123c464158bSHong Zhang @*/ 21247087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 2125c464158bSHong Zhang { 2126dfbe8321SBarry Smith PetscErrorCode ierr; 2127c464158bSHong Zhang 2128c464158bSHong Zhang PetscFunctionBegin; 2129f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 2130f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2131c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 2132ab93d7beSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 213349b5e25fSSatish Balay PetscFunctionReturn(0); 213449b5e25fSSatish Balay } 213549b5e25fSSatish Balay 21364a2ae208SSatish Balay #undef __FUNCT__ 21374a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ" 2138dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 213949b5e25fSSatish Balay { 214049b5e25fSSatish Balay Mat C; 214149b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 21426849ba73SBarry Smith PetscErrorCode ierr; 2143b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 214449b5e25fSSatish Balay 214549b5e25fSSatish Balay PetscFunctionBegin; 2146e32f2f54SBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix"); 214749b5e25fSSatish Balay 214849b5e25fSSatish Balay *B = 0; 2149ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2150d0f46423SBarry Smith ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr); 21518e9a0fb8SHong Zhang ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr); 21521d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 2153692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 2154692f9cbeSHong Zhang 2155273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 2156d5f3da31SBarry Smith C->factortype = A->factortype; 215749b5e25fSSatish Balay c->row = 0; 215849b5e25fSSatish Balay c->icol = 0; 215949b5e25fSSatish Balay c->saved_values = 0; 2160a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 216149b5e25fSSatish Balay C->assembled = PETSC_TRUE; 216249b5e25fSSatish Balay 21631e1e43feSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 21641e1e43feSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 216549b5e25fSSatish Balay c->bs2 = a->bs2; 216649b5e25fSSatish Balay c->mbs = a->mbs; 216749b5e25fSSatish Balay c->nbs = a->nbs; 216849b5e25fSSatish Balay 2169c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2170c760cd28SBarry Smith c->imax = a->imax; 2171c760cd28SBarry Smith c->ilen = a->ilen; 2172c760cd28SBarry Smith c->free_imax_ilen = PETSC_FALSE; 2173c760cd28SBarry Smith } else { 2174dcca6d9dSJed Brown ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr); 21753bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 217649b5e25fSSatish Balay for (i=0; i<mbs; i++) { 217749b5e25fSSatish Balay c->imax[i] = a->imax[i]; 217849b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 217949b5e25fSSatish Balay } 2180c760cd28SBarry Smith c->free_imax_ilen = PETSC_TRUE; 2181c760cd28SBarry Smith } 218249b5e25fSSatish Balay 218349b5e25fSSatish Balay /* allocate the matrix space */ 21844da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2185785e854fSJed Brown ierr = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr); 21863bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 218744e1c64aSLisandro Dalcin c->i = a->i; 218844e1c64aSLisandro Dalcin c->j = a->j; 21894da8f245SBarry Smith c->singlemalloc = PETSC_FALSE; 219044e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 21914da8f245SBarry Smith c->free_ij = PETSC_FALSE; 21924da8f245SBarry Smith c->parent = A; 21934da8f245SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 21944da8f245SBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21954da8f245SBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21964da8f245SBarry Smith } else { 2197dcca6d9dSJed Brown ierr = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr); 219813f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 21993bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 22004da8f245SBarry Smith c->singlemalloc = PETSC_TRUE; 220144e1c64aSLisandro Dalcin c->free_a = PETSC_TRUE; 22024da8f245SBarry Smith c->free_ij = PETSC_TRUE; 22034da8f245SBarry Smith } 220449b5e25fSSatish Balay if (mbs > 0) { 22054da8f245SBarry Smith if (cpvalues != MAT_SHARE_NONZERO_PATTERN) { 220613f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 22074da8f245SBarry Smith } 220849b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 220949b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 221049b5e25fSSatish Balay } else { 221149b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 221249b5e25fSSatish Balay } 2213a1c3900fSBarry Smith if (a->jshort) { 221444e1c64aSLisandro Dalcin /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */ 221544e1c64aSLisandro Dalcin /* if the parent matrix is reassembled, this child matrix will never notice */ 2216785e854fSJed Brown ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr); 22173bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr); 2218a1c3900fSBarry Smith ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr); 221926fbe8dcSKarl Rupp 22204da8f245SBarry Smith c->free_jshort = PETSC_TRUE; 22214da8f245SBarry Smith } 2222a1c3900fSBarry Smith } 222349b5e25fSSatish Balay 222449b5e25fSSatish Balay c->roworiented = a->roworiented; 222549b5e25fSSatish Balay c->nonew = a->nonew; 222649b5e25fSSatish Balay 222749b5e25fSSatish Balay if (a->diag) { 2228c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2229c760cd28SBarry Smith c->diag = a->diag; 2230c760cd28SBarry Smith c->free_diag = PETSC_FALSE; 2231c760cd28SBarry Smith } else { 2232785e854fSJed Brown ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr); 22333bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr); 223426fbe8dcSKarl Rupp for (i=0; i<mbs; i++) c->diag[i] = a->diag[i]; 2235c760cd28SBarry Smith c->free_diag = PETSC_TRUE; 2236c760cd28SBarry Smith } 223744e1c64aSLisandro Dalcin } 22386c6c5352SBarry Smith c->nz = a->nz; 2239f2cbd3d5SJed Brown c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 224049b5e25fSSatish Balay c->solve_work = 0; 224149b5e25fSSatish Balay c->mult_work = 0; 224226fbe8dcSKarl Rupp 224349b5e25fSSatish Balay *B = C; 2244140e18c1SBarry Smith ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 224549b5e25fSSatish Balay PetscFunctionReturn(0); 224649b5e25fSSatish Balay } 224749b5e25fSSatish Balay 22484a2ae208SSatish Balay #undef __FUNCT__ 22495bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ" 2250112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer) 22512f480046SShri Abhyankar { 22522f480046SShri Abhyankar Mat_SeqSBAIJ *a; 22532f480046SShri Abhyankar PetscErrorCode ierr; 22542f480046SShri Abhyankar int fd; 22552f480046SShri Abhyankar PetscMPIInt size; 22562f480046SShri Abhyankar PetscInt i,nz,header[4],*rowlengths=0,M,N,bs=1; 22572f480046SShri Abhyankar PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 22582f480046SShri Abhyankar PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols; 22592f480046SShri Abhyankar PetscInt *masked,nmask,tmp,bs2,ishift; 22602f480046SShri Abhyankar PetscScalar *aa; 2261ce94432eSBarry Smith MPI_Comm comm; 22622f480046SShri Abhyankar 22632f480046SShri Abhyankar PetscFunctionBegin; 2264ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 22650298fd71SBarry Smith ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr); 22662f480046SShri Abhyankar bs2 = bs*bs; 22672f480046SShri Abhyankar 22682f480046SShri Abhyankar ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 22692f480046SShri Abhyankar if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor"); 22702f480046SShri Abhyankar ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 22712f480046SShri Abhyankar ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 22722f480046SShri Abhyankar if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 22732f480046SShri Abhyankar M = header[1]; N = header[2]; nz = header[3]; 22742f480046SShri Abhyankar 22752f480046SShri Abhyankar if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 22762f480046SShri Abhyankar 22772f480046SShri Abhyankar if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices"); 22782f480046SShri Abhyankar 22792f480046SShri Abhyankar /* 22802f480046SShri Abhyankar This code adds extra rows to make sure the number of rows is 22812f480046SShri Abhyankar divisible by the blocksize 22822f480046SShri Abhyankar */ 22832f480046SShri Abhyankar mbs = M/bs; 22842f480046SShri Abhyankar extra_rows = bs - M + bs*(mbs); 22852f480046SShri Abhyankar if (extra_rows == bs) extra_rows = 0; 22862f480046SShri Abhyankar else mbs++; 22872f480046SShri Abhyankar if (extra_rows) { 22882f480046SShri Abhyankar ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr); 22892f480046SShri Abhyankar } 22902f480046SShri Abhyankar 22912f480046SShri Abhyankar /* Set global sizes if not already set */ 22922f480046SShri Abhyankar if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) { 22932f480046SShri Abhyankar ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr); 22942f480046SShri Abhyankar } else { /* Check if the matrix global sizes are correct */ 22952f480046SShri Abhyankar ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr); 22962f480046SShri 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); 22972f480046SShri Abhyankar } 22982f480046SShri Abhyankar 22992f480046SShri Abhyankar /* read in row lengths */ 2300785e854fSJed Brown ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr); 23012f480046SShri Abhyankar ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 23022f480046SShri Abhyankar for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 23032f480046SShri Abhyankar 23042f480046SShri Abhyankar /* read in column indices */ 2305785e854fSJed Brown ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr); 23062f480046SShri Abhyankar ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 23072f480046SShri Abhyankar for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 23082f480046SShri Abhyankar 23092f480046SShri Abhyankar /* loop over row lengths determining block row lengths */ 23101795a4d1SJed Brown ierr = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr); 2311dcca6d9dSJed Brown ierr = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr); 23122f480046SShri Abhyankar ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 23132f480046SShri Abhyankar rowcount = 0; 23142f480046SShri Abhyankar nzcount = 0; 23152f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23162f480046SShri Abhyankar nmask = 0; 23172f480046SShri Abhyankar for (j=0; j<bs; j++) { 23182f480046SShri Abhyankar kmax = rowlengths[rowcount]; 23192f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23202f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23212f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 23222f480046SShri Abhyankar } 23232f480046SShri Abhyankar rowcount++; 23242f480046SShri Abhyankar } 23252f480046SShri Abhyankar s_browlengths[i] += nmask; 23262f480046SShri Abhyankar 23272f480046SShri Abhyankar /* zero out the mask elements we set */ 23282f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23292f480046SShri Abhyankar } 23302f480046SShri Abhyankar 23312f480046SShri Abhyankar /* Do preallocation */ 23322f480046SShri Abhyankar ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr); 23332f480046SShri Abhyankar a = (Mat_SeqSBAIJ*)newmat->data; 23342f480046SShri Abhyankar 23352f480046SShri Abhyankar /* set matrix "i" values */ 23362f480046SShri Abhyankar a->i[0] = 0; 23372f480046SShri Abhyankar for (i=1; i<= mbs; i++) { 23382f480046SShri Abhyankar a->i[i] = a->i[i-1] + s_browlengths[i-1]; 23392f480046SShri Abhyankar a->ilen[i-1] = s_browlengths[i-1]; 23402f480046SShri Abhyankar } 23412f480046SShri Abhyankar a->nz = a->i[mbs]; 23422f480046SShri Abhyankar 23432f480046SShri Abhyankar /* read in nonzero values */ 2344785e854fSJed Brown ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr); 23452f480046SShri Abhyankar ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 23462f480046SShri Abhyankar for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 23472f480046SShri Abhyankar 23482f480046SShri Abhyankar /* set "a" and "j" values into matrix */ 23492f480046SShri Abhyankar nzcount = 0; jcount = 0; 23502f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23512f480046SShri Abhyankar nzcountb = nzcount; 23522f480046SShri Abhyankar nmask = 0; 23532f480046SShri Abhyankar for (j=0; j<bs; j++) { 23542f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23552f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23562f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23572f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 23582f480046SShri Abhyankar } 23592f480046SShri Abhyankar } 23602f480046SShri Abhyankar /* sort the masked values */ 23612f480046SShri Abhyankar ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 23622f480046SShri Abhyankar 23632f480046SShri Abhyankar /* set "j" values into matrix */ 23642f480046SShri Abhyankar maskcount = 1; 23652f480046SShri Abhyankar for (j=0; j<nmask; j++) { 23662f480046SShri Abhyankar a->j[jcount++] = masked[j]; 23672f480046SShri Abhyankar mask[masked[j]] = maskcount++; 23682f480046SShri Abhyankar } 23692f480046SShri Abhyankar 23702f480046SShri Abhyankar /* set "a" values into matrix */ 23712f480046SShri Abhyankar ishift = bs2*a->i[i]; 23722f480046SShri Abhyankar for (j=0; j<bs; j++) { 23732f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23742f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23752f480046SShri Abhyankar tmp = jj[nzcountb]/bs; /* block col. index */ 23762f480046SShri Abhyankar if (tmp >= i) { 23772f480046SShri Abhyankar block = mask[tmp] - 1; 23782f480046SShri Abhyankar point = jj[nzcountb] - bs*tmp; 23792f480046SShri Abhyankar idx = ishift + bs2*block + j + bs*point; 23802f480046SShri Abhyankar a->a[idx] = aa[nzcountb]; 23812f480046SShri Abhyankar } 23822f480046SShri Abhyankar nzcountb++; 23832f480046SShri Abhyankar } 23842f480046SShri Abhyankar } 23852f480046SShri Abhyankar /* zero out the mask elements we set */ 23862f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23872f480046SShri Abhyankar } 23882f480046SShri Abhyankar if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 23892f480046SShri Abhyankar 23902f480046SShri Abhyankar ierr = PetscFree(rowlengths);CHKERRQ(ierr); 23912f480046SShri Abhyankar ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 23922f480046SShri Abhyankar ierr = PetscFree(aa);CHKERRQ(ierr); 23932f480046SShri Abhyankar ierr = PetscFree(jj);CHKERRQ(ierr); 23942f480046SShri Abhyankar ierr = PetscFree2(mask,masked);CHKERRQ(ierr); 23952f480046SShri Abhyankar 23962f480046SShri Abhyankar ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23972f480046SShri Abhyankar ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23982f480046SShri Abhyankar PetscFunctionReturn(0); 23992f480046SShri Abhyankar } 24002f480046SShri Abhyankar 24012f480046SShri Abhyankar #undef __FUNCT__ 2402c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays" 2403c75a6043SHong Zhang /*@ 2404c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2405c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2406c75a6043SHong Zhang 2407c75a6043SHong Zhang Collective on MPI_Comm 2408c75a6043SHong Zhang 2409c75a6043SHong Zhang Input Parameters: 2410c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2411c75a6043SHong Zhang . bs - size of block 2412c75a6043SHong Zhang . m - number of rows 2413c75a6043SHong Zhang . n - number of columns 2414c75a6043SHong Zhang . i - row indices 2415c75a6043SHong Zhang . j - column indices 2416c75a6043SHong Zhang - a - matrix values 2417c75a6043SHong Zhang 2418c75a6043SHong Zhang Output Parameter: 2419c75a6043SHong Zhang . mat - the matrix 2420c75a6043SHong Zhang 2421dfb205c3SBarry Smith Level: advanced 2422c75a6043SHong Zhang 2423c75a6043SHong Zhang Notes: 2424c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2425c75a6043SHong Zhang once the matrix is destroyed 2426c75a6043SHong Zhang 2427c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2428c75a6043SHong Zhang 2429c75a6043SHong Zhang The i and j indices are 0 based 2430c75a6043SHong Zhang 2431dfb205c3SBarry 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 2432dfb205c3SBarry Smith it is the regular CSR format excluding the lower triangular elements. 2433dfb205c3SBarry Smith 243469b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ() 2435c75a6043SHong Zhang 2436c75a6043SHong Zhang @*/ 24377087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat) 2438c75a6043SHong Zhang { 2439c75a6043SHong Zhang PetscErrorCode ierr; 2440c75a6043SHong Zhang PetscInt ii; 2441c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2442c75a6043SHong Zhang 2443c75a6043SHong Zhang PetscFunctionBegin; 2444e32f2f54SBarry Smith if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 2445e32f2f54SBarry Smith if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2446c75a6043SHong Zhang 2447c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2448c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2449c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2450c75a6043SHong Zhang ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 2451c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2452dcca6d9dSJed Brown ierr = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr); 24533bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr); 2454c75a6043SHong Zhang 2455c75a6043SHong Zhang sbaij->i = i; 2456c75a6043SHong Zhang sbaij->j = j; 2457c75a6043SHong Zhang sbaij->a = a; 245826fbe8dcSKarl Rupp 2459c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2460c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2461e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2462e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2463c75a6043SHong Zhang 2464c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2465c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2466c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2467e32f2f54SBarry 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]); 2468c75a6043SHong Zhang #endif 2469c75a6043SHong Zhang } 2470c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2471c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2472e32f2f54SBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2473e32f2f54SBarry 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]); 2474c75a6043SHong Zhang } 2475c75a6043SHong Zhang #endif 2476c75a6043SHong Zhang 2477c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2478c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2479c75a6043SHong Zhang PetscFunctionReturn(0); 2480c75a6043SHong Zhang } 2481d06b337dSHong Zhang 2482d06b337dSHong Zhang 2483d06b337dSHong Zhang 248449b5e25fSSatish Balay 248549b5e25fSSatish Balay 2486