fix optimizer.py (#34431)

PaddlePaddle · Jul 28, 2021 · 0fb15d9 · 0fb15d9
1 parent 995195f
commit 0fb15d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
@@ -4637,7 +4637,7 @@ def _add_op_device_attr_for_op(self, op, idx, block):
         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 = [