conservative judgment (#32619)

056a2fca · wenbin · GitHub · 33703da8 · 056a2fca
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

paddle/fluid/inference/tensorrt/convert/elementwise_op.cc paddle/fluid/inference/tensorrt/convert/elementwise_op.cc +4 -0

未找到文件。
--- a/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc
@@ -25,6 +25,10 @@ static bool CheckDims(const nvinfer1::Dims& dims_x,
    return false;
  }
  for (int i = 0; i < dims_x.nbDims; i++) {
+    // conservative judgment
+    if (dims_x.d[i] == -1 || dims_y.d[i] == -1) {
+      return false;
+    }
    if (dims_x.d[i] != dims_y.d[i]) {
      return false;
    }