File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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" );
21+
2022namespace paddle {
2123namespace prim {
24+
2225bool PrimCommonUtils::IsBwdPrimEnabled () {
2326 return StaticCompositeContext::Instance ().IsBwdPrimEnabled ();
2427}
@@ -40,10 +43,12 @@ bool PrimCommonUtils::IsFwdPrimEnabled() {
4043}
4144
4245void PrimCommonUtils::SetFwdPrimEnabled (bool enable_prim) {
46+ VLOG (0 ) << " FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
4347 StaticCompositeContext::Instance ().SetFwdPrimEnabled (enable_prim);
4448}
4549
4650void PrimCommonUtils::SetAllPrimEnabled (bool enable_prim) {
51+ VLOG (0 ) << " FLAGS_prim_enabled ====================== " << FLAGS_prim_enabled;
4752 StaticCompositeContext::Instance ().SetAllPrimEnabled (enable_prim);
4853}
4954
You can’t perform that action at this time.
0 commit comments