From 019e443821fbb49ac8330c637c690aca0431d3c0 Mon Sep 17 00:00:00 2001
From: Shixiaowei02 <39303645+Shixiaowei02@users.noreply.github.com>
Date: Tue, 22 Jun 2021 05:47:51 +0000
Subject: [PATCH] fix bugs, test=develop

---
 paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc b/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc
index 50364d2d0109d..55ef6a492954e 100644
--- a/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/seqpool_cvm_concat_fuse_pass.cc
@@ -61,8 +61,10 @@ SeqPoolCVMConcatFusePass::SeqPoolCVMConcatFusePass() {
       .IsTensor()
       .End()
       .AddOutput("MaxIndex")
+      .IsOptional()
       .End()
       .AddAttr("pooltype")
+      .IsStringIn({"AVERAGE", "SUM", "SQRT", "LAST", "FIRST", "MAX"})
       .End()
       .AddAttr("pad_value")
       .End();
@@ -85,6 +87,7 @@ SeqPoolCVMConcatFusePass::SeqPoolCVMConcatFusePass() {
       .End()
       .AddInput("AxisTensor")
       .IsTensor()
+      .IsOptional()
       .End()
       .AddOutput("Out")
       .IsTensor()