1641875f9SMatthew G Knepley 2641875f9SMatthew G Knepley /* 3d515b9b4SShri Abhyankar Provides an interface to the CHOLMOD sparse solver available through SuiteSparse version 4.2.1 4641875f9SMatthew G Knepley 58999bf53SRichard Mills When built with PETSC_USE_64BIT_INDICES this will use Suitesparse_long as the 6641875f9SMatthew G Knepley integer type in UMFPACK, otherwise it will use int. This means 7641875f9SMatthew G Knepley all integers in this file as simply declared as PetscInt. Also it means 89e475b0dSSatish Balay that one cannot use 64BIT_INDICES on 32bit machines [as Suitesparse_long is 32bit only] 9641875f9SMatthew G Knepley 10641875f9SMatthew G Knepley */ 11641875f9SMatthew G Knepley 12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h> 13c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/cholmod/cholmodimpl.h> 14641875f9SMatthew G Knepley 15641875f9SMatthew G Knepley /* 16641875f9SMatthew G Knepley This is a terrible hack, but it allows the error handler to retain a context. 17641875f9SMatthew G Knepley Note that this hack really cannot be made both reentrant and concurrent. 18641875f9SMatthew G Knepley */ 19641875f9SMatthew G Knepley static Mat static_F; 20641875f9SMatthew G Knepley 21641875f9SMatthew G Knepley static void CholmodErrorHandler(int status,const char *file,int line,const char *message) 22641875f9SMatthew G Knepley { 23ec55ff42SBarry Smith PetscErrorCode ierr; 24641875f9SMatthew G Knepley 25641875f9SMatthew G Knepley PetscFunctionBegin; 26641875f9SMatthew G Knepley if (status > CHOLMOD_OK) { 27e49b6e0cSMatthew G. Knepley ierr = PetscInfo4(static_F,"CHOLMOD warning %d at %s:%d: %s\n",status,file,line,message);CHKERRV(ierr); 28641875f9SMatthew G Knepley } else if (status == CHOLMOD_OK) { /* Documentation says this can happen, but why? */ 29e49b6e0cSMatthew G. Knepley ierr = PetscInfo3(static_F,"CHOLMOD OK at %s:%d: %s\n",file,line,message);CHKERRV(ierr); 30641875f9SMatthew G Knepley } else { 31e49b6e0cSMatthew G. Knepley ierr = PetscErrorPrintf("CHOLMOD error %d at %s:%d: %s\n",status,file,line,message);CHKERRV(ierr); 32641875f9SMatthew G Knepley } 33641875f9SMatthew G Knepley PetscFunctionReturnVoid(); 34641875f9SMatthew G Knepley } 35641875f9SMatthew G Knepley 367087cfbeSBarry Smith PetscErrorCode CholmodStart(Mat F) 37641875f9SMatthew G Knepley { 38641875f9SMatthew G Knepley PetscErrorCode ierr; 396b8f6f9dSBarry Smith Mat_CHOLMOD *chol=(Mat_CHOLMOD*)F->data; 40641875f9SMatthew G Knepley cholmod_common *c; 41ace3abfcSBarry Smith PetscBool flg; 42641875f9SMatthew G Knepley 43641875f9SMatthew G Knepley PetscFunctionBegin; 44641875f9SMatthew G Knepley if (chol->common) PetscFunctionReturn(0); 45854ce69bSBarry Smith ierr = PetscMalloc1(1,&chol->common);CHKERRQ(ierr); 46641875f9SMatthew G Knepley ierr = !cholmod_X_start(chol->common);CHKERRQ(ierr); 4726fbe8dcSKarl Rupp 48641875f9SMatthew G Knepley c = chol->common; 49641875f9SMatthew G Knepley c->error_handler = CholmodErrorHandler; 50641875f9SMatthew G Knepley 51641875f9SMatthew G Knepley #define CHOLMOD_OPTION_DOUBLE(name,help) do { \ 52641875f9SMatthew G Knepley PetscReal tmp = (PetscReal)c->name; \ 538afaa268SBarry Smith ierr = PetscOptionsReal("-mat_cholmod_" #name,help,"None",tmp,&tmp,NULL);CHKERRQ(ierr); \ 54641875f9SMatthew G Knepley c->name = (double)tmp; \ 55641875f9SMatthew G Knepley } while (0) 5626fbe8dcSKarl Rupp 57641875f9SMatthew G Knepley #define CHOLMOD_OPTION_INT(name,help) do { \ 58641875f9SMatthew G Knepley PetscInt tmp = (PetscInt)c->name; \ 598afaa268SBarry Smith ierr = PetscOptionsInt("-mat_cholmod_" #name,help,"None",tmp,&tmp,NULL);CHKERRQ(ierr); \ 60641875f9SMatthew G Knepley c->name = (int)tmp; \ 61641875f9SMatthew G Knepley } while (0) 6226fbe8dcSKarl Rupp 63641875f9SMatthew G Knepley #define CHOLMOD_OPTION_SIZE_T(name,help) do { \ 6454b3d318SStefano Zampini PetscReal tmp = (PetscInt)c->name; \ 6554b3d318SStefano Zampini ierr = PetscOptionsReal("-mat_cholmod_" #name,help,"None",tmp,&tmp,NULL);CHKERRQ(ierr); \ 66ce94432eSBarry Smith if (tmp < 0) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ARG_OUTOFRANGE,"value must be positive"); \ 67641875f9SMatthew G Knepley c->name = (size_t)tmp; \ 68641875f9SMatthew G Knepley } while (0) 6926fbe8dcSKarl Rupp 70b9eaa5e8SBarry Smith #define CHOLMOD_OPTION_BOOL(name,help) do { \ 71ace3abfcSBarry Smith PetscBool tmp = (PetscBool) !!c->name; \ 728afaa268SBarry Smith ierr = PetscOptionsBool("-mat_cholmod_" #name,help,"None",tmp,&tmp,NULL);CHKERRQ(ierr); \ 73641875f9SMatthew G Knepley c->name = (int)tmp; \ 74641875f9SMatthew G Knepley } while (0) 75641875f9SMatthew G Knepley 76ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)F),((PetscObject)F)->prefix,"CHOLMOD Options","Mat");CHKERRQ(ierr); 7754b3d318SStefano Zampini CHOLMOD_OPTION_INT(nmethods,"Number of different ordering methods to try"); 7826fbe8dcSKarl Rupp 79b9eaa5e8SBarry Smith #if defined(PETSC_USE_SUITESPARSE_GPU) 80b9eaa5e8SBarry Smith c->useGPU = 1; 81b9eaa5e8SBarry Smith CHOLMOD_OPTION_INT(useGPU,"Use GPU for BLAS 1, otherwise 0"); 8254b3d318SStefano Zampini CHOLMOD_OPTION_SIZE_T(maxGpuMemBytes,"Maximum memory to allocate on the GPU"); 8354b3d318SStefano Zampini CHOLMOD_OPTION_DOUBLE(maxGpuMemFraction,"Fraction of available GPU memory to allocate"); 84b9eaa5e8SBarry Smith #endif 85b9eaa5e8SBarry Smith 8654b3d318SStefano Zampini /* CHOLMOD handles first-time packing and refactor-packing separately, but we usually want them to be the same. */ 8754b3d318SStefano Zampini chol->pack = (PetscBool)c->final_pack; 888afaa268SBarry Smith ierr = PetscOptionsBool("-mat_cholmod_pack","Pack factors after factorization [disable for frequent repeat factorization]","None",chol->pack,&chol->pack,NULL);CHKERRQ(ierr); 89641875f9SMatthew G Knepley c->final_pack = (int)chol->pack; 90641875f9SMatthew G Knepley 91641875f9SMatthew G Knepley CHOLMOD_OPTION_DOUBLE(dbound,"Minimum absolute value of diagonal entries of D"); 92641875f9SMatthew G Knepley CHOLMOD_OPTION_DOUBLE(grow0,"Global growth ratio when factors are modified"); 93641875f9SMatthew G Knepley CHOLMOD_OPTION_DOUBLE(grow1,"Column growth ratio when factors are modified"); 94641875f9SMatthew G Knepley CHOLMOD_OPTION_SIZE_T(grow2,"Affine column growth constant when factors are modified"); 95641875f9SMatthew G Knepley CHOLMOD_OPTION_SIZE_T(maxrank,"Max rank of update, larger values are faster but use more memory [2,4,8]"); 96641875f9SMatthew G Knepley { 97641875f9SMatthew G Knepley static const char *const list[] = {"SIMPLICIAL","AUTO","SUPERNODAL","MatCholmodFactorType","MAT_CHOLMOD_FACTOR_",0}; 988afaa268SBarry Smith ierr = PetscOptionsEnum("-mat_cholmod_factor","Factorization method","None",list,(PetscEnum)c->supernodal,(PetscEnum*)&c->supernodal,NULL);CHKERRQ(ierr); 99641875f9SMatthew G Knepley } 100641875f9SMatthew G Knepley if (c->supernodal) CHOLMOD_OPTION_DOUBLE(supernodal_switch,"flop/nnz_L threshold for switching to supernodal factorization"); 101b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(final_asis,"Leave factors \"as is\""); 102b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(final_pack,"Pack the columns when finished (use FALSE if the factors will be updated later)"); 103641875f9SMatthew G Knepley if (!c->final_asis) { 104b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(final_super,"Leave supernodal factors instead of converting to simplicial"); 105b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(final_ll,"Turn LDL' factorization into LL'"); 106b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(final_monotonic,"Ensure columns are monotonic when done"); 107b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(final_resymbol,"Remove numerically zero values resulting from relaxed supernodal amalgamation"); 108641875f9SMatthew G Knepley } 109641875f9SMatthew G Knepley { 110641875f9SMatthew G Knepley PetscReal tmp[] = {(PetscReal)c->zrelax[0],(PetscReal)c->zrelax[1],(PetscReal)c->zrelax[2]}; 111641875f9SMatthew G Knepley PetscInt n = 3; 112641875f9SMatthew G Knepley ierr = PetscOptionsRealArray("-mat_cholmod_zrelax","3 real supernodal relaxed amalgamation parameters","None",tmp,&n,&flg);CHKERRQ(ierr); 113ce94432eSBarry Smith if (flg && n != 3) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ARG_OUTOFRANGE,"must provide exactly 3 parameters to -mat_cholmod_zrelax"); 114641875f9SMatthew G Knepley if (flg) while (n--) c->zrelax[n] = (double)tmp[n]; 115641875f9SMatthew G Knepley } 116641875f9SMatthew G Knepley { 117641875f9SMatthew G Knepley PetscInt n,tmp[] = {(PetscInt)c->nrelax[0],(PetscInt)c->nrelax[1],(PetscInt)c->nrelax[2]}; 118641875f9SMatthew G Knepley ierr = PetscOptionsIntArray("-mat_cholmod_nrelax","3 size_t supernodal relaxed amalgamation parameters","None",tmp,&n,&flg);CHKERRQ(ierr); 119ce94432eSBarry Smith if (flg && n != 3) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ARG_OUTOFRANGE,"must provide exactly 3 parameters to -mat_cholmod_nrelax"); 120641875f9SMatthew G Knepley if (flg) while (n--) c->nrelax[n] = (size_t)tmp[n]; 121641875f9SMatthew G Knepley } 122b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(prefer_upper,"Work with upper triangular form [faster when using fill-reducing ordering, slower in natural ordering]"); 123b9eaa5e8SBarry Smith CHOLMOD_OPTION_BOOL(default_nesdis,"Use NESDIS instead of METIS for nested dissection"); 124641875f9SMatthew G Knepley CHOLMOD_OPTION_INT(print,"Verbosity level"); 125641875f9SMatthew G Knepley ierr = PetscOptionsEnd();CHKERRQ(ierr); 126641875f9SMatthew G Knepley PetscFunctionReturn(0); 127641875f9SMatthew G Knepley } 128641875f9SMatthew G Knepley 129*218db3c1SStefano Zampini static PetscErrorCode MatWrapCholmod_seqsbaij(Mat A,PetscBool values,cholmod_sparse *C,PetscBool *aijalloc,PetscBool *valloc) 130641875f9SMatthew G Knepley { 131641875f9SMatthew G Knepley Mat_SeqSBAIJ *sbaij = (Mat_SeqSBAIJ*)A->data; 132*218db3c1SStefano Zampini PetscBool vallocin = PETSC_FALSE; 133641875f9SMatthew G Knepley PetscErrorCode ierr; 134641875f9SMatthew G Knepley 135641875f9SMatthew G Knepley PetscFunctionBegin; 136641875f9SMatthew G Knepley ierr = PetscMemzero(C,sizeof(*C));CHKERRQ(ierr); 137641875f9SMatthew G Knepley /* CHOLMOD uses column alignment, SBAIJ stores the upper factor, so we pass it on as a lower factor, swapping the meaning of row and column */ 138641875f9SMatthew G Knepley C->nrow = (size_t)A->cmap->n; 139641875f9SMatthew G Knepley C->ncol = (size_t)A->rmap->n; 140641875f9SMatthew G Knepley C->nzmax = (size_t)sbaij->maxnz; 141641875f9SMatthew G Knepley C->p = sbaij->i; 142641875f9SMatthew G Knepley C->i = sbaij->j; 143*218db3c1SStefano Zampini if (values) { 144*218db3c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 145*218db3c1SStefano Zampini /* we need to pass CHOLMOD the conjugate matrix */ 146*218db3c1SStefano Zampini PetscScalar *v; 147*218db3c1SStefano Zampini PetscInt i; 148*218db3c1SStefano Zampini 149*218db3c1SStefano Zampini ierr = PetscMalloc1(sbaij->maxnz,&v);CHKERRQ(ierr); 150*218db3c1SStefano Zampini for (i = 0; i < sbaij->maxnz; i++) v[i] = PetscConj(sbaij->a[i]); 151*218db3c1SStefano Zampini C->x = v; 152*218db3c1SStefano Zampini vallocin = PETSC_TRUE; 153*218db3c1SStefano Zampini #else 154641875f9SMatthew G Knepley C->x = sbaij->a; 155*218db3c1SStefano Zampini #endif 156*218db3c1SStefano Zampini } 157641875f9SMatthew G Knepley C->stype = -1; 158641875f9SMatthew G Knepley C->itype = CHOLMOD_INT_TYPE; 159*218db3c1SStefano Zampini C->xtype = values ? CHOLMOD_SCALAR_TYPE : CHOLMOD_PATTERN; 160641875f9SMatthew G Knepley C->dtype = CHOLMOD_DOUBLE; 161641875f9SMatthew G Knepley C->sorted = 1; 162641875f9SMatthew G Knepley C->packed = 1; 163641875f9SMatthew G Knepley *aijalloc = PETSC_FALSE; 164*218db3c1SStefano Zampini *valloc = vallocin; 165641875f9SMatthew G Knepley PetscFunctionReturn(0); 166641875f9SMatthew G Knepley } 167641875f9SMatthew G Knepley 168*218db3c1SStefano Zampini #define GET_ARRAY_READ 0 169*218db3c1SStefano Zampini #define GET_ARRAY_WRITE 1 170*218db3c1SStefano Zampini 171*218db3c1SStefano Zampini static PetscErrorCode VecWrapCholmod(Vec X,PetscInt rw,cholmod_dense *Y) 172641875f9SMatthew G Knepley { 173641875f9SMatthew G Knepley PetscErrorCode ierr; 174*218db3c1SStefano Zampini PetscScalar *x; 175641875f9SMatthew G Knepley PetscInt n; 176641875f9SMatthew G Knepley 177641875f9SMatthew G Knepley PetscFunctionBegin; 178641875f9SMatthew G Knepley ierr = PetscMemzero(Y,sizeof(*Y));CHKERRQ(ierr); 179*218db3c1SStefano Zampini switch (rw) { 180*218db3c1SStefano Zampini case GET_ARRAY_READ: 181*218db3c1SStefano Zampini ierr = VecGetArrayRead(X,(const PetscScalar**)&x);CHKERRQ(ierr); 182*218db3c1SStefano Zampini break; 183*218db3c1SStefano Zampini case GET_ARRAY_WRITE: 184*218db3c1SStefano Zampini ierr = VecGetArrayWrite(X,&x);CHKERRQ(ierr); 185*218db3c1SStefano Zampini break; 186*218db3c1SStefano Zampini default: 187*218db3c1SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Case %D not handled",rw); 188*218db3c1SStefano Zampini break; 189*218db3c1SStefano Zampini } 190641875f9SMatthew G Knepley ierr = VecGetSize(X,&n);CHKERRQ(ierr); 19126fbe8dcSKarl Rupp 192*218db3c1SStefano Zampini Y->x = x; 193641875f9SMatthew G Knepley Y->nrow = n; 194641875f9SMatthew G Knepley Y->ncol = 1; 195641875f9SMatthew G Knepley Y->nzmax = n; 196641875f9SMatthew G Knepley Y->d = n; 197641875f9SMatthew G Knepley Y->xtype = CHOLMOD_SCALAR_TYPE; 198641875f9SMatthew G Knepley Y->dtype = CHOLMOD_DOUBLE; 199641875f9SMatthew G Knepley PetscFunctionReturn(0); 200641875f9SMatthew G Knepley } 201641875f9SMatthew G Knepley 202*218db3c1SStefano Zampini static PetscErrorCode VecUnWrapCholmod(Vec X,PetscInt rw,cholmod_dense *Y) 203d9ca1df4SBarry Smith { 204d9ca1df4SBarry Smith PetscErrorCode ierr; 205d9ca1df4SBarry Smith 206d9ca1df4SBarry Smith PetscFunctionBegin; 207*218db3c1SStefano Zampini switch (rw) { 208*218db3c1SStefano Zampini case GET_ARRAY_READ: 209*218db3c1SStefano Zampini ierr = VecRestoreArrayRead(X,(const PetscScalar**)&Y->x);CHKERRQ(ierr); 210*218db3c1SStefano Zampini break; 211*218db3c1SStefano Zampini case GET_ARRAY_WRITE: 212*218db3c1SStefano Zampini ierr = VecRestoreArrayWrite(X,(PetscScalar**)&Y->x);CHKERRQ(ierr); 213*218db3c1SStefano Zampini break; 214*218db3c1SStefano Zampini default: 215*218db3c1SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Case %D not handled",rw); 216*218db3c1SStefano Zampini break; 217*218db3c1SStefano Zampini } 218*218db3c1SStefano Zampini PetscFunctionReturn(0); 219*218db3c1SStefano Zampini } 220*218db3c1SStefano Zampini 221*218db3c1SStefano Zampini static PetscErrorCode MatDenseWrapCholmod(Mat X,PetscInt rw,cholmod_dense *Y) 222*218db3c1SStefano Zampini { 223*218db3c1SStefano Zampini PetscErrorCode ierr; 224*218db3c1SStefano Zampini PetscScalar *x; 225*218db3c1SStefano Zampini PetscInt m,n,lda; 226*218db3c1SStefano Zampini 227*218db3c1SStefano Zampini PetscFunctionBegin; 228*218db3c1SStefano Zampini ierr = PetscMemzero(Y,sizeof(*Y));CHKERRQ(ierr); 229*218db3c1SStefano Zampini switch (rw) { 230*218db3c1SStefano Zampini case GET_ARRAY_READ: 231*218db3c1SStefano Zampini ierr = MatDenseGetArrayRead(X,(const PetscScalar**)&x);CHKERRQ(ierr); 232*218db3c1SStefano Zampini break; 233*218db3c1SStefano Zampini case GET_ARRAY_WRITE: 234*218db3c1SStefano Zampini /* we don't have MatDenseGetArrayWrite */ 235*218db3c1SStefano Zampini ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr); 236*218db3c1SStefano Zampini break; 237*218db3c1SStefano Zampini default: 238*218db3c1SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Case %D not handled",rw); 239*218db3c1SStefano Zampini break; 240*218db3c1SStefano Zampini } 241*218db3c1SStefano Zampini ierr = MatDenseGetLDA(X,&lda);CHKERRQ(ierr); 242*218db3c1SStefano Zampini ierr = MatGetLocalSize(X,&m,&n);CHKERRQ(ierr); 243*218db3c1SStefano Zampini 244*218db3c1SStefano Zampini Y->x = x; 245*218db3c1SStefano Zampini Y->nrow = m; 246*218db3c1SStefano Zampini Y->ncol = n; 247*218db3c1SStefano Zampini Y->nzmax = lda*n; 248*218db3c1SStefano Zampini Y->d = lda; 249*218db3c1SStefano Zampini Y->xtype = CHOLMOD_SCALAR_TYPE; 250*218db3c1SStefano Zampini Y->dtype = CHOLMOD_DOUBLE; 251*218db3c1SStefano Zampini PetscFunctionReturn(0); 252*218db3c1SStefano Zampini } 253*218db3c1SStefano Zampini 254*218db3c1SStefano Zampini static PetscErrorCode MatDenseUnWrapCholmod(Mat X,PetscInt rw,cholmod_dense *Y) 255*218db3c1SStefano Zampini { 256*218db3c1SStefano Zampini PetscErrorCode ierr; 257*218db3c1SStefano Zampini 258*218db3c1SStefano Zampini PetscFunctionBegin; 259*218db3c1SStefano Zampini switch (rw) { 260*218db3c1SStefano Zampini case GET_ARRAY_READ: 261*218db3c1SStefano Zampini ierr = MatDenseRestoreArrayRead(X,(const PetscScalar**)&Y->x);CHKERRQ(ierr); 262*218db3c1SStefano Zampini break; 263*218db3c1SStefano Zampini case GET_ARRAY_WRITE: 264*218db3c1SStefano Zampini /* we don't have MatDenseRestoreArrayWrite */ 265*218db3c1SStefano Zampini ierr = MatDenseRestoreArray(X,(PetscScalar**)&Y->x);CHKERRQ(ierr); 266*218db3c1SStefano Zampini break; 267*218db3c1SStefano Zampini default: 268*218db3c1SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Case %D not handled",rw); 269*218db3c1SStefano Zampini break; 270*218db3c1SStefano Zampini } 271d9ca1df4SBarry Smith PetscFunctionReturn(0); 272d9ca1df4SBarry Smith } 273d9ca1df4SBarry Smith 274eb9872f6SBarry Smith PETSC_INTERN PetscErrorCode MatDestroy_CHOLMOD(Mat F) 275641875f9SMatthew G Knepley { 276641875f9SMatthew G Knepley PetscErrorCode ierr; 2776b8f6f9dSBarry Smith Mat_CHOLMOD *chol=(Mat_CHOLMOD*)F->data; 278641875f9SMatthew G Knepley 279641875f9SMatthew G Knepley PetscFunctionBegin; 280641875f9SMatthew G Knepley ierr = !cholmod_X_free_factor(&chol->factor,chol->common);CHKERRQ(ierr); 281641875f9SMatthew G Knepley ierr = !cholmod_X_finish(chol->common);CHKERRQ(ierr); 282641875f9SMatthew G Knepley ierr = PetscFree(chol->common);CHKERRQ(ierr); 283641875f9SMatthew G Knepley ierr = PetscFree(chol->matrix);CHKERRQ(ierr); 284*218db3c1SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)F,"MatFactorGetSolverType_C",NULL);CHKERRQ(ierr); 2856b8f6f9dSBarry Smith ierr = PetscFree(F->data);CHKERRQ(ierr); 286641875f9SMatthew G Knepley PetscFunctionReturn(0); 287641875f9SMatthew G Knepley } 288641875f9SMatthew G Knepley 289641875f9SMatthew G Knepley static PetscErrorCode MatSolve_CHOLMOD(Mat,Vec,Vec); 290*218db3c1SStefano Zampini static PetscErrorCode MatMatSolve_CHOLMOD(Mat,Mat,Mat); 291641875f9SMatthew G Knepley 292fcd503bcSBarry Smith /*static const char *const CholmodOrderingMethods[] = {"User","AMD","METIS","NESDIS(default)","Natural","NESDIS(small=20000)","NESDIS(small=4,no constrained)","NESDIS()"};*/ 293641875f9SMatthew G Knepley 294860c79edSBarry Smith static PetscErrorCode MatView_Info_CHOLMOD(Mat F,PetscViewer viewer) 295641875f9SMatthew G Knepley { 2966b8f6f9dSBarry Smith Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->data; 297641875f9SMatthew G Knepley const cholmod_common *c = chol->common; 298641875f9SMatthew G Knepley PetscErrorCode ierr; 299641875f9SMatthew G Knepley PetscInt i; 300641875f9SMatthew G Knepley 301641875f9SMatthew G Knepley PetscFunctionBegin; 302641875f9SMatthew G Knepley if (F->ops->solve != MatSolve_CHOLMOD) PetscFunctionReturn(0); 303641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"CHOLMOD run parameters:\n");CHKERRQ(ierr); 304641875f9SMatthew G Knepley ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 305641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Pack factors after symbolic factorization: %s\n",chol->pack ? "TRUE" : "FALSE");CHKERRQ(ierr); 306641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.dbound %g (Smallest absolute value of diagonal entries of D)\n",c->dbound);CHKERRQ(ierr); 307641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.grow0 %g\n",c->grow0);CHKERRQ(ierr); 308641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.grow1 %g\n",c->grow1);CHKERRQ(ierr); 309641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.grow2 %u\n",(unsigned)c->grow2);CHKERRQ(ierr); 310641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.maxrank %u\n",(unsigned)c->maxrank);CHKERRQ(ierr); 311641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.supernodal_switch %g\n",c->supernodal_switch);CHKERRQ(ierr); 312641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.supernodal %d\n",c->supernodal);CHKERRQ(ierr); 313641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_asis %d\n",c->final_asis);CHKERRQ(ierr); 314641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_super %d\n",c->final_super);CHKERRQ(ierr); 315641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_ll %d\n",c->final_ll);CHKERRQ(ierr); 316641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_pack %d\n",c->final_pack);CHKERRQ(ierr); 317641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_monotonic %d\n",c->final_monotonic);CHKERRQ(ierr); 318641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_resymbol %d\n",c->final_resymbol);CHKERRQ(ierr); 319641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.zrelax [%g,%g,%g]\n",c->zrelax[0],c->zrelax[1],c->zrelax[2]);CHKERRQ(ierr); 320641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.nrelax [%u,%u,%u]\n",(unsigned)c->nrelax[0],(unsigned)c->nrelax[1],(unsigned)c->nrelax[2]);CHKERRQ(ierr); 321641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.prefer_upper %d\n",c->prefer_upper);CHKERRQ(ierr); 322641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.print %d\n",c->print);CHKERRQ(ierr); 323641875f9SMatthew G Knepley for (i=0; i<c->nmethods; i++) { 324641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Ordering method %D%s:\n",i,i==c->selected ? " [SELECTED]" : "");CHKERRQ(ierr); 325641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer," lnz %g, fl %g, prune_dense %g, prune_dense2 %g\n", 326641875f9SMatthew G Knepley c->method[i].lnz,c->method[i].fl,c->method[i].prune_dense,c->method[i].prune_dense2);CHKERRQ(ierr); 327641875f9SMatthew G Knepley } 328641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.postorder %d\n",c->postorder);CHKERRQ(ierr); 329641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.default_nesdis %d (use NESDIS instead of METIS for nested dissection)\n",c->default_nesdis);CHKERRQ(ierr); 330641875f9SMatthew G Knepley /* Statistics */ 331641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.fl %g (flop count from most recent analysis)\n",c->fl);CHKERRQ(ierr); 332641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.lnz %g (fundamental nz in L)\n",c->lnz);CHKERRQ(ierr); 333641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.anz %g\n",c->anz);CHKERRQ(ierr); 334641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.modfl %g (flop count from most recent update)\n",c->modfl);CHKERRQ(ierr); 335641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.malloc_count %g (number of live objects)\n",(double)c->malloc_count);CHKERRQ(ierr);CHKERRQ(ierr); 336641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.memory_usage %g (peak memory usage in bytes)\n",(double)c->memory_usage);CHKERRQ(ierr);CHKERRQ(ierr); 337641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.memory_inuse %g (current memory usage in bytes)\n",(double)c->memory_inuse);CHKERRQ(ierr);CHKERRQ(ierr); 338641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.nrealloc_col %g (number of column reallocations)\n",c->nrealloc_col);CHKERRQ(ierr);CHKERRQ(ierr); 339641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.nrealloc_factor %g (number of factor reallocations due to column reallocations)\n",c->nrealloc_factor);CHKERRQ(ierr);CHKERRQ(ierr); 340641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.ndbounds_hit %g (number of times diagonal was modified by dbound)\n",c->ndbounds_hit);CHKERRQ(ierr);CHKERRQ(ierr); 341641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.rowfacfl %g (number of flops in last call to cholmod_rowfac)\n",c->rowfacfl);CHKERRQ(ierr);CHKERRQ(ierr); 342641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.aatfl %g (number of flops to compute A(:,f)*A(:,f)')\n",c->aatfl);CHKERRQ(ierr);CHKERRQ(ierr); 343b9eaa5e8SBarry Smith #if defined(PETSC_USE_SUITESPARSE_GPU) 344b9eaa5e8SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"Common.useGPU %d\n",c->useGPU);CHKERRQ(ierr);CHKERRQ(ierr); 345b9eaa5e8SBarry Smith #endif 346641875f9SMatthew G Knepley ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 347641875f9SMatthew G Knepley PetscFunctionReturn(0); 348641875f9SMatthew G Knepley } 349641875f9SMatthew G Knepley 350eb9872f6SBarry Smith PETSC_INTERN PetscErrorCode MatView_CHOLMOD(Mat F,PetscViewer viewer) 351641875f9SMatthew G Knepley { 352641875f9SMatthew G Knepley PetscErrorCode ierr; 353ace3abfcSBarry Smith PetscBool iascii; 354641875f9SMatthew G Knepley PetscViewerFormat format; 355641875f9SMatthew G Knepley 356641875f9SMatthew G Knepley PetscFunctionBegin; 357251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 358641875f9SMatthew G Knepley if (iascii) { 359641875f9SMatthew G Knepley ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 360641875f9SMatthew G Knepley if (format == PETSC_VIEWER_ASCII_INFO) { 361860c79edSBarry Smith ierr = MatView_Info_CHOLMOD(F,viewer);CHKERRQ(ierr); 362641875f9SMatthew G Knepley } 363641875f9SMatthew G Knepley } 364641875f9SMatthew G Knepley PetscFunctionReturn(0); 365641875f9SMatthew G Knepley } 366641875f9SMatthew G Knepley 367641875f9SMatthew G Knepley static PetscErrorCode MatSolve_CHOLMOD(Mat F,Vec B,Vec X) 368641875f9SMatthew G Knepley { 3696b8f6f9dSBarry Smith Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->data; 370*218db3c1SStefano Zampini cholmod_dense cholB,cholX,*X_handle,*Y_handle = NULL,*E_handle = NULL; 371641875f9SMatthew G Knepley PetscErrorCode ierr; 372641875f9SMatthew G Knepley 373641875f9SMatthew G Knepley PetscFunctionBegin; 374641875f9SMatthew G Knepley static_F = F; 375*218db3c1SStefano Zampini ierr = VecWrapCholmod(B,GET_ARRAY_READ,&cholB);CHKERRQ(ierr); 376*218db3c1SStefano Zampini ierr = VecWrapCholmod(X,GET_ARRAY_WRITE,&cholX);CHKERRQ(ierr); 377*218db3c1SStefano Zampini X_handle = &cholX; 378*218db3c1SStefano Zampini ierr = !cholmod_X_solve2(CHOLMOD_A,chol->factor,&cholB,NULL,&X_handle,NULL,&Y_handle,&E_handle,chol->common);CHKERRQ(ierr); 379*218db3c1SStefano Zampini ierr = !cholmod_X_free_dense(&Y_handle,chol->common);CHKERRQ(ierr); 380*218db3c1SStefano Zampini ierr = !cholmod_X_free_dense(&E_handle,chol->common);CHKERRQ(ierr); 381*218db3c1SStefano Zampini ierr = VecUnWrapCholmod(B,GET_ARRAY_READ,&cholB);CHKERRQ(ierr); 382*218db3c1SStefano Zampini ierr = VecUnWrapCholmod(X,GET_ARRAY_WRITE,&cholX);CHKERRQ(ierr); 383*218db3c1SStefano Zampini PetscFunctionReturn(0); 384*218db3c1SStefano Zampini } 385*218db3c1SStefano Zampini 386*218db3c1SStefano Zampini static PetscErrorCode MatMatSolve_CHOLMOD(Mat F,Mat B,Mat X) 387*218db3c1SStefano Zampini { 388*218db3c1SStefano Zampini Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->data; 389*218db3c1SStefano Zampini cholmod_dense cholB,cholX,*X_handle,*Y_handle = NULL,*E_handle = NULL; 390*218db3c1SStefano Zampini PetscErrorCode ierr; 391*218db3c1SStefano Zampini 392*218db3c1SStefano Zampini PetscFunctionBegin; 393*218db3c1SStefano Zampini static_F = F; 394*218db3c1SStefano Zampini ierr = MatDenseWrapCholmod(B,GET_ARRAY_READ,&cholB);CHKERRQ(ierr); 395*218db3c1SStefano Zampini ierr = MatDenseWrapCholmod(X,GET_ARRAY_WRITE,&cholX);CHKERRQ(ierr); 396*218db3c1SStefano Zampini X_handle = &cholX; 397*218db3c1SStefano Zampini ierr = !cholmod_X_solve2(CHOLMOD_A,chol->factor,&cholB,NULL,&X_handle,NULL,&Y_handle,&E_handle,chol->common);CHKERRQ(ierr); 398*218db3c1SStefano Zampini ierr = !cholmod_X_free_dense(&Y_handle,chol->common);CHKERRQ(ierr); 399*218db3c1SStefano Zampini ierr = !cholmod_X_free_dense(&E_handle,chol->common);CHKERRQ(ierr); 400*218db3c1SStefano Zampini ierr = MatDenseUnWrapCholmod(B,GET_ARRAY_READ,&cholB);CHKERRQ(ierr); 401*218db3c1SStefano Zampini ierr = MatDenseUnWrapCholmod(X,GET_ARRAY_WRITE,&cholX);CHKERRQ(ierr); 402641875f9SMatthew G Knepley PetscFunctionReturn(0); 403641875f9SMatthew G Knepley } 404641875f9SMatthew G Knepley 405641875f9SMatthew G Knepley static PetscErrorCode MatCholeskyFactorNumeric_CHOLMOD(Mat F,Mat A,const MatFactorInfo *info) 406641875f9SMatthew G Knepley { 4076b8f6f9dSBarry Smith Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->data; 408641875f9SMatthew G Knepley cholmod_sparse cholA; 409*218db3c1SStefano Zampini PetscBool aijalloc,valloc; 410641875f9SMatthew G Knepley PetscErrorCode ierr; 411641875f9SMatthew G Knepley 412641875f9SMatthew G Knepley PetscFunctionBegin; 413*218db3c1SStefano Zampini ierr = (*chol->Wrap)(A,PETSC_TRUE,&cholA,&aijalloc,&valloc);CHKERRQ(ierr); 414641875f9SMatthew G Knepley static_F = F; 415641875f9SMatthew G Knepley ierr = !cholmod_X_factorize(&cholA,chol->factor,chol->common); 416ce94432eSBarry Smith if (ierr) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD factorization failed with status %d",chol->common->status); 417ce94432eSBarry Smith if (chol->common->status == CHOLMOD_NOT_POSDEF) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_MAT_CH_ZRPVT,"CHOLMOD detected that the matrix is not positive definite, failure at column %u",(unsigned)chol->factor->minor); 418641875f9SMatthew G Knepley 419*218db3c1SStefano Zampini if (aijalloc) {ierr = PetscFree2(cholA.p,cholA.i);CHKERRQ(ierr);} 420*218db3c1SStefano Zampini if (valloc) {ierr = PetscFree(cholA.x);CHKERRQ(ierr);} 421*218db3c1SStefano Zampini #if defined(PETSC_USE_SUITESPARSE_GPU) 422*218db3c1SStefano Zampini ierr = PetscLogGpuTimeAdd(chol->common->CHOLMOD_GPU_GEMM_TIME + chol->common->CHOLMOD_GPU_SYRK_TIME + chol->common->CHOLMOD_GPU_TRSM_TIME + chol->common->CHOLMOD_GPU_POTRF_TIME);CHKERRQ(ierr); 423*218db3c1SStefano Zampini #endif 424641875f9SMatthew G Knepley 425641875f9SMatthew G Knepley F->ops->solve = MatSolve_CHOLMOD; 426641875f9SMatthew G Knepley F->ops->solvetranspose = MatSolve_CHOLMOD; 427*218db3c1SStefano Zampini F->ops->matsolve = MatMatSolve_CHOLMOD; 428*218db3c1SStefano Zampini F->ops->matsolvetranspose = MatMatSolve_CHOLMOD; 429641875f9SMatthew G Knepley PetscFunctionReturn(0); 430641875f9SMatthew G Knepley } 431641875f9SMatthew G Knepley 432eb9872f6SBarry Smith PETSC_INTERN PetscErrorCode MatCholeskyFactorSymbolic_CHOLMOD(Mat F,Mat A,IS perm,const MatFactorInfo *info) 433641875f9SMatthew G Knepley { 4346b8f6f9dSBarry Smith Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->data; 435641875f9SMatthew G Knepley PetscErrorCode ierr; 436641875f9SMatthew G Knepley cholmod_sparse cholA; 437*218db3c1SStefano Zampini PetscBool aijalloc,valloc; 438641875f9SMatthew G Knepley PetscInt *fset = 0; 439641875f9SMatthew G Knepley size_t fsize = 0; 440641875f9SMatthew G Knepley 441641875f9SMatthew G Knepley PetscFunctionBegin; 442*218db3c1SStefano Zampini ierr = (*chol->Wrap)(A,PETSC_FALSE,&cholA,&aijalloc,&valloc);CHKERRQ(ierr); 443641875f9SMatthew G Knepley static_F = F; 444641875f9SMatthew G Knepley if (chol->factor) { 445641875f9SMatthew G Knepley ierr = !cholmod_X_resymbol(&cholA,fset,fsize,(int)chol->pack,chol->factor,chol->common); 446ce94432eSBarry Smith if (ierr) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD analysis failed with status %d",chol->common->status); 447641875f9SMatthew G Knepley } else if (perm) { 448641875f9SMatthew G Knepley const PetscInt *ip; 449641875f9SMatthew G Knepley ierr = ISGetIndices(perm,&ip);CHKERRQ(ierr); 450641875f9SMatthew G Knepley chol->factor = cholmod_X_analyze_p(&cholA,(PetscInt*)ip,fset,fsize,chol->common); 451ce94432eSBarry Smith if (!chol->factor) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD analysis failed with status %d",chol->common->status); 452641875f9SMatthew G Knepley ierr = ISRestoreIndices(perm,&ip);CHKERRQ(ierr); 453641875f9SMatthew G Knepley } else { 454641875f9SMatthew G Knepley chol->factor = cholmod_X_analyze(&cholA,chol->common); 455ce94432eSBarry Smith if (!chol->factor) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD analysis failed with status %d",chol->common->status); 456641875f9SMatthew G Knepley } 457641875f9SMatthew G Knepley 458*218db3c1SStefano Zampini if (aijalloc) {ierr = PetscFree2(cholA.p,cholA.i);CHKERRQ(ierr);} 459*218db3c1SStefano Zampini if (valloc) {ierr = PetscFree(cholA.x);CHKERRQ(ierr);} 460641875f9SMatthew G Knepley 461641875f9SMatthew G Knepley F->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_CHOLMOD; 462641875f9SMatthew G Knepley PetscFunctionReturn(0); 463641875f9SMatthew G Knepley } 464641875f9SMatthew G Knepley 465ea799195SBarry Smith static PetscErrorCode MatFactorGetSolverType_seqsbaij_cholmod(Mat A,MatSolverType *type) 466641875f9SMatthew G Knepley { 467641875f9SMatthew G Knepley PetscFunctionBegin; 468641875f9SMatthew G Knepley *type = MATSOLVERCHOLMOD; 469641875f9SMatthew G Knepley PetscFunctionReturn(0); 470641875f9SMatthew G Knepley } 471641875f9SMatthew G Knepley 472*218db3c1SStefano Zampini PETSC_INTERN PetscErrorCode MatGetInfo_CHOLMOD(Mat F,MatInfoType flag,MatInfo *info) 473*218db3c1SStefano Zampini { 474*218db3c1SStefano Zampini Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->data; 475*218db3c1SStefano Zampini 476*218db3c1SStefano Zampini PetscFunctionBegin; 477*218db3c1SStefano Zampini info->block_size = 1.0; 478*218db3c1SStefano Zampini info->nz_allocated = chol->common->lnz; 479*218db3c1SStefano Zampini info->nz_used = chol->common->lnz; 480*218db3c1SStefano Zampini info->nz_unneeded = 0.0; 481*218db3c1SStefano Zampini info->assemblies = 0.0; 482*218db3c1SStefano Zampini info->mallocs = 0.0; 483*218db3c1SStefano Zampini info->memory = chol->common->memory_inuse; 484*218db3c1SStefano Zampini info->fill_ratio_given = 0; 485*218db3c1SStefano Zampini info->fill_ratio_needed = 0; 486*218db3c1SStefano Zampini info->factor_mallocs = chol->common->malloc_count; 487*218db3c1SStefano Zampini PetscFunctionReturn(0); 488*218db3c1SStefano Zampini } 489*218db3c1SStefano Zampini 490641875f9SMatthew G Knepley /*MC 491641875f9SMatthew G Knepley MATSOLVERCHOLMOD = "cholmod" - A matrix type providing direct solvers (Cholesky) for sequential matrices 492641875f9SMatthew G Knepley via the external package CHOLMOD. 493641875f9SMatthew G Knepley 494c2b89b5dSBarry Smith Use ./configure --download-suitesparse to install PETSc to use CHOLMOD 495c2b89b5dSBarry Smith 496f29f8b16SBarry Smith Use -pc_type cholesky -pc_factor_mat_solver_type cholmod to use this direct solver 497641875f9SMatthew G Knepley 498641875f9SMatthew G Knepley Consult CHOLMOD documentation for more information about the Common parameters 499641875f9SMatthew G Knepley which correspond to the options database keys below. 500641875f9SMatthew G Knepley 501641875f9SMatthew G Knepley Options Database Keys: 502e08999f5SMatthew G Knepley + -mat_cholmod_dbound <0> - Minimum absolute value of diagonal entries of D (None) 503e08999f5SMatthew G Knepley . -mat_cholmod_grow0 <1.2> - Global growth ratio when factors are modified (None) 504e08999f5SMatthew G Knepley . -mat_cholmod_grow1 <1.2> - Column growth ratio when factors are modified (None) 505e08999f5SMatthew G Knepley . -mat_cholmod_grow2 <5> - Affine column growth constant when factors are modified (None) 506e08999f5SMatthew G Knepley . -mat_cholmod_maxrank <8> - Max rank of update, larger values are faster but use more memory [2,4,8] (None) 507e08999f5SMatthew G Knepley . -mat_cholmod_factor <AUTO> - (choose one of) SIMPLICIAL AUTO SUPERNODAL 508e08999f5SMatthew G Knepley . -mat_cholmod_supernodal_switch <40> - flop/nnz_L threshold for switching to supernodal factorization (None) 509e08999f5SMatthew G Knepley . -mat_cholmod_final_asis <TRUE> - Leave factors "as is" (None) 510e08999f5SMatthew G Knepley . -mat_cholmod_final_pack <TRUE> - Pack the columns when finished (use FALSE if the factors will be updated later) (None) 511e08999f5SMatthew G Knepley . -mat_cholmod_zrelax <0.8> - 3 real supernodal relaxed amalgamation parameters (None) 512e08999f5SMatthew G Knepley . -mat_cholmod_nrelax <4> - 3 size_t supernodal relaxed amalgamation parameters (None) 513e08999f5SMatthew G Knepley . -mat_cholmod_prefer_upper <TRUE> - Work with upper triangular form (faster when using fill-reducing ordering, slower in natural ordering) (None) 514e08999f5SMatthew G Knepley - -mat_cholmod_print <3> - Verbosity level (None) 515641875f9SMatthew G Knepley 516641875f9SMatthew G Knepley Level: beginner 517641875f9SMatthew G Knepley 518a364b7d2SBarry Smith Note: CHOLMOD is part of SuiteSparse http://faculty.cse.tamu.edu/davis/suitesparse.html 519a364b7d2SBarry Smith 5203ca39a21SBarry Smith .seealso: PCCHOLESKY, PCFactorSetMatSolverType(), MatSolverType 521641875f9SMatthew G Knepley M*/ 522b2573a8aSBarry Smith 523db87b0f2SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat A,MatFactorType ftype,Mat *F) 524641875f9SMatthew G Knepley { 525641875f9SMatthew G Knepley Mat B; 526641875f9SMatthew G Knepley Mat_CHOLMOD *chol; 527641875f9SMatthew G Knepley PetscErrorCode ierr; 528641875f9SMatthew G Knepley PetscInt m=A->rmap->n,n=A->cmap->n,bs; 529*218db3c1SStefano Zampini const char *prefix; 530641875f9SMatthew G Knepley 531641875f9SMatthew G Knepley PetscFunctionBegin; 532641875f9SMatthew G Knepley ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 533ce94432eSBarry Smith if (bs != 1) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"CHOLMOD only supports block size=1, given %D",bs); 534*218db3c1SStefano Zampini #if defined(PETSC_USE_COMPLEX) 535*218db3c1SStefano Zampini if (!A->hermitian) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only for hermitian matrices"); 536*218db3c1SStefano Zampini #endif 537641875f9SMatthew G Knepley /* Create the factorization matrix F */ 538ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 539641875f9SMatthew G Knepley ierr = MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,m,n);CHKERRQ(ierr); 5406b8f6f9dSBarry Smith ierr = PetscStrallocpy("cholmod",&((PetscObject)B)->type_name);CHKERRQ(ierr); 541*218db3c1SStefano Zampini ierr = MatGetOptionsPrefix(A,&prefix);CHKERRQ(ierr); 542*218db3c1SStefano Zampini ierr = MatSetOptionsPrefix(B,prefix);CHKERRQ(ierr); 5436b8f6f9dSBarry Smith ierr = MatSetUp(B);CHKERRQ(ierr); 544b00a9115SJed Brown ierr = PetscNewLog(B,&chol);CHKERRQ(ierr); 54526fbe8dcSKarl Rupp 546641875f9SMatthew G Knepley chol->Wrap = MatWrapCholmod_seqsbaij; 5476b8f6f9dSBarry Smith B->data = chol; 548641875f9SMatthew G Knepley 549*218db3c1SStefano Zampini B->ops->getinfo = MatGetInfo_CHOLMOD; 550641875f9SMatthew G Knepley B->ops->view = MatView_CHOLMOD; 551641875f9SMatthew G Knepley B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_CHOLMOD; 552641875f9SMatthew G Knepley B->ops->destroy = MatDestroy_CHOLMOD; 5533ca39a21SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverType_C",MatFactorGetSolverType_seqsbaij_cholmod);CHKERRQ(ierr); 554641875f9SMatthew G Knepley B->factortype = MAT_FACTOR_CHOLESKY; 555*218db3c1SStefano Zampini B->assembled = PETSC_TRUE; 556641875f9SMatthew G Knepley B->preallocated = PETSC_TRUE; 557641875f9SMatthew G Knepley 558641875f9SMatthew G Knepley ierr = CholmodStart(B);CHKERRQ(ierr); 55900c67f3bSHong Zhang 56000c67f3bSHong Zhang ierr = PetscFree(B->solvertype);CHKERRQ(ierr); 56100c67f3bSHong Zhang ierr = PetscStrallocpy(MATSOLVERCHOLMOD,&B->solvertype);CHKERRQ(ierr); 56200c67f3bSHong Zhang 563641875f9SMatthew G Knepley *F = B; 564641875f9SMatthew G Knepley PetscFunctionReturn(0); 565641875f9SMatthew G Knepley } 566