[TRT] Fix TRT PadV2 LayerBuilder empty param->pads bug

Tencent · Aug 2, 2024 · 1d535ec · 1d535ec
1 parent debcb38
commit 1d535ec
Showing 1 changed file with 4 additions and 0 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
@@ -69,6 +69,10 @@ nvinfer1::DataType PadV2TRTPluginLayerBuilder::getOutputDataType(int index, cons
 
 ILayer* PadV2TRTPluginLayerBuilder::AddToNetwork(INetworkDefinition* network) noexcept {
     auto paramlist = dynamic_cast<PadLayerParam*>(param_);
+    if (paramlist->pads.empty()) {
+        LOGE("Error: PadV2 Layer got Empty params->pads.\n");
+        return nullptr;
+    }
 
     if (!UseTRTPaddingND(paramlist)) {
         return TensorRTPluginLayerBuilder::AddToNetwork(network);