diff --git a/src/operators/kernel/central-arm-func/pool_arm_func.h b/src/operators/kernel/central-arm-func/pool_arm_func.h index 95149732cab2b484bd35398c7b5ea3e51e9a2ea0..15a2ebc65ee8dc90e148dc13e44596d63abdf35c 100644 --- a/src/operators/kernel/central-arm-func/pool_arm_func.h +++ b/src/operators/kernel/central-arm-func/pool_arm_func.h @@ -96,7 +96,8 @@ void PoolCompute(const PoolParam ¶m) { paddings[1] == 0) { #if __ARM_NEON #if __aarch64__ - PoolBasic(pooling_type, ksize, strides, paddings, in_x, out); + PoolBasic(pooling_type, ksize, strides, paddings, in_x, + out); #else /// todo: fix bug in Pool2x2 if (pooling_type == "max") { @@ -106,7 +107,8 @@ void PoolCompute(const PoolParam ¶m) { } #endif #else - PoolBasic(pooling_type, ksize, strides, paddings, in_x, out); + PoolBasic(pooling_type, ksize, strides, paddings, in_x, + out); #endif // __ARM_NEON } else { diff --git a/test/operators/test_pool_op.cpp b/test/operators/test_pool_op.cpp index 48bf00c2ac910e4e75b53fb9ae184a222b60077f..5784ac065496ccfde73e516802fb2f79f622836f 100644 --- a/test/operators/test_pool_op.cpp +++ b/test/operators/test_pool_op.cpp @@ -202,6 +202,7 @@ int main(int argc, char *argv[]) { int in_channels = atoi(argv[1]); int in_height = atoi(argv[2]); int in_width = atoi(argv[3]); +#if __ARM_NEON // kernel = 3, pad = 1, stride = 1 LOG(paddle_mobile::kLOG_INFO) << "float, ceil_mode=false, pooling_type=max, kernel=3, pad=1, stride=1"; @@ -212,6 +213,7 @@ int main(int argc, char *argv[]) { << "float, ceil_mode=false, pooling_type=max, kernel=3, pad=0, stride=2"; paddle_mobile::TestPoolOp(in_channels, in_height, in_width); +#endif // kernel = 3, pad = 0, stride = 1 LOG(paddle_mobile::kLOG_INFO) << "int8_t, ceil_mode=false, pooling_type=max, kernel=3, pad=0, stride=1";