fix cumsum bug (#54160)

paddle/fluid/inference/tensorrt/convert/cumsum_op.cc

@@ -51,7 +51,7 @@ class CumsumOpConverter : public OpConverter {
             return Add1DConstantLayer(d, "", scalar);
           } else {
             nvinfer1::ITensor* inpShape = Shape(inpTensor);
-            return GetEleTensorOfShape(inpShape, d, scalar);
+            return GetEleTensorOfShape(inpShape, axis, scalar);
           }
         };
 

paddle/fluid/inference/tensorrt/convert/op_converter.h

@@ -529,6 +529,12 @@ class OpConverter {
   nvinfer1::ITensor* GetEleTensorOfShape(nvinfer1::ITensor* shape_tensor,
                                          int index,
                                          bool is_scalar = false) {
+    PADDLE_ENFORCE_GE(
+        index,
+        0,
+        platform::errors::PreconditionNotMet(
+            "The index should be greater or equal than 0, but got %d", index));
+
     auto* tensor =
         TRT_ENGINE_ADD_LAYER(engine_,
                              Gather,