Use TF logging for TensorRT informations
| 24 | |
| 25 | // Use TF logging for TensorRT informations |
| 26 | void Logger::log(Severity severity, const char* msg) noexcept { |
| 27 | // Suppress info-level messages |
| 28 | switch (severity) { |
| 29 | #if NV_TENSORRT_MAJOR > 5 || (NV_TENSORRT_MAJOR == 5 && NV_TENSORRT_MINOR >= 1) |
| 30 | case Severity::kVERBOSE: |
| 31 | #endif |
| 32 | case Severity::kINFO: { // Mark TRT info messages as debug! |
| 33 | VLOG(2) << name_ << " " << msg; |
| 34 | break; |
| 35 | } |
| 36 | case Severity::kWARNING: { |
| 37 | LOG(WARNING) << name_ << " " << msg; |
| 38 | break; |
| 39 | } |
| 40 | case Severity::kERROR: { |
| 41 | LOG(ERROR) << name_ << " " << msg; |
| 42 | break; |
| 43 | } |
| 44 | case Severity::kINTERNAL_ERROR: { |
| 45 | LOG(FATAL) << name_ << " " << msg; |
| 46 | break; |
| 47 | } |
| 48 | // This is useless for now. But would catch it in future if enum changes. It |
| 49 | // is always good to have default case! |
| 50 | default: { |
| 51 | LOG(FATAL) << name_ << "Got unknown severity level " << int(severity) |
| 52 | << " from TensorRT: " << msg; |
| 53 | break; |
| 54 | } |
| 55 | } |
| 56 | } |
| 57 | Logger* Logger::GetLogger() { |
| 58 | static Logger* logger = new Logger("DefaultLogger"); |
| 59 | return logger; |
no outgoing calls