diff --git a/paddle/fluid/operators/jit/gen/seqpool.cc b/paddle/fluid/operators/jit/gen/seqpool.cc index d9e5904add4486ddf126093865f7e0571c1909e4..ec8e4e9827441bc0a817c6da455cb9e530c8c1bf 100644 --- a/paddle/fluid/operators/jit/gen/seqpool.cc +++ b/paddle/fluid/operators/jit/gen/seqpool.cc @@ -66,7 +66,7 @@ class SeqPoolCreator : public JitCodeCreator { ((attr.w / YMM_FLOAT_BLOCK + 4 /* for rest */) * 4 /* load, mul and save */ + 256) * - 8; + 16; } std::unique_ptr CreateJitCode( const seq_pool_attr_t& attr) const override {