From f285f4c16212d6bfef772e6f74bf205b09f5e95c Mon Sep 17 00:00:00 2001
From: wenbin <wang3323032@qq.com>
Date: Tue, 27 Apr 2021 15:56:46 +0800
Subject: [PATCH] conservative judgment (#32556)

---
 paddle/fluid/inference/tensorrt/convert/elementwise_op.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc b/paddle/fluid/inference/tensorrt/convert/elementwise_op.cc
index 5419933e40..19d7951054 100644
--- 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;
     }
-- 
GitLab