From c828a4b0481463ed860407ca8f5f31cd96417c25 Mon Sep 17 00:00:00 2001 From: Dong Xu Date: Tue, 6 Aug 2024 19:19:18 +0800 Subject: [PATCH] [TRT] Disable out-dated TRT PadV2 Plugin --- .../network/tensorrt/layer_builder/pad_v2_layer_builder.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/source/tnn/network/tensorrt/layer_builder/pad_v2_layer_builder.cc b/source/tnn/network/tensorrt/layer_builder/pad_v2_layer_builder.cc index dbb3044fa..587a63246 100644 --- a/source/tnn/network/tensorrt/layer_builder/pad_v2_layer_builder.cc +++ b/source/tnn/network/tensorrt/layer_builder/pad_v2_layer_builder.cc @@ -74,9 +74,9 @@ ILayer* PadV2TRTPluginLayerBuilder::AddToNetwork(INetworkDefinition* network) no return nullptr; } - if (!UseTRTPaddingND(paramlist)) { - return TensorRTPluginLayerBuilder::AddToNetwork(network); - } + //if (!UseTRTPaddingND(paramlist)) { + // return TensorRTPluginLayerBuilder::AddToNetwork(network); + //} auto input_foreign_tensor = dynamic_cast(input_blobs_[0])->GetForeignTensor(); auto input_tensor = std::dynamic_pointer_cast(input_foreign_tensor)->GetTensor();