From e50489218dc2e7850326d42a22f8f83a46e44031 Mon Sep 17 00:00:00 2001 From: Trevor Morris Date: Thu, 11 Feb 2021 09:46:56 -0800 Subject: [PATCH] [BYOC][TRT] Fix small bug preventing TRT runtime compilation for versions < 6 (#7372) * Fix small bug preventing TRT runtime compilation for versions < 6 * Trigger ci --- 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