remove unused flag

PaddlePaddle · Jan 2, 2024 · 9dcaefc · 9dcaefc
1 parent ff4ec98
commit 9dcaefc
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/paddle/fluid/prim/utils/utils.cc b/paddle/fluid/prim/utils/utils.cc
@@ -13,17 +13,20 @@
 // limitations under the License.
 
 #include "paddle/fluid/prim/utils/utils.h"
+#include <sstream>
 #include "paddle/fluid/platform/flags.h"
 #include "paddle/fluid/prim/utils/static/static_global_utils.h"
 
 PADDLE_DEFINE_EXPORTED_bool(prim_enabled, false, "enable_prim or not");
-PADDLE_DEFINE_EXPORTED_string(prim_blacklist, "", "prim ops blacklist");
+PADDLE_DEFINE_EXPORTED_bool(prim_all, false, "enable prim_all or not");
+PADDLE_DEFINE_EXPORTED_bool(prim_forward, false, "enable prim_forward or not");
+PADDLE_DEFINE_EXPORTED_bool(prim_backward, false, "enable prim_backward not");
 
 namespace paddle {
 namespace prim {
-
 bool PrimCommonUtils::IsBwdPrimEnabled() {
-  return StaticCompositeContext::Instance().IsBwdPrimEnabled();
+  bool res = StaticCompositeContext::Instance().IsBwdPrimEnabled();
+  return res || FLAGS_prim_all || FLAGS_prim_backward;
 }
 
 void PrimCommonUtils::SetBwdPrimEnabled(bool enable_prim) {
@@ -39,16 +42,15 @@ void PrimCommonUtils::SetEagerPrimEnabled(bool enable_prim) {
 }
 
 bool PrimCommonUtils::IsFwdPrimEnabled() {
-  return StaticCompositeContext::Instance().IsFwdPrimEnabled();
+  bool res = StaticCompositeContext::Instance().IsFwdPrimEnabled();
+  return res || FLAGS_prim_all || FLAGS_prim_forward;
 }
 
 void PrimCommonUtils::SetFwdPrimEnabled(bool enable_prim) {
-  VLOG(0) << "FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
   StaticCompositeContext::Instance().SetFwdPrimEnabled(enable_prim);
 }
 
 void PrimCommonUtils::SetAllPrimEnabled(bool enable_prim) {
-  VLOG(0) << "FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
   StaticCompositeContext::Instance().SetAllPrimEnabled(enable_prim);
 }