Skip to content

Commit 93c02ba

Browse files
committed
remove unused flag
1 parent ff4ec98 commit 93c02ba

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

paddle/fluid/prim/utils/utils.cc

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,15 @@
1717
#include "paddle/fluid/prim/utils/static/static_global_utils.h"
1818

1919
PADDLE_DEFINE_EXPORTED_bool(prim_enabled, false, "enable_prim or not");
20-
PADDLE_DEFINE_EXPORTED_string(prim_blacklist, "", "prim ops blacklist");
20+
PADDLE_DEFINE_EXPORTED_bool(prim_all, false, "enable prim_all or not");
21+
PADDLE_DEFINE_EXPORTED_bool(prim_forward, false, "enable prim_forward or not");
22+
PADDLE_DEFINE_EXPORTED_bool(prim_backward, false, "enable prim_backward not");
2123

2224
namespace paddle {
2325
namespace prim {
24-
2526
bool PrimCommonUtils::IsBwdPrimEnabled() {
26-
return StaticCompositeContext::Instance().IsBwdPrimEnabled();
27+
bool res = StaticCompositeContext::Instance().IsBwdPrimEnabled();
28+
return res || FLAGS_prim_all || FLAGS_prim_backward;
2729
}
2830

2931
void PrimCommonUtils::SetBwdPrimEnabled(bool enable_prim) {
@@ -39,16 +41,15 @@ void PrimCommonUtils::SetEagerPrimEnabled(bool enable_prim) {
3941
}
4042

4143
bool PrimCommonUtils::IsFwdPrimEnabled() {
42-
return StaticCompositeContext::Instance().IsFwdPrimEnabled();
44+
bool res = StaticCompositeContext::Instance().IsFwdPrimEnabled();
45+
return res || FLAGS_prim_all || FLAGS_prim_forward;
4346
}
4447

4548
void PrimCommonUtils::SetFwdPrimEnabled(bool enable_prim) {
46-
VLOG(0) << "FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
4749
StaticCompositeContext::Instance().SetFwdPrimEnabled(enable_prim);
4850
}
4951

5052
void PrimCommonUtils::SetAllPrimEnabled(bool enable_prim) {
51-
VLOG(0) << "FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
5253
StaticCompositeContext::Instance().SetAllPrimEnabled(enable_prim);
5354
}
5455

0 commit comments

Comments
 (0)