fix multistep error

PaddlePaddle · Jul 25, 2023 · bda4d25 · bda4d25
1 parent 77f5e39
commit bda4d25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/legacy_test/test_lamb_op.py b/test/legacy_test/test_lamb_op.py
@@ -340,8 +340,8 @@ def test_check_output(self):
             self.inputs['Beta2Pow'] = convert_float_to_uint16(beta2_pow_out)
 
             # Randomize gradient for next step
-            self.inputs['Grad'] = np.random.uniform(-1, 1, (102, 105)).astype(
-                np.float32
+            self.inputs['Grad'] = convert_float_to_uint16(
+                np.random.uniform(-1, 1, (102, 105)).astype(np.float32)
             )