未验证 提交 3c1d9817 编写于 作者: H HappyAngel 提交者: GitHub

[arm] fix 3x3s2p0 avg pooling crash (#3754)

* fix pooling overflow, test=develop

* fix conflict test=develop
上级 5b091f91
......@@ -2701,7 +2701,9 @@ void pooling3x3s2p0_avg(const float* din,
w_unroll_size -= 1;
w_unroll_remian = wout - w_unroll_size * 4;
}
// do overflow process
w_unroll_size -= 1;
w_unroll_remian += 4;
auto zero_ptr =
static_cast<float*>(TargetMalloc(TARGET(kARM), win * sizeof(float)));
memset(zero_ptr, 0, win * sizeof(float));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册