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 5641875f9SMatthew G Knepley When build with PETSC_USE_64BIT_INDICES this will use UF_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 8641875f9SMatthew G Knepley that UMFPACK UL_Long version MUST be built with 64 bit integers when used. 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 #undef __FUNCT__ 22641875f9SMatthew G Knepley #define __FUNCT__ "CholmodErrorHandler" 23641875f9SMatthew G Knepley static void CholmodErrorHandler(int status,const char *file,int line,const char *message) 24641875f9SMatthew G Knepley { 25641875f9SMatthew G Knepley 26641875f9SMatthew G Knepley PetscFunctionBegin; 27641875f9SMatthew G Knepley if (status > CHOLMOD_OK) { 28641875f9SMatthew G Knepley PetscInfo4(static_F,"CHOLMOD warning %d at %s:%d: %s",status,file,line,message); 29641875f9SMatthew G Knepley } else if (status == CHOLMOD_OK) { /* Documentation says this can happen, but why? */ 30641875f9SMatthew G Knepley PetscInfo3(static_F,"CHOLMOD OK at %s:%d: %s",file,line,message); 31641875f9SMatthew G Knepley } else { 32641875f9SMatthew G Knepley PetscErrorPrintf("CHOLMOD error %d at %s:%d: %s\n",status,file,line,message); 33641875f9SMatthew G Knepley } 34641875f9SMatthew G Knepley PetscFunctionReturnVoid(); 35641875f9SMatthew G Knepley } 36641875f9SMatthew G Knepley 37641875f9SMatthew G Knepley #undef __FUNCT__ 38641875f9SMatthew G Knepley #define __FUNCT__ "CholmodStart" 397087cfbeSBarry Smith PetscErrorCode CholmodStart(Mat F) 40641875f9SMatthew G Knepley { 41641875f9SMatthew G Knepley PetscErrorCode ierr; 42641875f9SMatthew G Knepley Mat_CHOLMOD *chol=(Mat_CHOLMOD*)F->spptr; 43641875f9SMatthew G Knepley cholmod_common *c; 44ace3abfcSBarry Smith PetscBool flg; 45641875f9SMatthew G Knepley 46641875f9SMatthew G Knepley PetscFunctionBegin; 47641875f9SMatthew G Knepley if (chol->common) PetscFunctionReturn(0); 48641875f9SMatthew G Knepley ierr = PetscMalloc(sizeof(*chol->common),&chol->common);CHKERRQ(ierr); 49641875f9SMatthew G Knepley ierr = !cholmod_X_start(chol->common);CHKERRQ(ierr); 5026fbe8dcSKarl Rupp 51641875f9SMatthew G Knepley c = chol->common; 52641875f9SMatthew G Knepley c->error_handler = CholmodErrorHandler; 53641875f9SMatthew G Knepley 54641875f9SMatthew G Knepley #define CHOLMOD_OPTION_DOUBLE(name,help) do { \ 55641875f9SMatthew G Knepley PetscReal tmp = (PetscReal)c->name; \ 56641875f9SMatthew G Knepley ierr = PetscOptionsReal("-mat_cholmod_" #name,help,"None",tmp,&tmp,0);CHKERRQ(ierr); \ 57641875f9SMatthew G Knepley c->name = (double)tmp; \ 58641875f9SMatthew G Knepley } while (0) 5926fbe8dcSKarl Rupp 60641875f9SMatthew G Knepley #define CHOLMOD_OPTION_INT(name,help) do { \ 61641875f9SMatthew G Knepley PetscInt tmp = (PetscInt)c->name; \ 62641875f9SMatthew G Knepley ierr = PetscOptionsInt("-mat_cholmod_" #name,help,"None",tmp,&tmp,0);CHKERRQ(ierr); \ 63641875f9SMatthew G Knepley c->name = (int)tmp; \ 64641875f9SMatthew G Knepley } while (0) 6526fbe8dcSKarl Rupp 66641875f9SMatthew G Knepley #define CHOLMOD_OPTION_SIZE_T(name,help) do { \ 67641875f9SMatthew G Knepley PetscInt tmp = (PetscInt)c->name; \ 68641875f9SMatthew G Knepley ierr = PetscOptionsInt("-mat_cholmod_" #name,help,"None",tmp,&tmp,0);CHKERRQ(ierr); \ 69ce94432eSBarry Smith if (tmp < 0) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ARG_OUTOFRANGE,"value must be positive"); \ 70641875f9SMatthew G Knepley c->name = (size_t)tmp; \ 71641875f9SMatthew G Knepley } while (0) 7226fbe8dcSKarl Rupp 73641875f9SMatthew G Knepley #define CHOLMOD_OPTION_TRUTH(name,help) do { \ 74ace3abfcSBarry Smith PetscBool tmp = (PetscBool) !!c->name; \ 75acfcf0e5SJed Brown ierr = PetscOptionsBool("-mat_cholmod_" #name,help,"None",tmp,&tmp,0);CHKERRQ(ierr); \ 76641875f9SMatthew G Knepley c->name = (int)tmp; \ 77641875f9SMatthew G Knepley } while (0) 78641875f9SMatthew G Knepley 79ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)F),((PetscObject)F)->prefix,"CHOLMOD Options","Mat");CHKERRQ(ierr); 80641875f9SMatthew G Knepley /* CHOLMOD handles first-time packing and refactor-packing separately, but we usually want them to be the same. */ 81ace3abfcSBarry Smith chol->pack = (PetscBool)c->final_pack; 8226fbe8dcSKarl Rupp 83acfcf0e5SJed Brown ierr = PetscOptionsBool("-mat_cholmod_pack","Pack factors after factorization [disable for frequent repeat factorization]","None",chol->pack,&chol->pack,0);CHKERRQ(ierr); 84641875f9SMatthew G Knepley c->final_pack = (int)chol->pack; 85641875f9SMatthew G Knepley 86641875f9SMatthew G Knepley CHOLMOD_OPTION_DOUBLE(dbound,"Minimum absolute value of diagonal entries of D"); 87641875f9SMatthew G Knepley CHOLMOD_OPTION_DOUBLE(grow0,"Global growth ratio when factors are modified"); 88641875f9SMatthew G Knepley CHOLMOD_OPTION_DOUBLE(grow1,"Column growth ratio when factors are modified"); 89641875f9SMatthew G Knepley CHOLMOD_OPTION_SIZE_T(grow2,"Affine column growth constant when factors are modified"); 90641875f9SMatthew G Knepley CHOLMOD_OPTION_SIZE_T(maxrank,"Max rank of update, larger values are faster but use more memory [2,4,8]"); 91641875f9SMatthew G Knepley { 92641875f9SMatthew G Knepley static const char *const list[] = {"SIMPLICIAL","AUTO","SUPERNODAL","MatCholmodFactorType","MAT_CHOLMOD_FACTOR_",0}; 93641875f9SMatthew G Knepley PetscEnum choice = (PetscEnum)c->supernodal; 9426fbe8dcSKarl Rupp 95641875f9SMatthew G Knepley ierr = PetscOptionsEnum("-mat_cholmod_factor","Factorization method","None",list,(PetscEnum)c->supernodal,&choice,0);CHKERRQ(ierr); 96641875f9SMatthew G Knepley c->supernodal = (int)choice; 97641875f9SMatthew G Knepley } 98641875f9SMatthew G Knepley if (c->supernodal) CHOLMOD_OPTION_DOUBLE(supernodal_switch,"flop/nnz_L threshold for switching to supernodal factorization"); 99641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(final_asis,"Leave factors \"as is\""); 100641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(final_pack,"Pack the columns when finished (use FALSE if the factors will be updated later)"); 101641875f9SMatthew G Knepley if (!c->final_asis) { 102641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(final_super,"Leave supernodal factors instead of converting to simplicial"); 103641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(final_ll,"Turn LDL' factorization into LL'"); 104641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(final_monotonic,"Ensure columns are monotonic when done"); 105641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(final_resymbol,"Remove numerically zero values resulting from relaxed supernodal amalgamation"); 106641875f9SMatthew G Knepley } 107641875f9SMatthew G Knepley { 108641875f9SMatthew G Knepley PetscReal tmp[] = {(PetscReal)c->zrelax[0],(PetscReal)c->zrelax[1],(PetscReal)c->zrelax[2]}; 109641875f9SMatthew G Knepley PetscInt n = 3; 110641875f9SMatthew G Knepley ierr = PetscOptionsRealArray("-mat_cholmod_zrelax","3 real supernodal relaxed amalgamation parameters","None",tmp,&n,&flg);CHKERRQ(ierr); 111ce94432eSBarry Smith if (flg && n != 3) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ARG_OUTOFRANGE,"must provide exactly 3 parameters to -mat_cholmod_zrelax"); 112641875f9SMatthew G Knepley if (flg) while (n--) c->zrelax[n] = (double)tmp[n]; 113641875f9SMatthew G Knepley } 114641875f9SMatthew G Knepley { 115641875f9SMatthew G Knepley PetscInt n,tmp[] = {(PetscInt)c->nrelax[0],(PetscInt)c->nrelax[1],(PetscInt)c->nrelax[2]}; 116641875f9SMatthew G Knepley ierr = PetscOptionsIntArray("-mat_cholmod_nrelax","3 size_t supernodal relaxed amalgamation parameters","None",tmp,&n,&flg);CHKERRQ(ierr); 117ce94432eSBarry Smith if (flg && n != 3) SETERRQ(PetscObjectComm((PetscObject)F),PETSC_ERR_ARG_OUTOFRANGE,"must provide exactly 3 parameters to -mat_cholmod_nrelax"); 118641875f9SMatthew G Knepley if (flg) while (n--) c->nrelax[n] = (size_t)tmp[n]; 119641875f9SMatthew G Knepley } 120641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(prefer_upper,"Work with upper triangular form [faster when using fill-reducing ordering, slower in natural ordering]"); 121641875f9SMatthew G Knepley CHOLMOD_OPTION_TRUTH(default_nesdis,"Use NESDIS instead of METIS for nested dissection"); 122641875f9SMatthew G Knepley CHOLMOD_OPTION_INT(print,"Verbosity level"); 123641875f9SMatthew G Knepley ierr = PetscOptionsEnd();CHKERRQ(ierr); 124641875f9SMatthew G Knepley PetscFunctionReturn(0); 125641875f9SMatthew G Knepley } 126641875f9SMatthew G Knepley 127641875f9SMatthew G Knepley #undef __FUNCT__ 128641875f9SMatthew G Knepley #define __FUNCT__ "MatWrapCholmod_seqsbaij" 129ace3abfcSBarry Smith static PetscErrorCode MatWrapCholmod_seqsbaij(Mat A,PetscBool values,cholmod_sparse *C,PetscBool *aijalloc) 130641875f9SMatthew G Knepley { 131641875f9SMatthew G Knepley Mat_SeqSBAIJ *sbaij = (Mat_SeqSBAIJ*)A->data; 132641875f9SMatthew G Knepley PetscErrorCode ierr; 133641875f9SMatthew G Knepley 134641875f9SMatthew G Knepley PetscFunctionBegin; 135641875f9SMatthew G Knepley ierr = PetscMemzero(C,sizeof(*C));CHKERRQ(ierr); 136641875f9SMatthew 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 */ 137641875f9SMatthew G Knepley C->nrow = (size_t)A->cmap->n; 138641875f9SMatthew G Knepley C->ncol = (size_t)A->rmap->n; 139641875f9SMatthew G Knepley C->nzmax = (size_t)sbaij->maxnz; 140641875f9SMatthew G Knepley C->p = sbaij->i; 141641875f9SMatthew G Knepley C->i = sbaij->j; 142641875f9SMatthew G Knepley C->x = sbaij->a; 143641875f9SMatthew G Knepley C->stype = -1; 144641875f9SMatthew G Knepley C->itype = CHOLMOD_INT_TYPE; 145641875f9SMatthew G Knepley C->xtype = CHOLMOD_SCALAR_TYPE; 146641875f9SMatthew G Knepley C->dtype = CHOLMOD_DOUBLE; 147641875f9SMatthew G Knepley C->sorted = 1; 148641875f9SMatthew G Knepley C->packed = 1; 149641875f9SMatthew G Knepley *aijalloc = PETSC_FALSE; 150641875f9SMatthew G Knepley PetscFunctionReturn(0); 151641875f9SMatthew G Knepley } 152641875f9SMatthew G Knepley 153641875f9SMatthew G Knepley #undef __FUNCT__ 154641875f9SMatthew G Knepley #define __FUNCT__ "VecWrapCholmod" 155641875f9SMatthew G Knepley static PetscErrorCode VecWrapCholmod(Vec X,cholmod_dense *Y) 156641875f9SMatthew G Knepley { 157641875f9SMatthew G Knepley PetscErrorCode ierr; 158641875f9SMatthew G Knepley PetscScalar *x; 159641875f9SMatthew G Knepley PetscInt n; 160641875f9SMatthew G Knepley 161641875f9SMatthew G Knepley PetscFunctionBegin; 162641875f9SMatthew G Knepley ierr = PetscMemzero(Y,sizeof(*Y));CHKERRQ(ierr); 163641875f9SMatthew G Knepley ierr = VecGetArray(X,&x);CHKERRQ(ierr); 164641875f9SMatthew G Knepley ierr = VecGetSize(X,&n);CHKERRQ(ierr); 16526fbe8dcSKarl Rupp 166641875f9SMatthew G Knepley Y->x = (double*)x; 167641875f9SMatthew G Knepley Y->nrow = n; 168641875f9SMatthew G Knepley Y->ncol = 1; 169641875f9SMatthew G Knepley Y->nzmax = n; 170641875f9SMatthew G Knepley Y->d = n; 171641875f9SMatthew G Knepley Y->x = (double*)x; 172641875f9SMatthew G Knepley Y->xtype = CHOLMOD_SCALAR_TYPE; 173641875f9SMatthew G Knepley Y->dtype = CHOLMOD_DOUBLE; 174641875f9SMatthew G Knepley PetscFunctionReturn(0); 175641875f9SMatthew G Knepley } 176641875f9SMatthew G Knepley 177641875f9SMatthew G Knepley #undef __FUNCT__ 178641875f9SMatthew G Knepley #define __FUNCT__ "MatDestroy_CHOLMOD" 1797087cfbeSBarry Smith PetscErrorCode MatDestroy_CHOLMOD(Mat F) 180641875f9SMatthew G Knepley { 181641875f9SMatthew G Knepley PetscErrorCode ierr; 182641875f9SMatthew G Knepley Mat_CHOLMOD *chol=(Mat_CHOLMOD*)F->spptr; 183641875f9SMatthew G Knepley 184641875f9SMatthew G Knepley PetscFunctionBegin; 185bf0cc555SLisandro Dalcin if (chol) { 186641875f9SMatthew G Knepley ierr = !cholmod_X_free_factor(&chol->factor,chol->common);CHKERRQ(ierr); 187641875f9SMatthew G Knepley ierr = !cholmod_X_finish(chol->common);CHKERRQ(ierr); 188641875f9SMatthew G Knepley ierr = PetscFree(chol->common);CHKERRQ(ierr); 189641875f9SMatthew G Knepley ierr = PetscFree(chol->matrix);CHKERRQ(ierr); 190641875f9SMatthew G Knepley ierr = (*chol->Destroy)(F);CHKERRQ(ierr); 191bf0cc555SLisandro Dalcin } 192bf0cc555SLisandro Dalcin ierr = PetscFree(F->spptr);CHKERRQ(ierr); 193641875f9SMatthew G Knepley PetscFunctionReturn(0); 194641875f9SMatthew G Knepley } 195641875f9SMatthew G Knepley 196641875f9SMatthew G Knepley static PetscErrorCode MatSolve_CHOLMOD(Mat,Vec,Vec); 197641875f9SMatthew G Knepley 198*fcd503bcSBarry Smith /*static const char *const CholmodOrderingMethods[] = {"User","AMD","METIS","NESDIS(default)","Natural","NESDIS(small=20000)","NESDIS(small=4,no constrained)","NESDIS()"};*/ 199641875f9SMatthew G Knepley 200641875f9SMatthew G Knepley #undef __FUNCT__ 201641875f9SMatthew G Knepley #define __FUNCT__ "MatFactorInfo_CHOLMOD" 202641875f9SMatthew G Knepley static PetscErrorCode MatFactorInfo_CHOLMOD(Mat F,PetscViewer viewer) 203641875f9SMatthew G Knepley { 204641875f9SMatthew G Knepley Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->spptr; 205641875f9SMatthew G Knepley const cholmod_common *c = chol->common; 206641875f9SMatthew G Knepley PetscErrorCode ierr; 207641875f9SMatthew G Knepley PetscInt i; 208641875f9SMatthew G Knepley 209641875f9SMatthew G Knepley PetscFunctionBegin; 210641875f9SMatthew G Knepley if (F->ops->solve != MatSolve_CHOLMOD) PetscFunctionReturn(0); 211641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"CHOLMOD run parameters:\n");CHKERRQ(ierr); 212641875f9SMatthew G Knepley ierr = PetscViewerASCIIPushTab(viewer);CHKERRQ(ierr); 213641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Pack factors after symbolic factorization: %s\n",chol->pack ? "TRUE" : "FALSE");CHKERRQ(ierr); 214641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.dbound %g (Smallest absolute value of diagonal entries of D)\n",c->dbound);CHKERRQ(ierr); 215641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.grow0 %g\n",c->grow0);CHKERRQ(ierr); 216641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.grow1 %g\n",c->grow1);CHKERRQ(ierr); 217641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.grow2 %u\n",(unsigned)c->grow2);CHKERRQ(ierr); 218641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.maxrank %u\n",(unsigned)c->maxrank);CHKERRQ(ierr); 219641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.supernodal_switch %g\n",c->supernodal_switch);CHKERRQ(ierr); 220641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.supernodal %d\n",c->supernodal);CHKERRQ(ierr); 221641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_asis %d\n",c->final_asis);CHKERRQ(ierr); 222641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_super %d\n",c->final_super);CHKERRQ(ierr); 223641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_ll %d\n",c->final_ll);CHKERRQ(ierr); 224641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_pack %d\n",c->final_pack);CHKERRQ(ierr); 225641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_monotonic %d\n",c->final_monotonic);CHKERRQ(ierr); 226641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.final_resymbol %d\n",c->final_resymbol);CHKERRQ(ierr); 227641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.zrelax [%g,%g,%g]\n",c->zrelax[0],c->zrelax[1],c->zrelax[2]);CHKERRQ(ierr); 228641875f9SMatthew 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); 229641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.prefer_upper %d\n",c->prefer_upper);CHKERRQ(ierr); 230641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.print %d\n",c->print);CHKERRQ(ierr); 231641875f9SMatthew G Knepley for (i=0; i<c->nmethods; i++) { 232641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Ordering method %D%s:\n",i,i==c->selected ? " [SELECTED]" : "");CHKERRQ(ierr); 233641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer," lnz %g, fl %g, prune_dense %g, prune_dense2 %g\n", 234641875f9SMatthew G Knepley c->method[i].lnz,c->method[i].fl,c->method[i].prune_dense,c->method[i].prune_dense2);CHKERRQ(ierr); 235641875f9SMatthew G Knepley } 236641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.postorder %d\n",c->postorder);CHKERRQ(ierr); 237641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.default_nesdis %d (use NESDIS instead of METIS for nested dissection)\n",c->default_nesdis);CHKERRQ(ierr); 238641875f9SMatthew G Knepley /* Statistics */ 239641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.fl %g (flop count from most recent analysis)\n",c->fl);CHKERRQ(ierr); 240641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.lnz %g (fundamental nz in L)\n",c->lnz);CHKERRQ(ierr); 241641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.anz %g\n",c->anz);CHKERRQ(ierr); 242641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.modfl %g (flop count from most recent update)\n",c->modfl);CHKERRQ(ierr); 243641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.malloc_count %g (number of live objects)\n",(double)c->malloc_count);CHKERRQ(ierr);CHKERRQ(ierr); 244641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.memory_usage %g (peak memory usage in bytes)\n",(double)c->memory_usage);CHKERRQ(ierr);CHKERRQ(ierr); 245641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.memory_inuse %g (current memory usage in bytes)\n",(double)c->memory_inuse);CHKERRQ(ierr);CHKERRQ(ierr); 246641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.nrealloc_col %g (number of column reallocations)\n",c->nrealloc_col);CHKERRQ(ierr);CHKERRQ(ierr); 247641875f9SMatthew 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); 248641875f9SMatthew 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); 249641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.rowfacfl %g (number of flops in last call to cholmod_rowfac)\n",c->rowfacfl);CHKERRQ(ierr);CHKERRQ(ierr); 250641875f9SMatthew G Knepley ierr = PetscViewerASCIIPrintf(viewer,"Common.aatfl %g (number of flops to compute A(:,f)*A(:,f)')\n",c->aatfl);CHKERRQ(ierr);CHKERRQ(ierr); 251641875f9SMatthew G Knepley ierr = PetscViewerASCIIPopTab(viewer);CHKERRQ(ierr); 252641875f9SMatthew G Knepley PetscFunctionReturn(0); 253641875f9SMatthew G Knepley } 254641875f9SMatthew G Knepley 255641875f9SMatthew G Knepley #undef __FUNCT__ 256641875f9SMatthew G Knepley #define __FUNCT__ "MatView_CHOLMOD" 2577087cfbeSBarry Smith PetscErrorCode MatView_CHOLMOD(Mat F,PetscViewer viewer) 258641875f9SMatthew G Knepley { 259641875f9SMatthew G Knepley PetscErrorCode ierr; 260ace3abfcSBarry Smith PetscBool iascii; 261641875f9SMatthew G Knepley PetscViewerFormat format; 262641875f9SMatthew G Knepley 263641875f9SMatthew G Knepley PetscFunctionBegin; 264641875f9SMatthew G Knepley ierr = MatView_SeqSBAIJ(F,viewer);CHKERRQ(ierr); 265251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 266641875f9SMatthew G Knepley if (iascii) { 267641875f9SMatthew G Knepley ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 268641875f9SMatthew G Knepley if (format == PETSC_VIEWER_ASCII_INFO) { 269641875f9SMatthew G Knepley ierr = MatFactorInfo_CHOLMOD(F,viewer);CHKERRQ(ierr); 270641875f9SMatthew G Knepley } 271641875f9SMatthew G Knepley } 272641875f9SMatthew G Knepley PetscFunctionReturn(0); 273641875f9SMatthew G Knepley } 274641875f9SMatthew G Knepley 275641875f9SMatthew G Knepley #undef __FUNCT__ 276641875f9SMatthew G Knepley #define __FUNCT__ "MatSolve_CHOLMOD" 277641875f9SMatthew G Knepley static PetscErrorCode MatSolve_CHOLMOD(Mat F,Vec B,Vec X) 278641875f9SMatthew G Knepley { 279641875f9SMatthew G Knepley Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->spptr; 280641875f9SMatthew G Knepley cholmod_dense cholB,*cholX; 281641875f9SMatthew G Knepley PetscScalar *x; 282641875f9SMatthew G Knepley PetscErrorCode ierr; 283641875f9SMatthew G Knepley 284641875f9SMatthew G Knepley PetscFunctionBegin; 285641875f9SMatthew G Knepley ierr = VecWrapCholmod(B,&cholB);CHKERRQ(ierr); 286641875f9SMatthew G Knepley static_F = F; 287641875f9SMatthew G Knepley cholX = cholmod_X_solve(CHOLMOD_A,chol->factor,&cholB,chol->common); 288641875f9SMatthew G Knepley if (!cholX) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"CHOLMOD failed"); 289641875f9SMatthew G Knepley ierr = VecGetArray(X,&x);CHKERRQ(ierr); 290641875f9SMatthew G Knepley ierr = PetscMemcpy(x,cholX->x,cholX->nrow*sizeof(*x));CHKERRQ(ierr); 291641875f9SMatthew G Knepley ierr = !cholmod_X_free_dense(&cholX,chol->common);CHKERRQ(ierr); 292641875f9SMatthew G Knepley ierr = VecRestoreArray(X,&x);CHKERRQ(ierr); 293641875f9SMatthew G Knepley PetscFunctionReturn(0); 294641875f9SMatthew G Knepley } 295641875f9SMatthew G Knepley 296641875f9SMatthew G Knepley #undef __FUNCT__ 297641875f9SMatthew G Knepley #define __FUNCT__ "MatCholeskyFactorNumeric_CHOLMOD" 298641875f9SMatthew G Knepley static PetscErrorCode MatCholeskyFactorNumeric_CHOLMOD(Mat F,Mat A,const MatFactorInfo *info) 299641875f9SMatthew G Knepley { 300641875f9SMatthew G Knepley Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->spptr; 301641875f9SMatthew G Knepley cholmod_sparse cholA; 302ace3abfcSBarry Smith PetscBool aijalloc; 303641875f9SMatthew G Knepley PetscErrorCode ierr; 304641875f9SMatthew G Knepley 305641875f9SMatthew G Knepley PetscFunctionBegin; 306641875f9SMatthew G Knepley ierr = (*chol->Wrap)(A,PETSC_TRUE,&cholA,&aijalloc);CHKERRQ(ierr); 307641875f9SMatthew G Knepley static_F = F; 308641875f9SMatthew G Knepley ierr = !cholmod_X_factorize(&cholA,chol->factor,chol->common); 309ce94432eSBarry Smith if (ierr) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD factorization failed with status %d",chol->common->status); 310ce94432eSBarry 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); 311641875f9SMatthew G Knepley 312641875f9SMatthew G Knepley if (aijalloc) {ierr = PetscFree3(cholA.p,cholA.i,cholA.x);CHKERRQ(ierr);} 313641875f9SMatthew G Knepley 314641875f9SMatthew G Knepley F->ops->solve = MatSolve_CHOLMOD; 315641875f9SMatthew G Knepley F->ops->solvetranspose = MatSolve_CHOLMOD; 316641875f9SMatthew G Knepley PetscFunctionReturn(0); 317641875f9SMatthew G Knepley } 318641875f9SMatthew G Knepley 319641875f9SMatthew G Knepley #undef __FUNCT__ 320641875f9SMatthew G Knepley #define __FUNCT__ "MatCholeskyFactorSymbolic_CHOLMOD" 3217087cfbeSBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_CHOLMOD(Mat F,Mat A,IS perm,const MatFactorInfo *info) 322641875f9SMatthew G Knepley { 323641875f9SMatthew G Knepley Mat_CHOLMOD *chol = (Mat_CHOLMOD*)F->spptr; 324641875f9SMatthew G Knepley PetscErrorCode ierr; 325641875f9SMatthew G Knepley cholmod_sparse cholA; 326ace3abfcSBarry Smith PetscBool aijalloc; 327641875f9SMatthew G Knepley PetscInt *fset = 0; 328641875f9SMatthew G Knepley size_t fsize = 0; 329641875f9SMatthew G Knepley 330641875f9SMatthew G Knepley PetscFunctionBegin; 331641875f9SMatthew G Knepley ierr = (*chol->Wrap)(A,PETSC_FALSE,&cholA,&aijalloc);CHKERRQ(ierr); 332641875f9SMatthew G Knepley static_F = F; 333641875f9SMatthew G Knepley if (chol->factor) { 334641875f9SMatthew G Knepley ierr = !cholmod_X_resymbol(&cholA,fset,fsize,(int)chol->pack,chol->factor,chol->common); 335ce94432eSBarry Smith if (ierr) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD analysis failed with status %d",chol->common->status); 336641875f9SMatthew G Knepley } else if (perm) { 337641875f9SMatthew G Knepley const PetscInt *ip; 338641875f9SMatthew G Knepley ierr = ISGetIndices(perm,&ip);CHKERRQ(ierr); 339641875f9SMatthew G Knepley chol->factor = cholmod_X_analyze_p(&cholA,(PetscInt*)ip,fset,fsize,chol->common); 340ce94432eSBarry Smith if (!chol->factor) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD analysis failed with status %d",chol->common->status); 341641875f9SMatthew G Knepley ierr = ISRestoreIndices(perm,&ip);CHKERRQ(ierr); 342641875f9SMatthew G Knepley } else { 343641875f9SMatthew G Knepley chol->factor = cholmod_X_analyze(&cholA,chol->common); 344ce94432eSBarry Smith if (!chol->factor) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD analysis failed with status %d",chol->common->status); 345641875f9SMatthew G Knepley } 346641875f9SMatthew G Knepley 347641875f9SMatthew G Knepley if (aijalloc) {ierr = PetscFree3(cholA.p,cholA.i,cholA.x);CHKERRQ(ierr);} 348641875f9SMatthew G Knepley 349641875f9SMatthew G Knepley F->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_CHOLMOD; 350641875f9SMatthew G Knepley PetscFunctionReturn(0); 351641875f9SMatthew G Knepley } 352641875f9SMatthew G Knepley 353641875f9SMatthew G Knepley #undef __FUNCT__ 354641875f9SMatthew G Knepley #define __FUNCT__ "MatFactorGetSolverPackage_seqsbaij_cholmod" 355641875f9SMatthew G Knepley PetscErrorCode MatFactorGetSolverPackage_seqsbaij_cholmod(Mat A,const MatSolverPackage *type) 356641875f9SMatthew G Knepley { 357641875f9SMatthew G Knepley PetscFunctionBegin; 358641875f9SMatthew G Knepley *type = MATSOLVERCHOLMOD; 359641875f9SMatthew G Knepley PetscFunctionReturn(0); 360641875f9SMatthew G Knepley } 361641875f9SMatthew G Knepley 362641875f9SMatthew G Knepley /*MC 363641875f9SMatthew G Knepley MATSOLVERCHOLMOD = "cholmod" - A matrix type providing direct solvers (Cholesky) for sequential matrices 364641875f9SMatthew G Knepley via the external package CHOLMOD. 365641875f9SMatthew G Knepley 366f02a4714SShri ./configure --download-suitesparse to install PETSc to use CHOLMOD 367641875f9SMatthew G Knepley 368641875f9SMatthew G Knepley Consult CHOLMOD documentation for more information about the Common parameters 369641875f9SMatthew G Knepley which correspond to the options database keys below. 370641875f9SMatthew G Knepley 371641875f9SMatthew G Knepley Options Database Keys: 372e08999f5SMatthew G Knepley + -mat_cholmod_dbound <0> - Minimum absolute value of diagonal entries of D (None) 373e08999f5SMatthew G Knepley . -mat_cholmod_grow0 <1.2> - Global growth ratio when factors are modified (None) 374e08999f5SMatthew G Knepley . -mat_cholmod_grow1 <1.2> - Column growth ratio when factors are modified (None) 375e08999f5SMatthew G Knepley . -mat_cholmod_grow2 <5> - Affine column growth constant when factors are modified (None) 376e08999f5SMatthew G Knepley . -mat_cholmod_maxrank <8> - Max rank of update, larger values are faster but use more memory [2,4,8] (None) 377e08999f5SMatthew G Knepley . -mat_cholmod_factor <AUTO> - (choose one of) SIMPLICIAL AUTO SUPERNODAL 378e08999f5SMatthew G Knepley . -mat_cholmod_supernodal_switch <40> - flop/nnz_L threshold for switching to supernodal factorization (None) 379e08999f5SMatthew G Knepley . -mat_cholmod_final_asis <TRUE> - Leave factors "as is" (None) 380e08999f5SMatthew G Knepley . -mat_cholmod_final_pack <TRUE> - Pack the columns when finished (use FALSE if the factors will be updated later) (None) 381e08999f5SMatthew G Knepley . -mat_cholmod_zrelax <0.8> - 3 real supernodal relaxed amalgamation parameters (None) 382e08999f5SMatthew G Knepley . -mat_cholmod_nrelax <4> - 3 size_t supernodal relaxed amalgamation parameters (None) 383e08999f5SMatthew G Knepley . -mat_cholmod_prefer_upper <TRUE> - Work with upper triangular form (faster when using fill-reducing ordering, slower in natural ordering) (None) 384e08999f5SMatthew G Knepley - -mat_cholmod_print <3> - Verbosity level (None) 385641875f9SMatthew G Knepley 386641875f9SMatthew G Knepley Level: beginner 387641875f9SMatthew G Knepley 388641875f9SMatthew G Knepley .seealso: PCCHOLESKY, PCFactorSetMatSolverPackage(), MatSolverPackage 389641875f9SMatthew G Knepley M*/ 390b2573a8aSBarry Smith 391641875f9SMatthew G Knepley #undef __FUNCT__ 392641875f9SMatthew G Knepley #define __FUNCT__ "MatGetFactor_seqsbaij_cholmod" 3938cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat A,MatFactorType ftype,Mat *F) 394641875f9SMatthew G Knepley { 395641875f9SMatthew G Knepley Mat B; 396641875f9SMatthew G Knepley Mat_CHOLMOD *chol; 397641875f9SMatthew G Knepley PetscErrorCode ierr; 398641875f9SMatthew G Knepley PetscInt m=A->rmap->n,n=A->cmap->n,bs; 399641875f9SMatthew G Knepley 400641875f9SMatthew G Knepley PetscFunctionBegin; 401641875f9SMatthew G Knepley if (ftype != MAT_FACTOR_CHOLESKY) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"CHOLMOD cannot do %s factorization with SBAIJ, only %s", 402641875f9SMatthew G Knepley MatFactorTypes[ftype],MatFactorTypes[MAT_FACTOR_CHOLESKY]); 403641875f9SMatthew G Knepley ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 404ce94432eSBarry Smith if (bs != 1) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"CHOLMOD only supports block size=1, given %D",bs); 405641875f9SMatthew G Knepley /* Create the factorization matrix F */ 406ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 407641875f9SMatthew G Knepley ierr = MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,m,n);CHKERRQ(ierr); 408641875f9SMatthew G Knepley ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 4090298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(B,1,0,NULL);CHKERRQ(ierr); 410b00a9115SJed Brown ierr = PetscNewLog(B,&chol);CHKERRQ(ierr); 41126fbe8dcSKarl Rupp 412641875f9SMatthew G Knepley chol->Wrap = MatWrapCholmod_seqsbaij; 413641875f9SMatthew G Knepley chol->Destroy = MatDestroy_SeqSBAIJ; 414641875f9SMatthew G Knepley B->spptr = chol; 415641875f9SMatthew G Knepley 416641875f9SMatthew G Knepley B->ops->view = MatView_CHOLMOD; 417641875f9SMatthew G Knepley B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_CHOLMOD; 418641875f9SMatthew G Knepley B->ops->destroy = MatDestroy_CHOLMOD; 419bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_seqsbaij_cholmod);CHKERRQ(ierr); 420641875f9SMatthew G Knepley B->factortype = MAT_FACTOR_CHOLESKY; 421641875f9SMatthew G Knepley B->assembled = PETSC_TRUE; /* required by -ksp_view */ 422641875f9SMatthew G Knepley B->preallocated = PETSC_TRUE; 423641875f9SMatthew G Knepley 424641875f9SMatthew G Knepley ierr = CholmodStart(B);CHKERRQ(ierr); 425641875f9SMatthew G Knepley *F = B; 426641875f9SMatthew G Knepley PetscFunctionReturn(0); 427641875f9SMatthew G Knepley } 428