xref: /petsc/src/sys/objects/init.c (revision bf4d2887a0c7d252f9e728fc844ebcd5378c5580)
1e5c89e4eSSatish Balay /*
2e5c89e4eSSatish Balay 
3e5c89e4eSSatish Balay    This file defines part of the initialization of PETSc
4e5c89e4eSSatish Balay 
5540e20f2SPierre Jolivet   This file uses regular malloc and free because it cannot be 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>
13fdc842d1SBarry Smith #if defined(PETSC_USE_LOG)
14fdc842d1SBarry Smith PETSC_INTERN PetscErrorCode PetscLogInitialize(void);
15fdc842d1SBarry Smith #endif
16ef386f4bSSatish Balay 
17ba61063dSBarry Smith #if defined(PETSC_HAVE_SYS_SYSINFO_H)
1851d315f7SKerry Stevens #include <sys/sysinfo.h>
19ba61063dSBarry Smith #endif
20121deb67SSatish Balay #if defined(PETSC_HAVE_UNISTD_H)
2151d315f7SKerry Stevens #include <unistd.h>
22121deb67SSatish Balay #endif
2305035670SJunchao Zhang 
24d9b72601SDominic Meiser #if defined(PETSC_HAVE_CUDA)
25d9b72601SDominic Meiser   #include <cuda_runtime.h>
267fd2f626SJose E. Roman   #include <petsccublas.h>
2705035670SJunchao Zhang #endif
2805035670SJunchao Zhang 
2905035670SJunchao Zhang #if defined(PETSC_HAVE_HIP)
3005035670SJunchao Zhang   #include <hip/hip_runtime.h>
3105035670SJunchao Zhang #endif
3205035670SJunchao Zhang 
3305035670SJunchao Zhang #if defined(PETSC_HAVE_DEVICE)
34c2a741eeSJunchao Zhang   #if defined(PETSC_HAVE_OMPI_MAJOR_VERSION)
35c2a741eeSJunchao Zhang     #include "mpi-ext.h" /* Needed for OpenMPI CUDA-aware check */
36c2a741eeSJunchao Zhang   #endif
37d9b72601SDominic Meiser #endif
38555d055bSBarry Smith 
39f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL)
40f0a7718cSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLInit();
41f0a7718cSKarl Rupp #endif
42f0a7718cSKarl Rupp 
4305035670SJunchao Zhang 
44e5c89e4eSSatish Balay /* ------------------------Nasty global variables -------------------------------*/
45e5c89e4eSSatish Balay /*
46e5c89e4eSSatish Balay      Indicates if PETSc started up MPI, or it was
47e5c89e4eSSatish Balay    already started before PETSc was initialized.
48e5c89e4eSSatish Balay */
490cbf60d0SJose E. Roman PetscBool   PetscBeganMPI                 = PETSC_FALSE;
508ad20175SVaclav Hapla PetscBool   PetscErrorHandlingInitialized = PETSC_FALSE;
510cbf60d0SJose E. Roman PetscBool   PetscInitializeCalled         = PETSC_FALSE;
520cbf60d0SJose E. Roman PetscBool   PetscFinalizeCalled           = PETSC_FALSE;
53d6f2c3cbSBarry Smith 
547087cfbeSBarry Smith PetscMPIInt PetscGlobalRank               = -1;
557087cfbeSBarry Smith PetscMPIInt PetscGlobalSize               = -1;
56ba61063dSBarry Smith 
57c2b86a48SJunchao Zhang #if defined(PETSC_HAVE_KOKKOS)
58c2b86a48SJunchao Zhang PetscBool   PetscBeganKokkos              = PETSC_FALSE;
59c2b86a48SJunchao Zhang #endif
60c2b86a48SJunchao Zhang 
61c2a741eeSJunchao Zhang PetscBool   use_gpu_aware_mpi             = PETSC_TRUE;
62c2a741eeSJunchao Zhang 
6350f81f78SJed Brown #if defined(PETSC_HAVE_COMPLEX)
64e5c89e4eSSatish Balay #if defined(PETSC_COMPLEX_INSTANTIATE)
65e5c89e4eSSatish Balay template <> class std::complex<double>; /* instantiate complex template class */
66e5c89e4eSSatish Balay #endif
672c876bd9SBarry Smith #if !defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX)
68500d8756SSatish Balay MPI_Datatype MPIU_C_DOUBLE_COMPLEX;
69500d8756SSatish Balay MPI_Datatype MPIU_C_COMPLEX;
702c876bd9SBarry Smith #endif
718619c96cSJed Brown 
728619c96cSJed Brown /*MC
738619c96cSJed Brown    PETSC_i - the imaginary number i
748619c96cSJed Brown 
758619c96cSJed Brown    Synopsis:
768619c96cSJed Brown    #include <petscsys.h>
778619c96cSJed Brown    PetscComplex PETSC_i;
788619c96cSJed Brown 
798619c96cSJed Brown    Level: beginner
808619c96cSJed Brown 
818619c96cSJed Brown    Note:
828cd53115SBarry Smith    Complex numbers are automatically available if PETSc located a working complex implementation
838619c96cSJed Brown 
848619c96cSJed Brown .seealso: PetscRealPart(), PetscImaginaryPart(), PetscRealPartComplex(), PetscImaginaryPartComplex()
858619c96cSJed Brown M*/
8650f81f78SJed Brown PetscComplex PETSC_i;
87e5c89e4eSSatish Balay #endif
88ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128)
89c90a1750SBarry Smith MPI_Datatype MPIU___FLOAT128 = 0;
907c2de775SJed Brown #if defined(PETSC_HAVE_COMPLEX)
918c764dc5SJose Roman MPI_Datatype MPIU___COMPLEX128 = 0;
928c764dc5SJose Roman #endif
93570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16)
94570b7f6dSBarry Smith MPI_Datatype MPIU___FP16 = 0;
95c90a1750SBarry Smith #endif
967087cfbeSBarry Smith MPI_Datatype MPIU_2SCALAR = 0;
970354ff80SSatish Balay #if defined(PETSC_USE_64BIT_INDICES)
987087cfbeSBarry Smith MPI_Datatype MPIU_2INT = 0;
9944041f26SJed Brown #endif
1008ad47952SJed Brown MPI_Datatype MPIU_BOOL;
1018ad47952SJed Brown MPI_Datatype MPIU_ENUM;
1027cdaf61dSJed Brown MPI_Datatype MPIU_FORTRANADDR;
103e316c87fSJed Brown MPI_Datatype MPIU_SIZE_T;
10475567043SBarry Smith 
105e5c89e4eSSatish Balay /*
106e5c89e4eSSatish Balay        Function that is called to display all error messages
107e5c89e4eSSatish Balay */
1087087cfbeSBarry Smith PetscErrorCode (*PetscErrorPrintf)(const char [],...)          = PetscErrorPrintfDefault;
1097087cfbeSBarry Smith PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...)  = PetscHelpPrintfDefault;
1107087cfbeSBarry Smith PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list)    = PetscVFPrintfDefault;
111bab1f7e6SVictor Minden /*
1124cf1874eSKarl Rupp   This is needed to turn on/off GPU synchronization
1138b5db460SBarry Smith */
1144cf1874eSKarl Rupp PetscBool PetscViennaCLSynchronize = PETSC_FALSE;
115bab1f7e6SVictor Minden 
116e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
117e5c89e4eSSatish Balay /*
118e5c89e4eSSatish Balay    Optional file where all PETSc output from various prints is saved
119e5c89e4eSSatish Balay */
12095c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history;
1210298fd71SBarry Smith FILE *petsc_history = NULL;
122e5c89e4eSSatish Balay 
1237087cfbeSBarry Smith PetscErrorCode  PetscOpenHistoryFile(const char filename[],FILE **fd)
124e5c89e4eSSatish Balay {
125e5c89e4eSSatish Balay   PetscErrorCode ierr;
126e5c89e4eSSatish Balay   PetscMPIInt    rank,size;
127e5c89e4eSSatish Balay   char           pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64];
128e5c89e4eSSatish Balay   char           version[256];
129e5c89e4eSSatish Balay 
130e5c89e4eSSatish Balay   PetscFunctionBegin;
131e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
132e5c89e4eSSatish Balay   if (!rank) {
133e5c89e4eSSatish Balay     char        arch[10];
134f56c2debSBarry Smith     int         err;
135f56c2debSBarry Smith 
136e5c89e4eSSatish Balay     ierr = PetscGetArchType(arch,10);CHKERRQ(ierr);
137e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
138a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
139e5c89e4eSSatish Balay     ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
140e5c89e4eSSatish Balay     if (filename) {
141e5c89e4eSSatish Balay       ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr);
142e5c89e4eSSatish Balay     } else {
143589a23caSBarry Smith       ierr = PetscGetHomeDirectory(pfile,sizeof(pfile));CHKERRQ(ierr);
144589a23caSBarry Smith       ierr = PetscStrlcat(pfile,"/.petschistory",sizeof(pfile));CHKERRQ(ierr);
145e5c89e4eSSatish Balay       ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr);
146e5c89e4eSSatish Balay     }
147e5c89e4eSSatish Balay 
148a297a907SKarl Rupp     *fd = fopen(fname,"a");
149a297a907SKarl Rupp     if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname);
150a297a907SKarl Rupp 
151c0bb3764SVaclav Hapla     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"----------------------------------------\n");CHKERRQ(ierr);
152e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr);
153589a23caSBarry Smith     ierr = PetscGetProgramName(pname,sizeof(pname));CHKERRQ(ierr);
154e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr);
155c0bb3764SVaclav Hapla     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"----------------------------------------\n");CHKERRQ(ierr);
156a297a907SKarl Rupp 
157f56c2debSBarry Smith     err = fflush(*fd);
158e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
159e5c89e4eSSatish Balay   }
160e5c89e4eSSatish Balay   PetscFunctionReturn(0);
161e5c89e4eSSatish Balay }
162e5c89e4eSSatish Balay 
16395c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscCloseHistoryFile(FILE **fd)
164e5c89e4eSSatish Balay {
165e5c89e4eSSatish Balay   PetscErrorCode ierr;
166e5c89e4eSSatish Balay   PetscMPIInt    rank;
167e5c89e4eSSatish Balay   char           date[64];
168f56c2debSBarry Smith   int            err;
169e5c89e4eSSatish Balay 
170e5c89e4eSSatish Balay   PetscFunctionBegin;
171e5c89e4eSSatish Balay   ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr);
172e5c89e4eSSatish Balay   if (!rank) {
173e5c89e4eSSatish Balay     ierr = PetscGetDate(date,64);CHKERRQ(ierr);
174c0bb3764SVaclav Hapla     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"----------------------------------------\n");CHKERRQ(ierr);
175e5c89e4eSSatish Balay     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr);
176c0bb3764SVaclav Hapla     ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"----------------------------------------\n");CHKERRQ(ierr);
177f56c2debSBarry Smith     err  = fflush(*fd);
178e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file");
179f56c2debSBarry Smith     err = fclose(*fd);
180e32f2f54SBarry Smith     if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
181e5c89e4eSSatish Balay   }
182e5c89e4eSSatish Balay   PetscFunctionReturn(0);
183e5c89e4eSSatish Balay }
184e5c89e4eSSatish Balay 
185e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/
186e5c89e4eSSatish Balay 
187e5c89e4eSSatish Balay /*
188e5c89e4eSSatish Balay    This is ugly and probably belongs somewhere else, but I want to
189e5c89e4eSSatish Balay   be able to put a true MPI abort error handler with command line args.
190e5c89e4eSSatish Balay 
191e5c89e4eSSatish Balay     This is so MPI errors in the debugger will leave all the stack
1923c311c98SBarry Smith   frames. The default MP_Abort() cleans up and exits thus providing no useful information
1933c311c98SBarry Smith   in the debugger hence we call abort() instead of MPI_Abort().
194e5c89e4eSSatish Balay */
195e5c89e4eSSatish Balay 
19633c7d699SBarry Smith void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag,...)
197e5c89e4eSSatish Balay {
198e5c89e4eSSatish Balay   PetscFunctionBegin;
1993c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
200e5c89e4eSSatish Balay   abort();
201e5c89e4eSSatish Balay }
202e5c89e4eSSatish Balay 
20333c7d699SBarry Smith void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag,...)
204e5c89e4eSSatish Balay {
205e5c89e4eSSatish Balay   PetscErrorCode ierr;
206e5c89e4eSSatish Balay 
207e5c89e4eSSatish Balay   PetscFunctionBegin;
2083c311c98SBarry Smith   (*PetscErrorPrintf)("MPI error %d\n",*flag);
209e5c89e4eSSatish Balay   ierr = PetscAttachDebugger();
21041e02c4dSJunchao Zhang   if (ierr) PETSCABORT(*comm,*flag); /* hopeless so get out */
211e5c89e4eSSatish Balay }
212e5c89e4eSSatish Balay 
213e5c89e4eSSatish Balay /*@C
214e5c89e4eSSatish Balay    PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one
215e5c89e4eSSatish Balay      wishes a clean exit somewhere deep in the program.
216e5c89e4eSSatish Balay 
217e5c89e4eSSatish Balay    Collective on PETSC_COMM_WORLD
218e5c89e4eSSatish Balay 
219e5c89e4eSSatish Balay    Options Database Keys are the same as for PetscFinalize()
220e5c89e4eSSatish Balay 
221e5c89e4eSSatish Balay    Level: advanced
222e5c89e4eSSatish Balay 
223e5c89e4eSSatish Balay    Note:
224e5c89e4eSSatish Balay    See PetscInitialize() for more general runtime options.
225e5c89e4eSSatish Balay 
22688c29154SBarry Smith .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize()
227e5c89e4eSSatish Balay @*/
2287087cfbeSBarry Smith PetscErrorCode  PetscEnd(void)
229e5c89e4eSSatish Balay {
230e5c89e4eSSatish Balay   PetscFunctionBegin;
231e5c89e4eSSatish Balay   PetscFinalize();
232e5c89e4eSSatish Balay   exit(0);
233e5c89e4eSSatish Balay   return 0;
234e5c89e4eSSatish Balay }
235e5c89e4eSSatish Balay 
236ace3abfcSBarry Smith PetscBool PetscOptionsPublish = PETSC_FALSE;
23795c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseHBWMalloc_Private(void);
23895c0884eSLisandro Dalcin PETSC_INTERN PetscBool      petscsetmallocvisited;
239e5c89e4eSSatish Balay static       char           emacsmachinename[256];
240e5c89e4eSSatish Balay 
24102c9f0b5SLisandro Dalcin PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = NULL;
24202c9f0b5SLisandro Dalcin PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm)    = NULL;
243e5c89e4eSSatish Balay 
244e5c89e4eSSatish Balay /*@C
245e5c89e4eSSatish Balay    PetscSetHelpVersionFunctions - Sets functions that print help and version information
246e5c89e4eSSatish Balay    before the PETSc help and version information is printed. Must call BEFORE PetscInitialize().
247e5c89e4eSSatish Balay    This routine enables a "higher-level" package that uses PETSc to print its messages first.
248e5c89e4eSSatish Balay 
249e5c89e4eSSatish Balay    Input Parameter:
2500298fd71SBarry Smith +  help - the help function (may be NULL)
2510298fd71SBarry Smith -  version - the version function (may be NULL)
252e5c89e4eSSatish Balay 
253e5c89e4eSSatish Balay    Level: developer
254e5c89e4eSSatish Balay 
255e5c89e4eSSatish Balay @*/
2567087cfbeSBarry Smith PetscErrorCode  PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm))
257e5c89e4eSSatish Balay {
258e5c89e4eSSatish Balay   PetscFunctionBegin;
259e5c89e4eSSatish Balay   PetscExternalHelpFunction    = help;
260e5c89e4eSSatish Balay   PetscExternalVersionFunction = version;
261e5c89e4eSSatish Balay   PetscFunctionReturn(0);
262e5c89e4eSSatish Balay }
263e5c89e4eSSatish Balay 
26405df10baSBarry Smith #if defined(PETSC_USE_LOG)
26595c0884eSLisandro Dalcin PETSC_INTERN PetscBool   PetscObjectsLog;
26605df10baSBarry Smith #endif
26705df10baSBarry Smith 
268e781f417SStefano Zampini void PetscMPI_Comm_eh(MPI_Comm *comm, PetscMPIInt *err, ...)
269e781f417SStefano Zampini {
270e781f417SStefano Zampini   if (PetscUnlikely(*err)) {
271e781f417SStefano Zampini     PetscMPIInt len;
272e781f417SStefano Zampini     char        errstring[MPI_MAX_ERROR_STRING];
273e781f417SStefano Zampini 
274e781f417SStefano Zampini     MPI_Error_string(*err,errstring,&len);
275e781f417SStefano Zampini     PetscError(MPI_COMM_SELF,__LINE__,PETSC_FUNCTION_NAME,__FILE__,PETSC_MPI_ERROR_CODE,PETSC_ERROR_INITIAL,"Internal error in MPI: %s",errstring);
276e781f417SStefano Zampini   }
277e781f417SStefano Zampini   return;
278e781f417SStefano Zampini }
279e781f417SStefano Zampini 
28005035670SJunchao Zhang /* CUPM stands for 'CUDA Programming Model', which is implemented in either CUDA or HIP.
28105035670SJunchao Zhang    Use the following macros to define CUDA/HIP initialization related vars/routines.
28205035670SJunchao Zhang  */
28305035670SJunchao Zhang #if defined(PETSC_HAVE_CUDA)
28405035670SJunchao Zhang   typedef cudaError_t                             cupmError_t;
28505035670SJunchao Zhang   typedef struct cudaDeviceProp                   cupmDeviceProp;
28605035670SJunchao Zhang   #define cupmGetDeviceCount(x)                   cudaGetDeviceCount(x)
28705035670SJunchao Zhang   #define cupmGetDevice(x)                        cudaGetDevice(x)
28805035670SJunchao Zhang   #define cupmSetDevice(x)                        cudaSetDevice(x)
28905035670SJunchao Zhang   #define cupmSetDeviceFlags(x)                   cudaSetDeviceFlags(x)
29005035670SJunchao Zhang   #define cupmGetDeviceProperties(x,y)            cudaGetDeviceProperties(x,y)
29105035670SJunchao Zhang   #define cupmGetLastError()                      cudaGetLastError()
29205035670SJunchao Zhang   #define cupmDeviceMapHost                       cudaDeviceMapHost
29305035670SJunchao Zhang   #define cupmSuccess                             cudaSuccess
29405035670SJunchao Zhang   #define cupmErrorSetOnActiveProcess             cudaErrorSetOnActiveProcess
29505035670SJunchao Zhang   #define CHKERRCUPM(x)                           CHKERRCUDA(x)
29605035670SJunchao Zhang   #define PetscCUPMBLASInitializeHandle()         PetscCUBLASInitializeHandle()
29705035670SJunchao Zhang   #define PetscCUPMSOLVERDnInitializeHandle()     PetscCUSOLVERDnInitializeHandle()
29805035670SJunchao Zhang   #define PetscCUPMInitialize                     PetscCUDAInitialize
29905035670SJunchao Zhang   #define PetscCUPMInitialized                    PetscCUDAInitialized
30005035670SJunchao Zhang   #define PetscCUPMInitializeCheck                PetscCUDAInitializeCheck
30105035670SJunchao Zhang   #define PetscCUPMInitializeAndView              PetscCUDAInitializeAndView
30205035670SJunchao Zhang   #define PetscCUPMSynchronize                    PetscCUDASynchronize
30305035670SJunchao Zhang   #define PetscNotUseCUPM                         PetscNotUseCUDA
30405035670SJunchao Zhang   #define cupmOptionsStr                          "CUDA options"
3054dcd27cbSJunchao Zhang   #define cupmSetDeviceStr                        "-cuda_device"
30605035670SJunchao Zhang   #define cupmViewStr                             "-cuda_view"
30705035670SJunchao Zhang   #define cupmSynchronizeStr                      "-cuda_synchronize"
30805035670SJunchao Zhang   #define PetscCUPMInitializeStr                  "PetscCUDAInitialize"
30905035670SJunchao Zhang   #define PetscOptionsCheckCUPM                   PetscOptionsCheckCUDA
31005035670SJunchao Zhang   #define PetscMPICUPMAwarenessCheck              PetscMPICUDAAwarenessCheck
31105035670SJunchao Zhang   #include "cupminit.inc"
31205035670SJunchao Zhang #endif
31305035670SJunchao Zhang 
31405035670SJunchao Zhang #if defined(PETSC_HAVE_HIP)
31505035670SJunchao Zhang   typedef hipError_t                              cupmError_t;
31605035670SJunchao Zhang   typedef hipDeviceProp_t                         cupmDeviceProp;
31705035670SJunchao Zhang   #define cupmGetDeviceCount(x)                   hipGetDeviceCount(x)
31805035670SJunchao Zhang   #define cupmGetDevice(x)                        hipGetDevice(x)
31905035670SJunchao Zhang   #define cupmSetDevice(x)                        hipSetDevice(x)
32005035670SJunchao Zhang   #define cupmSetDeviceFlags(x)                   hipSetDeviceFlags(x)
32105035670SJunchao Zhang   #define cupmGetDeviceProperties(x,y)            hipGetDeviceProperties(x,y)
32205035670SJunchao Zhang   #define cupmGetLastError()                      hipGetLastError()
32305035670SJunchao Zhang   #define cupmDeviceMapHost                       hipDeviceMapHost
32405035670SJunchao Zhang   #define cupmSuccess                             hipSuccess
32505035670SJunchao Zhang   #define cupmErrorSetOnActiveProcess             hipErrorSetOnActiveProcess
32605035670SJunchao Zhang   #define CHKERRCUPM(x)                           CHKERRQ((x)==hipSuccess? 0:PETSC_ERR_LIB)
32705035670SJunchao Zhang   #define PetscCUPMBLASInitializeHandle()         0
32805035670SJunchao Zhang   #define PetscCUPMSOLVERDnInitializeHandle()     0
32905035670SJunchao Zhang   #define PetscCUPMInitialize                     PetscHIPInitialize
33005035670SJunchao Zhang   #define PetscCUPMInitialized                    PetscHIPInitialized
33105035670SJunchao Zhang   #define PetscCUPMInitializeCheck                PetscHIPInitializeCheck
33205035670SJunchao Zhang   #define PetscCUPMInitializeAndView              PetscHIPInitializeAndView
33305035670SJunchao Zhang   #define PetscCUPMSynchronize                    PetscHIPSynchronize
33405035670SJunchao Zhang   #define PetscNotUseCUPM                         PetscNotUseHIP
33505035670SJunchao Zhang   #define cupmOptionsStr                          "HIP options"
3364dcd27cbSJunchao Zhang   #define cupmSetDeviceStr                        "-hip_device"
33705035670SJunchao Zhang   #define cupmViewStr                             "-hip_view"
33805035670SJunchao Zhang   #define cupmSynchronizeStr                      "-hip_synchronize"
33905035670SJunchao Zhang   #define PetscCUPMInitializeStr                  "PetscHIPInitialize"
34005035670SJunchao Zhang   #define PetscOptionsCheckCUPM                   PetscOptionsCheckHIP
34105035670SJunchao Zhang   #define PetscMPICUPMAwarenessCheck              PetscMPIHIPAwarenessCheck
34205035670SJunchao Zhang   #include "cupminit.inc"
34305035670SJunchao Zhang #endif
34405035670SJunchao Zhang 
34557171095SVaclav Hapla PETSC_INTERN PetscErrorCode  PetscOptionsCheckInitial_Private(const char help[])
346e5c89e4eSSatish Balay {
347e5ed2c37SJose E. Roman   char              string[64];
348e5c89e4eSSatish Balay   MPI_Comm          comm = PETSC_COMM_WORLD;
34905035670SJunchao Zhang   PetscBool         flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flag,hasHelp,logView;
350e5c89e4eSSatish Balay   PetscErrorCode    ierr;
35167584ceeSBarry Smith   PetscReal         si;
35228559dc8SJed Brown   PetscInt          intensity;
353e5c89e4eSSatish Balay   int               i;
354e5c89e4eSSatish Balay   PetscMPIInt       rank;
355d314f959SVaclav Hapla   char              version[256];
3562479a3a6SBarry Smith #if defined(PETSC_USE_LOG)
357e5ed2c37SJose E. Roman   char              mname[PETSC_MAX_PATH_LEN];
3582479a3a6SBarry Smith   PetscViewerFormat format;
35967584ceeSBarry Smith   PetscBool         flg4 = PETSC_FALSE;
36067584ceeSBarry Smith #endif
3617381773fSBarry Smith 
362e5c89e4eSSatish Balay   PetscFunctionBegin;
36342218b76SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
364e5c89e4eSSatish Balay 
36567584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY)
36692f119d6SBarry Smith   if (!(PETSC_RUNNING_ON_VALGRIND)) {
367e5c89e4eSSatish Balay     /*
368e5c89e4eSSatish Balay       Setup the memory management; support for tracing malloc() usage
369e5c89e4eSSatish Balay     */
370244bdbccSBarry Smith     PetscBool         mdebug = PETSC_FALSE, eachcall = PETSC_FALSE, initializenan = PETSC_FALSE, mlog = PETSC_FALSE;
37192f119d6SBarry Smith 
37276bd3646SJed Brown     if (PetscDefined(USE_DEBUG)) {
37392f119d6SBarry Smith       mdebug        = PETSC_TRUE;
37492f119d6SBarry Smith       initializenan = PETSC_TRUE;
37592f119d6SBarry Smith       ierr   = PetscOptionsHasName(NULL,NULL,"-malloc_test",&flg1);CHKERRQ(ierr);
37676bd3646SJed Brown     } else {
37792f119d6SBarry Smith       /* don't warn about unused option */
37892f119d6SBarry Smith       ierr = PetscOptionsHasName(NULL,NULL,"-malloc_test",&flg1);CHKERRQ(ierr);
37992f119d6SBarry Smith       flg1 = PETSC_FALSE;
38076bd3646SJed Brown     }
38179dccf82SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_debug",&flg2,&flg3);CHKERRQ(ierr);
38292f119d6SBarry Smith     if (flg1 || flg2) {
38392f119d6SBarry Smith       mdebug        = PETSC_TRUE;
38492f119d6SBarry Smith       eachcall      = PETSC_TRUE;
38592f119d6SBarry Smith       initializenan = PETSC_TRUE;
38679dccf82SBarry Smith     } else if (flg3 && !flg2) {
38779dccf82SBarry Smith       mdebug        = PETSC_FALSE;
38879dccf82SBarry Smith       eachcall      = PETSC_FALSE;
38979dccf82SBarry Smith       initializenan = PETSC_FALSE;
390e5c89e4eSSatish Balay     }
39192f119d6SBarry Smith 
392608c71bfSMatthew G. Knepley     ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_requested_size",&flg1,&flg2);CHKERRQ(ierr);
393608c71bfSMatthew G. Knepley     if (flg2) {ierr = PetscMallocLogRequestedSizeSet(flg1);CHKERRQ(ierr);}
394608c71bfSMatthew G. Knepley 
39592f119d6SBarry Smith     ierr = PetscOptionsHasName(NULL,NULL,"-malloc_view",&mlog);CHKERRQ(ierr);
39692f119d6SBarry Smith     if (mlog) {
39792f119d6SBarry Smith       mdebug = PETSC_TRUE;
39892f119d6SBarry Smith     }
39979dccf82SBarry Smith     /* the next line is deprecated */
40092f119d6SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&mdebug,NULL);CHKERRQ(ierr);
40192f119d6SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_dump",&mdebug,NULL);CHKERRQ(ierr);
40292f119d6SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&mdebug,NULL);CHKERRQ(ierr);
40392f119d6SBarry Smith     if (mdebug) {
40492f119d6SBarry Smith       ierr = PetscMallocSetDebug(eachcall,initializenan);CHKERRQ(ierr);
40592f119d6SBarry Smith     }
40692f119d6SBarry Smith     if (mlog) {
40792f119d6SBarry Smith       PetscReal logthreshold = 0;
40892f119d6SBarry Smith       ierr = PetscOptionsGetReal(NULL,NULL,"-malloc_view_threshold",&logthreshold,NULL);CHKERRQ(ierr);
40992f119d6SBarry Smith       ierr = PetscMallocViewSet(logthreshold);CHKERRQ(ierr);
41092f119d6SBarry Smith     }
4118b254c29SBarry Smith #if defined(PETSC_USE_LOG)
4128b254c29SBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&PetscLogMemory,NULL);CHKERRQ(ierr);
4138b254c29SBarry Smith #endif
41492f119d6SBarry Smith   }
41592f119d6SBarry Smith 
416ba282f50SJed Brown   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_coalesce",&flg1,&flg2);CHKERRQ(ierr);
417ba282f50SJed Brown   if (flg2) {ierr = PetscMallocSetCoalesce(flg1);CHKERRQ(ierr);}
41890d69ab7SBarry Smith   flg1 = PETSC_FALSE;
4192f21b5c6SHong Zhang   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_hbw",&flg1,NULL);CHKERRQ(ierr);
420cf6f3811SHong Zhang   /* ignore this option if malloc is already set */
421cf6f3811SHong Zhang   if (flg1 && !petscsetmallocvisited) {ierr = PetscSetUseHBWMalloc_Private();CHKERRQ(ierr);}
422e5c89e4eSSatish Balay 
42390d69ab7SBarry Smith   flg1 = PETSC_FALSE;
424c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_info",&flg1,NULL);CHKERRQ(ierr);
4257783f70dSSatish Balay   if (!flg1) {
42690d69ab7SBarry Smith     flg1 = PETSC_FALSE;
427c5929fdfSBarry Smith     ierr = PetscOptionsGetBool(NULL,NULL,"-memory_view",&flg1,NULL);CHKERRQ(ierr);
4287783f70dSSatish Balay   }
429e5c89e4eSSatish Balay   if (flg1) {
430e5c89e4eSSatish Balay     ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr);
431e5c89e4eSSatish Balay   }
43267584ceeSBarry Smith #endif
433e5c89e4eSSatish Balay 
434b4427426SBarry Smith #if defined(PETSC_USE_LOG)
435c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-objects_dump",&PetscObjectsLog);CHKERRQ(ierr);
436b4427426SBarry Smith #endif
43705df10baSBarry Smith 
438e5c89e4eSSatish Balay   /*
439e5c89e4eSSatish Balay       Set the display variable for graphics
440e5c89e4eSSatish Balay   */
441e5c89e4eSSatish Balay   ierr = PetscSetDisplay();CHKERRQ(ierr);
442e5c89e4eSSatish Balay 
44351dcc849SKerry Stevens   /*
44457171095SVaclav Hapla      Print main application help message
44557171095SVaclav Hapla   */
44657171095SVaclav Hapla   ierr = PetscOptionsHasHelp(NULL,&hasHelp);CHKERRQ(ierr);
44757171095SVaclav Hapla   if (help && hasHelp) {
44857171095SVaclav Hapla     ierr = PetscPrintf(comm,help);CHKERRQ(ierr);
44957171095SVaclav Hapla     ierr = PetscPrintf(comm,"----------------------------------------\n");CHKERRQ(ierr);
45057171095SVaclav Hapla   }
45157171095SVaclav Hapla 
45257171095SVaclav Hapla   /*
453e5c89e4eSSatish Balay       Print the PETSc version information
454e5c89e4eSSatish Balay   */
455d314f959SVaclav Hapla   ierr = PetscOptionsHasName(NULL,NULL,"-version",&flg1);CHKERRQ(ierr);
456d314f959SVaclav Hapla   if (flg1 || hasHelp) {
457e5c89e4eSSatish Balay     /*
458e5c89e4eSSatish Balay        Print "higher-level" package version message
459e5c89e4eSSatish Balay     */
460e5c89e4eSSatish Balay     if (PetscExternalVersionFunction) {
461e5c89e4eSSatish Balay       ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr);
462e5c89e4eSSatish Balay     }
463e5c89e4eSSatish Balay 
464a523d312SBarry Smith     ierr = PetscGetVersion(version,256);CHKERRQ(ierr);
465e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr);
466e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr);
467e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr);
46884e42920SBarry Smith     ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr);
469e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr);
470e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr);
471c0bb3764SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm,"----------------------------------------\n");CHKERRQ(ierr);
4727ca660e7SBarry Smith   }
4737ca660e7SBarry Smith 
47494941ca7SBarry Smith   /*
47594941ca7SBarry Smith        Print "higher-level" package help message
47694941ca7SBarry Smith   */
47757171095SVaclav Hapla   if (hasHelp) {
478d314f959SVaclav Hapla     PetscBool hasHelpIntro;
479d314f959SVaclav Hapla 
48094941ca7SBarry Smith     if (PetscExternalHelpFunction) {
48194941ca7SBarry Smith       ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr);
48294941ca7SBarry Smith     }
483d314f959SVaclav Hapla     ierr = PetscOptionsHasHelpIntro_Internal(NULL,&hasHelpIntro);CHKERRQ(ierr);
484d314f959SVaclav Hapla     if (hasHelpIntro) {
485bdb346e9SBarry Smith       ierr = PetscOptionsDestroyDefault();CHKERRQ(ierr);
486008a6e76SBarry Smith       ierr = PetscFreeMPIResources();CHKERRQ(ierr);
4877ca660e7SBarry Smith       ierr = MPI_Finalize();CHKERRQ(ierr);
4887ca660e7SBarry Smith       exit(0);
4897ca660e7SBarry Smith     }
490e5c89e4eSSatish Balay   }
491e5c89e4eSSatish Balay 
492e5c89e4eSSatish Balay   /*
493e5c89e4eSSatish Balay       Setup the error handling
494e5c89e4eSSatish Balay   */
49590d69ab7SBarry Smith   flg1 = PETSC_FALSE;
496c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_abort",&flg1,NULL);CHKERRQ(ierr);
497b59baad1SJed Brown   if (flg1) {
49842218b76SBarry Smith     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr);
49902c9f0b5SLisandro Dalcin     ierr = PetscPushErrorHandler(PetscAbortErrorHandler,NULL);CHKERRQ(ierr);
500b59baad1SJed Brown   }
50190d69ab7SBarry Smith   flg1 = PETSC_FALSE;
502c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_mpiabort",&flg1,NULL);CHKERRQ(ierr);
50302c9f0b5SLisandro Dalcin   if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,NULL);CHKERRQ(ierr);}
50490d69ab7SBarry Smith   flg1 = PETSC_FALSE;
505c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_on_error",&flg1,NULL);CHKERRQ(ierr);
506e5c89e4eSSatish Balay   if (flg1) {
507d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr);
508e5c89e4eSSatish Balay   }
509e781f417SStefano Zampini   /* experimental */
510e781f417SStefano Zampini   flg1 = PETSC_FALSE;
511e781f417SStefano Zampini   ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_error_string",&flg1,NULL);CHKERRQ(ierr);
512e781f417SStefano Zampini   if (flg1) {
513e781f417SStefano Zampini     MPI_Errhandler eh;
514e781f417SStefano Zampini 
515e781f417SStefano Zampini     ierr = MPI_Comm_create_errhandler(PetscMPI_Comm_eh,&eh);CHKERRQ(ierr);
516e781f417SStefano Zampini     ierr = MPI_Comm_set_errhandler(comm,eh);CHKERRQ(ierr);
517e781f417SStefano Zampini     ierr = MPI_Errhandler_free(&eh);CHKERRQ(ierr);
518e781f417SStefano Zampini   }
51990d69ab7SBarry Smith   flg1 = PETSC_FALSE;
520c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-no_signal_handler",&flg1,NULL);CHKERRQ(ierr);
5218d359177SBarry Smith   if (!flg1) {ierr = PetscPushSignalHandler(PetscSignalHandlerDefault,(void*)0);CHKERRQ(ierr);}
522e5c89e4eSSatish Balay 
523e5c89e4eSSatish Balay   /*
524e5c89e4eSSatish Balay       Setup debugger information
525e5c89e4eSSatish Balay   */
526e5c89e4eSSatish Balay   ierr = PetscSetDefaultDebugger();CHKERRQ(ierr);
527589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_attach_debugger",string,sizeof(string),&flg1);CHKERRQ(ierr);
528e5c89e4eSSatish Balay   if (flg1) {
529e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
530e5c89e4eSSatish Balay 
531e5c89e4eSSatish Balay     ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
53233c7d699SBarry Smith     ierr = MPI_Comm_create_errhandler(Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr);
533d54338ecSKarl Rupp     ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
53402c9f0b5SLisandro Dalcin     ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,NULL);CHKERRQ(ierr);
535e5c89e4eSSatish Balay   }
536589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-debug_terminal",string,sizeof(string),&flg1);CHKERRQ(ierr);
5375e96ac45SJed Brown   if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); }
538589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-start_in_debugger",string,sizeof(string),&flg1);CHKERRQ(ierr);
539589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-stop_for_debugger",string,sizeof(string),&flg2);CHKERRQ(ierr);
540e5c89e4eSSatish Balay   if (flg1 || flg2) {
541e5c89e4eSSatish Balay     PetscMPIInt    size;
542*bf4d2887SBarry Smith     PetscInt       lsize,*ranks;
543e5c89e4eSSatish Balay     MPI_Errhandler err_handler;
544e5c89e4eSSatish Balay     /*
545e5c89e4eSSatish Balay        we have to make sure that all processors have opened
546e5c89e4eSSatish Balay        connections to all other processors, otherwise once the
547e5c89e4eSSatish Balay        debugger has stated it is likely to receive a SIGUSR1
548e5c89e4eSSatish Balay        and kill the program.
549e5c89e4eSSatish Balay     */
55042218b76SBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
551e5c89e4eSSatish Balay     if (size > 2) {
552533163c2SBarry Smith       PetscMPIInt dummy = 0;
553e5c89e4eSSatish Balay       MPI_Status  status;
554e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
555e5c89e4eSSatish Balay         if (rank != i) {
55642218b76SBarry Smith           ierr = MPI_Send(&dummy,1,MPI_INT,i,109,comm);CHKERRQ(ierr);
557e5c89e4eSSatish Balay         }
558e5c89e4eSSatish Balay       }
559e5c89e4eSSatish Balay       for (i=0; i<size; i++) {
560e5c89e4eSSatish Balay         if (rank != i) {
56142218b76SBarry Smith           ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,comm,&status);CHKERRQ(ierr);
562e5c89e4eSSatish Balay         }
563e5c89e4eSSatish Balay       }
564e5c89e4eSSatish Balay     }
565e5c89e4eSSatish Balay     /* check if this processor node should be in debugger */
566*bf4d2887SBarry Smith     ierr  = PetscMalloc1(size,&ranks);CHKERRQ(ierr);
567e5c89e4eSSatish Balay     lsize = size;
568*bf4d2887SBarry Smith     /* Deprecated in 3.14 */
569*bf4d2887SBarry Smith     ierr  = PetscOptionsGetIntArray(NULL,NULL,"-debugger_nodes",ranks,&lsize,&flag);CHKERRQ(ierr);
570*bf4d2887SBarry Smith     if (flag) {
571*bf4d2887SBarry Smith       const char * const quietopt="-options_suppress_deprecated_warnings";
572*bf4d2887SBarry Smith       char               msg[4096];
573*bf4d2887SBarry Smith       PetscBool          quiet = PETSC_FALSE;
574*bf4d2887SBarry Smith 
575*bf4d2887SBarry Smith       ierr = PetscOptionsGetBool(NULL,NULL,quietopt,&quiet,NULL);CHKERRQ(ierr);
576*bf4d2887SBarry Smith       if (!quiet) {
577*bf4d2887SBarry Smith         ierr = PetscStrcpy(msg,"** PETSc DEPRECATION WARNING ** : the option ");CHKERRQ(ierr);
578*bf4d2887SBarry Smith         ierr = PetscStrcat(msg,"-debugger_nodes");CHKERRQ(ierr);
579*bf4d2887SBarry Smith         ierr = PetscStrcat(msg," is deprecated as of version ");CHKERRQ(ierr);
580*bf4d2887SBarry Smith         ierr = PetscStrcat(msg,"3.14");CHKERRQ(ierr);
581*bf4d2887SBarry Smith         ierr = PetscStrcat(msg," and will be removed in a future release.");CHKERRQ(ierr);
582*bf4d2887SBarry Smith         ierr = PetscStrcat(msg," Please use the option ");CHKERRQ(ierr);
583*bf4d2887SBarry Smith         ierr = PetscStrcat(msg,"-debugger_ranks");CHKERRQ(ierr);
584*bf4d2887SBarry Smith         ierr = PetscStrcat(msg," instead.");CHKERRQ(ierr);
585*bf4d2887SBarry Smith         ierr = PetscStrcat(msg," (Silence this warning with ");CHKERRQ(ierr);
586*bf4d2887SBarry Smith         ierr = PetscStrcat(msg,quietopt);CHKERRQ(ierr);
587*bf4d2887SBarry Smith         ierr = PetscStrcat(msg,")\n");CHKERRQ(ierr);
588*bf4d2887SBarry Smith         ierr = PetscPrintf(comm,msg);CHKERRQ(ierr);
589*bf4d2887SBarry Smith       }
590*bf4d2887SBarry Smith     } else {
591*bf4d2887SBarry Smith       lsize = size;
592*bf4d2887SBarry Smith       ierr  = PetscOptionsGetIntArray(NULL,NULL,"-debugger_ranks",ranks,&lsize,&flag);CHKERRQ(ierr);
593*bf4d2887SBarry Smith     }
594e5c89e4eSSatish Balay     if (flag) {
595e5c89e4eSSatish Balay       for (i=0; i<lsize; i++) {
596*bf4d2887SBarry Smith         if (ranks[i] == rank) { flag = PETSC_FALSE; break; }
597e5c89e4eSSatish Balay       }
598e5c89e4eSSatish Balay     }
599e5c89e4eSSatish Balay     if (!flag) {
600e5c89e4eSSatish Balay       ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr);
60102c9f0b5SLisandro Dalcin       ierr = PetscPushErrorHandler(PetscAbortErrorHandler,NULL);CHKERRQ(ierr);
602e5c89e4eSSatish Balay       if (flg1) {
603e5c89e4eSSatish Balay         ierr = PetscAttachDebugger();CHKERRQ(ierr);
604e5c89e4eSSatish Balay       } else {
605e5c89e4eSSatish Balay         ierr = PetscStopForDebugger();CHKERRQ(ierr);
606e5c89e4eSSatish Balay       }
60733c7d699SBarry Smith       ierr = MPI_Comm_create_errhandler(Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr);
608d54338ecSKarl Rupp       ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr);
6092a2a2941SBarry Smith     } else {
6102a2a2941SBarry Smith       ierr = PetscWaitOnError();CHKERRQ(ierr);
611e5c89e4eSSatish Balay     }
612*bf4d2887SBarry Smith     ierr = PetscFree(ranks);CHKERRQ(ierr);
613e5c89e4eSSatish Balay   }
614e5c89e4eSSatish Balay 
615589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-on_error_emacs",emacsmachinename,sizeof(emacsmachinename),&flg1);CHKERRQ(ierr);
616cb9801acSJed Brown   if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);}
617e5c89e4eSSatish Balay 
618e5c89e4eSSatish Balay   /*
619e5c89e4eSSatish Balay         Setup profiling and logging
620e5c89e4eSSatish Balay   */
6216cf91177SBarry Smith #if defined(PETSC_USE_INFO)
6228bb29257SSatish Balay   {
623e94e781bSJacob Faibussowitsch     ierr = PetscInfoSetFromOptions(NULL);CHKERRQ(ierr);
624e5c89e4eSSatish Balay   }
625865f6aa8SSatish Balay #endif
626aba4c478SBarry Smith   ierr = PetscDetermineInitialFPTrap();
627cc9df77eSBarry Smith   flg1 = PETSC_FALSE;
628cc9df77eSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-fp_trap",&flg1,&flag);CHKERRQ(ierr);
629cc9df77eSBarry Smith   if (flag) {ierr = PetscSetFPTrap((PetscFPTrap)flg1);CHKERRQ(ierr);}
630cc9df77eSBarry Smith   ierr = PetscOptionsGetInt(NULL,NULL,"-check_pointer_intensity",&intensity,&flag);CHKERRQ(ierr);
631cc9df77eSBarry Smith   if (flag) {ierr = PetscCheckPointerSetIntensity(intensity);CHKERRQ(ierr);}
632865f6aa8SSatish Balay #if defined(PETSC_USE_LOG)
633865f6aa8SSatish Balay   mname[0] = 0;
634589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-history",mname,sizeof(mname),&flg1);CHKERRQ(ierr);
635865f6aa8SSatish Balay   if (flg1) {
636865f6aa8SSatish Balay     if (mname[0]) {
637f3dea69dSBarry Smith       ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr);
638865f6aa8SSatish Balay     } else {
639706d7a88SBarry Smith       ierr = PetscOpenHistoryFile(NULL,&petsc_history);CHKERRQ(ierr);
640865f6aa8SSatish Balay     }
641865f6aa8SSatish Balay   }
642217044c2SLisandro Dalcin 
643217044c2SLisandro Dalcin   ierr = PetscOptionsGetBool(NULL,NULL,"-log_sync",&PetscLogSyncOn,NULL);CHKERRQ(ierr);
644217044c2SLisandro Dalcin 
645e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
64690d69ab7SBarry Smith   flg1 = PETSC_FALSE;
647c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_mpe",&flg1);CHKERRQ(ierr);
648495fc317SBarry Smith   if (flg1) {ierr = PetscLogMPEBegin();CHKERRQ(ierr);}
649e5c89e4eSSatish Balay #endif
65090d69ab7SBarry Smith   flg1 = PETSC_FALSE;
65190d69ab7SBarry Smith   flg3 = PETSC_FALSE;
652c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-log_all",&flg1,NULL);CHKERRQ(ierr);
653c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
654e5c89e4eSSatish Balay   if (flg1)                      { ierr = PetscLogAllBegin();CHKERRQ(ierr); }
655bb1d7374SBarry Smith   else if (flg3)                 { ierr = PetscLogDefaultBegin();CHKERRQ(ierr);}
656e5c89e4eSSatish Balay 
657589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-log_trace",mname,sizeof(mname),&flg1);CHKERRQ(ierr);
658e5c89e4eSSatish Balay   if (flg1) {
659e5c89e4eSSatish Balay     char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN];
660e5c89e4eSSatish Balay     FILE *file;
661e5c89e4eSSatish Balay     if (mname[0]) {
6622e924ca5SSatish Balay       PetscSNPrintf(name,PETSC_MAX_PATH_LEN,"%s.%d",mname,rank);
663e5c89e4eSSatish Balay       ierr = PetscFixFilename(name,fname);CHKERRQ(ierr);
664e5c89e4eSSatish Balay       file = fopen(fname,"w");
665f3dea69dSBarry Smith       if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname);
666a297a907SKarl Rupp     } else file = PETSC_STDOUT;
667e5c89e4eSSatish Balay     ierr = PetscLogTraceBegin(file);CHKERRQ(ierr);
668e5c89e4eSSatish Balay   }
669bb1d7374SBarry Smith 
67016413a6aSBarry Smith   ierr = PetscOptionsGetViewer(comm,NULL,NULL,"-log_view",NULL,&format,&flg4);CHKERRQ(ierr);
671bb1d7374SBarry Smith   if (flg4) {
672bb1d7374SBarry Smith     if (format == PETSC_VIEWER_ASCII_XML) {
673bb1d7374SBarry Smith       ierr = PetscLogNestedBegin();CHKERRQ(ierr);
674bb1d7374SBarry Smith     } else {
675bb1d7374SBarry Smith       ierr = PetscLogDefaultBegin();CHKERRQ(ierr);
676bb1d7374SBarry Smith     }
677eccbb886SLisandro Dalcin   }
678eccbb886SLisandro Dalcin   if (flg4 && format == PETSC_VIEWER_ASCII_XML) {
679eccbb886SLisandro Dalcin     PetscReal threshold = PetscRealConstant(0.01);
680eccbb886SLisandro Dalcin     ierr = PetscOptionsGetReal(NULL,NULL,"-log_threshold",&threshold,&flg1);CHKERRQ(ierr);
681eccbb886SLisandro Dalcin     if (flg1) {ierr = PetscLogSetThreshold((PetscLogDouble)threshold,NULL);CHKERRQ(ierr);}
682bb1d7374SBarry Smith   }
683e5c89e4eSSatish Balay #endif
684e5c89e4eSSatish Balay 
685c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-saws_options",&PetscOptionsPublish,NULL);CHKERRQ(ierr);
68605035670SJunchao Zhang   ierr = PetscOptionsGetBool(NULL,NULL,"-use_gpu_aware_mpi",&use_gpu_aware_mpi,NULL);CHKERRQ(ierr);
6878d4e85a7SStefano Zampini   /*
68805035670SJunchao Zhang     If collecting logging information, by default, wait for device to complete its operations
6898d4e85a7SStefano Zampini     before returning to the CPU in order to get accurate timings of each event
6908d4e85a7SStefano Zampini   */
691cae85d06SJunchao Zhang   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&logView);CHKERRQ(ierr);
69205035670SJunchao Zhang   if (!logView) {ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&logView);CHKERRQ(ierr);}
6938d4e85a7SStefano Zampini 
69405035670SJunchao Zhang #if defined(PETSC_HAVE_CUDA)
69505035670SJunchao Zhang   ierr = PetscOptionsCheckCUDA(logView);CHKERRQ(ierr);
69605035670SJunchao Zhang #endif
697cae85d06SJunchao Zhang 
69805035670SJunchao Zhang #if defined(PETSC_HAVE_HIP)
69905035670SJunchao Zhang   ierr = PetscOptionsCheckHIP(logView);CHKERRQ(ierr);
7004b0a73b2SBarry Smith #endif
7014b0a73b2SBarry Smith 
702e5c89e4eSSatish Balay   /*
703e5c89e4eSSatish Balay        Print basic help message
704e5c89e4eSSatish Balay   */
70557171095SVaclav Hapla   if (hasHelp) {
706e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr);
707d314f959SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -version: prints PETSc version\n");CHKERRQ(ierr);
708d314f959SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -help intro: prints example description and PETSc version, and exits\n");CHKERRQ(ierr);
709d314f959SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -help: prints example description, PETSc version, and available options for used routines\n");CHKERRQ(ierr);
710301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr);
711301d30feSBarry Smith     ierr = (*PetscHelpPrintf)(comm,"       only when run in the debugger\n");CHKERRQ(ierr);
712e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
713e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start the debugger in new xterm\n");CHKERRQ(ierr);
714e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       unless noxterm is given\n");CHKERRQ(ierr);
715e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr);
716e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"       start all processes in the debugger\n");CHKERRQ(ierr);
717e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr);
718e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"    emacs jumps to error file\n");CHKERRQ(ierr);
719*bf4d2887SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -debugger_ranks [n1,n2,..] Ranks to start in debugger\n");CHKERRQ(ierr);
720e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr);
721e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr);
722e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"                      waits the delay for you to attach\n");CHKERRQ(ierr);
7231cda70a7SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -display display: Location where X window graphics and debuggers are displayed\n");CHKERRQ(ierr);
724e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr);
725e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr);
726e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr);
727e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm,"           note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr);
728e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr);
72979dccf82SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc: use PETSc error checking malloc (deprecated, use -malloc_debug)\n");CHKERRQ(ierr);
73079dccf82SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use PETSc error checking malloc (deprecated, use -malloc_debug no)\n");CHKERRQ(ierr);
7314161f2a3SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr);
73292f119d6SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_view <optional filename>: keeps log of all memory allocations, displays in PetscFinalize()\n");CHKERRQ(ierr);
73379dccf82SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -malloc_debug <true or false>: enables or disables extended checking for memory corruption\n");CHKERRQ(ierr);
73426a7e8d4SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -options_view: dump list of options inputted\n");CHKERRQ(ierr);
735e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr);
736e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr);
737e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr);
738e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr);
739a8c7a070SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr);
7400841954dSBarry Smith     ierr = (*PetscHelpPrintf)(comm," -memory_view: print memory usage at end of run\n");CHKERRQ(ierr);
741e5c89e4eSSatish Balay #if defined(PETSC_USE_LOG)
742e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr);
743185ae32fSMatthew G. Knepley     ierr = (*PetscHelpPrintf)(comm," -log_view [:filename:[format]]: logging objects and events\n");CHKERRQ(ierr);
744e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr);
74556071f75SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -log_exclude <list,of,classnames>: exclude given classes from logging\n");CHKERRQ(ierr);
746e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE)
747495fc317SBarry Smith     ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through Jumpshot\n");CHKERRQ(ierr);
748e5c89e4eSSatish Balay #endif
749e94e781bSJacob Faibussowitsch #endif
750e94e781bSJacob Faibussowitsch #if defined(PETSC_USE_INFO)
751fe9b927eSVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -info [filename][:[~]<list,of,classnames>[:[~]self]]: print verbose information\n");CHKERRQ(ierr);
752e5c89e4eSSatish Balay #endif
753e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr);
754c5b5d8d5SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -options_monitor: monitor options to standard output, including that set previously e.g. in option files\n");CHKERRQ(ierr);
755c5b5d8d5SVaclav Hapla     ierr = (*PetscHelpPrintf)(comm," -options_monitor_cancel: cancels all hardwired option monitors\n");CHKERRQ(ierr);
756e5c89e4eSSatish Balay     ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr);
757e5c89e4eSSatish Balay   }
758e5c89e4eSSatish Balay 
75974ba8654SBarry Smith #if defined(PETSC_HAVE_POPEN)
76074ba8654SBarry Smith   {
76174ba8654SBarry Smith   char machine[128];
762589a23caSBarry Smith   ierr = PetscOptionsGetString(NULL,NULL,"-popen_machine",machine,sizeof(machine),&flg1);CHKERRQ(ierr);
76374ba8654SBarry Smith   if (flg1) {
76474ba8654SBarry Smith     ierr = PetscPOpenSetMachine(machine);CHKERRQ(ierr);
76574ba8654SBarry Smith   }
76674ba8654SBarry Smith   }
76774ba8654SBarry Smith #endif
76874ba8654SBarry Smith 
769c5929fdfSBarry Smith   ierr = PetscOptionsGetReal(NULL,NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr);
770e5c89e4eSSatish Balay   if (flg1) {
771e5c89e4eSSatish Balay     ierr = PetscSleep(si);CHKERRQ(ierr);
772e5c89e4eSSatish Balay   }
773e5c89e4eSSatish Balay 
774fdc842d1SBarry Smith #if defined(PETSC_HAVE_VIENNACL)
775c5929fdfSBarry Smith   ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr);
776f14045dbSBarry Smith   if (!flg3) {
777c5929fdfSBarry Smith     ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&flg3);CHKERRQ(ierr);
778f14045dbSBarry Smith   }
779c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(NULL,NULL,"-viennacl_synchronize",&flg3,NULL);CHKERRQ(ierr);
780f14045dbSBarry Smith   PetscViennaCLSynchronize = flg3;
781fdc842d1SBarry Smith   ierr = PetscViennaCLInit();CHKERRQ(ierr);
7824cf1874eSKarl Rupp #endif
78382f73ecaSAlejandro Lamas Daviña 
784fdc842d1SBarry Smith   /*
785fdc842d1SBarry Smith      Creates the logging data structures; this is enabled even if logging is not turned on
786fdc842d1SBarry Smith      This is the last thing we do before returning to the user code to prevent having the
787fdc842d1SBarry Smith      logging numbers contaminated by any startup time associated with MPI and the GPUs
788fdc842d1SBarry Smith   */
789fdc842d1SBarry Smith #if defined(PETSC_USE_LOG)
790fdc842d1SBarry Smith   ierr = PetscLogInitialize();CHKERRQ(ierr);
791f0a7718cSKarl Rupp #endif
792f0a7718cSKarl Rupp 
793e5c89e4eSSatish Balay   PetscFunctionReturn(0);
794e5c89e4eSSatish Balay }
795