1 /* 2 3 This file defines part of the initialization of PETSc 4 5 This file uses regular malloc and free because it cannot known 6 what malloc is being used until it has already processed the input. 7 */ 8 9 #include <petscsys.h> /*I "petscsys.h" I*/ 10 11 #if defined(PETSC_HAVE_SYS_SYSINFO_H) 12 #include <sys/sysinfo.h> 13 #endif 14 #if defined(PETSC_HAVE_UNISTD_H) 15 #include <unistd.h> 16 #endif 17 #if defined(PETSC_HAVE_STDLIB_H) 18 #include <stdlib.h> 19 #endif 20 #if defined(PETSC_HAVE_MALLOC_H) 21 #include <malloc.h> 22 #endif 23 24 /* ------------------------Nasty global variables -------------------------------*/ 25 /* 26 Indicates if PETSc started up MPI, or it was 27 already started before PETSc was initialized. 28 */ 29 PetscBool PetscBeganMPI = PETSC_FALSE; 30 PetscBool PetscInitializeCalled = PETSC_FALSE; 31 PetscBool PetscFinalizeCalled = PETSC_FALSE; 32 PetscMPIInt PetscGlobalRank = -1; 33 PetscMPIInt PetscGlobalSize = -1; 34 35 #if defined(PETSC_USE_COMPLEX) 36 #if defined(PETSC_COMPLEX_INSTANTIATE) 37 template <> class std::complex<double>; /* instantiate complex template class */ 38 #endif 39 #if !defined(PETSC_HAVE_MPI_C_DOUBLE_COMPLEX) 40 MPI_Datatype MPIU_C_DOUBLE_COMPLEX; 41 MPI_Datatype MPIU_C_COMPLEX; 42 #endif 43 PetscScalar PETSC_i; 44 #else 45 PetscScalar PETSC_i = 0.0; 46 #endif 47 #if defined(PETSC_USE_REAL___FLOAT128) 48 MPI_Datatype MPIU___FLOAT128 = 0; 49 #endif 50 MPI_Datatype MPIU_2SCALAR = 0; 51 MPI_Datatype MPIU_2INT = 0; 52 53 /* 54 Function that is called to display all error messages 55 */ 56 PetscErrorCode (*PetscErrorPrintf)(const char [],...) = PetscErrorPrintfDefault; 57 PetscErrorCode (*PetscHelpPrintf)(MPI_Comm,const char [],...) = PetscHelpPrintfDefault; 58 #if defined(PETSC_HAVE_MATLAB_ENGINE) 59 PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list) = PetscVFPrintf_Matlab; 60 #else 61 PetscErrorCode (*PetscVFPrintf)(FILE*,const char[],va_list) = PetscVFPrintfDefault; 62 #endif 63 /* 64 This is needed to turn on/off cusp synchronization 65 */ 66 PetscBool PetscCUSPSynchronize = PETSC_FALSE; 67 68 /* pthread_key for PetscStack */ 69 #if defined(PETSC_HAVE_PTHREADCLASSES) && !defined(PETSC_PTHREAD_LOCAL) 70 pthread_key_t petscstack_key; 71 #endif 72 73 /* ------------------------------------------------------------------------------*/ 74 /* 75 Optional file where all PETSc output from various prints is saved 76 */ 77 FILE *petsc_history = PETSC_NULL; 78 79 #undef __FUNCT__ 80 #define __FUNCT__ "PetscOpenHistoryFile" 81 PetscErrorCode PetscOpenHistoryFile(const char filename[],FILE **fd) 82 { 83 PetscErrorCode ierr; 84 PetscMPIInt rank,size; 85 char pfile[PETSC_MAX_PATH_LEN],pname[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN],date[64]; 86 char version[256]; 87 88 PetscFunctionBegin; 89 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 90 if (!rank) { 91 char arch[10]; 92 int err; 93 PetscViewer viewer; 94 95 ierr = PetscGetArchType(arch,10);CHKERRQ(ierr); 96 ierr = PetscGetDate(date,64);CHKERRQ(ierr); 97 ierr = PetscGetVersion(version,256);CHKERRQ(ierr); 98 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 99 if (filename) { 100 ierr = PetscFixFilename(filename,fname);CHKERRQ(ierr); 101 } else { 102 ierr = PetscGetHomeDirectory(pfile,240);CHKERRQ(ierr); 103 ierr = PetscStrcat(pfile,"/.petschistory");CHKERRQ(ierr); 104 ierr = PetscFixFilename(pfile,fname);CHKERRQ(ierr); 105 } 106 107 *fd = fopen(fname,"a"); if (!fd) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Cannot open file: %s",fname); 108 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 109 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s %s\n",version,date);CHKERRQ(ierr); 110 ierr = PetscGetProgramName(pname,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 111 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"%s on a %s, %d proc. with options:\n",pname,arch,size);CHKERRQ(ierr); 112 ierr = PetscViewerASCIIOpenWithFILE(PETSC_COMM_WORLD,*fd,&viewer);CHKERRQ(ierr); 113 ierr = PetscOptionsView(viewer);CHKERRQ(ierr); 114 ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 115 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 116 err = fflush(*fd); 117 if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 118 } 119 PetscFunctionReturn(0); 120 } 121 122 #undef __FUNCT__ 123 #define __FUNCT__ "PetscCloseHistoryFile" 124 PetscErrorCode PetscCloseHistoryFile(FILE **fd) 125 { 126 PetscErrorCode ierr; 127 PetscMPIInt rank; 128 char date[64]; 129 int err; 130 131 PetscFunctionBegin; 132 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 133 if (!rank) { 134 ierr = PetscGetDate(date,64);CHKERRQ(ierr); 135 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 136 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"Finished at %s\n",date);CHKERRQ(ierr); 137 ierr = PetscFPrintf(PETSC_COMM_SELF,*fd,"---------------------------------------------------------\n");CHKERRQ(ierr); 138 err = fflush(*fd); 139 if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fflush() failed on file"); 140 err = fclose(*fd); 141 if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file"); 142 } 143 PetscFunctionReturn(0); 144 } 145 146 /* ------------------------------------------------------------------------------*/ 147 148 /* 149 This is ugly and probably belongs somewhere else, but I want to 150 be able to put a true MPI abort error handler with command line args. 151 152 This is so MPI errors in the debugger will leave all the stack 153 frames. The default MP_Abort() cleans up and exits thus providing no useful information 154 in the debugger hence we call abort() instead of MPI_Abort(). 155 */ 156 157 #undef __FUNCT__ 158 #define __FUNCT__ "Petsc_MPI_AbortOnError" 159 void Petsc_MPI_AbortOnError(MPI_Comm *comm,PetscMPIInt *flag) 160 { 161 PetscFunctionBegin; 162 (*PetscErrorPrintf)("MPI error %d\n",*flag); 163 abort(); 164 } 165 166 #undef __FUNCT__ 167 #define __FUNCT__ "Petsc_MPI_DebuggerOnError" 168 void Petsc_MPI_DebuggerOnError(MPI_Comm *comm,PetscMPIInt *flag) 169 { 170 PetscErrorCode ierr; 171 172 PetscFunctionBegin; 173 (*PetscErrorPrintf)("MPI error %d\n",*flag); 174 ierr = PetscAttachDebugger(); 175 if (ierr) { /* hopeless so get out */ 176 MPI_Abort(*comm,*flag); 177 } 178 } 179 180 #undef __FUNCT__ 181 #define __FUNCT__ "PetscEnd" 182 /*@C 183 PetscEnd - Calls PetscFinalize() and then ends the program. This is useful if one 184 wishes a clean exit somewhere deep in the program. 185 186 Collective on PETSC_COMM_WORLD 187 188 Options Database Keys are the same as for PetscFinalize() 189 190 Level: advanced 191 192 Note: 193 See PetscInitialize() for more general runtime options. 194 195 .seealso: PetscInitialize(), PetscOptionsView(), PetscMallocDump(), PetscMPIDump(), PetscFinalize() 196 @*/ 197 PetscErrorCode PetscEnd(void) 198 { 199 PetscFunctionBegin; 200 PetscFinalize(); 201 exit(0); 202 return 0; 203 } 204 205 PetscBool PetscOptionsPublish = PETSC_FALSE; 206 extern PetscErrorCode PetscSetUseTrMalloc_Private(void); 207 extern PetscBool petscsetmallocvisited; 208 static char emacsmachinename[256]; 209 210 PetscErrorCode (*PetscExternalVersionFunction)(MPI_Comm) = 0; 211 PetscErrorCode (*PetscExternalHelpFunction)(MPI_Comm) = 0; 212 213 #undef __FUNCT__ 214 #define __FUNCT__ "PetscSetHelpVersionFunctions" 215 /*@C 216 PetscSetHelpVersionFunctions - Sets functions that print help and version information 217 before the PETSc help and version information is printed. Must call BEFORE PetscInitialize(). 218 This routine enables a "higher-level" package that uses PETSc to print its messages first. 219 220 Input Parameter: 221 + help - the help function (may be PETSC_NULL) 222 - version - the version function (may be PETSC_NULL) 223 224 Level: developer 225 226 Concepts: package help message 227 228 @*/ 229 PetscErrorCode PetscSetHelpVersionFunctions(PetscErrorCode (*help)(MPI_Comm),PetscErrorCode (*version)(MPI_Comm)) 230 { 231 PetscFunctionBegin; 232 PetscExternalHelpFunction = help; 233 PetscExternalVersionFunction = version; 234 PetscFunctionReturn(0); 235 } 236 237 #if defined(PETSC_HAVE_CUDA) 238 #include <cublas.h> 239 #endif 240 241 #undef __FUNCT__ 242 #define __FUNCT__ "PetscOptionsCheckInitial_Private" 243 PetscErrorCode PetscOptionsCheckInitial_Private(void) 244 { 245 char string[64],mname[PETSC_MAX_PATH_LEN],*f; 246 MPI_Comm comm = PETSC_COMM_WORLD; 247 PetscBool flg1 = PETSC_FALSE,flg2 = PETSC_FALSE,flg3 = PETSC_FALSE,flg4 = PETSC_FALSE,flag; 248 PetscErrorCode ierr; 249 PetscReal si,logthreshold; 250 int i; 251 PetscMPIInt rank; 252 char version[256]; 253 #if defined(PETSC_USE_SERVER) 254 PetscBool flgz; 255 #endif 256 257 PetscFunctionBegin; 258 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 259 260 /* 261 Setup the memory management; support for tracing malloc() usage 262 */ 263 ierr = PetscOptionsHasName(PETSC_NULL,"-malloc_log",&flg3);CHKERRQ(ierr); 264 logthreshold = 0.0; 265 ierr = PetscOptionsGetReal(PETSC_NULL,"-malloc_log_threshold",&logthreshold,&flg1);CHKERRQ(ierr); 266 if (flg1) flg3 = PETSC_TRUE; 267 #if defined(PETSC_USE_DEBUG) 268 ierr = PetscOptionsGetBool(PETSC_NULL,"-malloc",&flg1,&flg2);CHKERRQ(ierr); 269 if ((!flg2 || flg1) && !petscsetmallocvisited) { 270 if (flg2 || !(PETSC_RUNNING_ON_VALGRIND)) { 271 /* turn off default -malloc if valgrind is being used */ 272 ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 273 } 274 } 275 #else 276 ierr = PetscOptionsGetBool(PETSC_NULL,"-malloc_dump",&flg1,PETSC_NULL);CHKERRQ(ierr); 277 ierr = PetscOptionsGetBool(PETSC_NULL,"-malloc",&flg2,PETSC_NULL);CHKERRQ(ierr); 278 if (flg1 || flg2 || flg3) {ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr);} 279 #endif 280 if (flg3) { 281 ierr = PetscMallocSetDumpLogThreshold((PetscLogDouble)logthreshold);CHKERRQ(ierr); 282 } 283 flg1 = PETSC_FALSE; 284 ierr = PetscOptionsGetBool(PETSC_NULL,"-malloc_debug",&flg1,PETSC_NULL);CHKERRQ(ierr); 285 if (flg1) { 286 ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 287 ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr); 288 } 289 flg1 = PETSC_FALSE; 290 ierr = PetscOptionsGetBool(PETSC_NULL,"-malloc_test",&flg1,PETSC_NULL);CHKERRQ(ierr); 291 #if defined(PETSC_USE_DEBUG) 292 if (flg1 && !PETSC_RUNNING_ON_VALGRIND) { 293 ierr = PetscSetUseTrMalloc_Private();CHKERRQ(ierr); 294 ierr = PetscMallocSetDumpLog();CHKERRQ(ierr); 295 ierr = PetscMallocDebug(PETSC_TRUE);CHKERRQ(ierr); 296 } 297 #endif 298 299 flg1 = PETSC_FALSE; 300 ierr = PetscOptionsGetBool(PETSC_NULL,"-malloc_info",&flg1,PETSC_NULL);CHKERRQ(ierr); 301 if (!flg1) { 302 flg1 = PETSC_FALSE; 303 ierr = PetscOptionsGetBool(PETSC_NULL,"-memory_info",&flg1,PETSC_NULL);CHKERRQ(ierr); 304 } 305 if (flg1) { 306 ierr = PetscMemorySetGetMaximumUsage();CHKERRQ(ierr); 307 } 308 309 /* 310 Set the display variable for graphics 311 */ 312 ierr = PetscSetDisplay();CHKERRQ(ierr); 313 314 /* 315 Print the PETSc version information 316 */ 317 ierr = PetscOptionsHasName(PETSC_NULL,"-v",&flg1);CHKERRQ(ierr); 318 ierr = PetscOptionsHasName(PETSC_NULL,"-version",&flg2);CHKERRQ(ierr); 319 ierr = PetscOptionsHasName(PETSC_NULL,"-help",&flg3);CHKERRQ(ierr); 320 if (flg1 || flg2 || flg3){ 321 322 /* 323 Print "higher-level" package version message 324 */ 325 if (PetscExternalVersionFunction) { 326 ierr = (*PetscExternalVersionFunction)(comm);CHKERRQ(ierr); 327 } 328 329 ierr = PetscGetVersion(version,256);CHKERRQ(ierr); 330 ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------\ 331 ------------------------------\n");CHKERRQ(ierr); 332 ierr = (*PetscHelpPrintf)(comm,"%s\n",version);CHKERRQ(ierr); 333 ierr = (*PetscHelpPrintf)(comm,"%s",PETSC_AUTHOR_INFO);CHKERRQ(ierr); 334 ierr = (*PetscHelpPrintf)(comm,"See docs/changes/index.html for recent updates.\n");CHKERRQ(ierr); 335 ierr = (*PetscHelpPrintf)(comm,"See docs/faq.html for problems.\n");CHKERRQ(ierr); 336 ierr = (*PetscHelpPrintf)(comm,"See docs/manualpages/index.html for help. \n");CHKERRQ(ierr); 337 ierr = (*PetscHelpPrintf)(comm,"Libraries linked from %s\n",PETSC_LIB_DIR);CHKERRQ(ierr); 338 ierr = (*PetscHelpPrintf)(comm,"--------------------------------------------\ 339 ------------------------------\n");CHKERRQ(ierr); 340 } 341 342 /* 343 Print "higher-level" package help message 344 */ 345 if (flg3){ 346 if (PetscExternalHelpFunction) { 347 ierr = (*PetscExternalHelpFunction)(comm);CHKERRQ(ierr); 348 } 349 } 350 351 /* 352 Setup the error handling 353 */ 354 flg1 = PETSC_FALSE; 355 ierr = PetscOptionsGetBool(PETSC_NULL,"-on_error_abort",&flg1,PETSC_NULL);CHKERRQ(ierr); 356 if (flg1) { 357 ierr = MPI_Errhandler_set(PETSC_COMM_WORLD,MPI_ERRORS_ARE_FATAL);CHKERRQ(ierr); 358 ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr); 359 } 360 flg1 = PETSC_FALSE; 361 ierr = PetscOptionsGetBool(PETSC_NULL,"-on_error_mpiabort",&flg1,PETSC_NULL);CHKERRQ(ierr); 362 if (flg1) { ierr = PetscPushErrorHandler(PetscMPIAbortErrorHandler,0);CHKERRQ(ierr);} 363 flg1 = PETSC_FALSE; 364 ierr = PetscOptionsGetBool(PETSC_NULL,"-mpi_return_on_error",&flg1,PETSC_NULL);CHKERRQ(ierr); 365 if (flg1) { 366 ierr = MPI_Errhandler_set(comm,MPI_ERRORS_RETURN);CHKERRQ(ierr); 367 } 368 flg1 = PETSC_FALSE; 369 ierr = PetscOptionsGetBool(PETSC_NULL,"-no_signal_handler",&flg1,PETSC_NULL);CHKERRQ(ierr); 370 if (!flg1) {ierr = PetscPushSignalHandler(PetscDefaultSignalHandler,(void*)0);CHKERRQ(ierr);} 371 flg1 = PETSC_FALSE; 372 ierr = PetscOptionsGetBool(PETSC_NULL,"-fp_trap",&flg1,PETSC_NULL);CHKERRQ(ierr); 373 if (flg1) {ierr = PetscSetFPTrap(PETSC_FP_TRAP_ON);CHKERRQ(ierr);} 374 375 /* 376 Setup debugger information 377 */ 378 ierr = PetscSetDefaultDebugger();CHKERRQ(ierr); 379 ierr = PetscOptionsGetString(PETSC_NULL,"-on_error_attach_debugger",string,64,&flg1);CHKERRQ(ierr); 380 if (flg1) { 381 MPI_Errhandler err_handler; 382 383 ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr); 384 ierr = MPI_Errhandler_create((MPI_Handler_function*)Petsc_MPI_DebuggerOnError,&err_handler);CHKERRQ(ierr); 385 ierr = MPI_Errhandler_set(comm,err_handler);CHKERRQ(ierr); 386 ierr = PetscPushErrorHandler(PetscAttachDebuggerErrorHandler,0);CHKERRQ(ierr); 387 } 388 ierr = PetscOptionsGetString(PETSC_NULL,"-debug_terminal",string,64,&flg1);CHKERRQ(ierr); 389 if (flg1) { ierr = PetscSetDebugTerminal(string);CHKERRQ(ierr); } 390 ierr = PetscOptionsGetString(PETSC_NULL,"-start_in_debugger",string,64,&flg1);CHKERRQ(ierr); 391 ierr = PetscOptionsGetString(PETSC_NULL,"-stop_for_debugger",string,64,&flg2);CHKERRQ(ierr); 392 if (flg1 || flg2) { 393 PetscMPIInt size; 394 PetscInt lsize,*nodes; 395 MPI_Errhandler err_handler; 396 /* 397 we have to make sure that all processors have opened 398 connections to all other processors, otherwise once the 399 debugger has stated it is likely to receive a SIGUSR1 400 and kill the program. 401 */ 402 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 403 if (size > 2) { 404 PetscMPIInt dummy = 0; 405 MPI_Status status; 406 for (i=0; i<size; i++) { 407 if (rank != i) { 408 ierr = MPI_Send(&dummy,1,MPI_INT,i,109,PETSC_COMM_WORLD);CHKERRQ(ierr); 409 } 410 } 411 for (i=0; i<size; i++) { 412 if (rank != i) { 413 ierr = MPI_Recv(&dummy,1,MPI_INT,i,109,PETSC_COMM_WORLD,&status);CHKERRQ(ierr); 414 } 415 } 416 } 417 /* check if this processor node should be in debugger */ 418 ierr = PetscMalloc(size*sizeof(PetscInt),&nodes);CHKERRQ(ierr); 419 lsize = size; 420 ierr = PetscOptionsGetIntArray(PETSC_NULL,"-debugger_nodes",nodes,&lsize,&flag);CHKERRQ(ierr); 421 if (flag) { 422 for (i=0; i<lsize; i++) { 423 if (nodes[i] == rank) { flag = PETSC_FALSE; break; } 424 } 425 } 426 if (!flag) { 427 ierr = PetscSetDebuggerFromString(string);CHKERRQ(ierr); 428 ierr = PetscPushErrorHandler(PetscAbortErrorHandler,0);CHKERRQ(ierr); 429 if (flg1) { 430 ierr = PetscAttachDebugger();CHKERRQ(ierr); 431 } else { 432 ierr = PetscStopForDebugger();CHKERRQ(ierr); 433 } 434 ierr = MPI_Errhandler_create((MPI_Handler_function*)Petsc_MPI_AbortOnError,&err_handler);CHKERRQ(ierr); 435 ierr = MPI_Errhandler_set(comm,err_handler);CHKERRQ(ierr); 436 } 437 ierr = PetscFree(nodes);CHKERRQ(ierr); 438 } 439 440 ierr = PetscOptionsGetString(PETSC_NULL,"-on_error_emacs",emacsmachinename,128,&flg1);CHKERRQ(ierr); 441 if (flg1 && !rank) {ierr = PetscPushErrorHandler(PetscEmacsClientErrorHandler,emacsmachinename);CHKERRQ(ierr);} 442 443 #if defined(PETSC_USE_SERVER) 444 ierr = PetscOptionsHasName(PETSC_NULL,"-server", &flgz);CHKERRQ(ierr); 445 if (flgz){ 446 PetscInt port = PETSC_DECIDE; 447 ierr = PetscOptionsGetInt(PETSC_NULL,"-server",&port,PETSC_NULL);CHKERRQ(ierr); 448 ierr = PetscWebServe(PETSC_COMM_WORLD,(int)port);CHKERRQ(ierr); 449 } 450 #endif 451 452 /* 453 Setup profiling and logging 454 */ 455 #if defined (PETSC_USE_INFO) 456 { 457 char logname[PETSC_MAX_PATH_LEN]; logname[0] = 0; 458 ierr = PetscOptionsGetString(PETSC_NULL,"-info",logname,250,&flg1);CHKERRQ(ierr); 459 if (flg1 && logname[0]) { 460 ierr = PetscInfoAllow(PETSC_TRUE,logname);CHKERRQ(ierr); 461 } else if (flg1) { 462 ierr = PetscInfoAllow(PETSC_TRUE,PETSC_NULL);CHKERRQ(ierr); 463 } 464 } 465 #endif 466 #if defined(PETSC_USE_LOG) 467 mname[0] = 0; 468 ierr = PetscOptionsGetString(PETSC_NULL,"-history",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr); 469 if (flg1) { 470 if (mname[0]) { 471 ierr = PetscOpenHistoryFile(mname,&petsc_history);CHKERRQ(ierr); 472 } else { 473 ierr = PetscOpenHistoryFile(0,&petsc_history);CHKERRQ(ierr); 474 } 475 } 476 #if defined(PETSC_HAVE_MPE) 477 flg1 = PETSC_FALSE; 478 ierr = PetscOptionsHasName(PETSC_NULL,"-log_mpe",&flg1);CHKERRQ(ierr); 479 if (flg1) PetscLogMPEBegin(); 480 #endif 481 flg1 = PETSC_FALSE; 482 flg2 = PETSC_FALSE; 483 flg3 = PETSC_FALSE; 484 ierr = PetscOptionsGetBool(PETSC_NULL,"-log_all",&flg1,PETSC_NULL);CHKERRQ(ierr); 485 ierr = PetscOptionsGetBool(PETSC_NULL,"-log",&flg2,PETSC_NULL);CHKERRQ(ierr); 486 ierr = PetscOptionsHasName(PETSC_NULL,"-log_summary",&flg3);CHKERRQ(ierr); 487 ierr = PetscOptionsHasName(PETSC_NULL,"-log_summary_python",&flg4);CHKERRQ(ierr); 488 if (flg1) { ierr = PetscLogAllBegin();CHKERRQ(ierr); } 489 else if (flg2 || flg3 || flg4) { ierr = PetscLogBegin();CHKERRQ(ierr);} 490 491 ierr = PetscOptionsGetString(PETSC_NULL,"-log_trace",mname,250,&flg1);CHKERRQ(ierr); 492 if (flg1) { 493 char name[PETSC_MAX_PATH_LEN],fname[PETSC_MAX_PATH_LEN]; 494 FILE *file; 495 if (mname[0]) { 496 sprintf(name,"%s.%d",mname,rank); 497 ierr = PetscFixFilename(name,fname);CHKERRQ(ierr); 498 file = fopen(fname,"w"); 499 if (!file) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_FILE_OPEN,"Unable to open trace file: %s",fname); 500 } else { 501 file = PETSC_STDOUT; 502 } 503 ierr = PetscLogTraceBegin(file);CHKERRQ(ierr); 504 } 505 #endif 506 507 /* 508 Setup building of stack frames for all function calls 509 */ 510 #if defined(PETSC_USE_DEBUG) 511 PetscThreadLocalRegister(petscstack); /* Creates petscstack_key if needed */ 512 ierr = PetscStackCreate();CHKERRQ(ierr); 513 #endif 514 515 ierr = PetscOptionsGetBool(PETSC_NULL,"-options_gui",&PetscOptionsPublish,PETSC_NULL);CHKERRQ(ierr); 516 517 #if defined(PETSC_HAVE_CUDA) 518 ierr = PetscOptionsHasName(PETSC_NULL,"-cuda_show_devices",&flg1);CHKERRQ(ierr); 519 if (flg1) { 520 struct cudaDeviceProp prop; 521 int devCount; 522 int device; 523 524 ierr = cudaGetDeviceCount(&devCount);CHKERRQ(ierr); 525 for(device = 0; device < devCount; ++device) { 526 ierr = cudaGetDeviceProperties(&prop, device);CHKERRQ(ierr); 527 ierr = PetscPrintf(PETSC_COMM_WORLD, "CUDA device %d: %s\n", device, prop.name);CHKERRQ(ierr); 528 } 529 } 530 { 531 int size; 532 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 533 if (size>1) { 534 int devCount, device, rank; 535 ierr = cudaGetDeviceCount(&devCount);CHKERRQ(ierr); 536 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 537 device = rank % devCount; 538 ierr = cudaSetDevice(device);CHKERRQ(ierr); 539 } 540 else { 541 int device; 542 /* the code below works for serial GPU simulations */ 543 ierr = PetscOptionsGetInt(PETSC_NULL,"-cuda_set_device", &device, &flg1);CHKERRQ(ierr); 544 if (flg1) { 545 ierr = cudaSetDevice(device);CHKERRQ(ierr); 546 } 547 } 548 } 549 #endif 550 551 552 /* 553 Print basic help message 554 */ 555 ierr = PetscOptionsHasName(PETSC_NULL,"-help",&flg1);CHKERRQ(ierr); 556 if (flg1) { 557 ierr = (*PetscHelpPrintf)(comm,"Options for all PETSc programs:\n");CHKERRQ(ierr); 558 ierr = (*PetscHelpPrintf)(comm," -help: prints help method for each option\n");CHKERRQ(ierr); 559 ierr = (*PetscHelpPrintf)(comm," -on_error_abort: cause an abort when an error is detected. Useful \n ");CHKERRQ(ierr); 560 ierr = (*PetscHelpPrintf)(comm," only when run in the debugger\n");CHKERRQ(ierr); 561 ierr = (*PetscHelpPrintf)(comm," -on_error_attach_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr); 562 ierr = (*PetscHelpPrintf)(comm," start the debugger in new xterm\n");CHKERRQ(ierr); 563 ierr = (*PetscHelpPrintf)(comm," unless noxterm is given\n");CHKERRQ(ierr); 564 ierr = (*PetscHelpPrintf)(comm," -start_in_debugger [gdb,dbx,xxgdb,ups,noxterm]\n");CHKERRQ(ierr); 565 ierr = (*PetscHelpPrintf)(comm," start all processes in the debugger\n");CHKERRQ(ierr); 566 ierr = (*PetscHelpPrintf)(comm," -on_error_emacs <machinename>\n");CHKERRQ(ierr); 567 ierr = (*PetscHelpPrintf)(comm," emacs jumps to error file\n");CHKERRQ(ierr); 568 ierr = (*PetscHelpPrintf)(comm," -debugger_nodes [n1,n2,..] Nodes to start in debugger\n");CHKERRQ(ierr); 569 ierr = (*PetscHelpPrintf)(comm," -debugger_pause [m] : delay (in seconds) to attach debugger\n");CHKERRQ(ierr); 570 ierr = (*PetscHelpPrintf)(comm," -stop_for_debugger : prints message on how to attach debugger manually\n");CHKERRQ(ierr); 571 ierr = (*PetscHelpPrintf)(comm," waits the delay for you to attach\n");CHKERRQ(ierr); 572 ierr = (*PetscHelpPrintf)(comm," -display display: Location where graphics and debuggers are displayed\n");CHKERRQ(ierr); 573 ierr = (*PetscHelpPrintf)(comm," -no_signal_handler: do not trap error signals\n");CHKERRQ(ierr); 574 ierr = (*PetscHelpPrintf)(comm," -mpi_return_on_error: MPI returns error code, rather than abort on internal error\n");CHKERRQ(ierr); 575 ierr = (*PetscHelpPrintf)(comm," -fp_trap: stop on floating point exceptions\n");CHKERRQ(ierr); 576 ierr = (*PetscHelpPrintf)(comm," note on IBM RS6000 this slows run greatly\n");CHKERRQ(ierr); 577 ierr = (*PetscHelpPrintf)(comm," -malloc_dump <optional filename>: dump list of unfreed memory at conclusion\n");CHKERRQ(ierr); 578 ierr = (*PetscHelpPrintf)(comm," -malloc: use our error checking malloc\n");CHKERRQ(ierr); 579 ierr = (*PetscHelpPrintf)(comm," -malloc no: don't use error checking malloc\n");CHKERRQ(ierr); 580 ierr = (*PetscHelpPrintf)(comm," -malloc_info: prints total memory usage\n");CHKERRQ(ierr); 581 ierr = (*PetscHelpPrintf)(comm," -malloc_log: keeps log of all memory allocations\n");CHKERRQ(ierr); 582 ierr = (*PetscHelpPrintf)(comm," -malloc_debug: enables extended checking for memory corruption\n");CHKERRQ(ierr); 583 ierr = (*PetscHelpPrintf)(comm," -options_table: dump list of options inputted\n");CHKERRQ(ierr); 584 ierr = (*PetscHelpPrintf)(comm," -options_left: dump list of unused options\n");CHKERRQ(ierr); 585 ierr = (*PetscHelpPrintf)(comm," -options_left no: don't dump list of unused options\n");CHKERRQ(ierr); 586 ierr = (*PetscHelpPrintf)(comm," -tmp tmpdir: alternative /tmp directory\n");CHKERRQ(ierr); 587 ierr = (*PetscHelpPrintf)(comm," -shared_tmp: tmp directory is shared by all processors\n");CHKERRQ(ierr); 588 ierr = (*PetscHelpPrintf)(comm," -not_shared_tmp: each processor has separate tmp directory\n");CHKERRQ(ierr); 589 ierr = (*PetscHelpPrintf)(comm," -memory_info: print memory usage at end of run\n");CHKERRQ(ierr); 590 ierr = (*PetscHelpPrintf)(comm," -server <port>: Run PETSc webserver (default port is 8080) see PetscWebServe()\n");CHKERRQ(ierr); 591 #if defined(PETSC_USE_LOG) 592 ierr = (*PetscHelpPrintf)(comm," -get_total_flops: total flops over all processors\n");CHKERRQ(ierr); 593 ierr = (*PetscHelpPrintf)(comm," -log[_all _summary _summary_python]: logging objects and events\n");CHKERRQ(ierr); 594 ierr = (*PetscHelpPrintf)(comm," -log_trace [filename]: prints trace of all PETSc calls\n");CHKERRQ(ierr); 595 #if defined(PETSC_HAVE_MPE) 596 ierr = (*PetscHelpPrintf)(comm," -log_mpe: Also create logfile viewable through upshot\n");CHKERRQ(ierr); 597 #endif 598 ierr = (*PetscHelpPrintf)(comm," -info <optional filename>: print informative messages about the calculations\n");CHKERRQ(ierr); 599 #endif 600 ierr = (*PetscHelpPrintf)(comm," -v: prints PETSc version number and release date\n");CHKERRQ(ierr); 601 ierr = (*PetscHelpPrintf)(comm," -options_file <file>: reads options from file\n");CHKERRQ(ierr); 602 ierr = (*PetscHelpPrintf)(comm," -petsc_sleep n: sleeps n seconds before running program\n");CHKERRQ(ierr); 603 ierr = (*PetscHelpPrintf)(comm,"-----------------------------------------------\n");CHKERRQ(ierr); 604 } 605 606 ierr = PetscOptionsGetReal(PETSC_NULL,"-petsc_sleep",&si,&flg1);CHKERRQ(ierr); 607 if (flg1) { 608 ierr = PetscSleep(si);CHKERRQ(ierr); 609 } 610 611 ierr = PetscOptionsGetString(PETSC_NULL,"-info_exclude",mname,PETSC_MAX_PATH_LEN,&flg1);CHKERRQ(ierr); 612 ierr = PetscStrstr(mname,"null",&f);CHKERRQ(ierr); 613 if (f) { 614 ierr = PetscInfoDeactivateClass(0);CHKERRQ(ierr); 615 } 616 617 #if defined(PETSC_HAVE_CUSP) 618 ierr = PetscOptionsHasName(PETSC_NULL,"-log_summary",&flg3);CHKERRQ(ierr); 619 if (flg3) flg1 = PETSC_TRUE; 620 else flg1 = PETSC_FALSE; 621 ierr = PetscOptionsGetBool(PETSC_NULL,"-cusp_synchronize",&flg1,PETSC_NULL);CHKERRQ(ierr); 622 if (flg1) PetscCUSPSynchronize = PETSC_TRUE; 623 #endif 624 625 PetscFunctionReturn(0); 626 } 627 628