add fuse padding zero

jack603047588 · Nov 5, 2021 · 4e577e7 · 4e577e7
1 parent eb0a839
commit 4e577e7
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/paddle/fluid/operators/pull_box_sparse_op.h b/paddle/fluid/operators/pull_box_sparse_op.h
@@ -22,6 +22,8 @@
 #include "paddle/fluid/framework/tensor.h"
 #include "paddle/fluid/operators/math/math_function.h"
 
+DECLARE_bool(enable_pull_box_padding_zero);
+
 namespace paddle {
 namespace operators {
 
@@ -123,8 +125,10 @@ static void PullBoxSparseFunctor(const framework::ExecutionContext &ctx) {
     const auto *slot = inputs[i];
     auto *output = outputs[i];
     if (slot->numel() == 0) {
-      // only support GPU
-      PaddingZeros<T>(ctx, output, batch_size, hidden_size);
+      if (FLAGS_enable_pull_box_padding_zero) {
+        // only support GPU
+        PaddingZeros<T>(ctx, output, batch_size, hidden_size);
+      }
       continue;
     }
     output->mutable_data<T>(ctx.GetPlace());

diff --git a/paddle/fluid/platform/flags.cc b/paddle/fluid/platform/flags.cc
@@ -604,3 +604,5 @@ DEFINE_bool(enable_pullpush_dedup_keys, false,
             "enable pull push dedup keys, default false");
 DEFINE_bool(enable_shuffle_by_searchid, false,
             "enable dualbox shuffle by searchid, default false");
+DEFINE_bool(enable_pull_box_padding_zero, true,
+            "enable pull box padding zero, default true");
diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py
@@ -266,6 +266,7 @@ def __bootstrap__():
             'enable_slotrecord_reset_shrink',
             'enable_pullpush_dedup_keys',
             'enable_shuffle_by_searchid',
+            'enable_pull_box_padding_zero',
         ]
     core.init_gflags(["--tryfromenv=" + ",".join(read_env_flags)])
     core.init_glog(sys.argv[0])