xref: /petsc/src/sys/objects/init.c (revision 95c0884e6f7665b705eebf88174e89dc920c2fc0)
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;
8075567043SBarry Smith 
81e5c89e4eSSatish Balay /*
82e5c89e4eSSatish Balay        Function that is called to display all error messages
83e5c89e4eSSatish Balay */
847087cfbeSBarry Smith PetscErrorCode (*PetscErrorPrintf)(const char [],...)          = PetscErrorPrintfDefault;
857087cfbeSBarry Smith PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...)  = PetscHelpPrintfDefault;
867087cfbeSBarry Smith PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list)    = PetscVFPrintfDefault;
87bab1f7e6SVictor Minden /*
884cf1874eSKarl Rupp   This is needed to turn on/off GPU synchronization
898b5db460SBarry Smith */
904cf1874eSKarl Rupp PetscBool PetscViennaCLSynchronize = PETSC_FALSE;
9182f73ecaSAlejandro Lamas Daviña PetscBool PetscCUDASynchronize = PETSC_FALSE;
92bab1f7e6SVictor Minden 
93e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
94e5c89e4eSSatish Balay /*
95e5c89e4eSSatish Balay    Optional file where all PETSc output from various prints is saved
96e5c89e4eSSatish Balay */
97*95c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history;
980298fd71SBarry Smith FILE *petsc_history = NULL;
99e5c89e4eSSatish Balay 
1007087cfbeSBarry Smith PetscErrorCode  PetscOpenHistoryFile(const char filename[],FILE **fd)
101e5c89e4eSSatish Balay {
102e5c89e4eSSatish Balay   PetscErrorCode ierr;
103e5c89e4eSSatish Balay   PetscMPIInt    rank,size;
104e5c89e4eSSatish Balay   char           pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64];
105e5c89e4eSSatish Balay   char           version[256];
106e5c89e4eSSatish Balay 
107e5c89e4eSSatish Balay   PetscFunctionBegin;
108e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
109e5c89e4eSSatish Balay   if (!rank) {
110e5c89e4eSSatish Balay     char        arch[10];
111f56c2debSBarry Smith     int         err;
112f56c2debSBarry Smith 
113e5c89e4eSSatish Balay     ierr = PetscGetArchType(arch,10);CHKERRQ(ierr);
114e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
115a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
116e5c89e4eSSatish Balay     ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
117e5c89e4eSSatish Balay     if (filename) {
118e5c89e4eSSatish Balay       ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr);
119e5c89e4eSSatish Balay     } else {
120e5c89e4eSSatish Balay       ierr = PetscGetHomeDirectory(pfile,240);CHKERRQ(ierr);
121e5c89e4eSSatish Balay       ierr = PetscStrcat(pfile,"/.petschistory");CHKERRQ(ierr);
122e5c89e4eSSatish Balay       ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr);
123e5c89e4eSSatish Balay     }
124e5c89e4eSSatish Balay 
125a297a907SKarl Rupp     *fd = fopen(fname,"a");
126a297a907SKarl Rupp     if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname);
127a297a907SKarl Rupp 
128e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
129e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr);
130e5c89e4eSSatish Balay     ierr = PetscGetProgramName(pname,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
131e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr);
132e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
133a297a907SKarl Rupp 
134f56c2debSBarry Smith     err = fflush(*fd);
135e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
136e5c89e4eSSatish Balay   }
137e5c89e4eSSatish Balay   PetscFunctionReturn(0);
138e5c89e4eSSatish Balay }
139e5c89e4eSSatish Balay 
140*95c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscCloseHistoryFile(FILE **fd)
141e5c89e4eSSatish Balay {
142e5c89e4eSSatish Balay   PetscErrorCode ierr;
143e5c89e4eSSatish Balay   PetscMPIInt    rank;
144e5c89e4eSSatish Balay   char           date[64];
145f56c2debSBarry Smith   int            err;
146e5c89e4eSSatish Balay 
147e5c89e4eSSatish Balay   PetscFunctionBegin;
148e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
149e5c89e4eSSatish Balay   if (!rank) {
150e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
151e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
152e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr);
153e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
154f56c2debSBarry Smith     err  = fflush(*fd);
155e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
156f56c2debSBarry Smith     err = fclose(*fd);
157e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
158e5c89e4eSSatish Balay   }
159e5c89e4eSSatish Balay   PetscFunctionReturn(0);
160e5c89e4eSSatish Balay }
161e5c89e4eSSatish Balay 
162e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
163e5c89e4eSSatish Balay 
164e5c89e4eSSatish Balay /*
165e5c89e4eSSatish Balay    This is ugly and probably belongs somewhere else, but I want to
166e5c89e4eSSatish Balay   be able to put a true MPI abort error handler with command line args.
167e5c89e4eSSatish Balay 
168e5c89e4eSSatish Balay     This is so MPI errors in the debugger will leave all the stack
1693c311c98SBarry Smith   frames. The default MP_Abort() cleans up and exits thus providing no useful information
1703c311c98SBarry Smith   in the debugger hence we call abort() instead of MPI_Abort().
171e5c89e4eSSatish Balay */
172e5c89e4eSSatish Balay 
173e5c89e4eSSatish Balay void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag)
174e5c89e4eSSatish Balay {
175e5c89e4eSSatish Balay   PetscFunctionBegin;
1763c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
177e5c89e4eSSatish Balay   abort();
178e5c89e4eSSatish Balay }
179e5c89e4eSSatish Balay 
180e5c89e4eSSatish Balay void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag)
181e5c89e4eSSatish Balay {
182e5c89e4eSSatish Balay   PetscErrorCode ierr;
183e5c89e4eSSatish Balay 
184e5c89e4eSSatish Balay   PetscFunctionBegin;
1853c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
186e5c89e4eSSatish Balay   ierr = PetscAttachDebugger();
187a297a907SKarl Rupp   if (ierr) MPI_Abort(*comm,*flag); /* hopeless so get out */
188e5c89e4eSSatish Balay }
189e5c89e4eSSatish Balay 
190e5c89e4eSSatish Balay /*@C
191e5c89e4eSSatish Balay    PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one
192e5c89e4eSSatish Balay      wishes a clean exit somewhere deep in the program.
193e5c89e4eSSatish Balay 
194e5c89e4eSSatish Balay    Collective on PETSC_COMM_WORLD
195e5c89e4eSSatish Balay 
196e5c89e4eSSatish Balay    Options Database Keys are the same as for PetscFinalize()
197e5c89e4eSSatish Balay 
198e5c89e4eSSatish Balay    Level: advanced
199e5c89e4eSSatish Balay 
200e5c89e4eSSatish Balay    Note:
201e5c89e4eSSatish Balay    See PetscInitialize() for more general runtime options.
202e5c89e4eSSatish Balay 
20388c29154SBarry Smith .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize()
204e5c89e4eSSatish Balay @*/
2057087cfbeSBarry Smith PetscErrorCode  PetscEnd(void)
206e5c89e4eSSatish Balay {
207e5c89e4eSSatish Balay   PetscFunctionBegin;
208e5c89e4eSSatish Balay   PetscFinalize();
209e5c89e4eSSatish Balay   exit(0);
210e5c89e4eSSatish Balay   return 0;
211e5c89e4eSSatish Balay }
212e5c89e4eSSatish Balay 
213ace3abfcSBarry Smith PetscBool PetscOptionsPublish = PETSC_FALSE;
214*95c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseTrMalloc_Private(void);
215*95c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseHBWMalloc_Private(void);
216*95c0884eSLisandro Dalcin PETSC_INTERN PetscBool      petscsetmallocvisited;
217e5c89e4eSSatish Balay static       char           emacsmachinename[256];
218e5c89e4eSSatish Balay 
219e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = 0;
220e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm)    = 0;
221e5c89e4eSSatish Balay 
222e5c89e4eSSatish Balay /*@C
223e5c89e4eSSatish Balay    PetscSetHelpVersionFunctions - Sets functions that print help and version information
224e5c89e4eSSatish Balay    before the PETSc help and version information is printed. Must call BEFORE PetscInitialize().
225e5c89e4eSSatish Balay    This routine enables a "higher-level" package that uses PETSc to print its messages first.
226e5c89e4eSSatish Balay 
227e5c89e4eSSatish Balay    Input Parameter:
2280298fd71SBarry Smith +  help - the help function (may be NULL)
2290298fd71SBarry Smith -  version - the version function (may be NULL)
230e5c89e4eSSatish Balay 
231e5c89e4eSSatish Balay    Level: developer
232e5c89e4eSSatish Balay 
233e5c89e4eSSatish Balay    Concepts: package help message
234e5c89e4eSSatish Balay 
235e5c89e4eSSatish Balay @*/
2367087cfbeSBarry Smith PetscErrorCode  PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm))
237e5c89e4eSSatish Balay {
238e5c89e4eSSatish Balay   PetscFunctionBegin;
239e5c89e4eSSatish Balay   PetscExternalHelpFunction    = help;
240e5c89e4eSSatish Balay   PetscExternalVersionFunction = version;
241e5c89e4eSSatish Balay   PetscFunctionReturn(0);
242e5c89e4eSSatish Balay }
243e5c89e4eSSatish Balay 
24405df10baSBarry Smith #if defined(PETSC_USE_LOG)
245*95c0884eSLisandro Dalcin PETSC_INTERN PetscBool   PetscObjectsLog;
24605df10baSBarry Smith #endif
24705df10baSBarry Smith 
248*95c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode  PetscOptionsCheckInitial_Private(void)
249e5c89e4eSSatish Balay {
250e5c89e4eSSatish Balay   char              string[64],mname[PETSC_MAX_PATH_LEN],*f;
251e5c89e4eSSatish Balay   MPI_Comm          comm = PETSC_COMM_WORLD;
25267584ceeSBarry Smith   PetscBool         flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flag;
253e5c89e4eSSatish Balay   PetscErrorCode    ierr;
25467584ceeSBarry Smith   PetscReal         si;
25528559dc8SJed Brown   PetscInt          intensity;
256e5c89e4eSSatish Balay   int               i;
257e5c89e4eSSatish Balay   PetscMPIInt       rank;
2587ca660e7SBarry Smith   char              version[256],helpoptions[256];
25967584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
26067584ceeSBarry Smith   PetscReal         logthreshold;
2612479a3a6SBarry Smith #endif
2622479a3a6SBarry Smith #if defined(PETSC_USE_LOG)
2632479a3a6SBarry Smith   PetscViewerFormat format;
26467584ceeSBarry Smith   PetscBool         flg4 = PETSC_FALSE;
26567584ceeSBarry Smith #endif
266e5c89e4eSSatish Balay 
267e5c89e4eSSatish Balay   PetscFunctionBegin;
26842218b76SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
269e5c89e4eSSatish Balay 
27067584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
271e5c89e4eSSatish Balay   /*
272e5c89e4eSSatish Balay       Setup the memory management; support for tracing malloc() usage
273e5c89e4eSSatish Balay   */
274c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-malloc_log",&flg3);CHKERRQ(ierr);
275574034a9SJed Brown   logthreshold = 0.0;
276c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-malloc_log_threshold",&logthreshold,&flg1);CHKERRQ(ierr);
277574034a9SJed Brown   if (flg1) flg3 = PETSC_TRUE;
2788bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG)
279c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg1,&flg2);CHKERRQ(ierr);
280e5c89e4eSSatish Balay   if ((!flg2 || flg1) && !petscsetmallocvisited) {
281dc92acbaSJed Brown     if (flg2 || !(PETSC_RUNNING_ON_VALGRIND)) {
282555d055bSBarry Smith       /* turn off default -malloc if valgrind is being used */
283e5c89e4eSSatish Balay       ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
284555d055bSBarry Smith     }
285e5c89e4eSSatish Balay   }
286e5c89e4eSSatish Balay #else
287c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_dump",&flg1,NULL);CHKERRQ(ierr);
288c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg2,NULL);CHKERRQ(ierr);
289e5c89e4eSSatish Balay   if (flg1 || flg2 || flg3) {ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);}
290e5c89e4eSSatish Balay #endif
291e5c89e4eSSatish Balay   if (flg3) {
292574034a9SJed Brown     ierr = PetscMallocSetDumpLogThreshold((PetscLogDouble)logthreshold);CHKERRQ(ierr);
293e5c89e4eSSatish Balay   }
294ba282f50SJed Brown   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_coalesce",&flg1,&flg2);CHKERRQ(ierr);
295ba282f50SJed Brown   if (flg2) {ierr = PetscMallocSetCoalesce(flg1);CHKERRQ(ierr);}
29690d69ab7SBarry Smith   flg1 = PETSC_FALSE;
297c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_debug",&flg1,NULL);CHKERRQ(ierr);
298e5c89e4eSSatish Balay   if (flg1) {
299e5c89e4eSSatish Balay     ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
300e5c89e4eSSatish Balay     ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr);
301e5c89e4eSSatish Balay   }
302dc92acbaSJed Brown   flg1 = PETSC_FALSE;
303c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_test",&flg1,NULL);CHKERRQ(ierr);
3048bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG)
305dc92acbaSJed Brown   if (flg1 && !PETSC_RUNNING_ON_VALGRIND) {
306dc92acbaSJed Brown     ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
307dc92acbaSJed Brown     ierr = PetscMallocSetDumpLog();CHKERRQ(ierr);
308dc92acbaSJed Brown     ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr);
309dc92acbaSJed Brown   }
310dc92acbaSJed Brown #endif
3112f21b5c6SHong Zhang   flg1 = PETSC_FALSE;
3122f21b5c6SHong Zhang   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_hbw",&flg1,NULL);CHKERRQ(ierr);
313cf6f3811SHong Zhang   /* ignore this option if malloc is already set */
314cf6f3811SHong Zhang   if (flg1 && !petscsetmallocvisited) {ierr = PetscSetUseHBWMalloc_Private();CHKERRQ(ierr);}
315e5c89e4eSSatish Balay 
31690d69ab7SBarry Smith   flg1 = PETSC_FALSE;
317c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_info",&flg1,NULL);CHKERRQ(ierr);
3187783f70dSSatish Balay   if (!flg1) {
31990d69ab7SBarry Smith     flg1 = PETSC_FALSE;
320c5929fdfSBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-memory_view",&flg1,NULL);CHKERRQ(ierr);
3217783f70dSSatish Balay   }
322e5c89e4eSSatish Balay   if (flg1) {
323e5c89e4eSSatish Balay     ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr);
324e5c89e4eSSatish Balay   }
32567584ceeSBarry Smith #endif
326e5c89e4eSSatish Balay 
327b4427426SBarry Smith #if defined(PETSC_USE_LOG)
328c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-objects_dump",&PetscObjectsLog);CHKERRQ(ierr);
329b4427426SBarry Smith #endif
33005df10baSBarry Smith 
331e5c89e4eSSatish Balay   /*
332e5c89e4eSSatish Balay       Set the display variable for graphics
333e5c89e4eSSatish Balay   */
334e5c89e4eSSatish Balay   ierr = PetscSetDisplay();CHKERRQ(ierr);
335e5c89e4eSSatish Balay 
33651dcc849SKerry Stevens   /*
337e5c89e4eSSatish Balay       Print the PETSc version information
338e5c89e4eSSatish Balay   */
339c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-v",&flg1);CHKERRQ(ierr);
340c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-version",&flg2);CHKERRQ(ierr);
341c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-help",&flg3);CHKERRQ(ierr);
342e5c89e4eSSatish Balay   if (flg1 || flg2 || flg3) {
343e5c89e4eSSatish Balay 
344e5c89e4eSSatish Balay     /*
345e5c89e4eSSatish Balay        Print "higher-level" package version message
346e5c89e4eSSatish Balay     */
347e5c89e4eSSatish Balay     if (PetscExternalVersionFunction) {
348e5c89e4eSSatish Balay       ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr);
349e5c89e4eSSatish Balay     }
350e5c89e4eSSatish Balay 
351a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
3527ca660e7SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr);
353e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr);
354e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr);
355e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr);
35684e42920SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr);
357e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr);
358e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr);
3597ca660e7SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr);
3607ca660e7SBarry Smith   }
3617ca660e7SBarry Smith 
36294941ca7SBarry Smith   /*
36394941ca7SBarry Smith        Print "higher-level" package help message
36494941ca7SBarry Smith   */
36594941ca7SBarry Smith   if (flg3) {
36694941ca7SBarry Smith     if (PetscExternalHelpFunction) {
36794941ca7SBarry Smith       ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr);
36894941ca7SBarry Smith     }
36994941ca7SBarry Smith   }
37094941ca7SBarry Smith 
3717ca660e7SBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-help",helpoptions,sizeof(helpoptions),&flg1);CHKERRQ(ierr);
3727ca660e7SBarry Smith   if (flg1) {
3737ca660e7SBarry Smith     ierr = PetscStrcmp(helpoptions,"intro",&flg2);CHKERRQ(ierr);
3747ca660e7SBarry Smith     if (flg2) {
375bdb346e9SBarry Smith       ierr = PetscOptionsDestroyDefault();CHKERRQ(ierr);
376008a6e76SBarry Smith       ierr = PetscFreeMPIResources();CHKERRQ(ierr);
3777ca660e7SBarry Smith       ierr = MPI_Finalize();CHKERRQ(ierr);
3787ca660e7SBarry Smith       exit(0);
3797ca660e7SBarry Smith     }
380e5c89e4eSSatish Balay   }
381e5c89e4eSSatish Balay 
382e5c89e4eSSatish Balay   /*
383e5c89e4eSSatish Balay       Setup the error handling
384e5c89e4eSSatish Balay   */
38590d69ab7SBarry Smith   flg1 = PETSC_FALSE;
386c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_abort",&flg1,NULL);CHKERRQ(ierr);
387b59baad1SJed Brown   if (flg1) {
38842218b76SBarry Smith     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr);
389b59baad1SJed Brown     ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr);
390b59baad1SJed Brown   }
39190d69ab7SBarry Smith   flg1 = PETSC_FALSE;
392c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_mpiabort",&flg1,NULL);CHKERRQ(ierr);
393cb9801acSJed Brown   if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,0);CHKERRQ(ierr);}
39490d69ab7SBarry Smith   flg1 = PETSC_FALSE;
395c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_on_error",&flg1,NULL);CHKERRQ(ierr);
396e5c89e4eSSatish Balay   if (flg1) {
397d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr);
398e5c89e4eSSatish Balay   }
39990d69ab7SBarry Smith   flg1 = PETSC_FALSE;
400c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-no_signal_handler",&flg1,NULL);CHKERRQ(ierr);
4018d359177SBarry Smith   if (!flg1) {ierr = PetscPushSignalHandler(PetscSignalHandlerDefault,(void*)0);CHKERRQ(ierr);}
40296cc47afSJed Brown   flg1 = PETSC_FALSE;
403c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-fp_trap",&flg1,NULL);CHKERRQ(ierr);
40496cc47afSJed Brown   if (flg1) {ierr = PetscSetFPTrap(PETSC_FP_TRAP_ON);CHKERRQ(ierr);}
405c5929fdfSBarry Smith   ierr = PetscOptionsGetInt(NULL,NULL,"-check_pointer_intensity",&intensity,&flag);CHKERRQ(ierr);
40628559dc8SJed Brown   if (flag) {ierr = PetscCheckPointerSetIntensity(intensity);CHKERRQ(ierr);}
407e5c89e4eSSatish Balay 
408e5c89e4eSSatish Balay   /*
409e5c89e4eSSatish Balay       Setup debugger information
410e5c89e4eSSatish Balay   */
411e5c89e4eSSatish Balay   ierr = PetscSetDefaultDebugger();CHKERRQ(ierr);
412c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_attach_debugger",string,64,&flg1);CHKERRQ(ierr);
413e5c89e4eSSatish Balay   if (flg1) {
414e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
415e5c89e4eSSatish Balay 
416e5c89e4eSSatish Balay     ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
417d54338ecSKarl Rupp     ierr = MPI_Comm_create_errhandler((MPI_Handler_function*)Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr);
418d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
419e5c89e4eSSatish Balay     ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,0);CHKERRQ(ierr);
420e5c89e4eSSatish Balay   }
421c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-debug_terminal",string,64,&flg1);CHKERRQ(ierr);
4225e96ac45SJed Brown   if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); }
423c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-start_in_debugger",string,64,&flg1);CHKERRQ(ierr);
424c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-stop_for_debugger",string,64,&flg2);CHKERRQ(ierr);
425e5c89e4eSSatish Balay   if (flg1 || flg2) {
426e5c89e4eSSatish Balay     PetscMPIInt    size;
427e5c89e4eSSatish Balay     PetscInt       lsize,*nodes;
428e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
429e5c89e4eSSatish Balay     /*
430e5c89e4eSSatish Balay        we have to make sure that all processors have opened
431e5c89e4eSSatish Balay        connections to all other processors, otherwise once the
432e5c89e4eSSatish Balay        debugger has stated it is likely to receive a SIGUSR1
433e5c89e4eSSatish Balay        and kill the program.
434e5c89e4eSSatish Balay     */
43542218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
436e5c89e4eSSatish Balay     if (size > 2) {
437533163c2SBarry Smith       PetscMPIInt dummy = 0;
438e5c89e4eSSatish Balay       MPI_Status  status;
439e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
440e5c89e4eSSatish Balay         if (rank != i) {
44142218b76SBarry Smith           ierr = MPI_Send(&dummy,1,MPI_INT,i,109,comm);CHKERRQ(ierr);
442e5c89e4eSSatish Balay         }
443e5c89e4eSSatish Balay       }
444e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
445e5c89e4eSSatish Balay         if (rank != i) {
44642218b76SBarry Smith           ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,comm,&status);CHKERRQ(ierr);
447e5c89e4eSSatish Balay         }
448e5c89e4eSSatish Balay       }
449e5c89e4eSSatish Balay     }
450e5c89e4eSSatish Balay     /* check if this processor node should be in debugger */
451785e854fSJed Brown     ierr  = PetscMalloc1(size,&nodes);CHKERRQ(ierr);
452e5c89e4eSSatish Balay     lsize = size;
453c5929fdfSBarry Smith     ierr  = PetscOptionsGetIntArray(NULL,NULL,"-debugger_nodes",nodes,&lsize,&flag);CHKERRQ(ierr);
454e5c89e4eSSatish Balay     if (flag) {
455e5c89e4eSSatish Balay       for (i=0; i<lsize; i++) {
456e5c89e4eSSatish Balay         if (nodes[i] == rank) { flag = PETSC_FALSE; break; }
457e5c89e4eSSatish Balay       }
458e5c89e4eSSatish Balay     }
459e5c89e4eSSatish Balay     if (!flag) {
460e5c89e4eSSatish Balay       ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
461e5c89e4eSSatish Balay       ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr);
462e5c89e4eSSatish Balay       if (flg1) {
463e5c89e4eSSatish Balay         ierr = PetscAttachDebugger();CHKERRQ(ierr);
464e5c89e4eSSatish Balay       } else {
465e5c89e4eSSatish Balay         ierr = PetscStopForDebugger();CHKERRQ(ierr);
466e5c89e4eSSatish Balay       }
467d54338ecSKarl Rupp       ierr = MPI_Comm_create_errhandler((MPI_Handler_function*)Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr);
468d54338ecSKarl Rupp       ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
469e5c89e4eSSatish Balay     }
470e5c89e4eSSatish Balay     ierr = PetscFree(nodes);CHKERRQ(ierr);
471e5c89e4eSSatish Balay   }
472e5c89e4eSSatish Balay 
473c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_emacs",emacsmachinename,128,&flg1);CHKERRQ(ierr);
474cb9801acSJed Brown   if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);}
475e5c89e4eSSatish Balay 
476e5c89e4eSSatish Balay   /*
477e5c89e4eSSatish Balay         Setup profiling and logging
478e5c89e4eSSatish Balay   */
4796cf91177SBarry Smith #if defined(PETSC_USE_INFO)
4808bb29257SSatish Balay   {
481e5c89e4eSSatish Balay     char logname[PETSC_MAX_PATH_LEN]; logname[0] = 0;
482c5929fdfSBarry Smith     ierr = PetscOptionsGetString(NULL,NULL,"-info",logname,250,&flg1);CHKERRQ(ierr);
4838bb29257SSatish Balay     if (flg1 && logname[0]) {
484fcc2139eSBarry Smith       ierr = PetscInfoAllow(PETSC_TRUE,logname);CHKERRQ(ierr);
4858bb29257SSatish Balay     } else if (flg1) {
4860298fd71SBarry Smith       ierr = PetscInfoAllow(PETSC_TRUE,NULL);CHKERRQ(ierr);
487e5c89e4eSSatish Balay     }
488e5c89e4eSSatish Balay   }
489865f6aa8SSatish Balay #endif
490865f6aa8SSatish Balay #if defined(PETSC_USE_LOG)
491865f6aa8SSatish Balay   mname[0] = 0;
492c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-history",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr);
493865f6aa8SSatish Balay   if (flg1) {
494865f6aa8SSatish Balay     if (mname[0]) {
495f3dea69dSBarry Smith       ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr);
496865f6aa8SSatish Balay     } else {
497706d7a88SBarry Smith       ierr = PetscOpenHistoryFile(NULL,&petsc_history);CHKERRQ(ierr);
498865f6aa8SSatish Balay     }
499865f6aa8SSatish Balay   }
500217044c2SLisandro Dalcin 
501217044c2SLisandro Dalcin   ierr = PetscOptionsGetBool(NULL,NULL,"-log_sync",&PetscLogSyncOn,NULL);CHKERRQ(ierr);
502217044c2SLisandro Dalcin 
503e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
50490d69ab7SBarry Smith   flg1 = PETSC_FALSE;
505c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_mpe",&flg1);CHKERRQ(ierr);
506495fc317SBarry Smith   if (flg1) {ierr = PetscLogMPEBegin();CHKERRQ(ierr);}
507e5c89e4eSSatish Balay #endif
50890d69ab7SBarry Smith   flg1 = PETSC_FALSE;
50990d69ab7SBarry Smith   flg3 = PETSC_FALSE;
510c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-log_all",&flg1,NULL);CHKERRQ(ierr);
511c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
512e5c89e4eSSatish Balay   if (flg1)                      { ierr = PetscLogAllBegin();CHKERRQ(ierr); }
513bb1d7374SBarry Smith   else if (flg3)                 { ierr = PetscLogDefaultBegin();CHKERRQ(ierr);}
514e5c89e4eSSatish Balay 
515c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-log_trace",mname,250,&flg1);CHKERRQ(ierr);
516e5c89e4eSSatish Balay   if (flg1) {
517e5c89e4eSSatish Balay     char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN];
518e5c89e4eSSatish Balay     FILE *file;
519e5c89e4eSSatish Balay     if (mname[0]) {
5202e924ca5SSatish Balay       PetscSNPrintf(name,PETSC_MAX_PATH_LEN,"%s.%d",mname,rank);
521e5c89e4eSSatish Balay       ierr = PetscFixFilename(name,fname);CHKERRQ(ierr);
522e5c89e4eSSatish Balay       file = fopen(fname,"w");
523f3dea69dSBarry Smith       if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname);
524a297a907SKarl Rupp     } else file = PETSC_STDOUT;
525e5c89e4eSSatish Balay     ierr = PetscLogTraceBegin(file);CHKERRQ(ierr);
526e5c89e4eSSatish Balay   }
527bb1d7374SBarry Smith 
52842218b76SBarry Smith   ierr = PetscOptionsGetViewer(comm,NULL,"-log_view",NULL,&format,&flg4);CHKERRQ(ierr);
529bb1d7374SBarry Smith   if (flg4) {
530bb1d7374SBarry Smith     if (format == PETSC_VIEWER_ASCII_XML) {
531bb1d7374SBarry Smith       ierr = PetscLogNestedBegin();CHKERRQ(ierr);
532bb1d7374SBarry Smith     } else {
533bb1d7374SBarry Smith       ierr = PetscLogDefaultBegin();CHKERRQ(ierr);
534bb1d7374SBarry Smith     }
535eccbb886SLisandro Dalcin   }
536eccbb886SLisandro Dalcin   if (flg4 && format == PETSC_VIEWER_ASCII_XML) {
537eccbb886SLisandro Dalcin     PetscReal threshold = PetscRealConstant(0.01);
538eccbb886SLisandro Dalcin     ierr = PetscOptionsGetReal(NULL,NULL,"-log_threshold",&threshold,&flg1);CHKERRQ(ierr);
539eccbb886SLisandro Dalcin     if (flg1) {ierr = PetscLogSetThreshold((PetscLogDouble)threshold,NULL);CHKERRQ(ierr);}
540bb1d7374SBarry Smith   }
541e5c89e4eSSatish Balay #endif
542e5c89e4eSSatish Balay 
543c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-saws_options",&PetscOptionsPublish,NULL);CHKERRQ(ierr);
544e5c89e4eSSatish Balay 
5454b0a73b2SBarry Smith #if defined(PETSC_HAVE_CUDA)
546c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-cuda_show_devices",&flg1);CHKERRQ(ierr);
5474b0a73b2SBarry Smith   if (flg1) {
5484b0a73b2SBarry Smith     struct cudaDeviceProp prop;
5494b0a73b2SBarry Smith     int                   devCount;
55016b6f2efSBarry Smith     PetscInt              device;
5512692e278SPaul Mullowney     cudaError_t           err = cudaSuccess;
552e09ad801SBarry Smith 
5532692e278SPaul Mullowney     err = cudaGetDeviceCount(&devCount);
554e09ad801SBarry Smith     if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err));
5554b0a73b2SBarry Smith     for (device = 0; device < devCount; ++device) {
55616b6f2efSBarry Smith       err = cudaGetDeviceProperties(&prop, (int)device);
557e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceProperties %s",cudaGetErrorString(err));
55842218b76SBarry Smith       ierr = PetscPrintf(comm, "CUDA device %D: %s\n", device, prop.name);CHKERRQ(ierr);
5594b0a73b2SBarry Smith     }
5604b0a73b2SBarry Smith   }
561bffec52eSKarl Rupp   if (!PetscCUDAInitialized) {
56216b6f2efSBarry Smith     PetscMPIInt size;
56342218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
564bd72cc96SPaul Mullowney     if (size>1) {
56516b6f2efSBarry Smith       int         devCount;
56616b6f2efSBarry Smith       PetscInt    device;
56716b6f2efSBarry Smith       PetscMPIInt rank;
5682692e278SPaul Mullowney       cudaError_t err = cudaSuccess;
5692692e278SPaul Mullowney 
5702692e278SPaul Mullowney       /* check to see if we force multiple ranks to hit the same GPU */
571c5929fdfSBarry Smith       ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr);
5722692e278SPaul Mullowney       if (flg1) {
57316b6f2efSBarry Smith         err = cudaSetDevice((int)device);
574e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
5752692e278SPaul Mullowney       } else {
576e09ad801SBarry Smith         /* we're not using the same GPU on multiple MPI threads. So try to allocated different   GPUs to different processes */
5772692e278SPaul Mullowney 
5782692e278SPaul Mullowney         /* First get the device count */
579e09ad801SBarry Smith         err   = cudaGetDeviceCount(&devCount);
580e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err));
5812692e278SPaul Mullowney 
5822692e278SPaul Mullowney         /* next determine the rank and then set the device via a mod */
58342218b76SBarry Smith         ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
584bd72cc96SPaul Mullowney         device = rank % devCount;
58516b6f2efSBarry Smith         err    = cudaSetDevice((int)device);
586e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
5872692e278SPaul Mullowney       }
5882692e278SPaul Mullowney 
5892692e278SPaul Mullowney       /* set the device flags so that it can map host memory ... do NOT throw exception on err!=cudaSuccess
5902692e278SPaul Mullowney        multiple devices may try to set the flags on the same device. So long as one of them succeeds, things
5912692e278SPaul Mullowney        are ok. */
5922692e278SPaul Mullowney       err = cudaSetDeviceFlags(cudaDeviceMapHost);
593e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err));
594a297a907SKarl Rupp     } else {
59516b6f2efSBarry Smith       PetscInt    device;
5962692e278SPaul Mullowney       cudaError_t err = cudaSuccess;
5972692e278SPaul Mullowney 
598f297a14cSJed Brown       /* the code below works for serial GPU simulations */
599c5929fdfSBarry Smith       ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr);
6004b0a73b2SBarry Smith       if (flg1) {
60116b6f2efSBarry Smith         err = cudaSetDevice((int)device);
602e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
6034b0a73b2SBarry Smith       }
6042692e278SPaul Mullowney 
6052692e278SPaul Mullowney       /* set the device flags so that it can map host memory ... here, we error check. */
6062692e278SPaul Mullowney       err = cudaSetDeviceFlags(cudaDeviceMapHost);
607e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err));
6084b0a73b2SBarry Smith     }
609bffec52eSKarl Rupp 
610bffec52eSKarl Rupp     PetscCUDAInitialized = PETSC_TRUE;
611bd72cc96SPaul Mullowney   }
6124b0a73b2SBarry Smith #endif
6134b0a73b2SBarry Smith 
6144b0a73b2SBarry Smith 
615e5c89e4eSSatish Balay   /*
616e5c89e4eSSatish Balay        Print basic help message
617e5c89e4eSSatish Balay   */
618c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-help",&flg1);CHKERRQ(ierr);
619e5c89e4eSSatish Balay   if (flg1) {
620e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr);
621301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -help: prints help method for each option\n");CHKERRQ(ierr);
622301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr);
623301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm,"       only when run in the debugger\n");CHKERRQ(ierr);
624e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
625e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start the debugger in new xterm\n");CHKERRQ(ierr);
626e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       unless noxterm is given\n");CHKERRQ(ierr);
627e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
628e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start all processes in the debugger\n");CHKERRQ(ierr);
629e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr);
630e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"    emacs jumps to error file\n");CHKERRQ(ierr);
631e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_nodes [n1,n2,..] Nodes to start in debugger\n");CHKERRQ(ierr);
632e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr);
633e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr);
634e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"                      waits the delay for you to attach\n");CHKERRQ(ierr);
6351cda70a7SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -display display: Location where X window graphics and debuggers are displayed\n");CHKERRQ(ierr);
636e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr);
637e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr);
638e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr);
639e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"           note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr);
640e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr);
641e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc: use our error checking malloc\n");CHKERRQ(ierr);
642e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use error checking malloc\n");CHKERRQ(ierr);
6434161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr);
6444161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_log: keeps log of all memory allocations\n");CHKERRQ(ierr);
645e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_debug: enables extended checking for memory corruption\n");CHKERRQ(ierr);
646e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_table: dump list of options inputted\n");CHKERRQ(ierr);
647e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr);
648e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr);
649e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr);
650e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr);
651a8c7a070SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr);
6520841954dSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -memory_view: print memory usage at end of run\n");CHKERRQ(ierr);
653e5c89e4eSSatish Balay #if defined(PETSC_USE_LOG)
654e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr);
655185ae32fSMatthew G. Knepley     ierr = (*PetscHelpPrintf)(comm," -log_view [:filename:[format]]: logging objects and events\n");CHKERRQ(ierr);
656e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr);
657e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
658495fc317SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through Jumpshot\n");CHKERRQ(ierr);
659e5c89e4eSSatish Balay #endif
6606cf91177SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -info <optional filename>: print informative messages about the calculations\n");CHKERRQ(ierr);
661e5c89e4eSSatish Balay #endif
662e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -v: prints PETSc version number and release date\n");CHKERRQ(ierr);
663e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr);
664e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr);
665e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"-----------------------------------------------\n");CHKERRQ(ierr);
666e5c89e4eSSatish Balay   }
667e5c89e4eSSatish Balay 
66874ba8654SBarry Smith #if defined(PETSC_HAVE_POPEN)
66974ba8654SBarry Smith   {
67074ba8654SBarry Smith   char machine[128];
671c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-popen_machine",machine,128,&flg1);CHKERRQ(ierr);
67274ba8654SBarry Smith   if (flg1) {
67374ba8654SBarry Smith     ierr = PetscPOpenSetMachine(machine);CHKERRQ(ierr);
67474ba8654SBarry Smith   }
67574ba8654SBarry Smith   }
67674ba8654SBarry Smith #endif
67774ba8654SBarry Smith 
678c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr);
679e5c89e4eSSatish Balay   if (flg1) {
680e5c89e4eSSatish Balay     ierr = PetscSleep(si);CHKERRQ(ierr);
681e5c89e4eSSatish Balay   }
682e5c89e4eSSatish Balay 
683c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr);
6842ea1be5dSBarry Smith   if (flg1) {
685e5c89e4eSSatish Balay     ierr = PetscStrstr(mname,"null",&f);CHKERRQ(ierr);
686e5c89e4eSSatish Balay     if (f) {
687ecd8bba6SJed Brown       ierr = PetscInfoDeactivateClass(0);CHKERRQ(ierr);
688e5c89e4eSSatish Balay     }
6892ea1be5dSBarry Smith   }
690827f890bSBarry Smith 
691e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_VECCUDA)
692c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
693f14045dbSBarry Smith   if (!flg3) {
694c5929fdfSBarry Smith     ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&flg3);CHKERRQ(ierr);
695f14045dbSBarry Smith   }
696bab1f7e6SVictor Minden #endif
697e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
698c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-viennacl_synchronize",&flg3,NULL);CHKERRQ(ierr);
699f14045dbSBarry Smith   PetscViennaCLSynchronize = flg3;
700b8ced49eSKarl Rupp #endif
701b8ced49eSKarl Rupp #if defined(PETSC_HAVE_VECCUDA)
70282f73ecaSAlejandro Lamas Daviña   ierr = PetscOptionsGetBool(NULL,NULL,"-cuda_synchronize",&flg3,NULL);CHKERRQ(ierr);
70382f73ecaSAlejandro Lamas Daviña   PetscCUDASynchronize = flg3;
7044cf1874eSKarl Rupp #endif
70582f73ecaSAlejandro Lamas Daviña 
706f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
707f0a7718cSKarl Rupp   ierr = PetscViennaCLInit();CHKERRQ(ierr);
708f0a7718cSKarl Rupp #endif
709f0a7718cSKarl Rupp 
710e5c89e4eSSatish Balay   PetscFunctionReturn(0);
711e5c89e4eSSatish Balay }
712