support dcu in eager (#43371)

ee605673 · Jiabin Yang · GitHub · a0d0bb63 · ee605673
显示空白变更内容
内联并排

Showing with 1 addition and 2 deletion

python/paddle/fluid/framework.py python/paddle/fluid/framework.py +1 -2

未找到文件。
--- 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