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