Skip to content

Commit 570110c

Browse files
committed
add cpp flag
1 parent 707caea commit 570110c

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

paddle/fluid/prim/utils/utils.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,11 @@
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");
21+
2022
namespace paddle {
2123
namespace prim {
24+
2225
bool PrimCommonUtils::IsBwdPrimEnabled() {
2326
return StaticCompositeContext::Instance().IsBwdPrimEnabled();
2427
}
@@ -40,10 +43,12 @@ bool PrimCommonUtils::IsFwdPrimEnabled() {
4043
}
4144

4245
void PrimCommonUtils::SetFwdPrimEnabled(bool enable_prim) {
46+
VLOG(0) << "FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
4347
StaticCompositeContext::Instance().SetFwdPrimEnabled(enable_prim);
4448
}
4549

4650
void PrimCommonUtils::SetAllPrimEnabled(bool enable_prim) {
51+
VLOG(0) << "FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
4752
StaticCompositeContext::Instance().SetAllPrimEnabled(enable_prim);
4853
}
4954

0 commit comments

Comments
 (0)