From 0fb15d9fcc2166db1ef1bdabbf86501c1f9a82b2 Mon Sep 17 00:00:00 2001
From: Roc <30228238+sljlp@users.noreply.github.com>
Date: Wed, 28 Jul 2021 15:31:01 +0800
Subject: [PATCH] fix optimizer.py (#34431)

---
 python/paddle/fluid/optimizer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
index 58ce5eb66a3..5193339a2f8 100755
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -4637,7 +4637,7 @@ class PipelineOptimizer(object):
         elif self._is_weight_decay_op(op) and op.type == 'scale':
             # set AdamW decay_coeff to device:all
             op._set_attr(self._op_device_key, f"{self._device}:all")
-        elif op.type == "alloc_float_status":
+        elif op.type == "alloc_float_status" or op.type == "clear_float_status":
             op._set_attr(self._op_device_key, f"{self._device}:all")
         else:
             other_known_ops = [
-- 
GitLab