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> 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 23d9b72601SDominic Meiser #if defined(PETSC_HAVE_CUDA) 24d9b72601SDominic Meiser #include <cuda_runtime.h> 257fd2f626SJose E. Roman #include <petsccublas.h> 26d9b72601SDominic Meiser #endif 27555d055bSBarry Smith 28f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL) 29f0a7718cSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLInit(); 30f0a7718cSKarl Rupp #endif 31f0a7718cSKarl Rupp 32e5c89e4eSSatish Balay /* ------------------------Nasty global variables -------------------------------*/ 33e5c89e4eSSatish Balay /* 34e5c89e4eSSatish Balay Indicates if PETSc started up MPI, or it was 35e5c89e4eSSatish Balay already started before PETSc was initialized. 36e5c89e4eSSatish Balay */ 370cbf60d0SJose E. Roman PetscBool PetscBeganMPI = PETSC_FALSE; 380cbf60d0SJose E. Roman PetscBool PetscInitializeCalled = PETSC_FALSE; 390cbf60d0SJose E. Roman PetscBool PetscFinalizeCalled = PETSC_FALSE; 40bffec52eSKarl Rupp PetscBool PetscCUDAInitialized = PETSC_FALSE; 41d6f2c3cbSBarry Smith 427087cfbeSBarry Smith PetscMPIInt PetscGlobalRank = -1; 437087cfbeSBarry Smith PetscMPIInt PetscGlobalSize = -1; 44ba61063dSBarry Smith 4550f81f78SJed Brown #if defined(PETSC_HAVE_COMPLEX) 46e5c89e4eSSatish Balay #if defined(PETSC_COMPLEX_INSTANTIATE) 47e5c89e4eSSatish Balay template <> class std::complex<double>; /* instantiate complex template class */ 48e5c89e4eSSatish Balay #endif 492c876bd9SBarry Smith #if !defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX) 50500d8756SSatish Balay MPI_Datatype MPIU_C_DOUBLE_COMPLEX; 51500d8756SSatish Balay MPI_Datatype MPIU_C_COMPLEX; 522c876bd9SBarry Smith #endif 538619c96cSJed Brown 548619c96cSJed Brown /*MC 558619c96cSJed Brown PETSC_i - the imaginary number i 568619c96cSJed Brown 578619c96cSJed Brown Synopsis: 588619c96cSJed Brown #include <petscsys.h> 598619c96cSJed Brown PetscComplex PETSC_i; 608619c96cSJed Brown 618619c96cSJed Brown Level: beginner 628619c96cSJed Brown 638619c96cSJed Brown Note: 648cd53115SBarry Smith Complex numbers are automatically available if PETSc located a working complex implementation 658619c96cSJed Brown 668619c96cSJed Brown .seealso: PetscRealPart(), PetscImaginaryPart(), PetscRealPartComplex(), PetscImaginaryPartComplex() 678619c96cSJed Brown M*/ 6850f81f78SJed Brown PetscComplex PETSC_i; 69e5c89e4eSSatish Balay #endif 70ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 71c90a1750SBarry Smith MPI_Datatype MPIU___FLOAT128 = 0; 727c2de775SJed Brown #if defined(PETSC_HAVE_COMPLEX) 738c764dc5SJose Roman MPI_Datatype MPIU___COMPLEX128 = 0; 748c764dc5SJose Roman #endif 75570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16) 76570b7f6dSBarry Smith MPI_Datatype MPIU___FP16 = 0; 77c90a1750SBarry Smith #endif 787087cfbeSBarry Smith MPI_Datatype MPIU_2SCALAR = 0; 790354ff80SSatish Balay #if defined(PETSC_USE_64BIT_INDICES) 807087cfbeSBarry Smith MPI_Datatype MPIU_2INT = 0; 8144041f26SJed Brown #endif 828ad47952SJed Brown MPI_Datatype MPIU_BOOL; 838ad47952SJed Brown MPI_Datatype MPIU_ENUM; 847cdaf61dSJed Brown MPI_Datatype MPIU_FORTRANADDR; 85e316c87fSJed Brown MPI_Datatype MPIU_SIZE_T; 8675567043SBarry Smith 87e5c89e4eSSatish Balay /* 88e5c89e4eSSatish Balay Function that is called to display all error messages 89e5c89e4eSSatish Balay */ 907087cfbeSBarry Smith PetscErrorCode (*PetscErrorPrintf)(const char [],...) = PetscErrorPrintfDefault; 917087cfbeSBarry Smith PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...) = PetscHelpPrintfDefault; 927087cfbeSBarry Smith PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list) = PetscVFPrintfDefault; 93bab1f7e6SVictor Minden /* 944cf1874eSKarl Rupp This is needed to turn on/off GPU synchronization 958b5db460SBarry Smith */ 964cf1874eSKarl Rupp PetscBool PetscViennaCLSynchronize = PETSC_FALSE; 9782f73ecaSAlejandro Lamas Daviña PetscBool PetscCUDASynchronize = PETSC_FALSE; 98bab1f7e6SVictor Minden 99e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/ 100e5c89e4eSSatish Balay /* 101e5c89e4eSSatish Balay Optional file where all PETSc output from various prints is saved 102e5c89e4eSSatish Balay */ 10395c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history; 1040298fd71SBarry Smith FILE *petsc_history = NULL; 105e5c89e4eSSatish Balay 1067087cfbeSBarry Smith PetscErrorCode PetscOpenHistoryFile(const char filename[],FILE **fd) 107e5c89e4eSSatish Balay { 108e5c89e4eSSatish Balay PetscErrorCode ierr; 109e5c89e4eSSatish Balay PetscMPIInt rank,size; 110e5c89e4eSSatish Balay char pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64]; 111e5c89e4eSSatish Balay char version[256]; 112e5c89e4eSSatish Balay 113e5c89e4eSSatish Balay PetscFunctionBegin; 114e5c89e4eSSatish Balay ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 115e5c89e4eSSatish Balay if (!rank) { 116e5c89e4eSSatish Balay char arch[10]; 117f56c2debSBarry Smith int err; 118f56c2debSBarry Smith 119e5c89e4eSSatish Balay ierr = PetscGetArchType(arch,10);CHKERRQ(ierr); 120e5c89e4eSSatish Balay ierr = PetscGetDate(date,64);CHKERRQ(ierr); 121a523d312SBarry Smith ierr = PetscGetVersion(version,256);CHKERRQ(ierr); 122e5c89e4eSSatish Balay ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 123e5c89e4eSSatish Balay if (filename) { 124e5c89e4eSSatish Balay ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr); 125e5c89e4eSSatish Balay } else { 126e5c89e4eSSatish Balay ierr = PetscGetHomeDirectory(pfile,240);CHKERRQ(ierr); 127e5c89e4eSSatish Balay ierr = PetscStrcat(pfile,"/.petschistory");CHKERRQ(ierr); 128e5c89e4eSSatish Balay ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr); 129e5c89e4eSSatish Balay } 130e5c89e4eSSatish Balay 131a297a907SKarl Rupp *fd = fopen(fname,"a"); 132a297a907SKarl Rupp if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname); 133a297a907SKarl Rupp 134e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 135e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr); 136e5c89e4eSSatish Balay ierr = PetscGetProgramName(pname,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 137e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr); 138e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 139a297a907SKarl Rupp 140f56c2debSBarry Smith err = fflush(*fd); 141e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 142e5c89e4eSSatish Balay } 143e5c89e4eSSatish Balay PetscFunctionReturn(0); 144e5c89e4eSSatish Balay } 145e5c89e4eSSatish Balay 14695c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscCloseHistoryFile(FILE **fd) 147e5c89e4eSSatish Balay { 148e5c89e4eSSatish Balay PetscErrorCode ierr; 149e5c89e4eSSatish Balay PetscMPIInt rank; 150e5c89e4eSSatish Balay char date[64]; 151f56c2debSBarry Smith int err; 152e5c89e4eSSatish Balay 153e5c89e4eSSatish Balay PetscFunctionBegin; 154e5c89e4eSSatish Balay ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 155e5c89e4eSSatish Balay if (!rank) { 156e5c89e4eSSatish Balay ierr = PetscGetDate(date,64);CHKERRQ(ierr); 157e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 158e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr); 159e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 160f56c2debSBarry Smith err = fflush(*fd); 161e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 162f56c2debSBarry Smith err = fclose(*fd); 163e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file"); 164e5c89e4eSSatish Balay } 165e5c89e4eSSatish Balay PetscFunctionReturn(0); 166e5c89e4eSSatish Balay } 167e5c89e4eSSatish Balay 168e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/ 169e5c89e4eSSatish Balay 170e5c89e4eSSatish Balay /* 171e5c89e4eSSatish Balay This is ugly and probably belongs somewhere else, but I want to 172e5c89e4eSSatish Balay be able to put a true MPI abort error handler with command line args. 173e5c89e4eSSatish Balay 174e5c89e4eSSatish Balay This is so MPI errors in the debugger will leave all the stack 1753c311c98SBarry Smith frames. The default MP_Abort() cleans up and exits thus providing no useful information 1763c311c98SBarry Smith in the debugger hence we call abort() instead of MPI_Abort(). 177e5c89e4eSSatish Balay */ 178e5c89e4eSSatish Balay 17933c7d699SBarry Smith void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag,...) 180e5c89e4eSSatish Balay { 181e5c89e4eSSatish Balay PetscFunctionBegin; 1823c311c98SBarry Smith (*PetscErrorPrintf)("MPI error %d\n",*flag); 183e5c89e4eSSatish Balay abort(); 184e5c89e4eSSatish Balay } 185e5c89e4eSSatish Balay 18633c7d699SBarry Smith void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag,...) 187e5c89e4eSSatish Balay { 188e5c89e4eSSatish Balay PetscErrorCode ierr; 189e5c89e4eSSatish Balay 190e5c89e4eSSatish Balay PetscFunctionBegin; 1913c311c98SBarry Smith (*PetscErrorPrintf)("MPI error %d\n",*flag); 192e5c89e4eSSatish Balay ierr = PetscAttachDebugger(); 193a297a907SKarl Rupp if (ierr) MPI_Abort(*comm,*flag); /* hopeless so get out */ 194e5c89e4eSSatish Balay } 195e5c89e4eSSatish Balay 1967381773fSBarry Smith #if defined(PETSC_HAVE_CUDA) 1977381773fSBarry Smith /*@C 1987381773fSBarry Smith PetscCUDAInitialize - Initializes the CUDA device and cuBLAS on the device 1997381773fSBarry Smith 2007381773fSBarry Smith Logically collective 2017381773fSBarry Smith 2027381773fSBarry Smith Input Parameter: 2037381773fSBarry Smith comm - the MPI communicator that will utilize the CUDA devices 2047381773fSBarry Smith 2057381773fSBarry Smith Options Database: 2069d90e779SBarry Smith + -cuda_initialize <default yes,no> - do the initialization in PetscInitialize(). If -cuda_initialize no is used then the default initialization is done automatically 2079d90e779SBarry Smith when the first CUDA call is made unless you call PetscCUDAInitialize() before any CUDA operations are performed 2087381773fSBarry Smith . -cuda_view - view information about the CUDA devices 2097381773fSBarry Smith . -cuda_synchronize - wait at the end of asynchronize CUDA calls so that their time gets credited to the current event; default with -log_view 2107381773fSBarry Smith - -cuda_set_device <gpu> - integer number of the device 2117381773fSBarry Smith 21290ea27d8SSatish Balay Level: beginner 21390ea27d8SSatish Balay 2147381773fSBarry Smith Notes: 2157381773fSBarry Smith Initializing cuBLAS takes about 1/2 second there it is done by default in PetscInitialize() before logging begins 2167381773fSBarry Smith 2177381773fSBarry Smith @*/ 2187381773fSBarry Smith PetscErrorCode PetscCUDAInitialize(MPI_Comm comm) 2197381773fSBarry Smith { 2207381773fSBarry Smith PetscErrorCode ierr; 2217381773fSBarry Smith PetscInt deviceOpt = 0; 2227381773fSBarry Smith PetscBool cuda_view_flag = PETSC_FALSE,flg; 2237381773fSBarry Smith struct cudaDeviceProp prop; 2247381773fSBarry Smith int devCount,device,devicecnt; 2257381773fSBarry Smith cudaError_t err = cudaSuccess; 2267381773fSBarry Smith PetscMPIInt rank,size; 2277381773fSBarry Smith 2287381773fSBarry Smith PetscFunctionBegin; 2297381773fSBarry Smith /* 2307381773fSBarry Smith If collecting logging information, by default, wait for GPU to complete its operations 2317381773fSBarry Smith before returning to the CPU in order to get accurate timings of each event 2327381773fSBarry Smith */ 2337381773fSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&PetscCUDASynchronize);CHKERRQ(ierr); 2347381773fSBarry Smith if (!PetscCUDASynchronize) { 2357381773fSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&PetscCUDASynchronize);CHKERRQ(ierr); 2367381773fSBarry Smith } 2377381773fSBarry Smith 2387381773fSBarry Smith ierr = PetscOptionsBegin(comm,NULL,"CUDA options","Sys");CHKERRQ(ierr); 2397381773fSBarry Smith ierr = PetscOptionsInt("-cuda_set_device","Set all MPI ranks to use the specified CUDA device",NULL,deviceOpt,&deviceOpt,&flg);CHKERRQ(ierr); 2407381773fSBarry Smith device = (int)deviceOpt; 2417381773fSBarry Smith ierr = PetscOptionsBool("-cuda_synchronize","Wait for the GPU to complete operations before returning to the CPU",NULL,PetscCUDASynchronize,&PetscCUDASynchronize,NULL);CHKERRQ(ierr); 2427381773fSBarry Smith ierr = PetscOptionsDeprecated("-cuda_show_devices","-cuda_view","3.12",NULL);CHKERRQ(ierr); 2437381773fSBarry Smith ierr = PetscOptionsName("-cuda_view","Display CUDA device information and assignments",NULL,&cuda_view_flag);CHKERRQ(ierr); 2447381773fSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 2457381773fSBarry Smith if (!PetscCUDAInitialized) { 2467381773fSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 2477381773fSBarry Smith 2487381773fSBarry Smith if (size>1 && !flg) { 2497381773fSBarry Smith /* check to see if we force multiple ranks to hit the same GPU */ 2507381773fSBarry Smith /* we're not using the same GPU on multiple MPI threads. So try to allocated different GPUs to different processes */ 2517381773fSBarry Smith 2527381773fSBarry Smith /* First get the device count */ 2537381773fSBarry Smith err = cudaGetDeviceCount(&devCount); 2547381773fSBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err)); 2557381773fSBarry Smith 2567381773fSBarry Smith /* next determine the rank and then set the device via a mod */ 2577381773fSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 2587381773fSBarry Smith device = rank % devCount; 2597381773fSBarry Smith } 2607381773fSBarry Smith err = cudaSetDevice(device); 2617381773fSBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err)); 2627381773fSBarry Smith 2637381773fSBarry Smith /* set the device flags so that it can map host memory */ 2647381773fSBarry Smith err = cudaSetDeviceFlags(cudaDeviceMapHost); 2657381773fSBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err)); 2667381773fSBarry Smith 2677381773fSBarry Smith ierr = PetscCUBLASInitializeHandle();CHKERRQ(ierr); 2683bbda9c2SStefano Zampini ierr = PetscCUSOLVERDnInitializeHandle();CHKERRQ(ierr); 2697381773fSBarry Smith PetscCUDAInitialized = PETSC_TRUE; 2707381773fSBarry Smith } 2717381773fSBarry Smith if (cuda_view_flag) { 2727381773fSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 2737381773fSBarry Smith err = cudaGetDeviceCount(&devCount); 2747381773fSBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err)); 2757381773fSBarry Smith for (devicecnt = 0; devicecnt < devCount; ++devicecnt) { 2767381773fSBarry Smith err = cudaGetDeviceProperties(&prop,devicecnt); 2777381773fSBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceProperties %s",cudaGetErrorString(err)); 2787381773fSBarry Smith ierr = PetscPrintf(comm, "CUDA device %d: %s\n", devicecnt, prop.name);CHKERRQ(ierr); 2797381773fSBarry Smith } 2807381773fSBarry Smith ierr = PetscSynchronizedPrintf(comm,"[%d] Using CUDA device %d.\n",rank,device);CHKERRQ(ierr); 2817381773fSBarry Smith ierr = PetscSynchronizedFlush(comm,PETSC_STDOUT);CHKERRQ(ierr); 2827381773fSBarry Smith } 2837381773fSBarry Smith PetscFunctionReturn(0); 2847381773fSBarry Smith } 2857381773fSBarry Smith #endif 2867381773fSBarry Smith 287e5c89e4eSSatish Balay /*@C 288e5c89e4eSSatish Balay PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one 289e5c89e4eSSatish Balay wishes a clean exit somewhere deep in the program. 290e5c89e4eSSatish Balay 291e5c89e4eSSatish Balay Collective on PETSC_COMM_WORLD 292e5c89e4eSSatish Balay 293e5c89e4eSSatish Balay Options Database Keys are the same as for PetscFinalize() 294e5c89e4eSSatish Balay 295e5c89e4eSSatish Balay Level: advanced 296e5c89e4eSSatish Balay 297e5c89e4eSSatish Balay Note: 298e5c89e4eSSatish Balay See PetscInitialize() for more general runtime options. 299e5c89e4eSSatish Balay 30088c29154SBarry Smith .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize() 301e5c89e4eSSatish Balay @*/ 3027087cfbeSBarry Smith PetscErrorCode PetscEnd(void) 303e5c89e4eSSatish Balay { 304e5c89e4eSSatish Balay PetscFunctionBegin; 305e5c89e4eSSatish Balay PetscFinalize(); 306e5c89e4eSSatish Balay exit(0); 307e5c89e4eSSatish Balay return 0; 308e5c89e4eSSatish Balay } 309e5c89e4eSSatish Balay 310ace3abfcSBarry Smith PetscBool PetscOptionsPublish = PETSC_FALSE; 31195c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseHBWMalloc_Private(void); 31295c0884eSLisandro Dalcin PETSC_INTERN PetscBool petscsetmallocvisited; 313e5c89e4eSSatish Balay static char emacsmachinename[256]; 314e5c89e4eSSatish Balay 315e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = 0; 316e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm) = 0; 317e5c89e4eSSatish Balay 318e5c89e4eSSatish Balay /*@C 319e5c89e4eSSatish Balay PetscSetHelpVersionFunctions - Sets functions that print help and version information 320e5c89e4eSSatish Balay before the PETSc help and version information is printed. Must call BEFORE PetscInitialize(). 321e5c89e4eSSatish Balay This routine enables a "higher-level" package that uses PETSc to print its messages first. 322e5c89e4eSSatish Balay 323e5c89e4eSSatish Balay Input Parameter: 3240298fd71SBarry Smith + help - the help function (may be NULL) 3250298fd71SBarry Smith - version - the version function (may be NULL) 326e5c89e4eSSatish Balay 327e5c89e4eSSatish Balay Level: developer 328e5c89e4eSSatish Balay 329e5c89e4eSSatish Balay @*/ 3307087cfbeSBarry Smith PetscErrorCode PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm)) 331e5c89e4eSSatish Balay { 332e5c89e4eSSatish Balay PetscFunctionBegin; 333e5c89e4eSSatish Balay PetscExternalHelpFunction = help; 334e5c89e4eSSatish Balay PetscExternalVersionFunction = version; 335e5c89e4eSSatish Balay PetscFunctionReturn(0); 336e5c89e4eSSatish Balay } 337e5c89e4eSSatish Balay 33805df10baSBarry Smith #if defined(PETSC_USE_LOG) 33995c0884eSLisandro Dalcin PETSC_INTERN PetscBool PetscObjectsLog; 34005df10baSBarry Smith #endif 34105df10baSBarry Smith 34295c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscOptionsCheckInitial_Private(void) 343e5c89e4eSSatish Balay { 344e5c89e4eSSatish Balay char string[64],mname[PETSC_MAX_PATH_LEN],*f; 345e5c89e4eSSatish Balay MPI_Comm comm = PETSC_COMM_WORLD; 34667584ceeSBarry Smith PetscBool flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flag; 347e5c89e4eSSatish Balay PetscErrorCode ierr; 34867584ceeSBarry Smith PetscReal si; 34928559dc8SJed Brown PetscInt intensity; 350e5c89e4eSSatish Balay int i; 351e5c89e4eSSatish Balay PetscMPIInt rank; 3527ca660e7SBarry Smith char version[256],helpoptions[256]; 3532479a3a6SBarry Smith #if defined(PETSC_USE_LOG) 3542479a3a6SBarry Smith PetscViewerFormat format; 35567584ceeSBarry Smith PetscBool flg4 = PETSC_FALSE; 35667584ceeSBarry Smith #endif 357012f15ecSRichard Tran Mills #if defined(PETSC_HAVE_CUDA) 3587381773fSBarry Smith PetscBool initCuda = PETSC_TRUE; 359012f15ecSRichard Tran Mills #endif 3607381773fSBarry Smith 361e5c89e4eSSatish Balay PetscFunctionBegin; 36242218b76SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 363e5c89e4eSSatish Balay 36467584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY) 36592f119d6SBarry Smith if (!(PETSC_RUNNING_ON_VALGRIND)) { 366e5c89e4eSSatish Balay /* 367e5c89e4eSSatish Balay Setup the memory management; support for tracing malloc() usage 368e5c89e4eSSatish Balay */ 369244bdbccSBarry Smith PetscBool mdebug = PETSC_FALSE, eachcall = PETSC_FALSE, initializenan = PETSC_FALSE, mlog = PETSC_FALSE; 37092f119d6SBarry Smith 3718bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG) 37292f119d6SBarry Smith mdebug = PETSC_TRUE; 37392f119d6SBarry Smith initializenan = PETSC_TRUE; 37492f119d6SBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-malloc_test",&flg1);CHKERRQ(ierr); 375e5c89e4eSSatish Balay #else 37692f119d6SBarry Smith /* don't warn about unused option */ 37792f119d6SBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-malloc_test",&flg1);CHKERRQ(ierr); 37892f119d6SBarry Smith flg1 = PETSC_FALSE; 379e5c89e4eSSatish Balay #endif 38079dccf82SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_debug",&flg2,&flg3);CHKERRQ(ierr); 38192f119d6SBarry Smith if (flg1 || flg2) { 38292f119d6SBarry Smith mdebug = PETSC_TRUE; 38392f119d6SBarry Smith eachcall = PETSC_TRUE; 38492f119d6SBarry Smith initializenan = PETSC_TRUE; 38579dccf82SBarry Smith } else if (flg3 && !flg2) { 38679dccf82SBarry Smith mdebug = PETSC_FALSE; 38779dccf82SBarry Smith eachcall = PETSC_FALSE; 38879dccf82SBarry Smith initializenan = PETSC_FALSE; 389e5c89e4eSSatish Balay } 39092f119d6SBarry Smith 39192f119d6SBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-malloc_view",&mlog);CHKERRQ(ierr); 39292f119d6SBarry Smith if (mlog) { 39392f119d6SBarry Smith mdebug = PETSC_TRUE; 39492f119d6SBarry Smith } 39579dccf82SBarry Smith /* the next line is deprecated */ 39692f119d6SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&mdebug,NULL);CHKERRQ(ierr); 39792f119d6SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_dump",&mdebug,NULL);CHKERRQ(ierr); 39892f119d6SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&mdebug,NULL);CHKERRQ(ierr); 39992f119d6SBarry Smith if (mdebug) { 40092f119d6SBarry Smith ierr = PetscMallocSetDebug(eachcall,initializenan);CHKERRQ(ierr); 40192f119d6SBarry Smith } 40292f119d6SBarry Smith if (mlog) { 40392f119d6SBarry Smith PetscReal logthreshold = 0; 40492f119d6SBarry Smith ierr = PetscOptionsGetReal(NULL,NULL,"-malloc_view_threshold",&logthreshold,NULL);CHKERRQ(ierr); 40592f119d6SBarry Smith ierr = PetscMallocViewSet(logthreshold);CHKERRQ(ierr); 40692f119d6SBarry Smith } 407*8b254c29SBarry Smith #if defined(PETSC_USE_LOG) 408*8b254c29SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&PetscLogMemory,NULL);CHKERRQ(ierr); 409*8b254c29SBarry Smith #endif 41092f119d6SBarry Smith } 41192f119d6SBarry Smith 412ba282f50SJed Brown ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_coalesce",&flg1,&flg2);CHKERRQ(ierr); 413ba282f50SJed Brown if (flg2) {ierr = PetscMallocSetCoalesce(flg1);CHKERRQ(ierr);} 41490d69ab7SBarry Smith flg1 = PETSC_FALSE; 4152f21b5c6SHong Zhang ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_hbw",&flg1,NULL);CHKERRQ(ierr); 416cf6f3811SHong Zhang /* ignore this option if malloc is already set */ 417cf6f3811SHong Zhang if (flg1 && !petscsetmallocvisited) {ierr = PetscSetUseHBWMalloc_Private();CHKERRQ(ierr);} 418e5c89e4eSSatish Balay 41990d69ab7SBarry Smith flg1 = PETSC_FALSE; 420c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_info",&flg1,NULL);CHKERRQ(ierr); 4217783f70dSSatish Balay if (!flg1) { 42290d69ab7SBarry Smith flg1 = PETSC_FALSE; 423c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-memory_view",&flg1,NULL);CHKERRQ(ierr); 4247783f70dSSatish Balay } 425e5c89e4eSSatish Balay if (flg1) { 426e5c89e4eSSatish Balay ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr); 427e5c89e4eSSatish Balay } 42867584ceeSBarry Smith #endif 429e5c89e4eSSatish Balay 430b4427426SBarry Smith #if defined(PETSC_USE_LOG) 431c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-objects_dump",&PetscObjectsLog);CHKERRQ(ierr); 432b4427426SBarry Smith #endif 43305df10baSBarry Smith 434e5c89e4eSSatish Balay /* 435e5c89e4eSSatish Balay Set the display variable for graphics 436e5c89e4eSSatish Balay */ 437e5c89e4eSSatish Balay ierr = PetscSetDisplay();CHKERRQ(ierr); 438e5c89e4eSSatish Balay 43951dcc849SKerry Stevens /* 440e5c89e4eSSatish Balay Print the PETSc version information 441e5c89e4eSSatish Balay */ 442c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-v",&flg1);CHKERRQ(ierr); 443c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-version",&flg2);CHKERRQ(ierr); 4442d747510SLisandro Dalcin ierr = PetscOptionsHasHelp(NULL,&flg3);CHKERRQ(ierr); 445e5c89e4eSSatish Balay if (flg1 || flg2 || flg3) { 446e5c89e4eSSatish Balay 447e5c89e4eSSatish Balay /* 448e5c89e4eSSatish Balay Print "higher-level" package version message 449e5c89e4eSSatish Balay */ 450e5c89e4eSSatish Balay if (PetscExternalVersionFunction) { 451e5c89e4eSSatish Balay ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr); 452e5c89e4eSSatish Balay } 453e5c89e4eSSatish Balay 454a523d312SBarry Smith ierr = PetscGetVersion(version,256);CHKERRQ(ierr); 4557ca660e7SBarry Smith ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr); 456e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr); 457e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr); 458e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr); 45984e42920SBarry Smith ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr); 460e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr); 461e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr); 4627ca660e7SBarry Smith ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr); 4637ca660e7SBarry Smith } 4647ca660e7SBarry Smith 46594941ca7SBarry Smith /* 46694941ca7SBarry Smith Print "higher-level" package help message 46794941ca7SBarry Smith */ 46894941ca7SBarry Smith if (flg3) { 46994941ca7SBarry Smith if (PetscExternalHelpFunction) { 47094941ca7SBarry Smith ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr); 47194941ca7SBarry Smith } 47294941ca7SBarry Smith } 47394941ca7SBarry Smith 4747ca660e7SBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-help",helpoptions,sizeof(helpoptions),&flg1);CHKERRQ(ierr); 4757ca660e7SBarry Smith if (flg1) { 4767ca660e7SBarry Smith ierr = PetscStrcmp(helpoptions,"intro",&flg2);CHKERRQ(ierr); 4777ca660e7SBarry Smith if (flg2) { 478bdb346e9SBarry Smith ierr = PetscOptionsDestroyDefault();CHKERRQ(ierr); 479008a6e76SBarry Smith ierr = PetscFreeMPIResources();CHKERRQ(ierr); 4807ca660e7SBarry Smith ierr = MPI_Finalize();CHKERRQ(ierr); 4817ca660e7SBarry Smith exit(0); 4827ca660e7SBarry Smith } 483e5c89e4eSSatish Balay } 484e5c89e4eSSatish Balay 485e5c89e4eSSatish Balay /* 486e5c89e4eSSatish Balay Setup the error handling 487e5c89e4eSSatish Balay */ 48890d69ab7SBarry Smith flg1 = PETSC_FALSE; 489c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_abort",&flg1,NULL);CHKERRQ(ierr); 490b59baad1SJed Brown if (flg1) { 49142218b76SBarry Smith ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr); 492b59baad1SJed Brown ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr); 493b59baad1SJed Brown } 49490d69ab7SBarry Smith flg1 = PETSC_FALSE; 495c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_mpiabort",&flg1,NULL);CHKERRQ(ierr); 496cb9801acSJed Brown if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,0);CHKERRQ(ierr);} 49790d69ab7SBarry Smith flg1 = PETSC_FALSE; 498c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_on_error",&flg1,NULL);CHKERRQ(ierr); 499e5c89e4eSSatish Balay if (flg1) { 500d54338ecSKarl Rupp ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr); 501e5c89e4eSSatish Balay } 50290d69ab7SBarry Smith flg1 = PETSC_FALSE; 503c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-no_signal_handler",&flg1,NULL);CHKERRQ(ierr); 5048d359177SBarry Smith if (!flg1) {ierr = PetscPushSignalHandler(PetscSignalHandlerDefault,(void*)0);CHKERRQ(ierr);} 50596cc47afSJed Brown flg1 = PETSC_FALSE; 506a9acdec7SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-fp_trap",&flg1,&flag);CHKERRQ(ierr); 507a9acdec7SBarry Smith if (flag) {ierr = PetscSetFPTrap((PetscFPTrap)flg1);CHKERRQ(ierr);} 508c5929fdfSBarry Smith ierr = PetscOptionsGetInt(NULL,NULL,"-check_pointer_intensity",&intensity,&flag);CHKERRQ(ierr); 50928559dc8SJed Brown if (flag) {ierr = PetscCheckPointerSetIntensity(intensity);CHKERRQ(ierr);} 510e5c89e4eSSatish Balay 511e5c89e4eSSatish Balay /* 512e5c89e4eSSatish Balay Setup debugger information 513e5c89e4eSSatish Balay */ 514e5c89e4eSSatish Balay ierr = PetscSetDefaultDebugger();CHKERRQ(ierr); 515c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-on_error_attach_debugger",string,64,&flg1);CHKERRQ(ierr); 516e5c89e4eSSatish Balay if (flg1) { 517e5c89e4eSSatish Balay MPI_Errhandler err_handler; 518e5c89e4eSSatish Balay 519e5c89e4eSSatish Balay ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr); 52033c7d699SBarry Smith ierr = MPI_Comm_create_errhandler(Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr); 521d54338ecSKarl Rupp ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr); 522e5c89e4eSSatish Balay ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,0);CHKERRQ(ierr); 523e5c89e4eSSatish Balay } 524c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-debug_terminal",string,64,&flg1);CHKERRQ(ierr); 5255e96ac45SJed Brown if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); } 526c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-start_in_debugger",string,64,&flg1);CHKERRQ(ierr); 527c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-stop_for_debugger",string,64,&flg2);CHKERRQ(ierr); 528e5c89e4eSSatish Balay if (flg1 || flg2) { 529e5c89e4eSSatish Balay PetscMPIInt size; 530e5c89e4eSSatish Balay PetscInt lsize,*nodes; 531e5c89e4eSSatish Balay MPI_Errhandler err_handler; 532e5c89e4eSSatish Balay /* 533e5c89e4eSSatish Balay we have to make sure that all processors have opened 534e5c89e4eSSatish Balay connections to all other processors, otherwise once the 535e5c89e4eSSatish Balay debugger has stated it is likely to receive a SIGUSR1 536e5c89e4eSSatish Balay and kill the program. 537e5c89e4eSSatish Balay */ 53842218b76SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 539e5c89e4eSSatish Balay if (size > 2) { 540533163c2SBarry Smith PetscMPIInt dummy = 0; 541e5c89e4eSSatish Balay MPI_Status status; 542e5c89e4eSSatish Balay for (i=0; i<size; i++) { 543e5c89e4eSSatish Balay if (rank != i) { 54442218b76SBarry Smith ierr = MPI_Send(&dummy,1,MPI_INT,i,109,comm);CHKERRQ(ierr); 545e5c89e4eSSatish Balay } 546e5c89e4eSSatish Balay } 547e5c89e4eSSatish Balay for (i=0; i<size; i++) { 548e5c89e4eSSatish Balay if (rank != i) { 54942218b76SBarry Smith ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,comm,&status);CHKERRQ(ierr); 550e5c89e4eSSatish Balay } 551e5c89e4eSSatish Balay } 552e5c89e4eSSatish Balay } 553e5c89e4eSSatish Balay /* check if this processor node should be in debugger */ 554785e854fSJed Brown ierr = PetscMalloc1(size,&nodes);CHKERRQ(ierr); 555e5c89e4eSSatish Balay lsize = size; 556c5929fdfSBarry Smith ierr = PetscOptionsGetIntArray(NULL,NULL,"-debugger_nodes",nodes,&lsize,&flag);CHKERRQ(ierr); 557e5c89e4eSSatish Balay if (flag) { 558e5c89e4eSSatish Balay for (i=0; i<lsize; i++) { 559e5c89e4eSSatish Balay if (nodes[i] == rank) { flag = PETSC_FALSE; break; } 560e5c89e4eSSatish Balay } 561e5c89e4eSSatish Balay } 562e5c89e4eSSatish Balay if (!flag) { 563e5c89e4eSSatish Balay ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr); 564e5c89e4eSSatish Balay ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr); 565e5c89e4eSSatish Balay if (flg1) { 566e5c89e4eSSatish Balay ierr = PetscAttachDebugger();CHKERRQ(ierr); 567e5c89e4eSSatish Balay } else { 568e5c89e4eSSatish Balay ierr = PetscStopForDebugger();CHKERRQ(ierr); 569e5c89e4eSSatish Balay } 57033c7d699SBarry Smith ierr = MPI_Comm_create_errhandler(Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr); 571d54338ecSKarl Rupp ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr); 572e5c89e4eSSatish Balay } 573e5c89e4eSSatish Balay ierr = PetscFree(nodes);CHKERRQ(ierr); 574e5c89e4eSSatish Balay } 575e5c89e4eSSatish Balay 576c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-on_error_emacs",emacsmachinename,128,&flg1);CHKERRQ(ierr); 577cb9801acSJed Brown if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);} 578e5c89e4eSSatish Balay 579e5c89e4eSSatish Balay /* 580e5c89e4eSSatish Balay Setup profiling and logging 581e5c89e4eSSatish Balay */ 5826cf91177SBarry Smith #if defined(PETSC_USE_INFO) 5838bb29257SSatish Balay { 584e5c89e4eSSatish Balay char logname[PETSC_MAX_PATH_LEN]; logname[0] = 0; 585c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-info",logname,250,&flg1);CHKERRQ(ierr); 5868bb29257SSatish Balay if (flg1 && logname[0]) { 587fcc2139eSBarry Smith ierr = PetscInfoAllow(PETSC_TRUE,logname);CHKERRQ(ierr); 5888bb29257SSatish Balay } else if (flg1) { 5890298fd71SBarry Smith ierr = PetscInfoAllow(PETSC_TRUE,NULL);CHKERRQ(ierr); 590e5c89e4eSSatish Balay } 591e5c89e4eSSatish Balay } 592865f6aa8SSatish Balay #endif 593865f6aa8SSatish Balay #if defined(PETSC_USE_LOG) 594865f6aa8SSatish Balay mname[0] = 0; 595c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-history",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr); 596865f6aa8SSatish Balay if (flg1) { 597865f6aa8SSatish Balay if (mname[0]) { 598f3dea69dSBarry Smith ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr); 599865f6aa8SSatish Balay } else { 600706d7a88SBarry Smith ierr = PetscOpenHistoryFile(NULL,&petsc_history);CHKERRQ(ierr); 601865f6aa8SSatish Balay } 602865f6aa8SSatish Balay } 603217044c2SLisandro Dalcin 604217044c2SLisandro Dalcin ierr = PetscOptionsGetBool(NULL,NULL,"-log_sync",&PetscLogSyncOn,NULL);CHKERRQ(ierr); 605217044c2SLisandro Dalcin 606e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE) 60790d69ab7SBarry Smith flg1 = PETSC_FALSE; 608c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_mpe",&flg1);CHKERRQ(ierr); 609495fc317SBarry Smith if (flg1) {ierr = PetscLogMPEBegin();CHKERRQ(ierr);} 610e5c89e4eSSatish Balay #endif 61190d69ab7SBarry Smith flg1 = PETSC_FALSE; 61290d69ab7SBarry Smith flg3 = PETSC_FALSE; 613c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-log_all",&flg1,NULL);CHKERRQ(ierr); 614c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr); 615e5c89e4eSSatish Balay if (flg1) { ierr = PetscLogAllBegin();CHKERRQ(ierr); } 616bb1d7374SBarry Smith else if (flg3) { ierr = PetscLogDefaultBegin();CHKERRQ(ierr);} 617e5c89e4eSSatish Balay 618c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-log_trace",mname,250,&flg1);CHKERRQ(ierr); 619e5c89e4eSSatish Balay if (flg1) { 620e5c89e4eSSatish Balay char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN]; 621e5c89e4eSSatish Balay FILE *file; 622e5c89e4eSSatish Balay if (mname[0]) { 6232e924ca5SSatish Balay PetscSNPrintf(name,PETSC_MAX_PATH_LEN,"%s.%d",mname,rank); 624e5c89e4eSSatish Balay ierr = PetscFixFilename(name,fname);CHKERRQ(ierr); 625e5c89e4eSSatish Balay file = fopen(fname,"w"); 626f3dea69dSBarry Smith if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname); 627a297a907SKarl Rupp } else file = PETSC_STDOUT; 628e5c89e4eSSatish Balay ierr = PetscLogTraceBegin(file);CHKERRQ(ierr); 629e5c89e4eSSatish Balay } 630bb1d7374SBarry Smith 63116413a6aSBarry Smith ierr = PetscOptionsGetViewer(comm,NULL,NULL,"-log_view",NULL,&format,&flg4);CHKERRQ(ierr); 632bb1d7374SBarry Smith if (flg4) { 633bb1d7374SBarry Smith if (format == PETSC_VIEWER_ASCII_XML) { 634bb1d7374SBarry Smith ierr = PetscLogNestedBegin();CHKERRQ(ierr); 635bb1d7374SBarry Smith } else { 636bb1d7374SBarry Smith ierr = PetscLogDefaultBegin();CHKERRQ(ierr); 637bb1d7374SBarry Smith } 638eccbb886SLisandro Dalcin } 639eccbb886SLisandro Dalcin if (flg4 && format == PETSC_VIEWER_ASCII_XML) { 640eccbb886SLisandro Dalcin PetscReal threshold = PetscRealConstant(0.01); 641eccbb886SLisandro Dalcin ierr = PetscOptionsGetReal(NULL,NULL,"-log_threshold",&threshold,&flg1);CHKERRQ(ierr); 642eccbb886SLisandro Dalcin if (flg1) {ierr = PetscLogSetThreshold((PetscLogDouble)threshold,NULL);CHKERRQ(ierr);} 643bb1d7374SBarry Smith } 644e5c89e4eSSatish Balay #endif 645e5c89e4eSSatish Balay 646c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-saws_options",&PetscOptionsPublish,NULL);CHKERRQ(ierr); 647e5c89e4eSSatish Balay 6484b0a73b2SBarry Smith #if defined(PETSC_HAVE_CUDA) 6497381773fSBarry Smith ierr = PetscOptionsBegin(comm,NULL,"CUDA initialize","Sys");CHKERRQ(ierr); 6507381773fSBarry Smith ierr = PetscOptionsBool("-cuda_initialize","Initialize the CUDA devices and cuBLAS during PetscInitialize()",NULL,initCuda,&initCuda,NULL);CHKERRQ(ierr); 651012f15ecSRichard Tran Mills ierr = PetscOptionsEnd();CHKERRQ(ierr); 6527381773fSBarry Smith if (initCuda) {ierr = PetscCUDAInitialize(PETSC_COMM_WORLD);CHKERRQ(ierr);} 6534b0a73b2SBarry Smith #endif 6544b0a73b2SBarry Smith 655e5c89e4eSSatish Balay /* 656e5c89e4eSSatish Balay Print basic help message 657e5c89e4eSSatish Balay */ 6582d747510SLisandro Dalcin ierr = PetscOptionsHasHelp(NULL,&flg1);CHKERRQ(ierr); 659e5c89e4eSSatish Balay if (flg1) { 660e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr); 661301d30feSBarry Smith ierr = (*PetscHelpPrintf)(comm," -help: prints help method for each option\n");CHKERRQ(ierr); 662301d30feSBarry Smith ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr); 663301d30feSBarry Smith ierr = (*PetscHelpPrintf)(comm," only when run in the debugger\n");CHKERRQ(ierr); 664e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr); 665e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," start the debugger in new xterm\n");CHKERRQ(ierr); 666e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," unless noxterm is given\n");CHKERRQ(ierr); 667e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr); 668e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," start all processes in the debugger\n");CHKERRQ(ierr); 669e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr); 670e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," emacs jumps to error file\n");CHKERRQ(ierr); 671e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -debugger_nodes [n1,n2,..] Nodes to start in debugger\n");CHKERRQ(ierr); 672e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr); 673e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr); 674e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," waits the delay for you to attach\n");CHKERRQ(ierr); 6751cda70a7SBarry Smith ierr = (*PetscHelpPrintf)(comm," -display display: Location where X window graphics and debuggers are displayed\n");CHKERRQ(ierr); 676e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr); 677e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr); 678e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr); 679e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr); 680e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr); 68179dccf82SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc: use PETSc error checking malloc (deprecated, use -malloc_debug)\n");CHKERRQ(ierr); 68279dccf82SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use PETSc error checking malloc (deprecated, use -malloc_debug no)\n");CHKERRQ(ierr); 6834161f2a3SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr); 68492f119d6SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc_view <optional filename>: keeps log of all memory allocations, displays in PetscFinalize()\n");CHKERRQ(ierr); 68579dccf82SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc_debug <true or false>: enables or disables extended checking for memory corruption\n");CHKERRQ(ierr); 68626a7e8d4SBarry Smith ierr = (*PetscHelpPrintf)(comm," -options_view: dump list of options inputted\n");CHKERRQ(ierr); 687e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr); 688e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr); 689e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr); 690e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr); 691a8c7a070SBarry Smith ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr); 6920841954dSBarry Smith ierr = (*PetscHelpPrintf)(comm," -memory_view: print memory usage at end of run\n");CHKERRQ(ierr); 693e5c89e4eSSatish Balay #if defined(PETSC_USE_LOG) 694e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr); 695185ae32fSMatthew G. Knepley ierr = (*PetscHelpPrintf)(comm," -log_view [:filename:[format]]: logging objects and events\n");CHKERRQ(ierr); 696e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr); 697e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE) 698495fc317SBarry Smith ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through Jumpshot\n");CHKERRQ(ierr); 699e5c89e4eSSatish Balay #endif 7006cf91177SBarry Smith ierr = (*PetscHelpPrintf)(comm," -info <optional filename>: print informative messages about the calculations\n");CHKERRQ(ierr); 701e5c89e4eSSatish Balay #endif 702e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -v: prints PETSc version number and release date\n");CHKERRQ(ierr); 703e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr); 704e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr); 705e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"-----------------------------------------------\n");CHKERRQ(ierr); 706e5c89e4eSSatish Balay } 707e5c89e4eSSatish Balay 70874ba8654SBarry Smith #if defined(PETSC_HAVE_POPEN) 70974ba8654SBarry Smith { 71074ba8654SBarry Smith char machine[128]; 711c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-popen_machine",machine,128,&flg1);CHKERRQ(ierr); 71274ba8654SBarry Smith if (flg1) { 71374ba8654SBarry Smith ierr = PetscPOpenSetMachine(machine);CHKERRQ(ierr); 71474ba8654SBarry Smith } 71574ba8654SBarry Smith } 71674ba8654SBarry Smith #endif 71774ba8654SBarry Smith 718c5929fdfSBarry Smith ierr = PetscOptionsGetReal(NULL,NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr); 719e5c89e4eSSatish Balay if (flg1) { 720e5c89e4eSSatish Balay ierr = PetscSleep(si);CHKERRQ(ierr); 721e5c89e4eSSatish Balay } 722e5c89e4eSSatish Balay 723c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr); 7242ea1be5dSBarry Smith if (flg1) { 725e5c89e4eSSatish Balay ierr = PetscStrstr(mname,"null",&f);CHKERRQ(ierr); 726e5c89e4eSSatish Balay if (f) { 727ecd8bba6SJed Brown ierr = PetscInfoDeactivateClass(0);CHKERRQ(ierr); 728e5c89e4eSSatish Balay } 7292ea1be5dSBarry Smith } 730827f890bSBarry Smith 731fdc842d1SBarry Smith #if defined(PETSC_HAVE_VIENNACL) 732c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr); 733f14045dbSBarry Smith if (!flg3) { 734c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&flg3);CHKERRQ(ierr); 735f14045dbSBarry Smith } 736c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-viennacl_synchronize",&flg3,NULL);CHKERRQ(ierr); 737f14045dbSBarry Smith PetscViennaCLSynchronize = flg3; 738fdc842d1SBarry Smith ierr = PetscViennaCLInit();CHKERRQ(ierr); 7394cf1874eSKarl Rupp #endif 74082f73ecaSAlejandro Lamas Daviña 741fdc842d1SBarry Smith /* 742fdc842d1SBarry Smith Creates the logging data structures; this is enabled even if logging is not turned on 743fdc842d1SBarry Smith This is the last thing we do before returning to the user code to prevent having the 744fdc842d1SBarry Smith logging numbers contaminated by any startup time associated with MPI and the GPUs 745fdc842d1SBarry Smith */ 746fdc842d1SBarry Smith #if defined(PETSC_USE_LOG) 747fdc842d1SBarry Smith ierr = PetscLogInitialize();CHKERRQ(ierr); 748f0a7718cSKarl Rupp #endif 749f0a7718cSKarl Rupp 750e5c89e4eSSatish Balay PetscFunctionReturn(0); 751e5c89e4eSSatish Balay } 752