[TRT] Add fallback to hand-written ROIAlign for TRT prev to 8.4

Tencent · Feb 27, 2024 · 70836d0 · 70836d0
1 parent 8934bc7
commit 70836d0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/source/tnn/network/tensorrt/layer_builder/roialign_layer_builder.cc b/source/tnn/network/tensorrt/layer_builder/roialign_layer_builder.cc
@@ -45,6 +45,7 @@ nvinfer1::DataType RoiAlignTRTPluginLayerBuilder::getOutputDataType(int index, c
 }
 
 ILayer* RoiAlignTRTPluginLayerBuilder::AddToNetwork(INetworkDefinition* network) noexcept {
+#if NV_TENSORRT_MAJOR * 10 + NV_TENSORRT_MINOR >= 84
     auto layer_param = dynamic_cast<RoiAlignLayerParam*>(param_);
     if (!layer_param) {
         LOGE("RoiAlignTRTPluginLayerBuilder: Unable to get layer param.");
@@ -86,6 +87,9 @@ ILayer* RoiAlignTRTPluginLayerBuilder::AddToNetwork(INetworkDefinition* network)
     }
 
     return layer;
+#else
+    return TensorRTPluginLayerBuilder::AddToNetwork(network);
+#endif
 }
 
 DimsExprs RoiAlignTRTPluginLayerBuilder::getOutputDimensions(int index, const nvinfer1::DimsExprs* inputs,