xref: /petsc/src/sys/objects/init.c (revision 7cdaf61d98bcd4bcae29d7d7ca8c82f2c6a9a029)
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;
80*7cdaf61dSJed Brown MPI_Datatype MPIU_FORTRANADDR;
8175567043SBarry Smith 
82e5c89e4eSSatish Balay /*
83e5c89e4eSSatish Balay        Function that is called to display all error messages
84e5c89e4eSSatish Balay */
857087cfbeSBarry Smith PetscErrorCode (*PetscErrorPrintf)(const char [],...)          = PetscErrorPrintfDefault;
867087cfbeSBarry Smith PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...)  = PetscHelpPrintfDefault;
877087cfbeSBarry Smith PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list)    = PetscVFPrintfDefault;
88bab1f7e6SVictor Minden /*
894cf1874eSKarl Rupp   This is needed to turn on/off GPU synchronization
908b5db460SBarry Smith */
914cf1874eSKarl Rupp PetscBool PetscViennaCLSynchronize = PETSC_FALSE;
9282f73ecaSAlejandro Lamas Daviña PetscBool PetscCUDASynchronize = PETSC_FALSE;
93bab1f7e6SVictor Minden 
94e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
95e5c89e4eSSatish Balay /*
96e5c89e4eSSatish Balay    Optional file where all PETSc output from various prints is saved
97e5c89e4eSSatish Balay */
9895c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history;
990298fd71SBarry Smith FILE *petsc_history = NULL;
100e5c89e4eSSatish Balay 
1017087cfbeSBarry Smith PetscErrorCode  PetscOpenHistoryFile(const char filename[],FILE **fd)
102e5c89e4eSSatish Balay {
103e5c89e4eSSatish Balay   PetscErrorCode ierr;
104e5c89e4eSSatish Balay   PetscMPIInt    rank,size;
105e5c89e4eSSatish Balay   char           pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64];
106e5c89e4eSSatish Balay   char           version[256];
107e5c89e4eSSatish Balay 
108e5c89e4eSSatish Balay   PetscFunctionBegin;
109e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
110e5c89e4eSSatish Balay   if (!rank) {
111e5c89e4eSSatish Balay     char        arch[10];
112f56c2debSBarry Smith     int         err;
113f56c2debSBarry Smith 
114e5c89e4eSSatish Balay     ierr = PetscGetArchType(arch,10);CHKERRQ(ierr);
115e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
116a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
117e5c89e4eSSatish Balay     ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
118e5c89e4eSSatish Balay     if (filename) {
119e5c89e4eSSatish Balay       ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr);
120e5c89e4eSSatish Balay     } else {
121e5c89e4eSSatish Balay       ierr = PetscGetHomeDirectory(pfile,240);CHKERRQ(ierr);
122e5c89e4eSSatish Balay       ierr = PetscStrcat(pfile,"/.petschistory");CHKERRQ(ierr);
123e5c89e4eSSatish Balay       ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr);
124e5c89e4eSSatish Balay     }
125e5c89e4eSSatish Balay 
126a297a907SKarl Rupp     *fd = fopen(fname,"a");
127a297a907SKarl Rupp     if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname);
128a297a907SKarl Rupp 
129e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
130e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr);
131e5c89e4eSSatish Balay     ierr = PetscGetProgramName(pname,PETSC_MAX_PATH_LEN);CHKERRQ(ierr);
132e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr);
133e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
134a297a907SKarl Rupp 
135f56c2debSBarry Smith     err = fflush(*fd);
136e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
137e5c89e4eSSatish Balay   }
138e5c89e4eSSatish Balay   PetscFunctionReturn(0);
139e5c89e4eSSatish Balay }
140e5c89e4eSSatish Balay 
14195c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscCloseHistoryFile(FILE **fd)
142e5c89e4eSSatish Balay {
143e5c89e4eSSatish Balay   PetscErrorCode ierr;
144e5c89e4eSSatish Balay   PetscMPIInt    rank;
145e5c89e4eSSatish Balay   char           date[64];
146f56c2debSBarry Smith   int            err;
147e5c89e4eSSatish Balay 
148e5c89e4eSSatish Balay   PetscFunctionBegin;
149e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
150e5c89e4eSSatish Balay   if (!rank) {
151e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
152e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
153e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr);
154e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr);
155f56c2debSBarry Smith     err  = fflush(*fd);
156e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
157f56c2debSBarry Smith     err = fclose(*fd);
158e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
159e5c89e4eSSatish Balay   }
160e5c89e4eSSatish Balay   PetscFunctionReturn(0);
161e5c89e4eSSatish Balay }
162e5c89e4eSSatish Balay 
163e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
164e5c89e4eSSatish Balay 
165e5c89e4eSSatish Balay /*
166e5c89e4eSSatish Balay    This is ugly and probably belongs somewhere else, but I want to
167e5c89e4eSSatish Balay   be able to put a true MPI abort error handler with command line args.
168e5c89e4eSSatish Balay 
169e5c89e4eSSatish Balay     This is so MPI errors in the debugger will leave all the stack
1703c311c98SBarry Smith   frames. The default MP_Abort() cleans up and exits thus providing no useful information
1713c311c98SBarry Smith   in the debugger hence we call abort() instead of MPI_Abort().
172e5c89e4eSSatish Balay */
173e5c89e4eSSatish Balay 
17433c7d699SBarry Smith void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag,...)
175e5c89e4eSSatish Balay {
176e5c89e4eSSatish Balay   PetscFunctionBegin;
1773c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
178e5c89e4eSSatish Balay   abort();
179e5c89e4eSSatish Balay }
180e5c89e4eSSatish Balay 
18133c7d699SBarry Smith void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag,...)
182e5c89e4eSSatish Balay {
183e5c89e4eSSatish Balay   PetscErrorCode ierr;
184e5c89e4eSSatish Balay 
185e5c89e4eSSatish Balay   PetscFunctionBegin;
1863c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
187e5c89e4eSSatish Balay   ierr = PetscAttachDebugger();
188a297a907SKarl Rupp   if (ierr) MPI_Abort(*comm,*flag); /* hopeless so get out */
189e5c89e4eSSatish Balay }
190e5c89e4eSSatish Balay 
191e5c89e4eSSatish Balay /*@C
192e5c89e4eSSatish Balay    PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one
193e5c89e4eSSatish Balay      wishes a clean exit somewhere deep in the program.
194e5c89e4eSSatish Balay 
195e5c89e4eSSatish Balay    Collective on PETSC_COMM_WORLD
196e5c89e4eSSatish Balay 
197e5c89e4eSSatish Balay    Options Database Keys are the same as for PetscFinalize()
198e5c89e4eSSatish Balay 
199e5c89e4eSSatish Balay    Level: advanced
200e5c89e4eSSatish Balay 
201e5c89e4eSSatish Balay    Note:
202e5c89e4eSSatish Balay    See PetscInitialize() for more general runtime options.
203e5c89e4eSSatish Balay 
20488c29154SBarry Smith .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize()
205e5c89e4eSSatish Balay @*/
2067087cfbeSBarry Smith PetscErrorCode  PetscEnd(void)
207e5c89e4eSSatish Balay {
208e5c89e4eSSatish Balay   PetscFunctionBegin;
209e5c89e4eSSatish Balay   PetscFinalize();
210e5c89e4eSSatish Balay   exit(0);
211e5c89e4eSSatish Balay   return 0;
212e5c89e4eSSatish Balay }
213e5c89e4eSSatish Balay 
214ace3abfcSBarry Smith PetscBool PetscOptionsPublish = PETSC_FALSE;
21595c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseTrMalloc_Private(void);
21695c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseHBWMalloc_Private(void);
21795c0884eSLisandro Dalcin PETSC_INTERN PetscBool      petscsetmallocvisited;
218e5c89e4eSSatish Balay static       char           emacsmachinename[256];
219e5c89e4eSSatish Balay 
220e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = 0;
221e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm)    = 0;
222e5c89e4eSSatish Balay 
223e5c89e4eSSatish Balay /*@C
224e5c89e4eSSatish Balay    PetscSetHelpVersionFunctions - Sets functions that print help and version information
225e5c89e4eSSatish Balay    before the PETSc help and version information is printed. Must call BEFORE PetscInitialize().
226e5c89e4eSSatish Balay    This routine enables a "higher-level" package that uses PETSc to print its messages first.
227e5c89e4eSSatish Balay 
228e5c89e4eSSatish Balay    Input Parameter:
2290298fd71SBarry Smith +  help - the help function (may be NULL)
2300298fd71SBarry Smith -  version - the version function (may be NULL)
231e5c89e4eSSatish Balay 
232e5c89e4eSSatish Balay    Level: developer
233e5c89e4eSSatish Balay 
234e5c89e4eSSatish Balay    Concepts: package help message
235e5c89e4eSSatish Balay 
236e5c89e4eSSatish Balay @*/
2377087cfbeSBarry Smith PetscErrorCode  PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm))
238e5c89e4eSSatish Balay {
239e5c89e4eSSatish Balay   PetscFunctionBegin;
240e5c89e4eSSatish Balay   PetscExternalHelpFunction    = help;
241e5c89e4eSSatish Balay   PetscExternalVersionFunction = version;
242e5c89e4eSSatish Balay   PetscFunctionReturn(0);
243e5c89e4eSSatish Balay }
244e5c89e4eSSatish Balay 
24505df10baSBarry Smith #if defined(PETSC_USE_LOG)
24695c0884eSLisandro Dalcin PETSC_INTERN PetscBool   PetscObjectsLog;
24705df10baSBarry Smith #endif
24805df10baSBarry Smith 
24995c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode  PetscOptionsCheckInitial_Private(void)
250e5c89e4eSSatish Balay {
251e5c89e4eSSatish Balay   char              string[64],mname[PETSC_MAX_PATH_LEN],*f;
252e5c89e4eSSatish Balay   MPI_Comm          comm = PETSC_COMM_WORLD;
25367584ceeSBarry Smith   PetscBool         flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flag;
254e5c89e4eSSatish Balay   PetscErrorCode    ierr;
25567584ceeSBarry Smith   PetscReal         si;
25628559dc8SJed Brown   PetscInt          intensity;
257e5c89e4eSSatish Balay   int               i;
258e5c89e4eSSatish Balay   PetscMPIInt       rank;
2597ca660e7SBarry Smith   char              version[256],helpoptions[256];
26067584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
26167584ceeSBarry Smith   PetscReal         logthreshold;
2622479a3a6SBarry Smith #endif
2632479a3a6SBarry Smith #if defined(PETSC_USE_LOG)
2642479a3a6SBarry Smith   PetscViewerFormat format;
26567584ceeSBarry Smith   PetscBool         flg4 = PETSC_FALSE;
26667584ceeSBarry Smith #endif
267e5c89e4eSSatish Balay 
268e5c89e4eSSatish Balay   PetscFunctionBegin;
26942218b76SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
270e5c89e4eSSatish Balay 
27167584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
272e5c89e4eSSatish Balay   /*
273e5c89e4eSSatish Balay       Setup the memory management; support for tracing malloc() usage
274e5c89e4eSSatish Balay   */
275c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-malloc_log",&flg3);CHKERRQ(ierr);
276574034a9SJed Brown   logthreshold = 0.0;
277c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-malloc_log_threshold",&logthreshold,&flg1);CHKERRQ(ierr);
278574034a9SJed Brown   if (flg1) flg3 = PETSC_TRUE;
2798bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG)
280c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg1,&flg2);CHKERRQ(ierr);
281e5c89e4eSSatish Balay   if ((!flg2 || flg1) && !petscsetmallocvisited) {
282dc92acbaSJed Brown     if (flg2 || !(PETSC_RUNNING_ON_VALGRIND)) {
283555d055bSBarry Smith       /* turn off default -malloc if valgrind is being used */
284e5c89e4eSSatish Balay       ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
285555d055bSBarry Smith     }
286e5c89e4eSSatish Balay   }
287e5c89e4eSSatish Balay #else
288c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_dump",&flg1,NULL);CHKERRQ(ierr);
289c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg2,NULL);CHKERRQ(ierr);
290e5c89e4eSSatish Balay   if (flg1 || flg2 || flg3) {ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);}
291e5c89e4eSSatish Balay #endif
292e5c89e4eSSatish Balay   if (flg3) {
293574034a9SJed Brown     ierr = PetscMallocSetDumpLogThreshold((PetscLogDouble)logthreshold);CHKERRQ(ierr);
294e5c89e4eSSatish Balay   }
295ba282f50SJed Brown   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_coalesce",&flg1,&flg2);CHKERRQ(ierr);
296ba282f50SJed Brown   if (flg2) {ierr = PetscMallocSetCoalesce(flg1);CHKERRQ(ierr);}
29790d69ab7SBarry Smith   flg1 = PETSC_FALSE;
298c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_debug",&flg1,NULL);CHKERRQ(ierr);
299e5c89e4eSSatish Balay   if (flg1) {
300e5c89e4eSSatish Balay     ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
301e5c89e4eSSatish Balay     ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr);
302e5c89e4eSSatish Balay   }
303dc92acbaSJed Brown   flg1 = PETSC_FALSE;
304c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_test",&flg1,NULL);CHKERRQ(ierr);
3058bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG)
306dc92acbaSJed Brown   if (flg1 && !PETSC_RUNNING_ON_VALGRIND) {
307dc92acbaSJed Brown     ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
308dc92acbaSJed Brown     ierr = PetscMallocSetDumpLog();CHKERRQ(ierr);
309dc92acbaSJed Brown     ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr);
310dc92acbaSJed Brown   }
311dc92acbaSJed Brown #endif
3122f21b5c6SHong Zhang   flg1 = PETSC_FALSE;
3132f21b5c6SHong Zhang   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_hbw",&flg1,NULL);CHKERRQ(ierr);
314cf6f3811SHong Zhang   /* ignore this option if malloc is already set */
315cf6f3811SHong Zhang   if (flg1 && !petscsetmallocvisited) {ierr = PetscSetUseHBWMalloc_Private();CHKERRQ(ierr);}
316e5c89e4eSSatish Balay 
31790d69ab7SBarry Smith   flg1 = PETSC_FALSE;
318c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_info",&flg1,NULL);CHKERRQ(ierr);
3197783f70dSSatish Balay   if (!flg1) {
32090d69ab7SBarry Smith     flg1 = PETSC_FALSE;
321c5929fdfSBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-memory_view",&flg1,NULL);CHKERRQ(ierr);
3227783f70dSSatish Balay   }
323e5c89e4eSSatish Balay   if (flg1) {
324e5c89e4eSSatish Balay     ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr);
325e5c89e4eSSatish Balay   }
32667584ceeSBarry Smith #endif
327e5c89e4eSSatish Balay 
328b4427426SBarry Smith #if defined(PETSC_USE_LOG)
329c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-objects_dump",&PetscObjectsLog);CHKERRQ(ierr);
330b4427426SBarry Smith #endif
33105df10baSBarry Smith 
332e5c89e4eSSatish Balay   /*
333e5c89e4eSSatish Balay       Set the display variable for graphics
334e5c89e4eSSatish Balay   */
335e5c89e4eSSatish Balay   ierr = PetscSetDisplay();CHKERRQ(ierr);
336e5c89e4eSSatish Balay 
33751dcc849SKerry Stevens   /*
338e5c89e4eSSatish Balay       Print the PETSc version information
339e5c89e4eSSatish Balay   */
340c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-v",&flg1);CHKERRQ(ierr);
341c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-version",&flg2);CHKERRQ(ierr);
3422d747510SLisandro Dalcin   ierr = PetscOptionsHasHelp(NULL,&flg3);CHKERRQ(ierr);
343e5c89e4eSSatish Balay   if (flg1 || flg2 || flg3) {
344e5c89e4eSSatish Balay 
345e5c89e4eSSatish Balay     /*
346e5c89e4eSSatish Balay        Print "higher-level" package version message
347e5c89e4eSSatish Balay     */
348e5c89e4eSSatish Balay     if (PetscExternalVersionFunction) {
349e5c89e4eSSatish Balay       ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr);
350e5c89e4eSSatish Balay     }
351e5c89e4eSSatish Balay 
352a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
3537ca660e7SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr);
354e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr);
355e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr);
356e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr);
35784e42920SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr);
358e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr);
359e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr);
3607ca660e7SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr);
3617ca660e7SBarry Smith   }
3627ca660e7SBarry Smith 
36394941ca7SBarry Smith   /*
36494941ca7SBarry Smith        Print "higher-level" package help message
36594941ca7SBarry Smith   */
36694941ca7SBarry Smith   if (flg3) {
36794941ca7SBarry Smith     if (PetscExternalHelpFunction) {
36894941ca7SBarry Smith       ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr);
36994941ca7SBarry Smith     }
37094941ca7SBarry Smith   }
37194941ca7SBarry Smith 
3727ca660e7SBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-help",helpoptions,sizeof(helpoptions),&flg1);CHKERRQ(ierr);
3737ca660e7SBarry Smith   if (flg1) {
3747ca660e7SBarry Smith     ierr = PetscStrcmp(helpoptions,"intro",&flg2);CHKERRQ(ierr);
3757ca660e7SBarry Smith     if (flg2) {
376bdb346e9SBarry Smith       ierr = PetscOptionsDestroyDefault();CHKERRQ(ierr);
377008a6e76SBarry Smith       ierr = PetscFreeMPIResources();CHKERRQ(ierr);
3787ca660e7SBarry Smith       ierr = MPI_Finalize();CHKERRQ(ierr);
3797ca660e7SBarry Smith       exit(0);
3807ca660e7SBarry Smith     }
381e5c89e4eSSatish Balay   }
382e5c89e4eSSatish Balay 
383e5c89e4eSSatish Balay   /*
384e5c89e4eSSatish Balay       Setup the error handling
385e5c89e4eSSatish Balay   */
38690d69ab7SBarry Smith   flg1 = PETSC_FALSE;
387c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_abort",&flg1,NULL);CHKERRQ(ierr);
388b59baad1SJed Brown   if (flg1) {
38942218b76SBarry Smith     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr);
390b59baad1SJed Brown     ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr);
391b59baad1SJed Brown   }
39290d69ab7SBarry Smith   flg1 = PETSC_FALSE;
393c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_mpiabort",&flg1,NULL);CHKERRQ(ierr);
394cb9801acSJed Brown   if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,0);CHKERRQ(ierr);}
39590d69ab7SBarry Smith   flg1 = PETSC_FALSE;
396c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_on_error",&flg1,NULL);CHKERRQ(ierr);
397e5c89e4eSSatish Balay   if (flg1) {
398d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr);
399e5c89e4eSSatish Balay   }
40090d69ab7SBarry Smith   flg1 = PETSC_FALSE;
401c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-no_signal_handler",&flg1,NULL);CHKERRQ(ierr);
4028d359177SBarry Smith   if (!flg1) {ierr = PetscPushSignalHandler(PetscSignalHandlerDefault,(void*)0);CHKERRQ(ierr);}
40396cc47afSJed Brown   flg1 = PETSC_FALSE;
404c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-fp_trap",&flg1,NULL);CHKERRQ(ierr);
40596cc47afSJed Brown   if (flg1) {ierr = PetscSetFPTrap(PETSC_FP_TRAP_ON);CHKERRQ(ierr);}
406c5929fdfSBarry Smith   ierr = PetscOptionsGetInt(NULL,NULL,"-check_pointer_intensity",&intensity,&flag);CHKERRQ(ierr);
40728559dc8SJed Brown   if (flag) {ierr = PetscCheckPointerSetIntensity(intensity);CHKERRQ(ierr);}
408e5c89e4eSSatish Balay 
409e5c89e4eSSatish Balay   /*
410e5c89e4eSSatish Balay       Setup debugger information
411e5c89e4eSSatish Balay   */
412e5c89e4eSSatish Balay   ierr = PetscSetDefaultDebugger();CHKERRQ(ierr);
413c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_attach_debugger",string,64,&flg1);CHKERRQ(ierr);
414e5c89e4eSSatish Balay   if (flg1) {
415e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
416e5c89e4eSSatish Balay 
417e5c89e4eSSatish Balay     ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
41833c7d699SBarry Smith     ierr = MPI_Comm_create_errhandler(Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr);
419d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
420e5c89e4eSSatish Balay     ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,0);CHKERRQ(ierr);
421e5c89e4eSSatish Balay   }
422c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-debug_terminal",string,64,&flg1);CHKERRQ(ierr);
4235e96ac45SJed Brown   if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); }
424c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-start_in_debugger",string,64,&flg1);CHKERRQ(ierr);
425c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-stop_for_debugger",string,64,&flg2);CHKERRQ(ierr);
426e5c89e4eSSatish Balay   if (flg1 || flg2) {
427e5c89e4eSSatish Balay     PetscMPIInt    size;
428e5c89e4eSSatish Balay     PetscInt       lsize,*nodes;
429e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
430e5c89e4eSSatish Balay     /*
431e5c89e4eSSatish Balay        we have to make sure that all processors have opened
432e5c89e4eSSatish Balay        connections to all other processors, otherwise once the
433e5c89e4eSSatish Balay        debugger has stated it is likely to receive a SIGUSR1
434e5c89e4eSSatish Balay        and kill the program.
435e5c89e4eSSatish Balay     */
43642218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
437e5c89e4eSSatish Balay     if (size > 2) {
438533163c2SBarry Smith       PetscMPIInt dummy = 0;
439e5c89e4eSSatish Balay       MPI_Status  status;
440e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
441e5c89e4eSSatish Balay         if (rank != i) {
44242218b76SBarry Smith           ierr = MPI_Send(&dummy,1,MPI_INT,i,109,comm);CHKERRQ(ierr);
443e5c89e4eSSatish Balay         }
444e5c89e4eSSatish Balay       }
445e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
446e5c89e4eSSatish Balay         if (rank != i) {
44742218b76SBarry Smith           ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,comm,&status);CHKERRQ(ierr);
448e5c89e4eSSatish Balay         }
449e5c89e4eSSatish Balay       }
450e5c89e4eSSatish Balay     }
451e5c89e4eSSatish Balay     /* check if this processor node should be in debugger */
452785e854fSJed Brown     ierr  = PetscMalloc1(size,&nodes);CHKERRQ(ierr);
453e5c89e4eSSatish Balay     lsize = size;
454c5929fdfSBarry Smith     ierr  = PetscOptionsGetIntArray(NULL,NULL,"-debugger_nodes",nodes,&lsize,&flag);CHKERRQ(ierr);
455e5c89e4eSSatish Balay     if (flag) {
456e5c89e4eSSatish Balay       for (i=0; i<lsize; i++) {
457e5c89e4eSSatish Balay         if (nodes[i] == rank) { flag = PETSC_FALSE; break; }
458e5c89e4eSSatish Balay       }
459e5c89e4eSSatish Balay     }
460e5c89e4eSSatish Balay     if (!flag) {
461e5c89e4eSSatish Balay       ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
462e5c89e4eSSatish Balay       ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr);
463e5c89e4eSSatish Balay       if (flg1) {
464e5c89e4eSSatish Balay         ierr = PetscAttachDebugger();CHKERRQ(ierr);
465e5c89e4eSSatish Balay       } else {
466e5c89e4eSSatish Balay         ierr = PetscStopForDebugger();CHKERRQ(ierr);
467e5c89e4eSSatish Balay       }
46833c7d699SBarry Smith       ierr = MPI_Comm_create_errhandler(Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr);
469d54338ecSKarl Rupp       ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
470e5c89e4eSSatish Balay     }
471e5c89e4eSSatish Balay     ierr = PetscFree(nodes);CHKERRQ(ierr);
472e5c89e4eSSatish Balay   }
473e5c89e4eSSatish Balay 
474c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_emacs",emacsmachinename,128,&flg1);CHKERRQ(ierr);
475cb9801acSJed Brown   if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);}
476e5c89e4eSSatish Balay 
477e5c89e4eSSatish Balay   /*
478e5c89e4eSSatish Balay         Setup profiling and logging
479e5c89e4eSSatish Balay   */
4806cf91177SBarry Smith #if defined(PETSC_USE_INFO)
4818bb29257SSatish Balay   {
482e5c89e4eSSatish Balay     char logname[PETSC_MAX_PATH_LEN]; logname[0] = 0;
483c5929fdfSBarry Smith     ierr = PetscOptionsGetString(NULL,NULL,"-info",logname,250,&flg1);CHKERRQ(ierr);
4848bb29257SSatish Balay     if (flg1 && logname[0]) {
485fcc2139eSBarry Smith       ierr = PetscInfoAllow(PETSC_TRUE,logname);CHKERRQ(ierr);
4868bb29257SSatish Balay     } else if (flg1) {
4870298fd71SBarry Smith       ierr = PetscInfoAllow(PETSC_TRUE,NULL);CHKERRQ(ierr);
488e5c89e4eSSatish Balay     }
489e5c89e4eSSatish Balay   }
490865f6aa8SSatish Balay #endif
491865f6aa8SSatish Balay #if defined(PETSC_USE_LOG)
492865f6aa8SSatish Balay   mname[0] = 0;
493c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-history",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr);
494865f6aa8SSatish Balay   if (flg1) {
495865f6aa8SSatish Balay     if (mname[0]) {
496f3dea69dSBarry Smith       ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr);
497865f6aa8SSatish Balay     } else {
498706d7a88SBarry Smith       ierr = PetscOpenHistoryFile(NULL,&petsc_history);CHKERRQ(ierr);
499865f6aa8SSatish Balay     }
500865f6aa8SSatish Balay   }
501217044c2SLisandro Dalcin 
502217044c2SLisandro Dalcin   ierr = PetscOptionsGetBool(NULL,NULL,"-log_sync",&PetscLogSyncOn,NULL);CHKERRQ(ierr);
503217044c2SLisandro Dalcin 
504e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
50590d69ab7SBarry Smith   flg1 = PETSC_FALSE;
506c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_mpe",&flg1);CHKERRQ(ierr);
507495fc317SBarry Smith   if (flg1) {ierr = PetscLogMPEBegin();CHKERRQ(ierr);}
508e5c89e4eSSatish Balay #endif
50990d69ab7SBarry Smith   flg1 = PETSC_FALSE;
51090d69ab7SBarry Smith   flg3 = PETSC_FALSE;
511c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-log_all",&flg1,NULL);CHKERRQ(ierr);
512c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
513e5c89e4eSSatish Balay   if (flg1)                      { ierr = PetscLogAllBegin();CHKERRQ(ierr); }
514bb1d7374SBarry Smith   else if (flg3)                 { ierr = PetscLogDefaultBegin();CHKERRQ(ierr);}
515e5c89e4eSSatish Balay 
516c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-log_trace",mname,250,&flg1);CHKERRQ(ierr);
517e5c89e4eSSatish Balay   if (flg1) {
518e5c89e4eSSatish Balay     char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN];
519e5c89e4eSSatish Balay     FILE *file;
520e5c89e4eSSatish Balay     if (mname[0]) {
5212e924ca5SSatish Balay       PetscSNPrintf(name,PETSC_MAX_PATH_LEN,"%s.%d",mname,rank);
522e5c89e4eSSatish Balay       ierr = PetscFixFilename(name,fname);CHKERRQ(ierr);
523e5c89e4eSSatish Balay       file = fopen(fname,"w");
524f3dea69dSBarry Smith       if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname);
525a297a907SKarl Rupp     } else file = PETSC_STDOUT;
526e5c89e4eSSatish Balay     ierr = PetscLogTraceBegin(file);CHKERRQ(ierr);
527e5c89e4eSSatish Balay   }
528bb1d7374SBarry Smith 
52916413a6aSBarry Smith   ierr = PetscOptionsGetViewer(comm,NULL,NULL,"-log_view",NULL,&format,&flg4);CHKERRQ(ierr);
530bb1d7374SBarry Smith   if (flg4) {
531bb1d7374SBarry Smith     if (format == PETSC_VIEWER_ASCII_XML) {
532bb1d7374SBarry Smith       ierr = PetscLogNestedBegin();CHKERRQ(ierr);
533bb1d7374SBarry Smith     } else {
534bb1d7374SBarry Smith       ierr = PetscLogDefaultBegin();CHKERRQ(ierr);
535bb1d7374SBarry Smith     }
536e3ed9ee7SBarry Smith     PetscLogMemory = PETSC_FALSE;
537e3ed9ee7SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&PetscLogMemory,NULL);CHKERRQ(ierr);
538e3ed9ee7SBarry Smith     if (PetscLogMemory) {
539e3ed9ee7SBarry Smith       ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);
540e3ed9ee7SBarry Smith     }
541eccbb886SLisandro Dalcin   }
542eccbb886SLisandro Dalcin   if (flg4 && format == PETSC_VIEWER_ASCII_XML) {
543eccbb886SLisandro Dalcin     PetscReal threshold = PetscRealConstant(0.01);
544eccbb886SLisandro Dalcin     ierr = PetscOptionsGetReal(NULL,NULL,"-log_threshold",&threshold,&flg1);CHKERRQ(ierr);
545eccbb886SLisandro Dalcin     if (flg1) {ierr = PetscLogSetThreshold((PetscLogDouble)threshold,NULL);CHKERRQ(ierr);}
546bb1d7374SBarry Smith   }
547e5c89e4eSSatish Balay #endif
548e5c89e4eSSatish Balay 
549c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-saws_options",&PetscOptionsPublish,NULL);CHKERRQ(ierr);
550e5c89e4eSSatish Balay 
5514b0a73b2SBarry Smith #if defined(PETSC_HAVE_CUDA)
552c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-cuda_show_devices",&flg1);CHKERRQ(ierr);
5534b0a73b2SBarry Smith   if (flg1) {
5544b0a73b2SBarry Smith     struct cudaDeviceProp prop;
5554b0a73b2SBarry Smith     int                   devCount;
55616b6f2efSBarry Smith     PetscInt              device;
5572692e278SPaul Mullowney     cudaError_t           err = cudaSuccess;
558e09ad801SBarry Smith 
5592692e278SPaul Mullowney     err = cudaGetDeviceCount(&devCount);
560e09ad801SBarry Smith     if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err));
5614b0a73b2SBarry Smith     for (device = 0; device < devCount; ++device) {
56216b6f2efSBarry Smith       err = cudaGetDeviceProperties(&prop, (int)device);
563e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceProperties %s",cudaGetErrorString(err));
56442218b76SBarry Smith       ierr = PetscPrintf(comm, "CUDA device %D: %s\n", device, prop.name);CHKERRQ(ierr);
5654b0a73b2SBarry Smith     }
5664b0a73b2SBarry Smith   }
567bffec52eSKarl Rupp   if (!PetscCUDAInitialized) {
56816b6f2efSBarry Smith     PetscMPIInt size;
56942218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
570bd72cc96SPaul Mullowney     if (size>1) {
57116b6f2efSBarry Smith       int         devCount;
57216b6f2efSBarry Smith       PetscInt    device;
57316b6f2efSBarry Smith       PetscMPIInt rank;
5742692e278SPaul Mullowney       cudaError_t err = cudaSuccess;
5752692e278SPaul Mullowney 
5762692e278SPaul Mullowney       /* check to see if we force multiple ranks to hit the same GPU */
577c5929fdfSBarry Smith       ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr);
5782692e278SPaul Mullowney       if (flg1) {
57916b6f2efSBarry Smith         err = cudaSetDevice((int)device);
580e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
5812692e278SPaul Mullowney       } else {
582e09ad801SBarry Smith         /* we're not using the same GPU on multiple MPI threads. So try to allocated different   GPUs to different processes */
5832692e278SPaul Mullowney 
5842692e278SPaul Mullowney         /* First get the device count */
585e09ad801SBarry Smith         err   = cudaGetDeviceCount(&devCount);
586e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err));
5872692e278SPaul Mullowney 
5882692e278SPaul Mullowney         /* next determine the rank and then set the device via a mod */
58942218b76SBarry Smith         ierr   = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
590bd72cc96SPaul Mullowney         device = rank % devCount;
59116b6f2efSBarry Smith         err    = cudaSetDevice((int)device);
592e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
5932692e278SPaul Mullowney       }
5942692e278SPaul Mullowney 
5952692e278SPaul Mullowney       /* set the device flags so that it can map host memory ... do NOT throw exception on err!=cudaSuccess
5962692e278SPaul Mullowney        multiple devices may try to set the flags on the same device. So long as one of them succeeds, things
5972692e278SPaul Mullowney        are ok. */
5982692e278SPaul Mullowney       err = cudaSetDeviceFlags(cudaDeviceMapHost);
599e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err));
600a297a907SKarl Rupp     } else {
60116b6f2efSBarry Smith       PetscInt    device;
6022692e278SPaul Mullowney       cudaError_t err = cudaSuccess;
6032692e278SPaul Mullowney 
604f297a14cSJed Brown       /* the code below works for serial GPU simulations */
605c5929fdfSBarry Smith       ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr);
6064b0a73b2SBarry Smith       if (flg1) {
60716b6f2efSBarry Smith         err = cudaSetDevice((int)device);
608e09ad801SBarry Smith         if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err));
6094b0a73b2SBarry Smith       }
6102692e278SPaul Mullowney 
6112692e278SPaul Mullowney       /* set the device flags so that it can map host memory ... here, we error check. */
6122692e278SPaul Mullowney       err = cudaSetDeviceFlags(cudaDeviceMapHost);
613e09ad801SBarry Smith       if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err));
6144b0a73b2SBarry Smith     }
615bffec52eSKarl Rupp 
616bffec52eSKarl Rupp     PetscCUDAInitialized = PETSC_TRUE;
617bd72cc96SPaul Mullowney   }
6184b0a73b2SBarry Smith #endif
6194b0a73b2SBarry Smith 
6204b0a73b2SBarry Smith 
621e5c89e4eSSatish Balay   /*
622e5c89e4eSSatish Balay        Print basic help message
623e5c89e4eSSatish Balay   */
6242d747510SLisandro Dalcin   ierr = PetscOptionsHasHelp(NULL,&flg1);CHKERRQ(ierr);
625e5c89e4eSSatish Balay   if (flg1) {
626e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr);
627301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -help: prints help method for each option\n");CHKERRQ(ierr);
628301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr);
629301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm,"       only when run in the debugger\n");CHKERRQ(ierr);
630e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
631e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start the debugger in new xterm\n");CHKERRQ(ierr);
632e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       unless noxterm is given\n");CHKERRQ(ierr);
633e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
634e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start all processes in the debugger\n");CHKERRQ(ierr);
635e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr);
636e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"    emacs jumps to error file\n");CHKERRQ(ierr);
637e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_nodes [n1,n2,..] Nodes to start in debugger\n");CHKERRQ(ierr);
638e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr);
639e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr);
640e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"                      waits the delay for you to attach\n");CHKERRQ(ierr);
6411cda70a7SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -display display: Location where X window graphics and debuggers are displayed\n");CHKERRQ(ierr);
642e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr);
643e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr);
644e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr);
645e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"           note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr);
646e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr);
647e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc: use our error checking malloc\n");CHKERRQ(ierr);
648e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use error checking malloc\n");CHKERRQ(ierr);
6494161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr);
6504161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_log: keeps log of all memory allocations\n");CHKERRQ(ierr);
651e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_debug: enables extended checking for memory corruption\n");CHKERRQ(ierr);
65226a7e8d4SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -options_view: dump list of options inputted\n");CHKERRQ(ierr);
653e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr);
654e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr);
655e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr);
656e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr);
657a8c7a070SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr);
6580841954dSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -memory_view: print memory usage at end of run\n");CHKERRQ(ierr);
659e5c89e4eSSatish Balay #if defined(PETSC_USE_LOG)
660e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr);
661185ae32fSMatthew G. Knepley     ierr = (*PetscHelpPrintf)(comm," -log_view [:filename:[format]]: logging objects and events\n");CHKERRQ(ierr);
662e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr);
663e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
664495fc317SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through Jumpshot\n");CHKERRQ(ierr);
665e5c89e4eSSatish Balay #endif
6666cf91177SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -info <optional filename>: print informative messages about the calculations\n");CHKERRQ(ierr);
667e5c89e4eSSatish Balay #endif
668e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -v: prints PETSc version number and release date\n");CHKERRQ(ierr);
669e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr);
670e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr);
671e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"-----------------------------------------------\n");CHKERRQ(ierr);
672e5c89e4eSSatish Balay   }
673e5c89e4eSSatish Balay 
67474ba8654SBarry Smith #if defined(PETSC_HAVE_POPEN)
67574ba8654SBarry Smith   {
67674ba8654SBarry Smith   char machine[128];
677c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-popen_machine",machine,128,&flg1);CHKERRQ(ierr);
67874ba8654SBarry Smith   if (flg1) {
67974ba8654SBarry Smith     ierr = PetscPOpenSetMachine(machine);CHKERRQ(ierr);
68074ba8654SBarry Smith   }
68174ba8654SBarry Smith   }
68274ba8654SBarry Smith #endif
68374ba8654SBarry Smith 
684c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr);
685e5c89e4eSSatish Balay   if (flg1) {
686e5c89e4eSSatish Balay     ierr = PetscSleep(si);CHKERRQ(ierr);
687e5c89e4eSSatish Balay   }
688e5c89e4eSSatish Balay 
689c5929fdfSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr);
6902ea1be5dSBarry Smith   if (flg1) {
691e5c89e4eSSatish Balay     ierr = PetscStrstr(mname,"null",&f);CHKERRQ(ierr);
692e5c89e4eSSatish Balay     if (f) {
693ecd8bba6SJed Brown       ierr = PetscInfoDeactivateClass(0);CHKERRQ(ierr);
694e5c89e4eSSatish Balay     }
6952ea1be5dSBarry Smith   }
696827f890bSBarry Smith 
697fd314934SBarry Smith #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA)
698c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
699f14045dbSBarry Smith   if (!flg3) {
700c5929fdfSBarry Smith     ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&flg3);CHKERRQ(ierr);
701f14045dbSBarry Smith   }
702bab1f7e6SVictor Minden #endif
703e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
704c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-viennacl_synchronize",&flg3,NULL);CHKERRQ(ierr);
705f14045dbSBarry Smith   PetscViennaCLSynchronize = flg3;
706b8ced49eSKarl Rupp #endif
707fd314934SBarry Smith #if defined(PETSC_HAVE_CUDA)
70882f73ecaSAlejandro Lamas Daviña   ierr = PetscOptionsGetBool(NULL,NULL,"-cuda_synchronize",&flg3,NULL);CHKERRQ(ierr);
70982f73ecaSAlejandro Lamas Daviña   PetscCUDASynchronize = flg3;
7104cf1874eSKarl Rupp #endif
71182f73ecaSAlejandro Lamas Daviña 
712f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
713f0a7718cSKarl Rupp   ierr = PetscViennaCLInit();CHKERRQ(ierr);
714f0a7718cSKarl Rupp #endif
715f0a7718cSKarl Rupp 
716e5c89e4eSSatish Balay   PetscFunctionReturn(0);
717e5c89e4eSSatish Balay }
718