xref: /petsc/src/sys/objects/init.c (revision e316c87f740c7e7b34401e9ee6f949125cc0e325)
1e5c89e4eSSatish Balay /*
2e5c89e4eSSatish Balay 
3e5c89e4eSSatish Balay    This file defines part of the initialization of PETSc
4e5c89e4eSSatish Balay 
5f2d66bcaSShri Abhyankar   This file uses regular malloc and free because it cannot known
6e5c89e4eSSatish Balay   what malloc is being used until it has already processed the input.
7e5c89e4eSSatish Balay */
8ef386f4bSSatish Balay 
9ef386f4bSSatish Balay #include <petscsys.h>        /*I  "petscsys.h"   I*/
10f67a399dSBarry Smith #include <petsc/private/petscimpl.h>
11022afb99SBarry Smith #include <petscvalgrind.h>
12665c2dedSJed Brown #include <petscviewer.h>
13ef386f4bSSatish Balay 
14ba61063dSBarry Smith #if defined(PETSC_HAVE_SYS_SYSINFO_H)
1551d315f7SKerry Stevens #include <sys/sysinfo.h>
16ba61063dSBarry Smith #endif
17121deb67SSatish Balay #if defined(PETSC_HAVE_UNISTD_H)
1851d315f7SKerry Stevens #include <unistd.h>
19121deb67SSatish Balay #endif
20d9b72601SDominic Meiser #if defined(PETSC_HAVE_CUDA)
21d9b72601SDominic Meiser #include <cuda_runtime.h>
22d9b72601SDominic Meiser #endif
23555d055bSBarry Smith 
24f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
25f0a7718cSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLInit();
26f0a7718cSKarl Rupp #endif
27f0a7718cSKarl Rupp 
28e5c89e4eSSatish Balay /* ------------------------Nasty global variables -------------------------------*/
29e5c89e4eSSatish Balay /*
30e5c89e4eSSatish Balay      Indicates if PETSc started up MPI, or it was
31e5c89e4eSSatish Balay    already started before PETSc was initialized.
32e5c89e4eSSatish Balay */
330cbf60d0SJose E. Roman PetscBool   PetscBeganMPI         = PETSC_FALSE;
340cbf60d0SJose E. Roman PetscBool   PetscInitializeCalled = PETSC_FALSE;
350cbf60d0SJose E. Roman PetscBool   PetscFinalizeCalled   = PETSC_FALSE;
36bffec52eSKarl Rupp PetscBool   PetscCUDAInitialized  = PETSC_FALSE;
37d6f2c3cbSBarry Smith 
387087cfbeSBarry Smith PetscMPIInt PetscGlobalRank       = -1;
397087cfbeSBarry Smith PetscMPIInt PetscGlobalSize       = -1;
40ba61063dSBarry Smith 
4150f81f78SJed Brown #if defined(PETSC_HAVE_COMPLEX)
42e5c89e4eSSatish Balay #if defined(PETSC_COMPLEX_INSTANTIATE)
43e5c89e4eSSatish Balay template <> class std::complex<double>; /* instantiate complex template class */
44e5c89e4eSSatish Balay #endif
452c876bd9SBarry Smith #if !defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX)
46500d8756SSatish Balay MPI_Datatype MPIU_C_DOUBLE_COMPLEX;
47500d8756SSatish Balay MPI_Datatype MPIU_C_COMPLEX;
482c876bd9SBarry Smith #endif
498619c96cSJed Brown 
508619c96cSJed Brown /*MC
518619c96cSJed Brown    PETSC_i - the imaginary number i
528619c96cSJed Brown 
538619c96cSJed Brown    Synopsis:
548619c96cSJed Brown    #include <petscsys.h>
558619c96cSJed Brown    PetscComplex PETSC_i;
568619c96cSJed Brown 
578619c96cSJed Brown    Level: beginner
588619c96cSJed Brown 
598619c96cSJed Brown    Note:
608cd53115SBarry Smith    Complex numbers are automatically available if PETSc located a working complex implementation
618619c96cSJed Brown 
628619c96cSJed Brown .seealso: PetscRealPart(), PetscImaginaryPart(), PetscRealPartComplex(), PetscImaginaryPartComplex()
638619c96cSJed Brown M*/
6450f81f78SJed Brown PetscComplex PETSC_i;
65e5c89e4eSSatish Balay #endif
66ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
67c90a1750SBarry Smith MPI_Datatype MPIU___FLOAT128 = 0;
687c2de775SJed Brown #if defined(PETSC_HAVE_COMPLEX)
698c764dc5SJose Roman MPI_Datatype MPIU___COMPLEX128 = 0;
708c764dc5SJose Roman #endif
71570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16)
72570b7f6dSBarry Smith MPI_Datatype MPIU___FP16 = 0;
73c90a1750SBarry Smith #endif
747087cfbeSBarry Smith MPI_Datatype MPIU_2SCALAR = 0;
7544041f26SJed Brown #if defined(PETSC_USE_64BIT_INDICES) || !defined(MPI_2INT)
767087cfbeSBarry Smith MPI_Datatype MPIU_2INT = 0;
7744041f26SJed Brown #endif
788ad47952SJed Brown MPI_Datatype MPIU_BOOL;
798ad47952SJed Brown MPI_Datatype MPIU_ENUM;
807cdaf61dSJed Brown MPI_Datatype MPIU_FORTRANADDR;
81*e316c87fSJed Brown MPI_Datatype MPIU_SIZE_T;
8275567043SBarry Smith 
83e5c89e4eSSatish Balay /*
84e5c89e4eSSatish Balay        Function that is called to display all error messages
85e5c89e4eSSatish Balay */
867087cfbeSBarry Smith PetscErrorCode (*PetscErrorPrintf)(const char [],...)          = PetscErrorPrintfDefault;
877087cfbeSBarry Smith PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...)  = PetscHelpPrintfDefault;
887087cfbeSBarry Smith PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list)    = PetscVFPrintfDefault;
89bab1f7e6SVictor Minden /*
904cf1874eSKarl Rupp   This is needed to turn on/off GPU synchronization
918b5db460SBarry Smith */
924cf1874eSKarl Rupp PetscBool PetscViennaCLSynchronize = PETSC_FALSE;
9382f73ecaSAlejandro Lamas Daviña PetscBool PetscCUDASynchronize = PETSC_FALSE;
94bab1f7e6SVictor Minden 
95e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
96e5c89e4eSSatish Balay /*
97e5c89e4eSSatish Balay    Optional file where all PETSc output from various prints is saved
98e5c89e4eSSatish Balay */
9995c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history;
1000298fd71SBarry Smith FILE *petsc_history = NULL;
101e5c89e4eSSatish Balay 
1027087cfbeSBarry Smith PetscErrorCode  PetscOpenHistoryFile(const char filename[],FILE **fd)
103e5c89e4eSSatish Balay {
104e5c89e4eSSatish Balay   PetscErrorCode ierr;
105e5c89e4eSSatish Balay   PetscMPIInt    rank,size;
106e5c89e4eSSatish Balay   char           pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64];
107e5c89e4eSSatish Balay   char           version[256];
108e5c89e4eSSatish Balay 
109e5c89e4eSSatish Balay   PetscFunctionBegin;
110e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
111e5c89e4eSSatish Balay   if (!rank) {
112e5c89e4eSSatish Balay     char        arch[10];
113f56c2debSBarry Smith     int         err;
114f56c2debSBarry Smith 
115e5c89e4eSSatish Balay     ierr = PetscGetArchType(arch,10);CHKERRQ(ierr);
116e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
117a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
118e5c89e4eSSatish Balay     ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
119e5c89e4eSSatish Balay     if (filename) {
120e5c89e4eSSatish Balay       ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr);
121e5c89e4eSSatish Balay     } else {
122e5c89e4eSSatish Balay       ierr = PetscGetHomeDirectory(pfile,240);CHKERRQ(ierr);
123e5c89e4eSSatish Balay       ierr = PetscStrcat(pfile,"/.petschistory");CHKERRQ(ierr);
124e5c89e4eSSatish Balay       ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr);
125e5c89e4eSSatish Balay     }
126e5c89e4eSSatish Balay 
127a297a907SKarl Rupp     *fd = fopen(fname,"a");
128a297a907SKarl Rupp     if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname);
129a297a907SKarl Rupp 
130e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
131e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr);
132e5c89e4eSSatish Balay     ierr = PetscGetProgramName(pname,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
133e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr);
134e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
135a297a907SKarl Rupp 
136f56c2debSBarry Smith     err = fflush(*fd);
137e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
138e5c89e4eSSatish Balay   }
139e5c89e4eSSatish Balay   PetscFunctionReturn(0);
140e5c89e4eSSatish Balay }
141e5c89e4eSSatish Balay 
14295c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscCloseHistoryFile(FILE **fd)
143e5c89e4eSSatish Balay {
144e5c89e4eSSatish Balay   PetscErrorCode ierr;
145e5c89e4eSSatish Balay   PetscMPIInt    rank;
146e5c89e4eSSatish Balay   char           date[64];
147f56c2debSBarry Smith   int            err;
148e5c89e4eSSatish Balay 
149e5c89e4eSSatish Balay   PetscFunctionBegin;
150e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
151e5c89e4eSSatish Balay   if (!rank) {
152e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
153e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
154e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr);
155e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
156f56c2debSBarry Smith     err  = fflush(*fd);
157e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
158f56c2debSBarry Smith     err = fclose(*fd);
159e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
160e5c89e4eSSatish Balay   }
161e5c89e4eSSatish Balay   PetscFunctionReturn(0);
162e5c89e4eSSatish Balay }
163e5c89e4eSSatish Balay 
164e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
165e5c89e4eSSatish Balay 
166e5c89e4eSSatish Balay /*
167e5c89e4eSSatish Balay    This is ugly and probably belongs somewhere else, but I want to
168e5c89e4eSSatish Balay   be able to put a true MPI abort error handler with command line args.
169e5c89e4eSSatish Balay 
170e5c89e4eSSatish Balay     This is so MPI errors in the debugger will leave all the stack
1713c311c98SBarry Smith   frames. The default MP_Abort() cleans up and exits thus providing no useful information
1723c311c98SBarry Smith   in the debugger hence we call abort() instead of MPI_Abort().
173e5c89e4eSSatish Balay */
174e5c89e4eSSatish Balay 
17533c7d699SBarry Smith void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag,...)
176e5c89e4eSSatish Balay {
177e5c89e4eSSatish Balay   PetscFunctionBegin;
1783c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
179e5c89e4eSSatish Balay   abort();
180e5c89e4eSSatish Balay }
181e5c89e4eSSatish Balay 
18233c7d699SBarry Smith void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag,...)
183e5c89e4eSSatish Balay {
184e5c89e4eSSatish Balay   PetscErrorCode ierr;
185e5c89e4eSSatish Balay 
186e5c89e4eSSatish Balay   PetscFunctionBegin;
1873c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
188e5c89e4eSSatish Balay   ierr = PetscAttachDebugger();
189a297a907SKarl Rupp   if (ierr) MPI_Abort(*comm,*flag); /* hopeless so get out */
190e5c89e4eSSatish Balay }
191e5c89e4eSSatish Balay 
192e5c89e4eSSatish Balay /*@C
193e5c89e4eSSatish Balay    PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one
194e5c89e4eSSatish Balay      wishes a clean exit somewhere deep in the program.
195e5c89e4eSSatish Balay 
196e5c89e4eSSatish Balay    Collective on PETSC_COMM_WORLD
197e5c89e4eSSatish Balay 
198e5c89e4eSSatish Balay    Options Database Keys are the same as for PetscFinalize()
199e5c89e4eSSatish Balay 
200e5c89e4eSSatish Balay    Level: advanced
201e5c89e4eSSatish Balay 
202e5c89e4eSSatish Balay    Note:
203e5c89e4eSSatish Balay    See PetscInitialize() for more general runtime options.
204e5c89e4eSSatish Balay 
20588c29154SBarry Smith .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize()
206e5c89e4eSSatish Balay @*/
2077087cfbeSBarry Smith PetscErrorCode  PetscEnd(void)
208e5c89e4eSSatish Balay {
209e5c89e4eSSatish Balay   PetscFunctionBegin;
210e5c89e4eSSatish Balay   PetscFinalize();
211e5c89e4eSSatish Balay   exit(0);
212e5c89e4eSSatish Balay   return 0;
213e5c89e4eSSatish Balay }
214e5c89e4eSSatish Balay 
215ace3abfcSBarry Smith PetscBool PetscOptionsPublish = PETSC_FALSE;
21695c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseTrMalloc_Private(void);
21795c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseHBWMalloc_Private(void);
21895c0884eSLisandro Dalcin PETSC_INTERN PetscBool      petscsetmallocvisited;
219e5c89e4eSSatish Balay static       char           emacsmachinename[256];
220e5c89e4eSSatish Balay 
221e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = 0;
222e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm)    = 0;
223e5c89e4eSSatish Balay 
224e5c89e4eSSatish Balay /*@C
225e5c89e4eSSatish Balay    PetscSetHelpVersionFunctions - Sets functions that print help and version information
226e5c89e4eSSatish Balay    before the PETSc help and version information is printed. Must call BEFORE PetscInitialize().
227e5c89e4eSSatish Balay    This routine enables a "higher-level" package that uses PETSc to print its messages first.
228e5c89e4eSSatish Balay 
229e5c89e4eSSatish Balay    Input Parameter:
2300298fd71SBarry Smith +  help - the help function (may be NULL)
2310298fd71SBarry Smith -  version - the version function (may be NULL)
232e5c89e4eSSatish Balay 
233e5c89e4eSSatish Balay    Level: developer
234e5c89e4eSSatish Balay 
235e5c89e4eSSatish Balay    Concepts: package help message
236e5c89e4eSSatish Balay 
237e5c89e4eSSatish Balay @*/
2387087cfbeSBarry Smith PetscErrorCode  PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm))
239e5c89e4eSSatish Balay {
240e5c89e4eSSatish Balay   PetscFunctionBegin;
241e5c89e4eSSatish Balay   PetscExternalHelpFunction    = help;
242e5c89e4eSSatish Balay   PetscExternalVersionFunction = version;
243e5c89e4eSSatish Balay   PetscFunctionReturn(0);
244e5c89e4eSSatish Balay }
245e5c89e4eSSatish Balay 
24605df10baSBarry Smith #if defined(PETSC_USE_LOG)
24795c0884eSLisandro Dalcin PETSC_INTERN PetscBool   PetscObjectsLog;
24805df10baSBarry Smith #endif
24905df10baSBarry Smith 
25095c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode  PetscOptionsCheckInitial_Private(void)
251e5c89e4eSSatish Balay {
252e5c89e4eSSatish Balay   char              string[64],mname[PETSC_MAX_PATH_LEN],*f;
253e5c89e4eSSatish Balay   MPI_Comm          comm = PETSC_COMM_WORLD;
25467584ceeSBarry Smith   PetscBool         flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flag;
255e5c89e4eSSatish Balay   PetscErrorCode    ierr;
25667584ceeSBarry Smith   PetscReal         si;
25728559dc8SJed Brown   PetscInt          intensity;
258e5c89e4eSSatish Balay   int               i;
259e5c89e4eSSatish Balay   PetscMPIInt       rank;
2607ca660e7SBarry Smith   char              version[256],helpoptions[256];
26167584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
26267584ceeSBarry Smith   PetscReal         logthreshold;
2632479a3a6SBarry Smith #endif
2642479a3a6SBarry Smith #if defined(PETSC_USE_LOG)
2652479a3a6SBarry Smith   PetscViewerFormat format;
26667584ceeSBarry Smith   PetscBool         flg4 = PETSC_FALSE;
26767584ceeSBarry Smith #endif
268e5c89e4eSSatish Balay 
269e5c89e4eSSatish Balay   PetscFunctionBegin;
27042218b76SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
271e5c89e4eSSatish Balay 
27267584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
273e5c89e4eSSatish Balay   /*
274e5c89e4eSSatish Balay       Setup the memory management; support for tracing malloc() usage
275e5c89e4eSSatish Balay   */
276c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-malloc_log",&flg3);CHKERRQ(ierr);
277574034a9SJed Brown   logthreshold = 0.0;
278c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-malloc_log_threshold",&logthreshold,&flg1);CHKERRQ(ierr);
279574034a9SJed Brown   if (flg1) flg3 = PETSC_TRUE;
2808bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG)
281c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg1,&flg2);CHKERRQ(ierr);
282e5c89e4eSSatish Balay   if ((!flg2 || flg1) && !petscsetmallocvisited) {
283dc92acbaSJed Brown     if (flg2 || !(PETSC_RUNNING_ON_VALGRIND)) {
284555d055bSBarry Smith       /* turn off default -malloc if valgrind is being used */
285e5c89e4eSSatish Balay       ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
286555d055bSBarry Smith     }
287e5c89e4eSSatish Balay   }
288e5c89e4eSSatish Balay #else
289c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_dump",&flg1,NULL);CHKERRQ(ierr);
290c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg2,NULL);CHKERRQ(ierr);
291e5c89e4eSSatish Balay   if (flg1 || flg2 || flg3) {ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);}
292e5c89e4eSSatish Balay #endif
293e5c89e4eSSatish Balay   if (flg3) {
294574034a9SJed Brown     ierr = PetscMallocSetDumpLogThreshold((PetscLogDouble)logthreshold);CHKERRQ(ierr);
295e5c89e4eSSatish Balay   }
296ba282f50SJed Brown   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_coalesce",&flg1,&flg2);CHKERRQ(ierr);
297ba282f50SJed Brown   if (flg2) {ierr = PetscMallocSetCoalesce(flg1);CHKERRQ(ierr);}
29890d69ab7SBarry Smith   flg1 = PETSC_FALSE;
299c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_debug",&flg1,NULL);CHKERRQ(ierr);
300e5c89e4eSSatish Balay   if (flg1) {
301e5c89e4eSSatish Balay     ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
302e5c89e4eSSatish Balay     ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr);
303e5c89e4eSSatish Balay   }
304dc92acbaSJed Brown   flg1 = PETSC_FALSE;
305c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_test",&flg1,NULL);CHKERRQ(ierr);
3068bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG)
307dc92acbaSJed Brown   if (flg1 && !PETSC_RUNNING_ON_VALGRIND) {
308dc92acbaSJed Brown     ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
309dc92acbaSJed Brown     ierr = PetscMallocSetDumpLog();CHKERRQ(ierr);
310dc92acbaSJed Brown     ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr);
311dc92acbaSJed Brown   }
312dc92acbaSJed Brown #endif
3132f21b5c6SHong Zhang   flg1 = PETSC_FALSE;
3142f21b5c6SHong Zhang   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_hbw",&flg1,NULL);CHKERRQ(ierr);
315cf6f3811SHong Zhang   /* ignore this option if malloc is already set */
316cf6f3811SHong Zhang   if (flg1 && !petscsetmallocvisited) {ierr = PetscSetUseHBWMalloc_Private();CHKERRQ(ierr);}
317e5c89e4eSSatish Balay 
31890d69ab7SBarry Smith   flg1 = PETSC_FALSE;
319c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_info",&flg1,NULL);CHKERRQ(ierr);
3207783f70dSSatish Balay   if (!flg1) {
32190d69ab7SBarry Smith     flg1 = PETSC_FALSE;
322c5929fdfSBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-memory_view",&flg1,NULL);CHKERRQ(ierr);
3237783f70dSSatish Balay   }
324e5c89e4eSSatish Balay   if (flg1) {
325e5c89e4eSSatish Balay     ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr);
326e5c89e4eSSatish Balay   }
32767584ceeSBarry Smith #endif
328e5c89e4eSSatish Balay 
329b4427426SBarry Smith #if defined(PETSC_USE_LOG)
330c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-objects_dump",&PetscObjectsLog);CHKERRQ(ierr);
331b4427426SBarry Smith #endif
33205df10baSBarry Smith 
333e5c89e4eSSatish Balay   /*
334e5c89e4eSSatish Balay       Set the display variable for graphics
335e5c89e4eSSatish Balay   */
336e5c89e4eSSatish Balay   ierr = PetscSetDisplay();CHKERRQ(ierr);
337e5c89e4eSSatish Balay 
33851dcc849SKerry Stevens   /*
339e5c89e4eSSatish Balay       Print the PETSc version information
340e5c89e4eSSatish Balay   */
341c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-v",&flg1);CHKERRQ(ierr);
342c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-version",&flg2);CHKERRQ(ierr);
3432d747510SLisandro Dalcin   ierr = PetscOptionsHasHelp(NULL,&flg3);CHKERRQ(ierr);
344e5c89e4eSSatish Balay   if (flg1 || flg2 || flg3) {
345e5c89e4eSSatish Balay 
346e5c89e4eSSatish Balay     /*
347e5c89e4eSSatish Balay        Print "higher-level" package version message
348e5c89e4eSSatish Balay     */
349e5c89e4eSSatish Balay     if (PetscExternalVersionFunction) {
350e5c89e4eSSatish Balay       ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr);
351e5c89e4eSSatish Balay     }
352e5c89e4eSSatish Balay 
353a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
3547ca660e7SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr);
355e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr);
356e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr);
357e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr);
35884e42920SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr);
359e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr);
360e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr);
3617ca660e7SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr);
3627ca660e7SBarry Smith   }
3637ca660e7SBarry Smith 
36494941ca7SBarry Smith   /*
36594941ca7SBarry Smith        Print "higher-level" package help message
36694941ca7SBarry Smith   */
36794941ca7SBarry Smith   if (flg3) {
36894941ca7SBarry Smith     if (PetscExternalHelpFunction) {
36994941ca7SBarry Smith       ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr);
37094941ca7SBarry Smith     }
37194941ca7SBarry Smith   }
37294941ca7SBarry Smith 
3737ca660e7SBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-help",helpoptions,sizeof(helpoptions),&flg1);CHKERRQ(ierr);
3747ca660e7SBarry Smith   if (flg1) {
3757ca660e7SBarry Smith     ierr = PetscStrcmp(helpoptions,"intro",&flg2);CHKERRQ(ierr);
3767ca660e7SBarry Smith     if (flg2) {
377bdb346e9SBarry Smith       ierr = PetscOptionsDestroyDefault();CHKERRQ(ierr);
378008a6e76SBarry Smith       ierr = PetscFreeMPIResources();CHKERRQ(ierr);
3797ca660e7SBarry Smith       ierr = MPI_Finalize();CHKERRQ(ierr);
3807ca660e7SBarry Smith       exit(0);
3817ca660e7SBarry Smith     }
382e5c89e4eSSatish Balay   }
383e5c89e4eSSatish Balay 
384e5c89e4eSSatish Balay   /*
385e5c89e4eSSatish Balay       Setup the error handling
386e5c89e4eSSatish Balay   */
38790d69ab7SBarry Smith   flg1 = PETSC_FALSE;
388c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_abort",&flg1,NULL);CHKERRQ(ierr);
389b59baad1SJed Brown   if (flg1) {
39042218b76SBarry Smith     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr);
391b59baad1SJed Brown     ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr);
392b59baad1SJed Brown   }
39390d69ab7SBarry Smith   flg1 = PETSC_FALSE;
394c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_mpiabort",&flg1,NULL);CHKERRQ(ierr);
395cb9801acSJed Brown   if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,0);CHKERRQ(ierr);}
39690d69ab7SBarry Smith   flg1 = PETSC_FALSE;
397c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_on_error",&flg1,NULL);CHKERRQ(ierr);
398e5c89e4eSSatish Balay   if (flg1) {
399d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr);
400e5c89e4eSSatish Balay   }
40190d69ab7SBarry Smith   flg1 = PETSC_FALSE;
402c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-no_signal_handler",&flg1,NULL);CHKERRQ(ierr);
4038d359177SBarry Smith   if (!flg1) {ierr = PetscPushSignalHandler(PetscSignalHandlerDefault,(void*)0);CHKERRQ(ierr);}
40496cc47afSJed Brown   flg1 = PETSC_FALSE;
405c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-fp_trap",&flg1,NULL);CHKERRQ(ierr);
40696cc47afSJed Brown   if (flg1) {ierr = PetscSetFPTrap(PETSC_FP_TRAP_ON);CHKERRQ(ierr);}
407c5929fdfSBarry Smith   ierr = PetscOptionsGetInt(NULL,NULL,"-check_pointer_intensity",&intensity,&flag);CHKERRQ(ierr);
40828559dc8SJed Brown   if (flag) {ierr = PetscCheckPointerSetIntensity(intensity);CHKERRQ(ierr);}
409e5c89e4eSSatish Balay 
410e5c89e4eSSatish Balay   /*
411e5c89e4eSSatish Balay       Setup debugger information
412e5c89e4eSSatish Balay   */
413e5c89e4eSSatish Balay   ierr = PetscSetDefaultDebugger();CHKERRQ(ierr);
414c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_attach_debugger",string,64,&flg1);CHKERRQ(ierr);
415e5c89e4eSSatish Balay   if (flg1) {
416e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
417e5c89e4eSSatish Balay 
418e5c89e4eSSatish Balay     ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
41933c7d699SBarry Smith     ierr = MPI_Comm_create_errhandler(Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr);
420d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
421e5c89e4eSSatish Balay     ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,0);CHKERRQ(ierr);
422e5c89e4eSSatish Balay   }
423c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-debug_terminal",string,64,&flg1);CHKERRQ(ierr);
4245e96ac45SJed Brown   if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); }
425c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-start_in_debugger",string,64,&flg1);CHKERRQ(ierr);
426c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-stop_for_debugger",string,64,&flg2);CHKERRQ(ierr);
427e5c89e4eSSatish Balay   if (flg1 || flg2) {
428e5c89e4eSSatish Balay     PetscMPIInt    size;
429e5c89e4eSSatish Balay     PetscInt       lsize,*nodes;
430e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
431e5c89e4eSSatish Balay     /*
432e5c89e4eSSatish Balay        we have to make sure that all processors have opened
433e5c89e4eSSatish Balay        connections to all other processors, otherwise once the
434e5c89e4eSSatish Balay        debugger has stated it is likely to receive a SIGUSR1
435e5c89e4eSSatish Balay        and kill the program.
436e5c89e4eSSatish Balay     */
43742218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
438e5c89e4eSSatish Balay     if (size > 2) {
439533163c2SBarry Smith       PetscMPIInt dummy = 0;
440e5c89e4eSSatish Balay       MPI_Status  status;
441e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
442e5c89e4eSSatish Balay         if (rank != i) {
44342218b76SBarry Smith           ierr = MPI_Send(&dummy,1,MPI_INT,i,109,comm);CHKERRQ(ierr);
444e5c89e4eSSatish Balay         }
445e5c89e4eSSatish Balay       }
446e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
447e5c89e4eSSatish Balay         if (rank != i) {
44842218b76SBarry Smith           ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,comm,&status);CHKERRQ(ierr);
449e5c89e4eSSatish Balay         }
450e5c89e4eSSatish Balay       }
451e5c89e4eSSatish Balay     }
452e5c89e4eSSatish Balay     /* check if this processor node should be in debugger */
453785e854fSJed Brown     ierr  = PetscMalloc1(size,&nodes);CHKERRQ(ierr);
454e5c89e4eSSatish Balay     lsize = size;
455c5929fdfSBarry Smith     ierr  = PetscOptionsGetIntArray(NULL,NULL,"-debugger_nodes",nodes,&lsize,&flag);CHKERRQ(ierr);
456e5c89e4eSSatish Balay     if (flag) {
457e5c89e4eSSatish Balay       for (i=0; i<lsize; i++) {
458e5c89e4eSSatish Balay         if (nodes[i] == rank) { flag = PETSC_FALSE; break; }
459e5c89e4eSSatish Balay       }
460e5c89e4eSSatish Balay     }
461e5c89e4eSSatish Balay     if (!flag) {
462e5c89e4eSSatish Balay       ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
463e5c89e4eSSatish Balay       ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr);
464e5c89e4eSSatish Balay       if (flg1) {
465e5c89e4eSSatish Balay         ierr = PetscAttachDebugger();CHKERRQ(ierr);
466e5c89e4eSSatish Balay       } else {
467e5c89e4eSSatish Balay         ierr = PetscStopForDebugger();CHKERRQ(ierr);
468e5c89e4eSSatish Balay       }
46933c7d699SBarry Smith       ierr = MPI_Comm_create_errhandler(Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr);
470d54338ecSKarl Rupp       ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
471e5c89e4eSSatish Balay     }
472e5c89e4eSSatish Balay     ierr = PetscFree(nodes);CHKERRQ(ierr);
473e5c89e4eSSatish Balay   }
474e5c89e4eSSatish Balay 
475c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_emacs",emacsmachinename,128,&flg1);CHKERRQ(ierr);
476cb9801acSJed Brown   if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);}
477e5c89e4eSSatish Balay 
478e5c89e4eSSatish Balay   /*
479e5c89e4eSSatish Balay         Setup profiling and logging
480e5c89e4eSSatish Balay   */
4816cf91177SBarry Smith #if defined(PETSC_USE_INFO)
4828bb29257SSatish Balay   {
483e5c89e4eSSatish Balay     char logname[PETSC_MAX_PATH_LEN]; logname[0] = 0;
484c5929fdfSBarry Smith     ierr = PetscOptionsGetString(NULL,NULL,"-info",logname,250,&flg1);CHKERRQ(ierr);
4858bb29257SSatish Balay     if (flg1 && logname[0]) {
486fcc2139eSBarry Smith       ierr = PetscInfoAllow(PETSC_TRUE,logname);CHKERRQ(ierr);
4878bb29257SSatish Balay     } else if (flg1) {
4880298fd71SBarry Smith       ierr = PetscInfoAllow(PETSC_TRUE,NULL);CHKERRQ(ierr);
489e5c89e4eSSatish Balay     }
490e5c89e4eSSatish Balay   }
491865f6aa8SSatish Balay #endif
492865f6aa8SSatish Balay #if defined(PETSC_USE_LOG)
493865f6aa8SSatish Balay   mname[0] = 0;
494c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-history",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr);
495865f6aa8SSatish Balay   if (flg1) {
496865f6aa8SSatish Balay     if (mname[0]) {
497f3dea69dSBarry Smith       ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr);
498865f6aa8SSatish Balay     } else {
499706d7a88SBarry Smith       ierr = PetscOpenHistoryFile(NULL,&petsc_history);CHKERRQ(ierr);
500865f6aa8SSatish Balay     }
501865f6aa8SSatish Balay   }
502217044c2SLisandro Dalcin 
503217044c2SLisandro Dalcin   ierr = PetscOptionsGetBool(NULL,NULL,"-log_sync",&PetscLogSyncOn,NULL);CHKERRQ(ierr);
504217044c2SLisandro Dalcin 
505e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
50690d69ab7SBarry Smith   flg1 = PETSC_FALSE;
507c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_mpe",&flg1);CHKERRQ(ierr);
508495fc317SBarry Smith   if (flg1) {ierr = PetscLogMPEBegin();CHKERRQ(ierr);}
509e5c89e4eSSatish Balay #endif
51090d69ab7SBarry Smith   flg1 = PETSC_FALSE;
51190d69ab7SBarry Smith   flg3 = PETSC_FALSE;
512c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-log_all",&flg1,NULL);CHKERRQ(ierr);
513c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
514e5c89e4eSSatish Balay   if (flg1)                      { ierr = PetscLogAllBegin();CHKERRQ(ierr); }
515bb1d7374SBarry Smith   else if (flg3)                 { ierr = PetscLogDefaultBegin();CHKERRQ(ierr);}
516e5c89e4eSSatish Balay 
517c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-log_trace",mname,250,&flg1);CHKERRQ(ierr);
518e5c89e4eSSatish Balay   if (flg1) {
519e5c89e4eSSatish Balay     char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN];
520e5c89e4eSSatish Balay     FILE *file;
521e5c89e4eSSatish Balay     if (mname[0]) {
5222e924ca5SSatish Balay       PetscSNPrintf(name,PETSC_MAX_PATH_LEN,"%s.%d",mname,rank);
523e5c89e4eSSatish Balay       ierr = PetscFixFilename(name,fname);CHKERRQ(ierr);
524e5c89e4eSSatish Balay       file = fopen(fname,"w");
525f3dea69dSBarry Smith       if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname);
526a297a907SKarl Rupp     } else file = PETSC_STDOUT;
527e5c89e4eSSatish Balay     ierr = PetscLogTraceBegin(file);CHKERRQ(ierr);
528e5c89e4eSSatish Balay   }
529bb1d7374SBarry Smith 
53016413a6aSBarry Smith   ierr = PetscOptionsGetViewer(comm,NULL,NULL,"-log_view",NULL,&format,&flg4);CHKERRQ(ierr);
531bb1d7374SBarry Smith   if (flg4) {
532bb1d7374SBarry Smith     if (format == PETSC_VIEWER_ASCII_XML) {
533bb1d7374SBarry Smith       ierr = PetscLogNestedBegin();CHKERRQ(ierr);
534bb1d7374SBarry Smith     } else {
535bb1d7374SBarry Smith       ierr = PetscLogDefaultBegin();CHKERRQ(ierr);
536bb1d7374SBarry Smith     }
537e3ed9ee7SBarry Smith     PetscLogMemory = PETSC_FALSE;
538e3ed9ee7SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&PetscLogMemory,NULL);CHKERRQ(ierr);
539e3ed9ee7SBarry Smith     if (PetscLogMemory) {
540e3ed9ee7SBarry Smith       ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
541e3ed9ee7SBarry Smith     }
542eccbb886SLisandro Dalcin   }
543eccbb886SLisandro Dalcin   if (flg4 && format == PETSC_VIEWER_ASCII_XML) {
544eccbb886SLisandro Dalcin     PetscReal threshold = PetscRealConstant(0.01);
545eccbb886SLisandro Dalcin     ierr = PetscOptionsGetReal(NULL,NULL,"-log_threshold",&threshold,&flg1);CHKERRQ(ierr);
546eccbb886SLisandro Dalcin     if (flg1) {ierr = PetscLogSetThreshold((PetscLogDouble)threshold,NULL);CHKERRQ(ierr);}
547bb1d7374SBarry Smith   }
548e5c89e4eSSatish Balay #endif
549e5c89e4eSSatish Balay 
550c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-saws_options",&PetscOptionsPublish,NULL);CHKERRQ(ierr);
551e5c89e4eSSatish Balay 
5524b0a73b2SBarry Smith #if defined(PETSC_HAVE_CUDA)
553c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-cuda_show_devices",&flg1);CHKERRQ(ierr);
5544b0a73b2SBarry Smith   if (flg1) {
5554b0a73b2SBarry Smith     struct cudaDeviceProp prop;
5564b0a73b2SBarry Smith     int                   devCount;
55716b6f2efSBarry Smith     PetscInt              device;
5582692e278SPaul Mullowney     cudaError_t           err = cudaSuccess;
559e09ad801SBarry Smith 
5602692e278SPaul Mullowney     err = cudaGetDeviceCount(&devCount);
561e09ad801SBarry Smith     if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err));
5624b0a73b2SBarry Smith     for (device = 0; device < devCount; ++device) {
56316b6f2efSBarry Smith       err = cudaGetDeviceProperties(&prop, (int)device);
564e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceProperties %s",cudaGetErrorString(err));
56542218b76SBarry Smith       ierr = PetscPrintf(comm, "CUDA device %D: %s\n", device, prop.name);CHKERRQ(ierr);
5664b0a73b2SBarry Smith     }
5674b0a73b2SBarry Smith   }
568bffec52eSKarl Rupp   if (!PetscCUDAInitialized) {
56916b6f2efSBarry Smith     PetscMPIInt size;
57042218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
571bd72cc96SPaul Mullowney     if (size>1) {
57216b6f2efSBarry Smith       int         devCount;
57316b6f2efSBarry Smith       PetscInt    device;
57416b6f2efSBarry Smith       PetscMPIInt rank;
5752692e278SPaul Mullowney       cudaError_t err = cudaSuccess;
5762692e278SPaul Mullowney 
5772692e278SPaul Mullowney       /* check to see if we force multiple ranks to hit the same GPU */
578c5929fdfSBarry Smith       ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr);
5792692e278SPaul Mullowney       if (flg1) {
58016b6f2efSBarry Smith         err = cudaSetDevice((int)device);
581e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
5822692e278SPaul Mullowney       } else {
583e09ad801SBarry Smith         /* we're not using the same GPU on multiple MPI threads. So try to allocated different   GPUs to different processes */
5842692e278SPaul Mullowney 
5852692e278SPaul Mullowney         /* First get the device count */
586e09ad801SBarry Smith         err   = cudaGetDeviceCount(&devCount);
587e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err));
5882692e278SPaul Mullowney 
5892692e278SPaul Mullowney         /* next determine the rank and then set the device via a mod */
59042218b76SBarry Smith         ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
591bd72cc96SPaul Mullowney         device = rank % devCount;
59216b6f2efSBarry Smith         err    = cudaSetDevice((int)device);
593e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
5942692e278SPaul Mullowney       }
5952692e278SPaul Mullowney 
5962692e278SPaul Mullowney       /* set the device flags so that it can map host memory ... do NOT throw exception on err!=cudaSuccess
5972692e278SPaul Mullowney        multiple devices may try to set the flags on the same device. So long as one of them succeeds, things
5982692e278SPaul Mullowney        are ok. */
5992692e278SPaul Mullowney       err = cudaSetDeviceFlags(cudaDeviceMapHost);
600e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err));
601a297a907SKarl Rupp     } else {
60216b6f2efSBarry Smith       PetscInt    device;
6032692e278SPaul Mullowney       cudaError_t err = cudaSuccess;
6042692e278SPaul Mullowney 
605f297a14cSJed Brown       /* the code below works for serial GPU simulations */
606c5929fdfSBarry Smith       ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr);
6074b0a73b2SBarry Smith       if (flg1) {
60816b6f2efSBarry Smith         err = cudaSetDevice((int)device);
609e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
6104b0a73b2SBarry Smith       }
6112692e278SPaul Mullowney 
6122692e278SPaul Mullowney       /* set the device flags so that it can map host memory ... here, we error check. */
6132692e278SPaul Mullowney       err = cudaSetDeviceFlags(cudaDeviceMapHost);
614e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err));
6154b0a73b2SBarry Smith     }
616bffec52eSKarl Rupp 
617bffec52eSKarl Rupp     PetscCUDAInitialized = PETSC_TRUE;
618bd72cc96SPaul Mullowney   }
6194b0a73b2SBarry Smith #endif
6204b0a73b2SBarry Smith 
6214b0a73b2SBarry Smith 
622e5c89e4eSSatish Balay   /*
623e5c89e4eSSatish Balay        Print basic help message
624e5c89e4eSSatish Balay   */
6252d747510SLisandro Dalcin   ierr = PetscOptionsHasHelp(NULL,&flg1);CHKERRQ(ierr);
626e5c89e4eSSatish Balay   if (flg1) {
627e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr);
628301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -help: prints help method for each option\n");CHKERRQ(ierr);
629301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr);
630301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm,"       only when run in the debugger\n");CHKERRQ(ierr);
631e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
632e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start the debugger in new xterm\n");CHKERRQ(ierr);
633e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       unless noxterm is given\n");CHKERRQ(ierr);
634e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
635e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start all processes in the debugger\n");CHKERRQ(ierr);
636e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr);
637e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"    emacs jumps to error file\n");CHKERRQ(ierr);
638e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_nodes [n1,n2,..] Nodes to start in debugger\n");CHKERRQ(ierr);
639e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr);
640e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr);
641e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"                      waits the delay for you to attach\n");CHKERRQ(ierr);
6421cda70a7SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -display display: Location where X window graphics and debuggers are displayed\n");CHKERRQ(ierr);
643e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr);
644e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr);
645e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr);
646e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"           note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr);
647e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr);
648e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc: use our error checking malloc\n");CHKERRQ(ierr);
649e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use error checking malloc\n");CHKERRQ(ierr);
6504161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr);
6514161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_log: keeps log of all memory allocations\n");CHKERRQ(ierr);
652e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_debug: enables extended checking for memory corruption\n");CHKERRQ(ierr);
65326a7e8d4SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -options_view: dump list of options inputted\n");CHKERRQ(ierr);
654e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr);
655e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr);
656e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr);
657e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr);
658a8c7a070SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr);
6590841954dSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -memory_view: print memory usage at end of run\n");CHKERRQ(ierr);
660e5c89e4eSSatish Balay #if defined(PETSC_USE_LOG)
661e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr);
662185ae32fSMatthew G. Knepley     ierr = (*PetscHelpPrintf)(comm," -log_view [:filename:[format]]: logging objects and events\n");CHKERRQ(ierr);
663e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr);
664e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
665495fc317SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through Jumpshot\n");CHKERRQ(ierr);
666e5c89e4eSSatish Balay #endif
6676cf91177SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -info <optional filename>: print informative messages about the calculations\n");CHKERRQ(ierr);
668e5c89e4eSSatish Balay #endif
669e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -v: prints PETSc version number and release date\n");CHKERRQ(ierr);
670e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr);
671e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr);
672e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"-----------------------------------------------\n");CHKERRQ(ierr);
673e5c89e4eSSatish Balay   }
674e5c89e4eSSatish Balay 
67574ba8654SBarry Smith #if defined(PETSC_HAVE_POPEN)
67674ba8654SBarry Smith   {
67774ba8654SBarry Smith   char machine[128];
678c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-popen_machine",machine,128,&flg1);CHKERRQ(ierr);
67974ba8654SBarry Smith   if (flg1) {
68074ba8654SBarry Smith     ierr = PetscPOpenSetMachine(machine);CHKERRQ(ierr);
68174ba8654SBarry Smith   }
68274ba8654SBarry Smith   }
68374ba8654SBarry Smith #endif
68474ba8654SBarry Smith 
685c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr);
686e5c89e4eSSatish Balay   if (flg1) {
687e5c89e4eSSatish Balay     ierr = PetscSleep(si);CHKERRQ(ierr);
688e5c89e4eSSatish Balay   }
689e5c89e4eSSatish Balay 
690c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr);
6912ea1be5dSBarry Smith   if (flg1) {
692e5c89e4eSSatish Balay     ierr = PetscStrstr(mname,"null",&f);CHKERRQ(ierr);
693e5c89e4eSSatish Balay     if (f) {
694ecd8bba6SJed Brown       ierr = PetscInfoDeactivateClass(0);CHKERRQ(ierr);
695e5c89e4eSSatish Balay     }
6962ea1be5dSBarry Smith   }
697827f890bSBarry Smith 
698fd314934SBarry Smith #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
699c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
700f14045dbSBarry Smith   if (!flg3) {
701c5929fdfSBarry Smith     ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&flg3);CHKERRQ(ierr);
702f14045dbSBarry Smith   }
703bab1f7e6SVictor Minden #endif
704e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
705c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-viennacl_synchronize",&flg3,NULL);CHKERRQ(ierr);
706f14045dbSBarry Smith   PetscViennaCLSynchronize = flg3;
707b8ced49eSKarl Rupp #endif
708fd314934SBarry Smith #if defined(PETSC_HAVE_CUDA)
70982f73ecaSAlejandro Lamas Daviña   ierr = PetscOptionsGetBool(NULL,NULL,"-cuda_synchronize",&flg3,NULL);CHKERRQ(ierr);
71082f73ecaSAlejandro Lamas Daviña   PetscCUDASynchronize = flg3;
7114cf1874eSKarl Rupp #endif
71282f73ecaSAlejandro Lamas Daviña 
713f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
714f0a7718cSKarl Rupp   ierr = PetscViennaCLInit();CHKERRQ(ierr);
715f0a7718cSKarl Rupp #endif
716f0a7718cSKarl Rupp 
717e5c89e4eSSatish Balay   PetscFunctionReturn(0);
718e5c89e4eSSatish Balay }
719