diff --git a/src/tensorrt/impl/tensorrt_opr.cpp b/src/tensorrt/impl/tensorrt_opr.cpp index 009b36aa4fb3a7c70421d1d3922988de100d0434..9e1a630ada0745dbef4b032483d26917bf4734b0 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(