From 7be637a7ad1f584d9c8858eab68ba128a5b6d09a Mon Sep 17 00:00:00 2001
From: zhoutianzi666 <39978853+zhoutianzi666@users.noreply.github.com>
Date: Fri, 8 Jul 2022 14:06:42 +0800
Subject: [PATCH] [Paddle-TRT] fix_pool (#44139)

* fix_pool

* fix pool_op_plugin
---
 paddle/fluid/inference/tensorrt/plugin/pool_op_plugin.cu | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/plugin/pool_op_plugin.cu b/paddle/fluid/inference/tensorrt/plugin/pool_op_plugin.cu
index 69c317781e..21eb89d135 100644
--- a/paddle/fluid/inference/tensorrt/plugin/pool_op_plugin.cu
+++ b/paddle/fluid/inference/tensorrt/plugin/pool_op_plugin.cu
@@ -184,11 +184,6 @@ nvinfer1::DimsExprs PoolPluginDynamic::getOutputDimensions(
                     platform::errors::InvalidArgument(
                         "The Split plugin should be only one input."));
 
-  PADDLE_ENFORCE_EQ(
-      inputs[0].d[1]->isConstant(),
-      true,
-      platform::errors::InvalidArgument("The channel dimension should be "
-                                        "static, but we found it's dynamic."));
   nvinfer1::DimsExprs output(inputs[0]);
   if (is_global_ && !adaptive_) {
     output.d[2] = expr_builder.constant(1);
-- 
GitLab