add user_defined_grads

PaddlePaddle · May 20, 2023 · 6348897 · 6348897
1 parent 1f74545
commit 6348897
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_lerp_op.py b/python/paddle/fluid/tests/unittests/test_lerp_op.py
@@ -15,11 +15,7 @@
 import unittest
 
 import numpy as np
-from eager_op_test import (
- OpTest,
- convert_float_to_uint16,
- convert_uint16_to_float,
-)
+from eager_op_test import OpTest, convert_float_to_uint16
 
 import paddle
 from paddle.fluid import core
@@ -237,12 +233,9 @@ def setUp(self):
  self.init_shape()
  self.init_xyshape()
  self.init_wshape()
- x = np.arange(1.0, 101.0).astype(np.float32).reshape(self.xshape)
- y = np.full(100, 10.0).astype(np.float32).reshape(self.yshape)
- w = np.random.random(self.wshape).astype(np.float32)
- x = convert_uint16_to_float(convert_float_to_uint16(x))
- y = convert_uint16_to_float(convert_float_to_uint16(y))
- w = convert_uint16_to_float(convert_float_to_uint16(w))
+ x = np.arange(1.0, 101.0).astype("float32").reshape(self.xshape)
+ y = np.full(100, 10.0).astype("float32").reshape(self.yshape)
+ w = np.random.random(self.wshape).astype("float32")
  self.inputs = {
  'X': convert_float_to_uint16(x),
  'Y': convert_float_to_uint16(y),
@@ -266,7 +259,12 @@ def test_check_output(self):
 
  def test_check_grad(self):
  place = core.CUDAPlace(0)
- self.check_grad_with_place(place, ['X', 'Y'], 'Out')
+ self.check_grad_with_place(
+ place,
+ ['X', 'Y'],
+ 'Out',
+ user_defined_grads=[np.zeros(self.xshape), np.ones(self.yshape)],
+ )
 
 
 if __name__ == "__main__":