【Prim】Add gather vjp #50305
【Prim】Add gather vjp #50305
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
… add_gather_vjp
… add_gather_vjp
… add_gather_vjp
| def test_cinn(self): | ||
| if core.is_compiled_with_cuda() and ( | ||
| framework._current_expected_place() == core.CPUPlace() | ||
| ): |
There was a problem hiding this comment.
cinn test only on cpu?
There was a problem hiding this comment.
cinn will crash, I just disable it and reopen it when we fix cinn
| np.random.rand(3, 88, 10), | ||
| 4, | ||
| ), | ||
| ], |
| dy_res[i].numpy(), | ||
| rtol=1e-6, | ||
| atol=1e-6, | ||
| ) |
There was a problem hiding this comment.
when test fp64 1e-6 should be change
There was a problem hiding this comment.
This case will not test fp64
| ) | ||
| paddle.enable_static() | ||
|
|
||
| def test_tanh_grad_comp(self): |
There was a problem hiding this comment.
change tanh to gather
| np.testing.assert_allclose( | ||
| actual=dx, | ||
| desired=ddx, | ||
| rtol=1e-6, |
There was a problem hiding this comment.
change 1e-6 when test fp64
| np.testing.assert_allclose( | ||
| actual=actual(self.primal0, self.index, self.axis), | ||
| desired=desired(self.primal0, self.index, self.axis), | ||
| rtol=1e-6, |
There was a problem hiding this comment.
change 1e-6 when test FP64
There was a problem hiding this comment.
since atol is 0, maybe they are equal
| @@ -0,0 +1,344 @@ | |||
| # Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. | |||
| # def is_inplace(self): | ||
| # if self.inplace_map | ||
| # return True | ||
| # return False |
| @@ -0,0 +1,116 @@ | |||
| # Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. | |||
| @@ -0,0 +1,236 @@ | |||
| # Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved. | |||
PR types
Others
PR changes
Others
Describe
This PR support gather vjp in prim