1 2 /* 3 Code that allows one to set the error handlers 4 */ 5 #include <petsc/private/petscimpl.h> /*I "petscsys.h" I*/ 6 #include <petscviewer.h> 7 8 /* A table of Petsc source files containing calls to PETSCABORT. We assume this table will 9 stay stable for a while. When things changed, we just need to add new files to the table. 10 */ 11 static const char* PetscAbortSourceFiles[] = { 12 "Souce code of main", /* 0 */ 13 "Not Found", /* 1, not found in petsc, but may be in users' code if they called PETSCABORT. */ 14 "sys/error/adebug.c", 15 "src/sys/error/errstop.c", 16 "sys/error/fp.c", 17 "sys/error/signal.c", /* 5 */ 18 "sys/ftn-custom/zutils.c", 19 "sys/logging/utils/stagelog.c", 20 "sys/mpiuni/mpitime.c", 21 "sys/objects/init.c", 22 "sys/objects/pinit.c", /* 10 */ 23 "vec/vec/interface/dlregisvec.c", 24 "vec/vec/utils/comb.c" 25 }; 26 27 /* Find index of the soure file where a PETSCABORT was called. */ 28 PetscErrorCode PetscAbortFindSourceFile_Private(const char* filepath, PetscInt *idx) 29 { 30 PetscErrorCode ierr; 31 PetscInt i,n = sizeof(PetscAbortSourceFiles)/sizeof(PetscAbortSourceFiles[0]); 32 PetscBool match; 33 char subpath[256]; 34 35 PetscFunctionBegin; 36 PetscValidIntPointer(idx,2); 37 *idx = 1; 38 for (i=2; i<n; i++) { 39 ierr = PetscFixFilename(PetscAbortSourceFiles[i],subpath);CHKERRQ(ierr); 40 ierr = PetscStrendswith(filepath,subpath,&match);CHKERRQ(ierr); 41 if (match) {*idx = i; break;} 42 } 43 PetscFunctionReturn(0); 44 } 45 46 typedef struct _EH *EH; 47 struct _EH { 48 PetscErrorCode (*handler)(MPI_Comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*); 49 void *ctx; 50 EH previous; 51 }; 52 53 static EH eh = NULL; 54 55 /*@C 56 PetscEmacsClientErrorHandler - Error handler that uses the emacsclient program to 57 load the file where the error occured. Then calls the "previous" error handler. 58 59 Not Collective 60 61 Input Parameters: 62 + comm - communicator over which error occured 63 . line - the line number of the error (indicated by __LINE__) 64 . file - the file in which the error was detected (indicated by __FILE__) 65 . mess - an error text string, usually just printed to the screen 66 . n - the generic error number 67 . p - specific error number 68 - ctx - error handler context 69 70 Options Database Key: 71 . -on_error_emacs <machinename> 72 73 Level: developer 74 75 Notes: 76 You must put (server-start) in your .emacs file for the emacsclient software to work 77 78 Most users need not directly employ this routine and the other error 79 handlers, but can instead use the simplified interface SETERRQ, which has 80 the calling sequence 81 $ SETERRQ(PETSC_COMM_SELF,number,p,mess) 82 83 Notes for experienced users: 84 Use PetscPushErrorHandler() to set the desired error handler. 85 86 Developer Note: 87 Since this is an error handler it cannot call CHKERRQ(); thus we just return if an error is detected. 88 89 90 .seealso: PetscError(), PetscPushErrorHandler(), PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), 91 PetscAbortErrorHandler(), PetscMPIAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscReturnErrorHandler() 92 @*/ 93 PetscErrorCode PetscEmacsClientErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 94 { 95 PetscErrorCode ierr; 96 char command[PETSC_MAX_PATH_LEN]; 97 const char *pdir; 98 FILE *fp; 99 100 PetscFunctionBegin; 101 ierr = PetscGetPetscDir(&pdir);if (ierr) PetscFunctionReturn(ierr); 102 sprintf(command,"cd %s; emacsclient --no-wait +%d %s\n",pdir,line,file); 103 #if defined(PETSC_HAVE_POPEN) 104 ierr = PetscPOpen(MPI_COMM_WORLD,(char*)ctx,command,"r",&fp);if (ierr) PetscFunctionReturn(ierr); 105 ierr = PetscPClose(MPI_COMM_WORLD,fp);if (ierr) PetscFunctionReturn(ierr); 106 #else 107 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP_SYS,"Cannot run external programs on this machine"); 108 #endif 109 ierr = PetscPopErrorHandler();if (ierr) PetscFunctionReturn(ierr); /* remove this handler from the stack of handlers */ 110 if (!eh) { 111 ierr = PetscTraceBackErrorHandler(comm,line,fun,file,n,p,mess,NULL);if (ierr) PetscFunctionReturn(ierr); 112 } else { 113 ierr = (*eh->handler)(comm,line,fun,file,n,p,mess,eh->ctx);if (ierr) PetscFunctionReturn(ierr); 114 } 115 PetscFunctionReturn(ierr); 116 } 117 118 /*@C 119 PetscPushErrorHandler - Sets a routine to be called on detection of errors. 120 121 Not Collective 122 123 Input Parameters: 124 + handler - error handler routine 125 - ctx - optional handler context that contains information needed by the handler (for 126 example file pointers for error messages etc.) 127 128 Calling sequence of handler: 129 $ int handler(MPI_Comm comm,int line,char *func,char *file,PetscErrorCode n,int p,char *mess,void *ctx); 130 131 + comm - communicator over which error occured 132 . line - the line number of the error (indicated by __LINE__) 133 . file - the file in which the error was detected (indicated by __FILE__) 134 . n - the generic error number (see list defined in include/petscerror.h) 135 . p - PETSC_ERROR_INITIAL if error just detected, otherwise PETSC_ERROR_REPEAT 136 . mess - an error text string, usually just printed to the screen 137 - ctx - the error handler context 138 139 Options Database Keys: 140 + -on_error_attach_debugger <noxterm,gdb or dbx> 141 - -on_error_abort 142 143 Level: intermediate 144 145 Notes: 146 The currently available PETSc error handlers include PetscTraceBackErrorHandler(), 147 PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), and PetscMPIAbortErrorHandler(), PetscReturnErrorHandler(). 148 149 Fortran Notes: 150 You can only push one error handler from Fortran before poping it. 151 152 .seealso: PetscPopErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscAbortErrorHandler(), PetscTraceBackErrorHandler(), PetscPushSignalHandler() 153 154 @*/ 155 PetscErrorCode PetscPushErrorHandler(PetscErrorCode (*handler)(MPI_Comm comm,int,const char*,const char*,PetscErrorCode,PetscErrorType,const char*,void*),void *ctx) 156 { 157 EH neweh; 158 PetscErrorCode ierr; 159 160 PetscFunctionBegin; 161 ierr = PetscNew(&neweh);CHKERRQ(ierr); 162 if (eh) neweh->previous = eh; 163 else neweh->previous = NULL; 164 neweh->handler = handler; 165 neweh->ctx = ctx; 166 eh = neweh; 167 PetscFunctionReturn(0); 168 } 169 170 /*@ 171 PetscPopErrorHandler - Removes the latest error handler that was 172 pushed with PetscPushErrorHandler(). 173 174 Not Collective 175 176 Level: intermediate 177 178 .seealso: PetscPushErrorHandler() 179 @*/ 180 PetscErrorCode PetscPopErrorHandler(void) 181 { 182 EH tmp; 183 PetscErrorCode ierr; 184 185 PetscFunctionBegin; 186 if (!eh) PetscFunctionReturn(0); 187 tmp = eh; 188 eh = eh->previous; 189 ierr = PetscFree(tmp);CHKERRQ(ierr); 190 PetscFunctionReturn(0); 191 } 192 193 /*@C 194 PetscReturnErrorHandler - Error handler that causes a return without printing an error message. 195 196 Not Collective 197 198 Input Parameters: 199 + comm - communicator over which error occurred 200 . line - the line number of the error (indicated by __LINE__) 201 . file - the file in which the error was detected (indicated by __FILE__) 202 . mess - an error text string, usually just printed to the screen 203 . n - the generic error number 204 . p - specific error number 205 - ctx - error handler context 206 207 Level: developer 208 209 Notes: 210 Most users need not directly employ this routine and the other error 211 handlers, but can instead use the simplified interface SETERRQ, which has 212 the calling sequence 213 $ SETERRQ(comm,number,mess) 214 215 PetscIgnoreErrorHandler() does the same thing as this function, but is deprecated, you should use this function. 216 217 Use PetscPushErrorHandler() to set the desired error handler. 218 219 .seealso: PetscPushErrorHandler(), PetscPopErrorHandler(), PetscError(), PetscAbortErrorHandler(), PetscMPIAbortErrorHandler(), PetscTraceBackErrorHandler(), 220 PetscAttachDebuggerErrorHandler(), PetscEmacsClientErrorHandler() 221 @*/ 222 PetscErrorCode PetscReturnErrorHandler(MPI_Comm comm,int line,const char *fun,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,void *ctx) 223 { 224 PetscFunctionBegin; 225 PetscFunctionReturn(n); 226 } 227 228 static char PetscErrorBaseMessage[1024]; 229 /* 230 The numerical values for these are defined in include/petscerror.h; any changes 231 there must also be made here 232 */ 233 static const char *PetscErrorStrings[] = { 234 /*55 */ "Out of memory", 235 "No support for this operation for this object type", 236 "No support for this operation on this system", 237 /*58 */ "Operation done in wrong order", 238 /*59 */ "Signal received", 239 /*60 */ "Nonconforming object sizes", 240 "Argument aliasing not permitted", 241 "Invalid argument", 242 /*63 */ "Argument out of range", 243 "Corrupt argument: https://www.mcs.anl.gov/petsc/documentation/faq.html#valgrind", 244 "Unable to open file", 245 "Read from file failed", 246 "Write to file failed", 247 "Invalid pointer", 248 /*69 */ "Arguments must have same type", 249 /*70 */ "Attempt to use a pointer that does not point to a valid accessible location", 250 /*71 */ "Zero pivot in LU factorization: https://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 251 /*72 */ "Floating point exception", 252 /*73 */ "Object is in wrong state", 253 "Corrupted Petsc object", 254 "Arguments are incompatible", 255 "Error in external library", 256 /*77 */ "Petsc has generated inconsistent data", 257 "Memory corruption: https://www.mcs.anl.gov/petsc/documentation/installation.html#valgrind", 258 "Unexpected data in file", 259 /*80 */ "Arguments must have same communicators", 260 /*81 */ "Zero pivot in Cholesky factorization: https://www.mcs.anl.gov/petsc/documentation/faq.html#zeropivot", 261 " ", 262 " ", 263 "Overflow in integer operation: https://www.mcs.anl.gov/petsc/documentation/faq.html#64-bit-indices", 264 /*85 */ "Null argument, when expecting valid pointer", 265 /*86 */ "Unknown type. Check for miss-spelling or missing package: https://www.mcs.anl.gov/petsc/documentation/installation.html#external", 266 /*87 */ "MPI library at runtime is not compatible with MPI used at compile time", 267 /*88 */ "Error in system call", 268 /*89 */ "Object Type not set: https://www.mcs.anl.gov/petsc/documentation/faq.html#objecttypenotset", 269 /*90 */ " ", 270 /* */ " ", 271 /*92 */ "See https://www.mcs.anl.gov/petsc/documentation/linearsolvertable.html for possible LU and Cholesky solvers", 272 /*93 */ "You cannot overwrite this option since that will conflict with other previously set options", 273 /*94 */ "Example/application run with number of MPI ranks it does not support", 274 /*95 */ "Missing or incorrect user input ", 275 /*96 */ "GPU resources unavailable ", 276 }; 277 278 /*@C 279 PetscErrorMessage - returns the text string associated with a PETSc error code. 280 281 Not Collective 282 283 Input Parameter: 284 . errnum - the error code 285 286 Output Parameter: 287 + text - the error message (NULL if not desired) 288 - specific - the specific error message that was set with SETERRxxx() or PetscError(). (NULL if not desired) 289 290 Level: developer 291 292 .seealso: PetscPushErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscError(), SETERRQ(), CHKERRQ() 293 PetscAbortErrorHandler(), PetscTraceBackErrorHandler() 294 @*/ 295 PetscErrorCode PetscErrorMessage(int errnum,const char *text[],char **specific) 296 { 297 PetscFunctionBegin; 298 if (text && errnum > PETSC_ERR_MIN_VALUE && errnum < PETSC_ERR_MAX_VALUE) *text = PetscErrorStrings[errnum-PETSC_ERR_MIN_VALUE-1]; 299 else if (text) *text = NULL; 300 301 if (specific) *specific = PetscErrorBaseMessage; 302 PetscFunctionReturn(0); 303 } 304 305 #if defined(PETSC_CLANGUAGE_CXX) 306 /* C++ exceptions are formally not allowed to propagate through extern "C" code. In practice, far too much software 307 * would be broken if implementations did not handle it it some common cases. However, keep in mind 308 * 309 * Rule 62. Don't allow exceptions to propagate across module boundaries 310 * 311 * in "C++ Coding Standards" by Sutter and Alexandrescu. (This accounts for part of the ongoing C++ binary interface 312 * instability.) Having PETSc raise errors as C++ exceptions was probably misguided and should eventually be removed. 313 * 314 * Here is the problem: You have a C++ function call a PETSc function, and you would like to maintain the error message 315 * and stack information from the PETSc error. You could make everyone write exactly this code in their C++, but that 316 * seems crazy to me. 317 */ 318 #include <sstream> 319 #include <stdexcept> 320 static void PetscCxxErrorThrow() { 321 const char *str; 322 if (eh && eh->ctx) { 323 std::ostringstream *msg; 324 msg = (std::ostringstream*) eh->ctx; 325 str = msg->str().c_str(); 326 } else str = "Error detected in C PETSc"; 327 328 throw std::runtime_error(str); 329 } 330 #endif 331 332 /*@C 333 PetscError - Routine that is called when an error has been detected, usually called through the macro SETERRQ(PETSC_COMM_SELF,). 334 335 Collective on comm 336 337 Input Parameters: 338 + comm - communicator over which error occurred. ALL ranks of this communicator MUST call this routine 339 . line - the line number of the error (indicated by __LINE__) 340 . func - the function name in which the error was detected 341 . file - the file in which the error was detected (indicated by __FILE__) 342 . n - the generic error number 343 . p - PETSC_ERROR_INITIAL indicates the error was initially detected, PETSC_ERROR_REPEAT indicates this is a traceback from a previously detected error 344 - mess - formatted message string - aka printf 345 346 Options Database: 347 + -error_output_stdout - output the error messages to stdout instead of the default stderr 348 - -error_output_none - do not output the error messages 349 350 Level: intermediate 351 352 Notes: 353 PETSc error handling is done with error return codes. A non-zero return indicates an error was detected. Errors are generally not something that the code 354 can recover from. Note that numerical errors (potential divide by zero, for example) are not managed by the error return codes; they are managed via, for example, 355 KSPGetConvergedReason() that indicates if the solve was successful or not. The option -ksp_error_if_not_converged, for example, turns numerical failures into 356 hard errors managed via PetscError(). 357 358 PETSc provides a rich supply of error handlers, see the list below, and users can also provide their own error handlers. 359 360 Most users need not directly use this routine and the error handlers, but 361 can instead use the simplified interface SETERRQ, which has the calling 362 sequence 363 $ SETERRQ(comm,n,mess) 364 365 Fortran Note: 366 This routine is used differently from Fortran 367 $ PetscError(MPI_Comm comm,PetscErrorCode n,PetscErrorType p,char *message) 368 369 Set the error handler with PetscPushErrorHandler(). 370 371 Developer Note: Since this is called after an error condition it should not be calling any error handlers (currently it ignores any error codes) 372 BUT this routine does call regular PETSc functions that may call error handlers, this is problematic and could be fixed by never calling other PETSc routines 373 but this annoying. 374 375 .seealso: PetscErrorCode, PetscPushErrorHandler(), PetscPopErrorHandler(), PetscTraceBackErrorHandler(), PetscAbortErrorHandler(), PetscMPIAbortErrorHandler(), 376 PetscReturnErrorHandler(), PetscAttachDebuggerErrorHandler(), PetscEmacsClientErrorHandler(), 377 SETERRQ(), CHKERRQ(), CHKMEMQ, SETERRQ1(), SETERRQ2(), PetscErrorMessage(), PETSCABORT() 378 @*/ 379 PetscErrorCode PetscError(MPI_Comm comm,int line,const char *func,const char *file,PetscErrorCode n,PetscErrorType p,const char *mess,...) 380 { 381 va_list Argp; 382 size_t fullLength; 383 char buf[2048],*lbuf = NULL; 384 PetscBool ismain; 385 PetscErrorCode ierr; 386 387 PetscFunctionBegin; 388 if (!func) func = "User provided function"; 389 if (!file) file = "User file"; 390 if (comm == MPI_COMM_NULL) comm = PETSC_COMM_SELF; 391 392 /* Compose the message evaluating the print format */ 393 if (mess) { 394 va_start(Argp,mess); 395 PetscVSNPrintf(buf,2048,mess,&fullLength,Argp); 396 va_end(Argp); 397 lbuf = buf; 398 if (p == PETSC_ERROR_INITIAL) PetscStrncpy(PetscErrorBaseMessage,lbuf,1023); 399 } 400 401 if (!eh) ierr = PetscTraceBackErrorHandler(comm,line,func,file,n,p,lbuf,NULL); 402 else ierr = (*eh->handler)(comm,line,func,file,n,p,lbuf,eh->ctx); 403 404 /* 405 If this is called from the main() routine we call MPI_Abort() instead of 406 return to allow the parallel program to be properly shutdown. 407 408 Does not call PETSCABORT() since that would provide the wrong source file and line number information 409 */ 410 PetscStrncmp(func,"main",4,&ismain); 411 if (ismain) { 412 PetscMPIInt errcode; 413 errcode = (PetscMPIInt)(0 + line*1000 + ierr); 414 if (petscwaitonerror) {PetscSleep(1000);} 415 MPI_Abort(comm,errcode); 416 } 417 418 #if defined(PETSC_CLANGUAGE_CXX) 419 if (p == PETSC_ERROR_IN_CXX) { 420 PetscCxxErrorThrow(); 421 } 422 #endif 423 PetscFunctionReturn(ierr); 424 } 425 426 /* -------------------------------------------------------------------------*/ 427 428 /*@C 429 PetscIntView - Prints an array of integers; useful for debugging. 430 431 Collective on PetscViewer 432 433 Input Parameters: 434 + N - number of integers in array 435 . idx - array of integers 436 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 437 438 Level: intermediate 439 440 Developer Notes: 441 idx cannot be const because may be passed to binary viewer where byte swapping is done 442 443 .seealso: PetscRealView() 444 @*/ 445 PetscErrorCode PetscIntView(PetscInt N,const PetscInt idx[],PetscViewer viewer) 446 { 447 PetscErrorCode ierr; 448 PetscMPIInt rank,size; 449 PetscInt j,i,n = N/20,p = N % 20; 450 PetscBool iascii,isbinary; 451 MPI_Comm comm; 452 453 PetscFunctionBegin; 454 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 455 PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 456 if (N) PetscValidIntPointer(idx,2); 457 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 458 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 459 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 460 461 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 462 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 463 if (iascii) { 464 ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 465 for (i=0; i<n; i++) { 466 if (size > 1) { 467 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D:", rank, 20*i);CHKERRQ(ierr); 468 } else { 469 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*i);CHKERRQ(ierr); 470 } 471 for (j=0; j<20; j++) { 472 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[i*20+j]);CHKERRQ(ierr); 473 } 474 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 475 } 476 if (p) { 477 if (size > 1) { 478 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %D:",rank ,20*n);CHKERRQ(ierr); 479 } else { 480 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D:",20*n);CHKERRQ(ierr); 481 } 482 for (i=0; i<p; i++) { ierr = PetscViewerASCIISynchronizedPrintf(viewer," %D",idx[20*n+i]);CHKERRQ(ierr);} 483 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 484 } 485 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 486 ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 487 } else if (isbinary) { 488 PetscMPIInt *sizes,Ntotal,*displs,NN; 489 PetscInt *array; 490 491 ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 492 493 if (size > 1) { 494 if (rank) { 495 ierr = MPI_Gather(&NN,1,MPI_INT,NULL,0,MPI_INT,0,comm);CHKERRQ(ierr); 496 ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,NULL,NULL,NULL,MPIU_INT,0,comm);CHKERRQ(ierr); 497 } else { 498 ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 499 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 500 Ntotal = sizes[0]; 501 ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 502 displs[0] = 0; 503 for (i=1; i<size; i++) { 504 Ntotal += sizes[i]; 505 displs[i] = displs[i-1] + sizes[i-1]; 506 } 507 ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 508 ierr = MPI_Gatherv((void*)idx,NN,MPIU_INT,array,sizes,displs,MPIU_INT,0,comm);CHKERRQ(ierr); 509 ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_INT);CHKERRQ(ierr); 510 ierr = PetscFree(sizes);CHKERRQ(ierr); 511 ierr = PetscFree(displs);CHKERRQ(ierr); 512 ierr = PetscFree(array);CHKERRQ(ierr); 513 } 514 } else { 515 ierr = PetscViewerBinaryWrite(viewer,idx,N,PETSC_INT);CHKERRQ(ierr); 516 } 517 } else { 518 const char *tname; 519 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 520 SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 521 } 522 PetscFunctionReturn(0); 523 } 524 525 /*@C 526 PetscRealView - Prints an array of doubles; useful for debugging. 527 528 Collective on PetscViewer 529 530 Input Parameters: 531 + N - number of PetscReal in array 532 . idx - array of PetscReal 533 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 534 535 Level: intermediate 536 537 Developer Notes: 538 idx cannot be const because may be passed to binary viewer where byte swapping is done 539 540 .seealso: PetscIntView() 541 @*/ 542 PetscErrorCode PetscRealView(PetscInt N,const PetscReal idx[],PetscViewer viewer) 543 { 544 PetscErrorCode ierr; 545 PetscMPIInt rank,size; 546 PetscInt j,i,n = N/5,p = N % 5; 547 PetscBool iascii,isbinary; 548 MPI_Comm comm; 549 550 PetscFunctionBegin; 551 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 552 PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,3); 553 PetscValidScalarPointer(idx,2); 554 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 555 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 556 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 557 558 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 559 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 560 if (iascii) { 561 PetscInt tab; 562 563 ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 564 ierr = PetscViewerASCIIGetTab(viewer, &tab);CHKERRQ(ierr); 565 for (i=0; i<n; i++) { 566 ierr = PetscViewerASCIISetTab(viewer, tab);CHKERRQ(ierr); 567 if (size > 1) { 568 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,(int)5*i);CHKERRQ(ierr); 569 } else { 570 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*i);CHKERRQ(ierr); 571 } 572 ierr = PetscViewerASCIISetTab(viewer, 0);CHKERRQ(ierr); 573 for (j=0; j<5; j++) { 574 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*5+j]);CHKERRQ(ierr); 575 } 576 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 577 } 578 if (p) { 579 ierr = PetscViewerASCIISetTab(viewer, tab);CHKERRQ(ierr); 580 if (size > 1) { 581 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,(int)5*n);CHKERRQ(ierr); 582 } else { 583 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",(int)5*n);CHKERRQ(ierr); 584 } 585 ierr = PetscViewerASCIISetTab(viewer, 0);CHKERRQ(ierr); 586 for (i=0; i<p; i++) { PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[5*n+i]);CHKERRQ(ierr);} 587 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 588 } 589 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 590 ierr = PetscViewerASCIISetTab(viewer, tab);CHKERRQ(ierr); 591 ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 592 } else if (isbinary) { 593 PetscMPIInt *sizes,*displs, Ntotal,NN; 594 PetscReal *array; 595 596 ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 597 598 if (size > 1) { 599 if (rank) { 600 ierr = MPI_Gather(&NN,1,MPI_INT,NULL,0,MPI_INT,0,comm);CHKERRQ(ierr); 601 ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,NULL,NULL,NULL,MPIU_REAL,0,comm);CHKERRQ(ierr); 602 } else { 603 ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 604 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 605 Ntotal = sizes[0]; 606 ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 607 displs[0] = 0; 608 for (i=1; i<size; i++) { 609 Ntotal += sizes[i]; 610 displs[i] = displs[i-1] + sizes[i-1]; 611 } 612 ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 613 ierr = MPI_Gatherv((PetscReal*)idx,NN,MPIU_REAL,array,sizes,displs,MPIU_REAL,0,comm);CHKERRQ(ierr); 614 ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_REAL);CHKERRQ(ierr); 615 ierr = PetscFree(sizes);CHKERRQ(ierr); 616 ierr = PetscFree(displs);CHKERRQ(ierr); 617 ierr = PetscFree(array);CHKERRQ(ierr); 618 } 619 } else { 620 ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_REAL);CHKERRQ(ierr); 621 } 622 } else { 623 const char *tname; 624 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 625 SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 626 } 627 PetscFunctionReturn(0); 628 } 629 630 /*@C 631 PetscScalarView - Prints an array of scalars; useful for debugging. 632 633 Collective on PetscViewer 634 635 Input Parameters: 636 + N - number of scalars in array 637 . idx - array of scalars 638 - viewer - location to print array, PETSC_VIEWER_STDOUT_WORLD, PETSC_VIEWER_STDOUT_SELF or 0 639 640 Level: intermediate 641 642 Developer Notes: 643 idx cannot be const because may be passed to binary viewer where byte swapping is done 644 645 .seealso: PetscIntView(), PetscRealView() 646 @*/ 647 PetscErrorCode PetscScalarView(PetscInt N,const PetscScalar idx[],PetscViewer viewer) 648 { 649 PetscErrorCode ierr; 650 PetscMPIInt rank,size; 651 PetscInt j,i,n = N/3,p = N % 3; 652 PetscBool iascii,isbinary; 653 MPI_Comm comm; 654 655 PetscFunctionBegin; 656 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 657 PetscValidHeader(viewer,3); 658 if (N) PetscValidScalarPointer(idx,2); 659 ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 660 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 661 ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 662 663 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 664 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 665 if (iascii) { 666 ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 667 for (i=0; i<n; i++) { 668 if (size > 1) { 669 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,3*i);CHKERRQ(ierr); 670 } else { 671 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*i);CHKERRQ(ierr); 672 } 673 for (j=0; j<3; j++) { 674 #if defined(PETSC_USE_COMPLEX) 675 ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[i*3+j]),(double)PetscImaginaryPart(idx[i*3+j]));CHKERRQ(ierr); 676 #else 677 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[i*3+j]);CHKERRQ(ierr); 678 #endif 679 } 680 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 681 } 682 if (p) { 683 if (size > 1) { 684 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] %2d:",rank ,3*n);CHKERRQ(ierr); 685 } else { 686 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%2d:",3*n);CHKERRQ(ierr); 687 } 688 for (i=0; i<p; i++) { 689 #if defined(PETSC_USE_COMPLEX) 690 ierr = PetscViewerASCIISynchronizedPrintf(viewer," (%12.4e,%12.4e)", (double)PetscRealPart(idx[n*3+i]),(double)PetscImaginaryPart(idx[n*3+i]));CHKERRQ(ierr); 691 #else 692 ierr = PetscViewerASCIISynchronizedPrintf(viewer," %12.4e",(double)idx[3*n+i]);CHKERRQ(ierr); 693 #endif 694 } 695 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 696 } 697 ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 698 ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 699 } else if (isbinary) { 700 PetscMPIInt *sizes,Ntotal,*displs,NN; 701 PetscScalar *array; 702 703 ierr = PetscMPIIntCast(N,&NN);CHKERRQ(ierr); 704 705 if (size > 1) { 706 if (rank) { 707 ierr = MPI_Gather(&NN,1,MPI_INT,NULL,0,MPI_INT,0,comm);CHKERRQ(ierr); 708 ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,NULL,NULL,NULL,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 709 } else { 710 ierr = PetscMalloc1(size,&sizes);CHKERRQ(ierr); 711 ierr = MPI_Gather(&NN,1,MPI_INT,sizes,1,MPI_INT,0,comm);CHKERRQ(ierr); 712 Ntotal = sizes[0]; 713 ierr = PetscMalloc1(size,&displs);CHKERRQ(ierr); 714 displs[0] = 0; 715 for (i=1; i<size; i++) { 716 Ntotal += sizes[i]; 717 displs[i] = displs[i-1] + sizes[i-1]; 718 } 719 ierr = PetscMalloc1(Ntotal,&array);CHKERRQ(ierr); 720 ierr = MPI_Gatherv((void*)idx,NN,MPIU_SCALAR,array,sizes,displs,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 721 ierr = PetscViewerBinaryWrite(viewer,array,Ntotal,PETSC_SCALAR);CHKERRQ(ierr); 722 ierr = PetscFree(sizes);CHKERRQ(ierr); 723 ierr = PetscFree(displs);CHKERRQ(ierr); 724 ierr = PetscFree(array);CHKERRQ(ierr); 725 } 726 } else { 727 ierr = PetscViewerBinaryWrite(viewer,(void*) idx,N,PETSC_SCALAR);CHKERRQ(ierr); 728 } 729 } else { 730 const char *tname; 731 ierr = PetscObjectGetName((PetscObject)viewer,&tname);CHKERRQ(ierr); 732 SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot handle that PetscViewer of type %s",tname); 733 } 734 PetscFunctionReturn(0); 735 } 736 737 #if defined(PETSC_HAVE_CUDA) 738 #include <petsccublas.h> 739 PETSC_EXTERN const char* PetscCUBLASGetErrorName(cublasStatus_t status) 740 { 741 switch(status) { 742 #if (CUDART_VERSION >= 8000) /* At least CUDA 8.0 of Sep. 2016 had these */ 743 case CUBLAS_STATUS_SUCCESS: return "CUBLAS_STATUS_SUCCESS"; 744 case CUBLAS_STATUS_NOT_INITIALIZED: return "CUBLAS_STATUS_NOT_INITIALIZED"; 745 case CUBLAS_STATUS_ALLOC_FAILED: return "CUBLAS_STATUS_ALLOC_FAILED"; 746 case CUBLAS_STATUS_INVALID_VALUE: return "CUBLAS_STATUS_INVALID_VALUE"; 747 case CUBLAS_STATUS_ARCH_MISMATCH: return "CUBLAS_STATUS_ARCH_MISMATCH"; 748 case CUBLAS_STATUS_MAPPING_ERROR: return "CUBLAS_STATUS_MAPPING_ERROR"; 749 case CUBLAS_STATUS_EXECUTION_FAILED: return "CUBLAS_STATUS_EXECUTION_FAILED"; 750 case CUBLAS_STATUS_INTERNAL_ERROR: return "CUBLAS_STATUS_INTERNAL_ERROR"; 751 case CUBLAS_STATUS_NOT_SUPPORTED: return "CUBLAS_STATUS_NOT_SUPPORTED"; 752 case CUBLAS_STATUS_LICENSE_ERROR: return "CUBLAS_STATUS_LICENSE_ERROR"; 753 #endif 754 default: return "unknown error"; 755 } 756 } 757 #endif 758