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

…ions < 6 (apache#7372) * Fix small bug preventing TRT runtime compilation for versions < 6 * Trigger ci
Lokiiiiii · Mar 1, 2021 · c8a71ed · c8a71ed
1 parent 103c522
commit c8a71ed
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