Fix untest error in test_adamax_op (#51692)

python/paddle/fluid/optimizer.py

@@ -2809,7 +2809,7 @@ class AdamaxOptimizer(Optimizer):
             with param.block.program._optimized_guard(
                 [param, grad]
             ), name_scope('adamx'):
-                beta1_pow_acc = self._get_accumulator(
+                beta1_pow_acc = self._get_accumulator_master(
                     self._beta1_pow_acc_str, param
                 )
                 if in_dygraph_mode():

python/paddle/fluid/tests/unittests/test_adamax_op.py

@@ -31,6 +31,7 @@ def adamx_wrapper(
     beta1=0.78,
     beta2=0.899,
     epsilon=1e-5,
+    find_master=False,
 ):
     return paddle._C_ops.adamax_(
         param,
@@ -43,6 +44,7 @@ def adamx_wrapper(
         beta1,
         beta2,
         epsilon,
+        find_master,
     )