From cc7fa9ad3764be4dd4cc83a5cfc802d4752983aa Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Fri, 29 Jan 2021 18:41:24 +0000 Subject: [PATCH 1/2] Fix small bug preventing TRT runtime compilation for versions < 6 --- src/runtime/contrib/tensorrt/tensorrt_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/runtime/contrib/tensorrt/tensorrt_ops.cc b/src/runtime/contrib/tensorrt/tensorrt_ops.cc index 69bb1dccfb62..824178eaa619 100644 --- a/src/runtime/contrib/tensorrt/tensorrt_ops.cc +++ b/src/runtime/contrib/tensorrt/tensorrt_ops.cc @@ -447,7 +447,7 @@ class BatchNormOpConverter : public TensorRTOpConverter { nvinfer1::IScaleLayer* scale_layer = params->network->addScaleNd( *input, nvinfer1::ScaleMode::kCHANNEL, weight_shift, weight_scale, power, channel_dim); #else - ICHECK_EQ(input->getDimensions().nbDims(), 3); + ICHECK_EQ(input->getDimensions().nbDims, 3); nvinfer1::IScaleLayer* scale_layer = params->network->addScale( *input, nvinfer1::ScaleMode::kCHANNEL, weight_shift, weight_scale, power); #endif From b7910fa3090652379927fa9c908dcf97907ffbd4 Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Thu, 11 Feb 2021 00:29:25 +0000 Subject: [PATCH 2/2] Trigger ci