diff --git a/python/paddle/fluid/framework.py b/python/paddle/fluid/framework.py index a790feb9af29c7af58fb99b673153ce861674502..44ef1ff5ae6db55b716fd69b9c60c13dc5c2a879 100644 --- a/python/paddle/fluid/framework.py +++ b/python/paddle/fluid/framework.py @@ -178,8 +178,7 @@ def _fallback_legacy_dygraph(): need_fallback = False # Only enable eager on CPU/GPU is_not_support = core.is_compiled_with_xpu() or core.is_compiled_with_npu( - ) or core.is_compiled_with_ipu() or core.is_compiled_with_mlu( - ) or core.is_compiled_with_rocm() + ) or core.is_compiled_with_ipu() or core.is_compiled_with_mlu() if _in_eager_mode_ and is_not_support: # switch into legacy dygraph mode