Fix

Author: co63oc

python/paddle/incubate/optimizer/lars_momentum.py

@@ -189,7 +189,22 @@ def _append_optimize_op(self, block, param_and_grad):
             inputs["MasterParam"] = master_weight
             outputs["MasterParamOut"] = master_weight
 
-        if in_dynamic_mode() or in_pir_mode():
+        if in_dynamic_mode():
+            _, _, _ = _C_ops.lars_momentum_(
+                [param_and_grad[0]],
+                [param_and_grad[1]],
+                [velocity_acc],
+                [lr],
+                master_weight,
+                self._momentum,
+                self._lars_coeff,
+                [_lars_weight_decay],
+                self._epsilon,
+                find_master,
+                self._rescale_grad,
+            )
+            return None
+        elif in_pir_mode():
             if isinstance(master_weight, pir.Value):
                 master_weight = [master_weight]
             _, _, _ = _C_ops.lars_momentum_(