From c8adfb3451281aad51e1398023d5af80b31fa7f3 Mon Sep 17 00:00:00 2001
From: nhzlx <zlx_hg@163.com>
Date: Wed, 25 Jul 2018 12:32:47 +0000
Subject: [PATCH] add paddle_enforce

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

diff --git a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
index c5852bdb532..11cad953618 100644
--- a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
@@ -29,6 +29,8 @@ class Pool2dOpConverter : public OpConverter {
         << "convert a fluid pool2d op to tensorrt pool2d layer without bias";
     framework::OpDesc op_desc(op, nullptr);
     // Declare inputs
+    PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);
+    PADDLE_ENFORCE_EQ(op_desc.Output("Out").size(), 1);
     auto* input1 = engine_->GetITensor(op_desc.Input("X")[0]);
 
     std::string pool_type =
-- 
GitLab