Sparsity support (#36413)

* add pool2d convert test * modify error * modify error * modify error * modify error * modify error * modify error * sparsity support

Sparsity support (#36413)
* add pool2d convert test * modify error * modify error * modify error * modify error * modify error * modify error * sparsity support
b857d755 · JingZhuangzhuang · GitHub · 7f5128f4 · b857d755
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/inference/tensorrt/engine.cc paddle/fluid/inference/tensorrt/engine.cc +1 -0

未找到文件。
--- a/paddle/fluid/inference/tensorrt/engine.cc
+++ b/paddle/fluid/inference/tensorrt/engine.cc
@@ -225,6 +225,7 @@ void TensorRTEngine::FreezeNetwork() {
  infer_engine_.reset(infer_builder_->buildEngineWithConfig(
      *network(), *infer_builder_config_));
 #else
+  infer_builder_config_->setFlag(nvinfer1::BuilderFlag::kSPARSE_WEIGHTS);
  infer_ptr<nvinfer1::IHostMemory> plan(infer_builder_->buildSerializedNetwork(
      *network(), *infer_builder_config_));
  infer_ptr<nvinfer1::IRuntime> runtime(createInferRuntime(&logger_));