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