Update test_lerp_op.py

modify the input dtype
PaddlePaddle · Apr 22, 2023 · 739d6e7 · 739d6e7
1 parent d390a49
commit 739d6e7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_lerp_op.py b/python/paddle/fluid/tests/unittests/test_lerp_op.py
@@ -231,9 +231,9 @@ def setUp(self):
  self.python_api = paddle.lerp
  self.init_dtype()
  self.init_shape()
- x = np.arange(1.0, 101.0).astype(np.float64).reshape(self.shape)
- y = np.full(100, 10.0).astype(np.float64).reshape(self.shape)
- w = np.asarray([0.5]).astype(np.float64)
+ x = np.arange(1.0, 101.0).astype(np.float32).reshape(self.shape)
+ y = np.full(100, 10.0).astype(np.float32).reshape(self.shape)
+ w = np.asarray([0.5]).astype(np.float32)
  self.inputs = {
  'X': convert_float_to_uint16(x),
  'Y': convert_float_to_uint16(y),