fix_arguments (#49186)

b0e9e48d · xiaoxiaohehe001 · GitHub · e4857e0b · b0e9e48d
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/inference/api/analysis_predictor.cc paddle/fluid/inference/api/analysis_predictor.cc +3 -1

未找到文件。
--- a/paddle/fluid/inference/api/analysis_predictor.cc
+++ b/paddle/fluid/inference/api/analysis_predictor.cc
@@ -1365,7 +1365,9 @@ void AnalysisPredictor::OptimizeInferenceProgram() {
 #if defined(_WIN32)
  argument_->PartiallyRelease();
 #else
-  if (config_.mkldnn_enabled() || config_.tensorrt_engine_enabled()) {
+  if (config_.mkldnn_enabled() ||
+      (config_.tensorrt_engine_enabled() &&
+       config_.tensorrt_precision_mode_ == AnalysisConfig::Precision::kInt8)) {
    argument_->PartiallyRelease();
  } else {
    argument_.reset(nullptr);