未验证 提交 4d495329 编写于 作者: X xiebaiyuan 提交者: GitHub

remove 256 enforce of in (#3477)

* fix bilinear_interp build err on mali,test=develop

* remove 256 enforce of in.. test=develop

* remove 256 enforce of in ,code style.. test=develop
上级 f882f9fd
......@@ -30,8 +30,6 @@ bool InstanceNormKernel<GPU_CL, float>::Init(InstanceNormParam<GPU_CL> *param) {
build_options = "-DLOCAL_MEM_128";
} else if (h == 64) {
build_options = "-DLOCAL_MEM_64";
} else if (h > 256) {
PADDLE_MOBILE_THROW_EXCEPTION("instance norm unsupported input height");
}
this->cl_helper_.AddKernel("instancenorm", "instancenorm_kernel.cl",
build_options);
......
......@@ -26,13 +26,11 @@ bool InstanceNormReluKernel<GPU_CL, float>::Init(
FusionInstanceNormReluParam<GPU_CL> *param) {
auto &dims = param->Out()->dims();
const int h = dims[2];
std::string build_options = "-DRELU";
std::string build_options = " -DRELU";
if (h == 128) {
build_options += " -DLOCAL_MEM_128";
} else if (h == 64) {
build_options += " -DLOCAL_MEM_64";
} else if (h > 256) {
PADDLE_MOBILE_THROW_EXCEPTION("instance norm unsupported input height");
}
this->cl_helper_.AddKernel("instancenorm", "instancenorm_kernel.cl",
build_options);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册