Skip to content

[Prim][PIR] Set prim gflag for pure cpp #60505

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 15 commits into from
Jan 3, 2024
Merged

[Prim][PIR] Set prim gflag for pure cpp #60505

Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
9961d37
inference support decomp
cyber-pioneer Dec 22, 2023
b0dcaba
polish code
cyber-pioneer Dec 25, 2023
531cc0b
Merge branch 'develop' of github.com:PaddlePaddle/Paddle into infer
cyber-pioneer Dec 26, 2023
fce1a00
add decomp base define
cyber-pioneer Dec 26, 2023
9a4f119
add decomp base define2
cyber-pioneer Dec 26, 2023
28648e1
change decomp infer
cyber-pioneer Dec 26, 2023
df1fc17
fix symbol overload
cyber-pioneer Dec 26, 2023
1c94e01
fix test case
cyber-pioneer Dec 26, 2023
ed442b5
debug
cyber-pioneer Dec 27, 2023
f14766f
debug
cyber-pioneer Dec 27, 2023
b576640
decomp add debug info
cyber-pioneer Jan 2, 2024
707caea
Merge branch 'develop' of github.com:PaddlePaddle/Paddle into infer
cyber-pioneer Jan 2, 2024
570110c
add cpp flag
cyber-pioneer Jan 2, 2024
ff4ec98
revert
cyber-pioneer Jan 2, 2024
93c02ba
remove unused flag
cyber-pioneer Jan 2, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions paddle/fluid/prim/utils/utils.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,15 @@
#include "paddle/fluid/prim/utils/static/static_global_utils.h"

PADDLE_DEFINE_EXPORTED_bool(prim_enabled, false, "enable_prim or not");
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) {
Expand All @@ -36,7 +41,8 @@ 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) {
Expand Down