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