未验证 提交 a9d17eef 编写于 作者: X xiaogang 提交者: GitHub

[ARM] fix 5x5s2_int8 compute bug, test=develop (#3079)

上级 1ae03f55
...@@ -102,7 +102,7 @@ void conv_depthwise_5x5s2_int8(Dtype* dout, ...@@ -102,7 +102,7 @@ void conv_depthwise_5x5s2_int8(Dtype* dout,
if (h + hout_r_block > hout) { if (h + hout_r_block > hout) {
h_kernel = hout - h; h_kernel = hout - h;
} }
int hs = h - padh; int hs = h * 2 - padh;
int he = hs + h_kernel * 2 + 3; int he = hs + h_kernel * 2 + 3;
#pragma omp parallel for num_threads(threads) #pragma omp parallel for num_threads(threads)
......
...@@ -502,7 +502,7 @@ TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) { ...@@ -502,7 +502,7 @@ TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) {
std::vector<DDim> dims; std::vector<DDim> dims;
DDim weights_dim({c, 1, 5, 5}); DDim weights_dim({c, 1, 5, 5});
for (auto& batch : {1, 2}) { for (auto& batch : {1, 2}) {
for (auto& h : {1, 3, 15, 33}) { for (auto& h : {1, 3, 15, 33, 112, 224}) {
dims.push_back(DDim({batch, c, h, h})); dims.push_back(DDim({batch, c, h, h}));
} }
} }
...@@ -514,7 +514,7 @@ TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) { ...@@ -514,7 +514,7 @@ TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) {
{1, 1}, {1, 1},
flag_bias, flag_bias,
flag_relu, flag_relu,
{4}, {1, 4},
{FLAGS_power_mode}); {FLAGS_power_mode});
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册