[Eager] Enable_legacy_for_mkldnn_ops_on_off (#42648)

6c696db1 · Weilong Wu · GitHub · 27acc6c3 · 6c696db1
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

python/paddle/fluid/tests/unittests/mkldnn/check_flags_mkldnn_ops_on_off.py ...d/tests/unittests/mkldnn/check_flags_mkldnn_ops_on_off.py +2 -0

未找到文件。
--- a/python/paddle/fluid/tests/unittests/mkldnn/check_flags_mkldnn_ops_on_off.py
+++ b/python/paddle/fluid/tests/unittests/mkldnn/check_flags_mkldnn_ops_on_off.py
@@ -20,6 +20,8 @@ import paddle.fluid as fluid
 import os
 from paddle.fluid.layer_helper import LayerHelper
 from paddle.fluid.framework import _global_flags
+from paddle.fluid.framework import _enable_legacy_dygraph
+_enable_legacy_dygraph()


 def check():