Fix random failure in test_trt_convert_strided_slice (#69573)

anderson101866 · web-flow · commit ebce58b344a5 · 2024-11-21T20:27:58.000+08:00
The test will produce several error log like:
API Usage Error (Tensor slice_output_data_subgraph_4 has axis 3 with inherently negative length. Proven upper bound is -111894960. Network must have an instance where axis has non-negative length.)

, which is uninitialized memory issue due to TRT10 alias Dims to Dims64,
 but TRT8 alias Dims to Dims32.
diff --git a/paddle/fluid/inference/tensorrt/convert/slice_op.cc b/paddle/fluid/inference/tensorrt/convert/slice_op.cc
@@ -47,7 +47,9 @@ class SliceOpConverter : public OpConverter {
     auto* shape_tensor = Shape(input);
     nvinfer1::Dims trt_start_dims;
     trt_start_dims.nbDims = input_dims.nbDims;
-    memset(trt_start_dims.d, 0, sizeof(int32_t) * input_dims.nbDims);
+    memset(trt_start_dims.d,
+           0,
+           sizeof(trt_start_dims.d[0]) * nvinfer1::Dims::MAX_DIMS);
     nvinfer1::Dims trt_size_dims = trt_start_dims;
     nvinfer1::Dims trt_step_dims = trt_start_dims;
     for (int i = 0; i < trt_step_dims.nbDims; i++) trt_step_dims.d[i] = 1;
diff --git a/paddle/fluid/inference/tensorrt/convert/strided_slice_op.cc b/paddle/fluid/inference/tensorrt/convert/strided_slice_op.cc
@@ -46,7 +46,9 @@ class StridedSliceOpConverter : public OpConverter {
     auto nchw_input_dims = input->getDimensions();
     nvinfer1::Dims trt_start_dims;
     trt_start_dims.nbDims = nchw_input_dims.nbDims;
-    memset(trt_start_dims.d, 0, sizeof(int32_t) * nchw_input_dims.nbDims);
+    memset(trt_start_dims.d,
+           0,
+           sizeof(trt_start_dims.d[0]) * nvinfer1::Dims::MAX_DIMS);
     nvinfer1::Dims trt_size_dims = trt_start_dims;
     nvinfer1::Dims trt_end_dims = trt_start_dims;
     nvinfer1::Dims trt_step_dims = trt_start_dims;
