p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!27506 [GraphKernel] fix probabilistic failure case. 

Merge pull request !27506 from chenlei_autodiff/fix_ci
This commit is contained in:
i-robot 2021-12-15 02:20:34 +00:00 committed by Gitee
parent 83fd1e3303 7a8cfd6048
commit f889165dc1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/st/ops/graph_kernel/test_fuse.py
View File

@ -33,7 +33,9 @@ class Net(Cell):
def construct(self, x, y, z):
sub_res = self.sub(x, y)
mul_res = self.mul(sub_res, x)
sqrt_grad_res = self.sqrt_grad(mul_res, z)
mul_square_res = P.Square()(mul_res)
add_one_res = self.add(mul_square_res, 1)
sqrt_grad_res = self.sqrt_grad(add_one_res, z)
square_res = P.Square()(sqrt_grad_res)
add_res = self.add(sqrt_grad_res, square_res)
add1_res = self.add(add_res, add_res)