support dcu in eager (#43371)

PaddlePaddle · Jun 13, 2022 · ee60567 · ee60567
1 parent a0d0bb6
commit ee60567
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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