1e5c89e4eSSatish Balay /* 2e5c89e4eSSatish Balay 3e5c89e4eSSatish Balay This file defines part of the initialization of PETSc 4e5c89e4eSSatish Balay 5f2d66bcaSShri Abhyankar This file uses regular malloc and free because it cannot known 6e5c89e4eSSatish Balay what malloc is being used until it has already processed the input. 7e5c89e4eSSatish Balay */ 8ef386f4bSSatish Balay 9ef386f4bSSatish Balay #include <petscsys.h> /*I "petscsys.h" I*/ 10f67a399dSBarry Smith #include <petsc/private/petscimpl.h> 11022afb99SBarry Smith #include <petscvalgrind.h> 12665c2dedSJed Brown #include <petscviewer.h> 13ef386f4bSSatish Balay 14ba61063dSBarry Smith #if defined(PETSC_HAVE_SYS_SYSINFO_H) 1551d315f7SKerry Stevens #include <sys/sysinfo.h> 16ba61063dSBarry Smith #endif 17121deb67SSatish Balay #if defined(PETSC_HAVE_UNISTD_H) 1851d315f7SKerry Stevens #include <unistd.h> 19121deb67SSatish Balay #endif 20d9b72601SDominic Meiser #if defined(PETSC_HAVE_CUDA) 21d9b72601SDominic Meiser #include <cuda_runtime.h> 22d9b72601SDominic Meiser #endif 23555d055bSBarry Smith 24f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL) 25f0a7718cSKarl Rupp PETSC_EXTERN PetscErrorCode PetscViennaCLInit(); 26f0a7718cSKarl Rupp #endif 27f0a7718cSKarl Rupp 28e5c89e4eSSatish Balay /* ------------------------Nasty global variables -------------------------------*/ 29e5c89e4eSSatish Balay /* 30e5c89e4eSSatish Balay Indicates if PETSc started up MPI, or it was 31e5c89e4eSSatish Balay already started before PETSc was initialized. 32e5c89e4eSSatish Balay */ 330cbf60d0SJose E. Roman PetscBool PetscBeganMPI = PETSC_FALSE; 340cbf60d0SJose E. Roman PetscBool PetscInitializeCalled = PETSC_FALSE; 350cbf60d0SJose E. Roman PetscBool PetscFinalizeCalled = PETSC_FALSE; 36bffec52eSKarl Rupp PetscBool PetscCUDAInitialized = PETSC_FALSE; 37d6f2c3cbSBarry Smith 387087cfbeSBarry Smith PetscMPIInt PetscGlobalRank = -1; 397087cfbeSBarry Smith PetscMPIInt PetscGlobalSize = -1; 40ba61063dSBarry Smith 4150f81f78SJed Brown #if defined(PETSC_HAVE_COMPLEX) 42e5c89e4eSSatish Balay #if defined(PETSC_COMPLEX_INSTANTIATE) 43e5c89e4eSSatish Balay template <> class std::complex<double>; /* instantiate complex template class */ 44e5c89e4eSSatish Balay #endif 452c876bd9SBarry Smith #if !defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX) 46500d8756SSatish Balay MPI_Datatype MPIU_C_DOUBLE_COMPLEX; 47500d8756SSatish Balay MPI_Datatype MPIU_C_COMPLEX; 482c876bd9SBarry Smith #endif 498619c96cSJed Brown 508619c96cSJed Brown /*MC 518619c96cSJed Brown PETSC_i - the imaginary number i 528619c96cSJed Brown 538619c96cSJed Brown Synopsis: 548619c96cSJed Brown #include <petscsys.h> 558619c96cSJed Brown PetscComplex PETSC_i; 568619c96cSJed Brown 578619c96cSJed Brown Level: beginner 588619c96cSJed Brown 598619c96cSJed Brown Note: 608cd53115SBarry Smith Complex numbers are automatically available if PETSc located a working complex implementation 618619c96cSJed Brown 628619c96cSJed Brown .seealso: PetscRealPart(), PetscImaginaryPart(), PetscRealPartComplex(), PetscImaginaryPartComplex() 638619c96cSJed Brown M*/ 6450f81f78SJed Brown PetscComplex PETSC_i; 65e5c89e4eSSatish Balay #endif 66ce63c4c1SBarry Smith #if defined(PETSC_USE_REAL___FLOAT128) 67c90a1750SBarry Smith MPI_Datatype MPIU___FLOAT128 = 0; 687c2de775SJed Brown #if defined(PETSC_HAVE_COMPLEX) 698c764dc5SJose Roman MPI_Datatype MPIU___COMPLEX128 = 0; 708c764dc5SJose Roman #endif 71570b7f6dSBarry Smith #elif defined(PETSC_USE_REAL___FP16) 72570b7f6dSBarry Smith MPI_Datatype MPIU___FP16 = 0; 73c90a1750SBarry Smith #endif 747087cfbeSBarry Smith MPI_Datatype MPIU_2SCALAR = 0; 7544041f26SJed Brown #if defined(PETSC_USE_64BIT_INDICES) || !defined(MPI_2INT) 767087cfbeSBarry Smith MPI_Datatype MPIU_2INT = 0; 7744041f26SJed Brown #endif 788ad47952SJed Brown MPI_Datatype MPIU_BOOL; 798ad47952SJed Brown MPI_Datatype MPIU_ENUM; 8075567043SBarry Smith 81e5c89e4eSSatish Balay /* 82e5c89e4eSSatish Balay Function that is called to display all error messages 83e5c89e4eSSatish Balay */ 847087cfbeSBarry Smith PetscErrorCode (*PetscErrorPrintf)(const char [],...) = PetscErrorPrintfDefault; 857087cfbeSBarry Smith PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...) = PetscHelpPrintfDefault; 867087cfbeSBarry Smith PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list) = PetscVFPrintfDefault; 87bab1f7e6SVictor Minden /* 884cf1874eSKarl Rupp This is needed to turn on/off GPU synchronization 898b5db460SBarry Smith */ 904cf1874eSKarl Rupp PetscBool PetscViennaCLSynchronize = PETSC_FALSE; 9182f73ecaSAlejandro Lamas Daviña PetscBool PetscCUDASynchronize = PETSC_FALSE; 92bab1f7e6SVictor Minden 93e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/ 94e5c89e4eSSatish Balay /* 95e5c89e4eSSatish Balay Optional file where all PETSc output from various prints is saved 96e5c89e4eSSatish Balay */ 9795c0884eSLisandro Dalcin PETSC_INTERN FILE *petsc_history; 980298fd71SBarry Smith FILE *petsc_history = NULL; 99e5c89e4eSSatish Balay 1007087cfbeSBarry Smith PetscErrorCode PetscOpenHistoryFile(const char filename[],FILE **fd) 101e5c89e4eSSatish Balay { 102e5c89e4eSSatish Balay PetscErrorCode ierr; 103e5c89e4eSSatish Balay PetscMPIInt rank,size; 104e5c89e4eSSatish Balay char pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64]; 105e5c89e4eSSatish Balay char version[256]; 106e5c89e4eSSatish Balay 107e5c89e4eSSatish Balay PetscFunctionBegin; 108e5c89e4eSSatish Balay ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 109e5c89e4eSSatish Balay if (!rank) { 110e5c89e4eSSatish Balay char arch[10]; 111f56c2debSBarry Smith int err; 112f56c2debSBarry Smith 113e5c89e4eSSatish Balay ierr = PetscGetArchType(arch,10);CHKERRQ(ierr); 114e5c89e4eSSatish Balay ierr = PetscGetDate(date,64);CHKERRQ(ierr); 115a523d312SBarry Smith ierr = PetscGetVersion(version,256);CHKERRQ(ierr); 116e5c89e4eSSatish Balay ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 117e5c89e4eSSatish Balay if (filename) { 118e5c89e4eSSatish Balay ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr); 119e5c89e4eSSatish Balay } else { 120e5c89e4eSSatish Balay ierr = PetscGetHomeDirectory(pfile,240);CHKERRQ(ierr); 121e5c89e4eSSatish Balay ierr = PetscStrcat(pfile,"/.petschistory");CHKERRQ(ierr); 122e5c89e4eSSatish Balay ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr); 123e5c89e4eSSatish Balay } 124e5c89e4eSSatish Balay 125a297a907SKarl Rupp *fd = fopen(fname,"a"); 126a297a907SKarl Rupp if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname); 127a297a907SKarl Rupp 128e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 129e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr); 130e5c89e4eSSatish Balay ierr = PetscGetProgramName(pname,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 131e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr); 132e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 133a297a907SKarl Rupp 134f56c2debSBarry Smith err = fflush(*fd); 135e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 136e5c89e4eSSatish Balay } 137e5c89e4eSSatish Balay PetscFunctionReturn(0); 138e5c89e4eSSatish Balay } 139e5c89e4eSSatish Balay 14095c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscCloseHistoryFile(FILE **fd) 141e5c89e4eSSatish Balay { 142e5c89e4eSSatish Balay PetscErrorCode ierr; 143e5c89e4eSSatish Balay PetscMPIInt rank; 144e5c89e4eSSatish Balay char date[64]; 145f56c2debSBarry Smith int err; 146e5c89e4eSSatish Balay 147e5c89e4eSSatish Balay PetscFunctionBegin; 148e5c89e4eSSatish Balay ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 149e5c89e4eSSatish Balay if (!rank) { 150e5c89e4eSSatish Balay ierr = PetscGetDate(date,64);CHKERRQ(ierr); 151e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 152e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr); 153e5c89e4eSSatish Balay ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 154f56c2debSBarry Smith err = fflush(*fd); 155e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 156f56c2debSBarry Smith err = fclose(*fd); 157e32f2f54SBarry Smith if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file"); 158e5c89e4eSSatish Balay } 159e5c89e4eSSatish Balay PetscFunctionReturn(0); 160e5c89e4eSSatish Balay } 161e5c89e4eSSatish Balay 162e5c89e4eSSatish Balay /* ------------------------------------------------------------------------------*/ 163e5c89e4eSSatish Balay 164e5c89e4eSSatish Balay /* 165e5c89e4eSSatish Balay This is ugly and probably belongs somewhere else, but I want to 166e5c89e4eSSatish Balay be able to put a true MPI abort error handler with command line args. 167e5c89e4eSSatish Balay 168e5c89e4eSSatish Balay This is so MPI errors in the debugger will leave all the stack 1693c311c98SBarry Smith frames. The default MP_Abort() cleans up and exits thus providing no useful information 1703c311c98SBarry Smith in the debugger hence we call abort() instead of MPI_Abort(). 171e5c89e4eSSatish Balay */ 172e5c89e4eSSatish Balay 17333c7d699SBarry Smith void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag,...) 174e5c89e4eSSatish Balay { 175e5c89e4eSSatish Balay PetscFunctionBegin; 1763c311c98SBarry Smith (*PetscErrorPrintf)("MPI error %d\n",*flag); 177e5c89e4eSSatish Balay abort(); 178e5c89e4eSSatish Balay } 179e5c89e4eSSatish Balay 18033c7d699SBarry Smith void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag,...) 181e5c89e4eSSatish Balay { 182e5c89e4eSSatish Balay PetscErrorCode ierr; 183e5c89e4eSSatish Balay 184e5c89e4eSSatish Balay PetscFunctionBegin; 1853c311c98SBarry Smith (*PetscErrorPrintf)("MPI error %d\n",*flag); 186e5c89e4eSSatish Balay ierr = PetscAttachDebugger(); 187a297a907SKarl Rupp if (ierr) MPI_Abort(*comm,*flag); /* hopeless so get out */ 188e5c89e4eSSatish Balay } 189e5c89e4eSSatish Balay 190e5c89e4eSSatish Balay /*@C 191e5c89e4eSSatish Balay PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one 192e5c89e4eSSatish Balay wishes a clean exit somewhere deep in the program. 193e5c89e4eSSatish Balay 194e5c89e4eSSatish Balay Collective on PETSC_COMM_WORLD 195e5c89e4eSSatish Balay 196e5c89e4eSSatish Balay Options Database Keys are the same as for PetscFinalize() 197e5c89e4eSSatish Balay 198e5c89e4eSSatish Balay Level: advanced 199e5c89e4eSSatish Balay 200e5c89e4eSSatish Balay Note: 201e5c89e4eSSatish Balay See PetscInitialize() for more general runtime options. 202e5c89e4eSSatish Balay 20388c29154SBarry Smith .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize() 204e5c89e4eSSatish Balay @*/ 2057087cfbeSBarry Smith PetscErrorCode PetscEnd(void) 206e5c89e4eSSatish Balay { 207e5c89e4eSSatish Balay PetscFunctionBegin; 208e5c89e4eSSatish Balay PetscFinalize(); 209e5c89e4eSSatish Balay exit(0); 210e5c89e4eSSatish Balay return 0; 211e5c89e4eSSatish Balay } 212e5c89e4eSSatish Balay 213ace3abfcSBarry Smith PetscBool PetscOptionsPublish = PETSC_FALSE; 21495c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseTrMalloc_Private(void); 21595c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscSetUseHBWMalloc_Private(void); 21695c0884eSLisandro Dalcin PETSC_INTERN PetscBool petscsetmallocvisited; 217e5c89e4eSSatish Balay static char emacsmachinename[256]; 218e5c89e4eSSatish Balay 219e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = 0; 220e5c89e4eSSatish Balay PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm) = 0; 221e5c89e4eSSatish Balay 222e5c89e4eSSatish Balay /*@C 223e5c89e4eSSatish Balay PetscSetHelpVersionFunctions - Sets functions that print help and version information 224e5c89e4eSSatish Balay before the PETSc help and version information is printed. Must call BEFORE PetscInitialize(). 225e5c89e4eSSatish Balay This routine enables a "higher-level" package that uses PETSc to print its messages first. 226e5c89e4eSSatish Balay 227e5c89e4eSSatish Balay Input Parameter: 2280298fd71SBarry Smith + help - the help function (may be NULL) 2290298fd71SBarry Smith - version - the version function (may be NULL) 230e5c89e4eSSatish Balay 231e5c89e4eSSatish Balay Level: developer 232e5c89e4eSSatish Balay 233e5c89e4eSSatish Balay Concepts: package help message 234e5c89e4eSSatish Balay 235e5c89e4eSSatish Balay @*/ 2367087cfbeSBarry Smith PetscErrorCode PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm)) 237e5c89e4eSSatish Balay { 238e5c89e4eSSatish Balay PetscFunctionBegin; 239e5c89e4eSSatish Balay PetscExternalHelpFunction = help; 240e5c89e4eSSatish Balay PetscExternalVersionFunction = version; 241e5c89e4eSSatish Balay PetscFunctionReturn(0); 242e5c89e4eSSatish Balay } 243e5c89e4eSSatish Balay 24405df10baSBarry Smith #if defined(PETSC_USE_LOG) 24595c0884eSLisandro Dalcin PETSC_INTERN PetscBool PetscObjectsLog; 24605df10baSBarry Smith #endif 24705df10baSBarry Smith 24895c0884eSLisandro Dalcin PETSC_INTERN PetscErrorCode PetscOptionsCheckInitial_Private(void) 249e5c89e4eSSatish Balay { 250e5c89e4eSSatish Balay char string[64],mname[PETSC_MAX_PATH_LEN],*f; 251e5c89e4eSSatish Balay MPI_Comm comm = PETSC_COMM_WORLD; 25267584ceeSBarry Smith PetscBool flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flag; 253e5c89e4eSSatish Balay PetscErrorCode ierr; 25467584ceeSBarry Smith PetscReal si; 25528559dc8SJed Brown PetscInt intensity; 256e5c89e4eSSatish Balay int i; 257e5c89e4eSSatish Balay PetscMPIInt rank; 2587ca660e7SBarry Smith char version[256],helpoptions[256]; 25967584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY) 26067584ceeSBarry Smith PetscReal logthreshold; 2612479a3a6SBarry Smith #endif 2622479a3a6SBarry Smith #if defined(PETSC_USE_LOG) 2632479a3a6SBarry Smith PetscViewerFormat format; 26467584ceeSBarry Smith PetscBool flg4 = PETSC_FALSE; 26567584ceeSBarry Smith #endif 266e5c89e4eSSatish Balay 267e5c89e4eSSatish Balay PetscFunctionBegin; 26842218b76SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 269e5c89e4eSSatish Balay 27067584ceeSBarry Smith #if !defined(PETSC_HAVE_THREADSAFETY) 271e5c89e4eSSatish Balay /* 272e5c89e4eSSatish Balay Setup the memory management; support for tracing malloc() usage 273e5c89e4eSSatish Balay */ 274c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-malloc_log",&flg3);CHKERRQ(ierr); 275574034a9SJed Brown logthreshold = 0.0; 276c5929fdfSBarry Smith ierr = PetscOptionsGetReal(NULL,NULL,"-malloc_log_threshold",&logthreshold,&flg1);CHKERRQ(ierr); 277574034a9SJed Brown if (flg1) flg3 = PETSC_TRUE; 2788bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG) 279c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg1,&flg2);CHKERRQ(ierr); 280e5c89e4eSSatish Balay if ((!flg2 || flg1) && !petscsetmallocvisited) { 281dc92acbaSJed Brown if (flg2 || !(PETSC_RUNNING_ON_VALGRIND)) { 282555d055bSBarry Smith /* turn off default -malloc if valgrind is being used */ 283e5c89e4eSSatish Balay ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 284555d055bSBarry Smith } 285e5c89e4eSSatish Balay } 286e5c89e4eSSatish Balay #else 287c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_dump",&flg1,NULL);CHKERRQ(ierr); 288c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc",&flg2,NULL);CHKERRQ(ierr); 289e5c89e4eSSatish Balay if (flg1 || flg2 || flg3) {ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);} 290e5c89e4eSSatish Balay #endif 291e5c89e4eSSatish Balay if (flg3) { 292574034a9SJed Brown ierr = PetscMallocSetDumpLogThreshold((PetscLogDouble)logthreshold);CHKERRQ(ierr); 293e5c89e4eSSatish Balay } 294ba282f50SJed Brown ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_coalesce",&flg1,&flg2);CHKERRQ(ierr); 295ba282f50SJed Brown if (flg2) {ierr = PetscMallocSetCoalesce(flg1);CHKERRQ(ierr);} 29690d69ab7SBarry Smith flg1 = PETSC_FALSE; 297c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_debug",&flg1,NULL);CHKERRQ(ierr); 298e5c89e4eSSatish Balay if (flg1) { 299e5c89e4eSSatish Balay ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 300e5c89e4eSSatish Balay ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr); 301e5c89e4eSSatish Balay } 302dc92acbaSJed Brown flg1 = PETSC_FALSE; 303c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_test",&flg1,NULL);CHKERRQ(ierr); 3048bf1f09cSShri Abhyankar #if defined(PETSC_USE_DEBUG) 305dc92acbaSJed Brown if (flg1 && !PETSC_RUNNING_ON_VALGRIND) { 306dc92acbaSJed Brown ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 307dc92acbaSJed Brown ierr = PetscMallocSetDumpLog();CHKERRQ(ierr); 308dc92acbaSJed Brown ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr); 309dc92acbaSJed Brown } 310dc92acbaSJed Brown #endif 3112f21b5c6SHong Zhang flg1 = PETSC_FALSE; 3122f21b5c6SHong Zhang ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_hbw",&flg1,NULL);CHKERRQ(ierr); 313cf6f3811SHong Zhang /* ignore this option if malloc is already set */ 314cf6f3811SHong Zhang if (flg1 && !petscsetmallocvisited) {ierr = PetscSetUseHBWMalloc_Private();CHKERRQ(ierr);} 315e5c89e4eSSatish Balay 31690d69ab7SBarry Smith flg1 = PETSC_FALSE; 317c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-malloc_info",&flg1,NULL);CHKERRQ(ierr); 3187783f70dSSatish Balay if (!flg1) { 31990d69ab7SBarry Smith flg1 = PETSC_FALSE; 320c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-memory_view",&flg1,NULL);CHKERRQ(ierr); 3217783f70dSSatish Balay } 322e5c89e4eSSatish Balay if (flg1) { 323e5c89e4eSSatish Balay ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr); 324e5c89e4eSSatish Balay } 32567584ceeSBarry Smith #endif 326e5c89e4eSSatish Balay 327b4427426SBarry Smith #if defined(PETSC_USE_LOG) 328c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-objects_dump",&PetscObjectsLog);CHKERRQ(ierr); 329b4427426SBarry Smith #endif 33005df10baSBarry Smith 331e5c89e4eSSatish Balay /* 332e5c89e4eSSatish Balay Set the display variable for graphics 333e5c89e4eSSatish Balay */ 334e5c89e4eSSatish Balay ierr = PetscSetDisplay();CHKERRQ(ierr); 335e5c89e4eSSatish Balay 33651dcc849SKerry Stevens /* 337e5c89e4eSSatish Balay Print the PETSc version information 338e5c89e4eSSatish Balay */ 339c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-v",&flg1);CHKERRQ(ierr); 340c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-version",&flg2);CHKERRQ(ierr); 3412d747510SLisandro Dalcin ierr = PetscOptionsHasHelp(NULL,&flg3);CHKERRQ(ierr); 342e5c89e4eSSatish Balay if (flg1 || flg2 || flg3) { 343e5c89e4eSSatish Balay 344e5c89e4eSSatish Balay /* 345e5c89e4eSSatish Balay Print "higher-level" package version message 346e5c89e4eSSatish Balay */ 347e5c89e4eSSatish Balay if (PetscExternalVersionFunction) { 348e5c89e4eSSatish Balay ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr); 349e5c89e4eSSatish Balay } 350e5c89e4eSSatish Balay 351a523d312SBarry Smith ierr = PetscGetVersion(version,256);CHKERRQ(ierr); 3527ca660e7SBarry Smith ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr); 353e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr); 354e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr); 355e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr); 35684e42920SBarry Smith ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr); 357e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr); 358e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr); 3597ca660e7SBarry Smith ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------------------------------------\n");CHKERRQ(ierr); 3607ca660e7SBarry Smith } 3617ca660e7SBarry Smith 36294941ca7SBarry Smith /* 36394941ca7SBarry Smith Print "higher-level" package help message 36494941ca7SBarry Smith */ 36594941ca7SBarry Smith if (flg3) { 36694941ca7SBarry Smith if (PetscExternalHelpFunction) { 36794941ca7SBarry Smith ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr); 36894941ca7SBarry Smith } 36994941ca7SBarry Smith } 37094941ca7SBarry Smith 3717ca660e7SBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-help",helpoptions,sizeof(helpoptions),&flg1);CHKERRQ(ierr); 3727ca660e7SBarry Smith if (flg1) { 3737ca660e7SBarry Smith ierr = PetscStrcmp(helpoptions,"intro",&flg2);CHKERRQ(ierr); 3747ca660e7SBarry Smith if (flg2) { 375bdb346e9SBarry Smith ierr = PetscOptionsDestroyDefault();CHKERRQ(ierr); 376008a6e76SBarry Smith ierr = PetscFreeMPIResources();CHKERRQ(ierr); 3777ca660e7SBarry Smith ierr = MPI_Finalize();CHKERRQ(ierr); 3787ca660e7SBarry Smith exit(0); 3797ca660e7SBarry Smith } 380e5c89e4eSSatish Balay } 381e5c89e4eSSatish Balay 382e5c89e4eSSatish Balay /* 383e5c89e4eSSatish Balay Setup the error handling 384e5c89e4eSSatish Balay */ 38590d69ab7SBarry Smith flg1 = PETSC_FALSE; 386c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_abort",&flg1,NULL);CHKERRQ(ierr); 387b59baad1SJed Brown if (flg1) { 38842218b76SBarry Smith ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr); 389b59baad1SJed Brown ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr); 390b59baad1SJed Brown } 39190d69ab7SBarry Smith flg1 = PETSC_FALSE; 392c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-on_error_mpiabort",&flg1,NULL);CHKERRQ(ierr); 393cb9801acSJed Brown if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,0);CHKERRQ(ierr);} 39490d69ab7SBarry Smith flg1 = PETSC_FALSE; 395c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-mpi_return_on_error",&flg1,NULL);CHKERRQ(ierr); 396e5c89e4eSSatish Balay if (flg1) { 397d54338ecSKarl Rupp ierr = MPI_Comm_set_errhandler(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr); 398e5c89e4eSSatish Balay } 39990d69ab7SBarry Smith flg1 = PETSC_FALSE; 400c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-no_signal_handler",&flg1,NULL);CHKERRQ(ierr); 4018d359177SBarry Smith if (!flg1) {ierr = PetscPushSignalHandler(PetscSignalHandlerDefault,(void*)0);CHKERRQ(ierr);} 40296cc47afSJed Brown flg1 = PETSC_FALSE; 403c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-fp_trap",&flg1,NULL);CHKERRQ(ierr); 40496cc47afSJed Brown if (flg1) {ierr = PetscSetFPTrap(PETSC_FP_TRAP_ON);CHKERRQ(ierr);} 405c5929fdfSBarry Smith ierr = PetscOptionsGetInt(NULL,NULL,"-check_pointer_intensity",&intensity,&flag);CHKERRQ(ierr); 40628559dc8SJed Brown if (flag) {ierr = PetscCheckPointerSetIntensity(intensity);CHKERRQ(ierr);} 407e5c89e4eSSatish Balay 408e5c89e4eSSatish Balay /* 409e5c89e4eSSatish Balay Setup debugger information 410e5c89e4eSSatish Balay */ 411e5c89e4eSSatish Balay ierr = PetscSetDefaultDebugger();CHKERRQ(ierr); 412c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-on_error_attach_debugger",string,64,&flg1);CHKERRQ(ierr); 413e5c89e4eSSatish Balay if (flg1) { 414e5c89e4eSSatish Balay MPI_Errhandler err_handler; 415e5c89e4eSSatish Balay 416e5c89e4eSSatish Balay ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr); 41733c7d699SBarry Smith ierr = MPI_Comm_create_errhandler(Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr); 418d54338ecSKarl Rupp ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr); 419e5c89e4eSSatish Balay ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,0);CHKERRQ(ierr); 420e5c89e4eSSatish Balay } 421c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-debug_terminal",string,64,&flg1);CHKERRQ(ierr); 4225e96ac45SJed Brown if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); } 423c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-start_in_debugger",string,64,&flg1);CHKERRQ(ierr); 424c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-stop_for_debugger",string,64,&flg2);CHKERRQ(ierr); 425e5c89e4eSSatish Balay if (flg1 || flg2) { 426e5c89e4eSSatish Balay PetscMPIInt size; 427e5c89e4eSSatish Balay PetscInt lsize,*nodes; 428e5c89e4eSSatish Balay MPI_Errhandler err_handler; 429e5c89e4eSSatish Balay /* 430e5c89e4eSSatish Balay we have to make sure that all processors have opened 431e5c89e4eSSatish Balay connections to all other processors, otherwise once the 432e5c89e4eSSatish Balay debugger has stated it is likely to receive a SIGUSR1 433e5c89e4eSSatish Balay and kill the program. 434e5c89e4eSSatish Balay */ 43542218b76SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 436e5c89e4eSSatish Balay if (size > 2) { 437533163c2SBarry Smith PetscMPIInt dummy = 0; 438e5c89e4eSSatish Balay MPI_Status status; 439e5c89e4eSSatish Balay for (i=0; i<size; i++) { 440e5c89e4eSSatish Balay if (rank != i) { 44142218b76SBarry Smith ierr = MPI_Send(&dummy,1,MPI_INT,i,109,comm);CHKERRQ(ierr); 442e5c89e4eSSatish Balay } 443e5c89e4eSSatish Balay } 444e5c89e4eSSatish Balay for (i=0; i<size; i++) { 445e5c89e4eSSatish Balay if (rank != i) { 44642218b76SBarry Smith ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,comm,&status);CHKERRQ(ierr); 447e5c89e4eSSatish Balay } 448e5c89e4eSSatish Balay } 449e5c89e4eSSatish Balay } 450e5c89e4eSSatish Balay /* check if this processor node should be in debugger */ 451785e854fSJed Brown ierr = PetscMalloc1(size,&nodes);CHKERRQ(ierr); 452e5c89e4eSSatish Balay lsize = size; 453c5929fdfSBarry Smith ierr = PetscOptionsGetIntArray(NULL,NULL,"-debugger_nodes",nodes,&lsize,&flag);CHKERRQ(ierr); 454e5c89e4eSSatish Balay if (flag) { 455e5c89e4eSSatish Balay for (i=0; i<lsize; i++) { 456e5c89e4eSSatish Balay if (nodes[i] == rank) { flag = PETSC_FALSE; break; } 457e5c89e4eSSatish Balay } 458e5c89e4eSSatish Balay } 459e5c89e4eSSatish Balay if (!flag) { 460e5c89e4eSSatish Balay ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr); 461e5c89e4eSSatish Balay ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr); 462e5c89e4eSSatish Balay if (flg1) { 463e5c89e4eSSatish Balay ierr = PetscAttachDebugger();CHKERRQ(ierr); 464e5c89e4eSSatish Balay } else { 465e5c89e4eSSatish Balay ierr = PetscStopForDebugger();CHKERRQ(ierr); 466e5c89e4eSSatish Balay } 46733c7d699SBarry Smith ierr = MPI_Comm_create_errhandler(Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr); 468d54338ecSKarl Rupp ierr = MPI_Comm_set_errhandler(comm,err_handler);CHKERRQ(ierr); 469e5c89e4eSSatish Balay } 470e5c89e4eSSatish Balay ierr = PetscFree(nodes);CHKERRQ(ierr); 471e5c89e4eSSatish Balay } 472e5c89e4eSSatish Balay 473c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-on_error_emacs",emacsmachinename,128,&flg1);CHKERRQ(ierr); 474cb9801acSJed Brown if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);} 475e5c89e4eSSatish Balay 476e5c89e4eSSatish Balay /* 477e5c89e4eSSatish Balay Setup profiling and logging 478e5c89e4eSSatish Balay */ 4796cf91177SBarry Smith #if defined(PETSC_USE_INFO) 4808bb29257SSatish Balay { 481e5c89e4eSSatish Balay char logname[PETSC_MAX_PATH_LEN]; logname[0] = 0; 482c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-info",logname,250,&flg1);CHKERRQ(ierr); 4838bb29257SSatish Balay if (flg1 && logname[0]) { 484fcc2139eSBarry Smith ierr = PetscInfoAllow(PETSC_TRUE,logname);CHKERRQ(ierr); 4858bb29257SSatish Balay } else if (flg1) { 4860298fd71SBarry Smith ierr = PetscInfoAllow(PETSC_TRUE,NULL);CHKERRQ(ierr); 487e5c89e4eSSatish Balay } 488e5c89e4eSSatish Balay } 489865f6aa8SSatish Balay #endif 490865f6aa8SSatish Balay #if defined(PETSC_USE_LOG) 491865f6aa8SSatish Balay mname[0] = 0; 492c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-history",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr); 493865f6aa8SSatish Balay if (flg1) { 494865f6aa8SSatish Balay if (mname[0]) { 495f3dea69dSBarry Smith ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr); 496865f6aa8SSatish Balay } else { 497706d7a88SBarry Smith ierr = PetscOpenHistoryFile(NULL,&petsc_history);CHKERRQ(ierr); 498865f6aa8SSatish Balay } 499865f6aa8SSatish Balay } 500217044c2SLisandro Dalcin 501217044c2SLisandro Dalcin ierr = PetscOptionsGetBool(NULL,NULL,"-log_sync",&PetscLogSyncOn,NULL);CHKERRQ(ierr); 502217044c2SLisandro Dalcin 503e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE) 50490d69ab7SBarry Smith flg1 = PETSC_FALSE; 505c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_mpe",&flg1);CHKERRQ(ierr); 506495fc317SBarry Smith if (flg1) {ierr = PetscLogMPEBegin();CHKERRQ(ierr);} 507e5c89e4eSSatish Balay #endif 50890d69ab7SBarry Smith flg1 = PETSC_FALSE; 50990d69ab7SBarry Smith flg3 = PETSC_FALSE; 510c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-log_all",&flg1,NULL);CHKERRQ(ierr); 511c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr); 512e5c89e4eSSatish Balay if (flg1) { ierr = PetscLogAllBegin();CHKERRQ(ierr); } 513bb1d7374SBarry Smith else if (flg3) { ierr = PetscLogDefaultBegin();CHKERRQ(ierr);} 514e5c89e4eSSatish Balay 515c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-log_trace",mname,250,&flg1);CHKERRQ(ierr); 516e5c89e4eSSatish Balay if (flg1) { 517e5c89e4eSSatish Balay char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN]; 518e5c89e4eSSatish Balay FILE *file; 519e5c89e4eSSatish Balay if (mname[0]) { 5202e924ca5SSatish Balay PetscSNPrintf(name,PETSC_MAX_PATH_LEN,"%s.%d",mname,rank); 521e5c89e4eSSatish Balay ierr = PetscFixFilename(name,fname);CHKERRQ(ierr); 522e5c89e4eSSatish Balay file = fopen(fname,"w"); 523f3dea69dSBarry Smith if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname); 524a297a907SKarl Rupp } else file = PETSC_STDOUT; 525e5c89e4eSSatish Balay ierr = PetscLogTraceBegin(file);CHKERRQ(ierr); 526e5c89e4eSSatish Balay } 527bb1d7374SBarry Smith 52816413a6aSBarry Smith ierr = PetscOptionsGetViewer(comm,NULL,NULL,"-log_view",NULL,&format,&flg4);CHKERRQ(ierr); 529bb1d7374SBarry Smith if (flg4) { 530bb1d7374SBarry Smith if (format == PETSC_VIEWER_ASCII_XML) { 531bb1d7374SBarry Smith ierr = PetscLogNestedBegin();CHKERRQ(ierr); 532bb1d7374SBarry Smith } else { 533bb1d7374SBarry Smith ierr = PetscLogDefaultBegin();CHKERRQ(ierr); 534bb1d7374SBarry Smith } 535e3ed9ee7SBarry Smith PetscLogMemory = PETSC_FALSE; 536e3ed9ee7SBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-log_view_memory",&PetscLogMemory,NULL);CHKERRQ(ierr); 537e3ed9ee7SBarry Smith if (PetscLogMemory) { 538e3ed9ee7SBarry Smith ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 539e3ed9ee7SBarry Smith } 540eccbb886SLisandro Dalcin } 541eccbb886SLisandro Dalcin if (flg4 && format == PETSC_VIEWER_ASCII_XML) { 542eccbb886SLisandro Dalcin PetscReal threshold = PetscRealConstant(0.01); 543eccbb886SLisandro Dalcin ierr = PetscOptionsGetReal(NULL,NULL,"-log_threshold",&threshold,&flg1);CHKERRQ(ierr); 544eccbb886SLisandro Dalcin if (flg1) {ierr = PetscLogSetThreshold((PetscLogDouble)threshold,NULL);CHKERRQ(ierr);} 545bb1d7374SBarry Smith } 546e5c89e4eSSatish Balay #endif 547e5c89e4eSSatish Balay 548c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-saws_options",&PetscOptionsPublish,NULL);CHKERRQ(ierr); 549e5c89e4eSSatish Balay 5504b0a73b2SBarry Smith #if defined(PETSC_HAVE_CUDA) 551bffec52eSKarl Rupp if (!PetscCUDAInitialized) { 55216b6f2efSBarry Smith PetscMPIInt size; 55342218b76SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 554bd72cc96SPaul Mullowney if (size>1) { 55516b6f2efSBarry Smith int devCount; 556*fef7099cSRichard Tran Mills int device; 55716b6f2efSBarry Smith PetscMPIInt rank; 5582692e278SPaul Mullowney cudaError_t err = cudaSuccess; 5592692e278SPaul Mullowney 5602692e278SPaul Mullowney /* check to see if we force multiple ranks to hit the same GPU */ 561c5929fdfSBarry Smith ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr); 5622692e278SPaul Mullowney if (flg1) { 56316b6f2efSBarry Smith err = cudaSetDevice((int)device); 564e09ad801SBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err)); 5652692e278SPaul Mullowney } else { 566e09ad801SBarry Smith /* we're not using the same GPU on multiple MPI threads. So try to allocated different GPUs to different processes */ 5672692e278SPaul Mullowney 5682692e278SPaul Mullowney /* First get the device count */ 569e09ad801SBarry Smith err = cudaGetDeviceCount(&devCount); 570e09ad801SBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err)); 5712692e278SPaul Mullowney 5722692e278SPaul Mullowney /* next determine the rank and then set the device via a mod */ 57342218b76SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 574bd72cc96SPaul Mullowney device = rank % devCount; 57516b6f2efSBarry Smith err = cudaSetDevice((int)device); 576e09ad801SBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err)); 5772692e278SPaul Mullowney } 5782692e278SPaul Mullowney 5792692e278SPaul Mullowney /* set the device flags so that it can map host memory ... do NOT throw exception on err!=cudaSuccess 5802692e278SPaul Mullowney multiple devices may try to set the flags on the same device. So long as one of them succeeds, things 5812692e278SPaul Mullowney are ok. */ 5822692e278SPaul Mullowney err = cudaSetDeviceFlags(cudaDeviceMapHost); 583e09ad801SBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err)); 584a297a907SKarl Rupp } else { 58516b6f2efSBarry Smith PetscInt device; 5862692e278SPaul Mullowney cudaError_t err = cudaSuccess; 5872692e278SPaul Mullowney 588f297a14cSJed Brown /* the code below works for serial GPU simulations */ 589c5929fdfSBarry Smith ierr = PetscOptionsGetInt(NULL,NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr); 5904b0a73b2SBarry Smith if (flg1) { 59116b6f2efSBarry Smith err = cudaSetDevice((int)device); 592e09ad801SBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDevice %s",cudaGetErrorString(err)); 5934b0a73b2SBarry Smith } 5942692e278SPaul Mullowney 5952692e278SPaul Mullowney /* set the device flags so that it can map host memory ... here, we error check. */ 5962692e278SPaul Mullowney err = cudaSetDeviceFlags(cudaDeviceMapHost); 597e09ad801SBarry Smith if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaSetDeviceFlags %s",cudaGetErrorString(err)); 5984b0a73b2SBarry Smith } 599bffec52eSKarl Rupp 600bffec52eSKarl Rupp PetscCUDAInitialized = PETSC_TRUE; 601bd72cc96SPaul Mullowney } 602*fef7099cSRichard Tran Mills ierr = PetscOptionsBegin(comm,NULL,"CUDA options",NULL);CHKERRQ(ierr); 603*fef7099cSRichard Tran Mills ierr = PetscOptionsDeprecated("-cuda_show_devices","-cuda_view","3.12",NULL);CHKERRQ(ierr); 604*fef7099cSRichard Tran Mills ierr = PetscOptionsName("-cuda_view","Display CUDA device information and assignments",NULL,&flg1);CHKERRQ(ierr); 605*fef7099cSRichard Tran Mills ierr = PetscOptionsEnd();CHKERRQ(ierr); 6067be52123SRichard Tran Mills if (flg1) { 6077be52123SRichard Tran Mills struct cudaDeviceProp prop; 6087be52123SRichard Tran Mills int devCount; 6097be52123SRichard Tran Mills PetscInt device; 6107be52123SRichard Tran Mills cudaError_t err = cudaSuccess; 6117be52123SRichard Tran Mills 6127be52123SRichard Tran Mills err = cudaGetDeviceCount(&devCount); 6137be52123SRichard Tran Mills if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceCount %s",cudaGetErrorString(err)); 6147be52123SRichard Tran Mills for (device = 0; device < devCount; ++device) { 6157be52123SRichard Tran Mills err = cudaGetDeviceProperties(&prop, (int)device); 6167be52123SRichard Tran Mills if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDeviceProperties %s",cudaGetErrorString(err)); 617*fef7099cSRichard Tran Mills ierr = PetscPrintf(comm, "CUDA device %d: %s\n", device, prop.name);CHKERRQ(ierr); 6187be52123SRichard Tran Mills } 6197be52123SRichard Tran Mills err = cudaGetDevice(&device); 6207be52123SRichard Tran Mills if (err != cudaSuccess) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SYS,"error in cudaGetDevice %s",cudaGetErrorString(err)); 6217be52123SRichard Tran Mills ierr = PetscSynchronizedPrintf(PETSC_COMM_WORLD,"[%d] Using CUDA device %d.\n",rank,device);CHKERRQ(ierr); 6227be52123SRichard Tran Mills ierr = PetscSynchronizedFlush(PETSC_COMM_WORLD,PETSC_STDOUT);CHKERRQ(ierr); 6237be52123SRichard Tran Mills } 6247be52123SRichard Tran Mills 6254b0a73b2SBarry Smith #endif 6264b0a73b2SBarry Smith 6274b0a73b2SBarry Smith 628e5c89e4eSSatish Balay /* 629e5c89e4eSSatish Balay Print basic help message 630e5c89e4eSSatish Balay */ 6312d747510SLisandro Dalcin ierr = PetscOptionsHasHelp(NULL,&flg1);CHKERRQ(ierr); 632e5c89e4eSSatish Balay if (flg1) { 633e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr); 634301d30feSBarry Smith ierr = (*PetscHelpPrintf)(comm," -help: prints help method for each option\n");CHKERRQ(ierr); 635301d30feSBarry Smith ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr); 636301d30feSBarry Smith ierr = (*PetscHelpPrintf)(comm," only when run in the debugger\n");CHKERRQ(ierr); 637e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr); 638e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," start the debugger in new xterm\n");CHKERRQ(ierr); 639e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," unless noxterm is given\n");CHKERRQ(ierr); 640e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr); 641e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," start all processes in the debugger\n");CHKERRQ(ierr); 642e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr); 643e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," emacs jumps to error file\n");CHKERRQ(ierr); 644e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -debugger_nodes [n1,n2,..] Nodes to start in debugger\n");CHKERRQ(ierr); 645e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr); 646e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr); 647e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," waits the delay for you to attach\n");CHKERRQ(ierr); 6481cda70a7SBarry Smith ierr = (*PetscHelpPrintf)(comm," -display display: Location where X window graphics and debuggers are displayed\n");CHKERRQ(ierr); 649e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr); 650e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr); 651e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr); 652e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr); 653e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr); 654e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -malloc: use our error checking malloc\n");CHKERRQ(ierr); 655e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use error checking malloc\n");CHKERRQ(ierr); 6564161f2a3SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr); 6574161f2a3SBarry Smith ierr = (*PetscHelpPrintf)(comm," -malloc_log: keeps log of all memory allocations\n");CHKERRQ(ierr); 658e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -malloc_debug: enables extended checking for memory corruption\n");CHKERRQ(ierr); 65926a7e8d4SBarry Smith ierr = (*PetscHelpPrintf)(comm," -options_view: dump list of options inputted\n");CHKERRQ(ierr); 660e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr); 661e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr); 662e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr); 663e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr); 664a8c7a070SBarry Smith ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr); 6650841954dSBarry Smith ierr = (*PetscHelpPrintf)(comm," -memory_view: print memory usage at end of run\n");CHKERRQ(ierr); 666e5c89e4eSSatish Balay #if defined(PETSC_USE_LOG) 667e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr); 668185ae32fSMatthew G. Knepley ierr = (*PetscHelpPrintf)(comm," -log_view [:filename:[format]]: logging objects and events\n");CHKERRQ(ierr); 669e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr); 670e5c89e4eSSatish Balay #if defined(PETSC_HAVE_MPE) 671495fc317SBarry Smith ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through Jumpshot\n");CHKERRQ(ierr); 672e5c89e4eSSatish Balay #endif 6736cf91177SBarry Smith ierr = (*PetscHelpPrintf)(comm," -info <optional filename>: print informative messages about the calculations\n");CHKERRQ(ierr); 674e5c89e4eSSatish Balay #endif 675e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -v: prints PETSc version number and release date\n");CHKERRQ(ierr); 676e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr); 677e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr); 678e5c89e4eSSatish Balay ierr = (*PetscHelpPrintf)(comm,"-----------------------------------------------\n");CHKERRQ(ierr); 679e5c89e4eSSatish Balay } 680e5c89e4eSSatish Balay 68174ba8654SBarry Smith #if defined(PETSC_HAVE_POPEN) 68274ba8654SBarry Smith { 68374ba8654SBarry Smith char machine[128]; 684c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-popen_machine",machine,128,&flg1);CHKERRQ(ierr); 68574ba8654SBarry Smith if (flg1) { 68674ba8654SBarry Smith ierr = PetscPOpenSetMachine(machine);CHKERRQ(ierr); 68774ba8654SBarry Smith } 68874ba8654SBarry Smith } 68974ba8654SBarry Smith #endif 69074ba8654SBarry Smith 691c5929fdfSBarry Smith ierr = PetscOptionsGetReal(NULL,NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr); 692e5c89e4eSSatish Balay if (flg1) { 693e5c89e4eSSatish Balay ierr = PetscSleep(si);CHKERRQ(ierr); 694e5c89e4eSSatish Balay } 695e5c89e4eSSatish Balay 696c5929fdfSBarry Smith ierr = PetscOptionsGetString(NULL,NULL,"-info_exclude",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr); 6972ea1be5dSBarry Smith if (flg1) { 698e5c89e4eSSatish Balay ierr = PetscStrstr(mname,"null",&f);CHKERRQ(ierr); 699e5c89e4eSSatish Balay if (f) { 700ecd8bba6SJed Brown ierr = PetscInfoDeactivateClass(0);CHKERRQ(ierr); 701e5c89e4eSSatish Balay } 7022ea1be5dSBarry Smith } 703827f890bSBarry Smith 704fd314934SBarry Smith #if defined(PETSC_HAVE_VIENNACL) || defined(PETSC_HAVE_CUDA) 705c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_summary",&flg3);CHKERRQ(ierr); 706f14045dbSBarry Smith if (!flg3) { 707c5929fdfSBarry Smith ierr = PetscOptionsHasName(NULL,NULL,"-log_view",&flg3);CHKERRQ(ierr); 708f14045dbSBarry Smith } 709bab1f7e6SVictor Minden #endif 710e9e886b6SKarl Rupp #if defined(PETSC_HAVE_VIENNACL) 711c5929fdfSBarry Smith ierr = PetscOptionsGetBool(NULL,NULL,"-viennacl_synchronize",&flg3,NULL);CHKERRQ(ierr); 712f14045dbSBarry Smith PetscViennaCLSynchronize = flg3; 713b8ced49eSKarl Rupp #endif 714fd314934SBarry Smith #if defined(PETSC_HAVE_CUDA) 71582f73ecaSAlejandro Lamas Daviña ierr = PetscOptionsGetBool(NULL,NULL,"-cuda_synchronize",&flg3,NULL);CHKERRQ(ierr); 71682f73ecaSAlejandro Lamas Daviña PetscCUDASynchronize = flg3; 7174cf1874eSKarl Rupp #endif 71882f73ecaSAlejandro Lamas Daviña 719f0a7718cSKarl Rupp #if defined(PETSC_HAVE_VIENNACL) 720f0a7718cSKarl Rupp ierr = PetscViennaCLInit();CHKERRQ(ierr); 721f0a7718cSKarl Rupp #endif 722f0a7718cSKarl Rupp 723e5c89e4eSSatish Balay PetscFunctionReturn(0); 724e5c89e4eSSatish Balay } 725