MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / log

Method log

tensorflow/compiler/tf2tensorrt/utils/trt_logger.cc:26–56  ·  view source on GitHub ↗

Use TF logging for TensorRT informations

Source from the content-addressed store, hash-verified

24
25// Use TF logging for TensorRT informations
26void 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}
57Logger* Logger::GetLogger() {
58 static Logger* logger = new Logger("DefaultLogger");
59 return logger;

Callers 8

testFloatOpsMethod · 0.45
testComplexOpsMethod · 0.45
testSoftplusMethod · 0.45
testIntOpsMethod · 0.45
ForwardMethod · 0.45
CompiledFunctionMethod · 0.45

Calls

no outgoing calls

Tested by 8

testFloatOpsMethod · 0.36
testComplexOpsMethod · 0.36
testSoftplusMethod · 0.36
testIntOpsMethod · 0.36
ForwardMethod · 0.36
CompiledFunctionMethod · 0.36