diff --git a/paddle/fluid/inference/tensorrt/engine.cc b/paddle/fluid/inference/tensorrt/engine.cc index 33386c746ae5a581af84b805377e6ada670e93ba..00a6b2ffbf92307dc9ee877b3fbc64b0142ecb9c 100644 --- a/paddle/fluid/inference/tensorrt/engine.cc +++ b/paddle/fluid/inference/tensorrt/engine.cc @@ -90,7 +90,9 @@ void TensorRTEngine::FreezeNetwork() { bool enable_int8 = (precision_ == AnalysisConfig::Precision::kInt8); if (enable_int8) { - infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kFP16); + if (!use_dla_) { + infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kFP16); + } infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kINT8); if (calibrator_) {