fix round op typos

This commit is contained in:
xcnick 2021-05-06 22:44:21 +08:00
parent c92d4f36aa
commit 0e61cabb68
2 changed files with 5 additions and 2 deletions

View File

@ -153,6 +153,9 @@ def test_rint():
np.testing.assert_almost_equal(output.asnumpy(), expect_output)
@pytest.mark.level0
@pytest.mark.platform_x86_cpu
@pytest.mark.env_onecard
def test_round():
net = RoundNet()

View File

@ -49,12 +49,12 @@ def generate_testcases(nptype):
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_sign_float32():
def test_round_float32():
generate_testcases(np.float32)
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_sign_float16():
def test_round_float16():
generate_testcases(np.float16)