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

!9107 ps testcast 

From: @chopin_syp
Reviewed-by: @xsmq,@kisnwang
Signed-off-by: @kisnwang
This commit is contained in:
mindspore-ci-bot 2020-11-27 16:31:54 +08:00 committed by Gitee
parent 67505ef198 6da2169439
commit 715ca637e1
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/ps/part_ps/test_ps_embedding_heterogeneous_conv2d_adam.py
View File

@ -173,7 +173,9 @@ class NetFactory:
def part_cmp(self):
part_ps = self.part_ps_impl()
no_ps = self.no_ps_impl()
assert np.allclose(no_ps, part_ps, rtol=1.0e-5, atol=1.0e-5)
print(part_ps)
print(no_ps)
assert np.allclose(no_ps, part_ps, rtol=1.0e-4, atol=1.0e-4)
if __name__ == "__main__":