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

!49079 修改norm算子测试样例 

Merge pull request !49079 from hujingsong/master
This commit is contained in:
i-robot 2023-02-20 02:56:46 +00:00 committed by Gitee
parent 70248f06da ab916de938
commit 5446cd1d44
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 25 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tests/st/ops/test_func_norm.py
View File

@ -59,14 +59,6 @@ def test_norm_normal(mode):
expect_output4 = np.array(9.0)
assert np.allclose(output4.asnumpy(), expect_output4)
output5 = net(a, -float('inf'))
expect_output5 = np.array(0.0)
assert np.allclose(output5.asnumpy(), expect_output5)
output6 = net(b, -float('inf'))
expect_output6 = np.array(2.0)
assert np.allclose(output6.asnumpy(), expect_output6)
output7 = net(a, 1)
expect_output7 = np.array(20.0)
assert np.allclose(output7.asnumpy(), expect_output7)
@ -125,3 +117,28 @@ def test_norm_normal(mode):
output21 = net(d[0, :, :]).asnumpy(), net(d[1, :, :]).asnumpy()
expect_output21 = np.array([3.7416575, 11.224972])
assert np.allclose(output21, expect_output21)
@pytest.mark.level0
@pytest.mark.platform_x86_cpu
@pytest.mark.platform_arm_cpu
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
@pytest.mark.parametrize('mode', [ms.GRAPH_MODE, ms.PYNATIVE_MODE])
def test_norm_normal2(mode):
"""
Feature: norm
Description: Verify the result of norm
Expectation: success
"""
ms.set_context(mode=mode)
net = Net()
a = ops.arange(9, dtype=ms.float32) - 4
b = a.reshape((3, 3))
output5 = net(a, -float('inf'))
expect_output5 = np.array(0.0)
assert np.allclose(output5.asnumpy(), expect_output5)
output6 = net(b, -float('inf'))
expect_output6 = np.array(2.0)
assert np.allclose(output6.asnumpy(), expect_output6)