1717#include " paddle/fluid/prim/utils/static/static_global_utils.h"
1818
1919PADDLE_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
2224namespace paddle {
2325namespace prim {
24-
2526bool PrimCommonUtils::IsBwdPrimEnabled () {
26- return StaticCompositeContext::Instance ().IsBwdPrimEnabled ();
27+ bool res = StaticCompositeContext::Instance ().IsBwdPrimEnabled ();
28+ return res || FLAGS_prim_all || FLAGS_prim_backward;
2729}
2830
2931void PrimCommonUtils::SetBwdPrimEnabled (bool enable_prim) {
@@ -39,16 +41,15 @@ void PrimCommonUtils::SetEagerPrimEnabled(bool enable_prim) {
3941}
4042
4143bool PrimCommonUtils::IsFwdPrimEnabled () {
42- return StaticCompositeContext::Instance ().IsFwdPrimEnabled ();
44+ bool res = StaticCompositeContext::Instance ().IsFwdPrimEnabled ();
45+ return res || FLAGS_prim_all || FLAGS_prim_forward;
4346}
4447
4548void PrimCommonUtils::SetFwdPrimEnabled (bool enable_prim) {
46- VLOG (0 ) << " FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
4749 StaticCompositeContext::Instance ().SetFwdPrimEnabled (enable_prim);
4850}
4951
5052void 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