fix code too big test=develop (#19111)

Fix seq_pool failed when input dims is too large. Resolve issue #3023
6 years ago · 945f3cf631
parent 4a959883e7
commit 945f3cf631
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/jit/gen/seqpool.cc
+++ b/paddle/fluid/operators/jit/gen/seqpool.cc
@ -66,7 +66,7 @@ class SeqPoolCreator : public JitCodeCreator<seq_pool_attr_t> {
           ((attr.w / YMM_FLOAT_BLOCK + 4 /* for rest */) *
                4 /* load, mul and save */ +
            256) *
-               8;
+               16;
  }
  std::unique_ptr<GenBase> CreateJitCode(
      const seq_pool_attr_t& attr) const override {