[TRT] Add TRT Layer delete when layer creation return non-zero

Tencent · Jun 18, 2024 · debcb38 · debcb38
1 parent baff1f6
commit debcb38
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/tnn/network/tensorrt/tensorrt_network.cc b/source/tnn/network/tensorrt/tensorrt_network.cc
@@ -513,6 +513,7 @@ Status TensorRTNetwork_::InitLayers(NetStructure *net_structure, NetResource *ne
         if (ret != TNN_OK) {
             LOGE("Error Init layer: %s layer type: %d (err: %d or 0x%X)\n", cur_layer->GetLayerName().c_str(),
                 type, (int)ret, (int)ret);
+            delete cur_layer;
             return ret;
         }