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