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

fix ci error

This commit is contained in:
jiaorui 2022-06-10 12:08:52 +08:00
parent 93eaf5512e
commit ed237f96c5
1 changed files with 117 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

123
tests/st/ops/ascend/test_deformable_conv.py
View File

@ -23,7 +23,7 @@ from mindspore import Tensor
context.set_context(device_target="Ascend")
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_arm_ascend_training
@pytest.mark.env_onecard
@ -38,7 +38,7 @@ def test_deformable_conv2d():
stride_h, stride_w = 1, 1
pad_h, pad_w = 0, 0
dilation_h, dilation_w = 1, 1
# x shape [1, 8, 1, 2]
# x shape [1, 64, 1, 2]
x = np.array([[[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
@ -46,9 +46,65 @@ def test_deformable_conv2d():
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[1.41634355, -0.05626683]],
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[1.41634355, -0.05626683]],
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[1.41634355, -0.05626683]],
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[1.41634355, -0.05626683]],
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[1.41634355, -0.05626683]],
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[1.41634355, -0.05626683]],
[[-0.5475785, -2.9879797]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[1.41675785, -0.25626683]],
[[-0.41675785, 1.79979878]],
[[0.4543585, -0.9792279]],
[[-1.5435465, -0.79898799]],
[[1.41634355, -0.05626683]],
[[-0.41675785, -0.05626683]],
[[0.5645654, 0.4656564]]]]).astype(np.float32)
x = Tensor(x, mstype.float32)
# weight shape [1, 8, 1, 1]
# weight shape [1, 64, 1, 1]
weight = np.array([[[[-2.1361961]],
[[-1.767576]],
[[0.454354]],
@ -56,6 +112,62 @@ def test_deformable_conv2d():
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-1.767576]],
[[0.454354]],
[[-2.1361961]],
[[0.56756]],
[[-0.80899]],
[[0.767676]],
[[-0.80899]],
[[0.767676]],
[[-1.675756]]]]).astype(np.float32)
weight = Tensor(weight, mstype.float32)
# offsets shape [1, 3, 1, 2]
@ -64,8 +176,7 @@ def test_deformable_conv2d():
[[-1.2452881, -1.0579522]]]]).astype(np.float32)
offsets = Tensor(offsets, mstype.float32)
out = deformable_conv2d(x, weight, offsets, (kh, kw), (1, 1, stride_h, stride_w), (pad_h, pad_h, pad_w, pad_w),
data_format="NCHW", dilations=(1, 1, dilation_h, dilation_w),
deformable_groups=deformable_group)
dilations=(1, 1, dilation_h, dilation_w), deformable_groups=deformable_group)
# expected output: [1, 1, 1, 2]
expected = np.array([[[[0.05569747, 0.82974637]]]]).astype(np.float32)
expected = np.array([[[[-0.45592457, 7.7579584]]]]).astype(np.float32)
assert np.allclose(out.asnumpy(), expected, 0.0001, 0.0001)