xref: /petsc/src/mat/impls/sbaij/seq/cholmod/sbaijcholmod.c (revision ce94432eddcd14845bc7e8083b7f8ea723b9bf7d)
1641875f9SMatthew G Knepley 
2641875f9SMatthew G Knepley /*
3641875f9SMatthew G Knepley    Provides an interface to the CHOLMOD 1.7.1 sparse solver
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); \
69*ce94432eSBarry 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 
79*ce94432eSBarry 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);
111*ce94432eSBarry 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);
117*ce94432eSBarry 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 
198641875f9SMatthew G Knepley 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);
309*ce94432eSBarry Smith   if (ierr) SETERRQ1(PetscObjectComm((PetscObject)F),PETSC_ERR_LIB,"CHOLMOD factorization failed with status %d",chol->common->status);
310*ce94432eSBarry 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);
335*ce94432eSBarry 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);
340*ce94432eSBarry 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);
344*ce94432eSBarry 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 EXTERN_C_BEGIN
354641875f9SMatthew G Knepley #undef __FUNCT__
355641875f9SMatthew G Knepley #define __FUNCT__ "MatFactorGetSolverPackage_seqsbaij_cholmod"
356641875f9SMatthew G Knepley PetscErrorCode MatFactorGetSolverPackage_seqsbaij_cholmod(Mat A,const MatSolverPackage *type)
357641875f9SMatthew G Knepley {
358641875f9SMatthew G Knepley   PetscFunctionBegin;
359641875f9SMatthew G Knepley   *type = MATSOLVERCHOLMOD;
360641875f9SMatthew G Knepley   PetscFunctionReturn(0);
361641875f9SMatthew G Knepley }
362641875f9SMatthew G Knepley EXTERN_C_END
363641875f9SMatthew G Knepley 
364641875f9SMatthew G Knepley /*MC
365641875f9SMatthew G Knepley   MATSOLVERCHOLMOD = "cholmod" - A matrix type providing direct solvers (Cholesky) for sequential matrices
366641875f9SMatthew G Knepley   via the external package CHOLMOD.
367641875f9SMatthew G Knepley 
368641875f9SMatthew G Knepley   ./configure --download-cholmod to install PETSc to use CHOLMOD
369641875f9SMatthew G Knepley 
370641875f9SMatthew G Knepley   Consult CHOLMOD documentation for more information about the Common parameters
371641875f9SMatthew G Knepley   which correspond to the options database keys below.
372641875f9SMatthew G Knepley 
373641875f9SMatthew G Knepley   Options Database Keys:
374e08999f5SMatthew G Knepley + -mat_cholmod_dbound <0>          - Minimum absolute value of diagonal entries of D (None)
375e08999f5SMatthew G Knepley . -mat_cholmod_grow0 <1.2>         - Global growth ratio when factors are modified (None)
376e08999f5SMatthew G Knepley . -mat_cholmod_grow1 <1.2>         - Column growth ratio when factors are modified (None)
377e08999f5SMatthew G Knepley . -mat_cholmod_grow2 <5>           - Affine column growth constant when factors are modified (None)
378e08999f5SMatthew G Knepley . -mat_cholmod_maxrank <8>         - Max rank of update, larger values are faster but use more memory [2,4,8] (None)
379e08999f5SMatthew G Knepley . -mat_cholmod_factor <AUTO>       - (choose one of) SIMPLICIAL AUTO SUPERNODAL
380e08999f5SMatthew G Knepley . -mat_cholmod_supernodal_switch <40> - flop/nnz_L threshold for switching to supernodal factorization (None)
381e08999f5SMatthew G Knepley . -mat_cholmod_final_asis <TRUE>   - Leave factors "as is" (None)
382e08999f5SMatthew G Knepley . -mat_cholmod_final_pack <TRUE>   - Pack the columns when finished (use FALSE if the factors will be updated later) (None)
383e08999f5SMatthew G Knepley . -mat_cholmod_zrelax <0.8>        - 3 real supernodal relaxed amalgamation parameters (None)
384e08999f5SMatthew G Knepley . -mat_cholmod_nrelax <4>          - 3 size_t supernodal relaxed amalgamation parameters (None)
385e08999f5SMatthew G Knepley . -mat_cholmod_prefer_upper <TRUE> - Work with upper triangular form (faster when using fill-reducing ordering, slower in natural ordering) (None)
386e08999f5SMatthew G Knepley - -mat_cholmod_print <3>           - Verbosity level (None)
387641875f9SMatthew G Knepley 
388641875f9SMatthew G Knepley    Level: beginner
389641875f9SMatthew G Knepley 
390641875f9SMatthew G Knepley .seealso: PCCHOLESKY, PCFactorSetMatSolverPackage(), MatSolverPackage
391641875f9SMatthew G Knepley M*/
392641875f9SMatthew G Knepley EXTERN_C_BEGIN
393641875f9SMatthew G Knepley #undef __FUNCT__
394641875f9SMatthew G Knepley #define __FUNCT__ "MatGetFactor_seqsbaij_cholmod"
395641875f9SMatthew G Knepley PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat A,MatFactorType ftype,Mat *F)
396641875f9SMatthew G Knepley {
397641875f9SMatthew G Knepley   Mat            B;
398641875f9SMatthew G Knepley   Mat_CHOLMOD    *chol;
399641875f9SMatthew G Knepley   PetscErrorCode ierr;
400641875f9SMatthew G Knepley   PetscInt       m=A->rmap->n,n=A->cmap->n,bs;
401641875f9SMatthew G Knepley 
402641875f9SMatthew G Knepley   PetscFunctionBegin;
403641875f9SMatthew G Knepley   if (ftype != MAT_FACTOR_CHOLESKY) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"CHOLMOD cannot do %s factorization with SBAIJ, only %s",
404641875f9SMatthew G Knepley                                              MatFactorTypes[ftype],MatFactorTypes[MAT_FACTOR_CHOLESKY]);
405641875f9SMatthew G Knepley   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
406*ce94432eSBarry Smith   if (bs != 1) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"CHOLMOD only supports block size=1, given %D",bs);
407641875f9SMatthew G Knepley   /* Create the factorization matrix F */
408*ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
409641875f9SMatthew G Knepley   ierr = MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,m,n);CHKERRQ(ierr);
410641875f9SMatthew G Knepley   ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
4110298fd71SBarry Smith   ierr = MatSeqSBAIJSetPreallocation(B,1,0,NULL);CHKERRQ(ierr);
412641875f9SMatthew G Knepley   ierr = PetscNewLog(B,Mat_CHOLMOD,&chol);CHKERRQ(ierr);
41326fbe8dcSKarl Rupp 
414641875f9SMatthew G Knepley   chol->Wrap    = MatWrapCholmod_seqsbaij;
415641875f9SMatthew G Knepley   chol->Destroy = MatDestroy_SeqSBAIJ;
416641875f9SMatthew G Knepley   B->spptr      = chol;
417641875f9SMatthew G Knepley 
418641875f9SMatthew G Knepley   B->ops->view                   = MatView_CHOLMOD;
419641875f9SMatthew G Knepley   B->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_CHOLMOD;
420641875f9SMatthew G Knepley   B->ops->destroy                = MatDestroy_CHOLMOD;
421641875f9SMatthew G Knepley   ierr                           = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_seqsbaij_cholmod",MatFactorGetSolverPackage_seqsbaij_cholmod);CHKERRQ(ierr);
422641875f9SMatthew G Knepley   B->factortype                  = MAT_FACTOR_CHOLESKY;
423641875f9SMatthew G Knepley   B->assembled                   = PETSC_TRUE; /* required by -ksp_view */
424641875f9SMatthew G Knepley   B->preallocated                = PETSC_TRUE;
425641875f9SMatthew G Knepley 
426641875f9SMatthew G Knepley   ierr = CholmodStart(B);CHKERRQ(ierr);
427641875f9SMatthew G Knepley   *F   = B;
428641875f9SMatthew G Knepley   PetscFunctionReturn(0);
429641875f9SMatthew G Knepley }
430641875f9SMatthew G Knepley EXTERN_C_END
431