From 765c9d354c220658530b9e955bcd36fca671faef Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 8 Apr 2020 16:29:02 +0800 Subject: [PATCH] fix(mgb/trt): change log level of trt log GitOrigin-RevId: 5f5495e303e4a78b02ad78f3f9ee6a725009ef52 --- src/tensorrt/impl/tensorrt_opr.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/tensorrt/impl/tensorrt_opr.cpp b/src/tensorrt/impl/tensorrt_opr.cpp index 009b36aa4..9e1a630ad 100644 --- a/src/tensorrt/impl/tensorrt_opr.cpp +++ b/src/tensorrt/impl/tensorrt_opr.cpp @@ -72,13 +72,13 @@ void TensorRTOpr::Logger::log(nvinfer1::ILogger::Severity severity, const char* msg) { switch (severity) { case Severity::kINTERNAL_ERROR: - mgb_log_error("TRT_INTERNAL_ERROR: %s", msg); + mgb_log("TRT_INTERNAL_ERROR: %s", msg); return; case Severity::kERROR: - mgb_log_error("TRT_ERROR: %s", msg); + mgb_log("TRT_ERROR: %s", msg); return; case Severity::kWARNING: - mgb_log_warn("TRT_WARNING: %s", msg); + mgb_log("TRT_WARNING: %s", msg); return; case Severity::kINFO: mgb_log_debug("TRT_INFO: %s", msg); @@ -167,7 +167,7 @@ void TensorRTManager::exec(cg::SingleCNOperatorNodeBase* opr, CompNode comp_node_check, nvinfer1::ICudaEngine* engine, size_t batch) { - + auto comp_node = opr->comp_node(); // ICudaEngine is bound to the currently active device comp_node.activate(); @@ -360,7 +360,7 @@ SymbolVarArray TensorRTOpr::make( std::shared_ptr network, TensorRTGraphFeatureBits feature_bits, std::shared_ptr gpu_allocator, const SymbolVarArray& src, - std::shared_ptr engine, + std::shared_ptr engine, const OperatorNodeConfig& config) { VarNodeArray var_node_array = cg::to_var_node_array(src); auto tensor_rt_opr = std::make_unique( -- GitLab