xref: /petsc/src/mat/impls/sbaij/seq/cholmod/sbaijcholmod.c (revision 218db3c1f1549ab3d3718007615fce10255b2d0c)
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