diff --git a/paddle/fluid/lite/kernels/arm/pool_compute_test.cc b/paddle/fluid/lite/kernels/arm/pool_compute_test.cc index b024ccef9d526d56bcf52c1600940ff0804eaf1f..844976b963b547d7458f73bec4777281f26bc52a 100644 --- a/paddle/fluid/lite/kernels/arm/pool_compute_test.cc +++ b/paddle/fluid/lite/kernels/arm/pool_compute_test.cc @@ -272,4 +272,4 @@ TEST(pool, retrive_op) { } // namespace lite } // namespace paddle -USE_LITE_KERNEL(pool, kARM, kFloat, kNCHW, def); +USE_LITE_KERNEL(pool2d, kARM, kFloat, kNCHW, def); diff --git a/paddle/fluid/lite/operators/pool_op_test.cc b/paddle/fluid/lite/operators/pool_op_test.cc index 9ab2865f1d04f2ca173b9d2f5f7d9e457f6754e8..01cb69d30762746fb544808fe72de019a3273025 100644 --- a/paddle/fluid/lite/operators/pool_op_test.cc +++ b/paddle/fluid/lite/operators/pool_op_test.cc @@ -86,5 +86,5 @@ TEST(pool_op_lite, test) { } // namespace paddle #ifdef LITE_WITH_ARM -USE_LITE_KERNEL(pool, kARM, kFloat, kNCHW, def); +USE_LITE_KERNEL(pool2d, kARM, kFloat, kNCHW, def); #endif