[BYOC][TRT] Fix small bug preventing TRT runtime compilation for vers…

…ions < 6 (#7372) * Fix small bug preventing TRT runtime compilation for versions < 6 * Trigger ci
apache · Feb 11, 2021 · 6b58321 · 6b58321
1 parent d05d75d
commit 6b58321
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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