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