!35737 takedown scatter_nd_update and sparse_segment_mean testcases

Merge pull request !35737 from yanghaoran/takedown_testcases
This commit is contained in:
i-robot 2022-06-10 06:44:02 +00:00 committed by Gitee
commit 187b902cf3
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions

View File

@ -156,7 +156,7 @@ def test_op4(dtype):
assert np.allclose(out.asnumpy(), np.array(expect, dtype=dtype))
@pytest.mark.level0
@pytest.mark.level1
@pytest.mark.platform_x86_cpu
@pytest.mark.env_onecard
@pytest.mark.parametrize('dtype', [np.float32, np.float64])

View File

@ -42,7 +42,7 @@ def sparse_segment_mean_numpy(x, indices, segment_ids):
return output
@pytest.mark.level0
@pytest.mark.level1
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
@pytest.mark.parametrize('data_type, index_type, error', [(np.float32, np.int32, 3), (np.float64, np.int64, 5)])