diff --git a/paddle/fluid/lite/arm/math/CMakeLists.txt b/paddle/fluid/lite/arm/math/CMakeLists.txt index 32f367f703e6cdf1484a2bf2e53edcf38f879357..8f4fa4d9ecc4bf458aeab75ea8c46853f17f683c 100644 --- a/paddle/fluid/lite/arm/math/CMakeLists.txt +++ b/paddle/fluid/lite/arm/math/CMakeLists.txt @@ -37,6 +37,8 @@ cc_library(math_arm SRCS dropout.cc gemm_prepacked_int8.cc gemv_arm_int8.cc + conv3x3s1_direct_int8.cc + conv3x3s2_direct_int8.cc DEPS ${lite_kernel_deps} eigen3 framework_proto_lite) # TODO(TJ): fix me do not deps proto diff --git a/paddle/fluid/lite/kernels/arm/conv_compute.cc b/paddle/fluid/lite/kernels/arm/conv_compute.cc index 811a4341f16bf72a3ec76bb88e2259ee9ca546bc..93bafd8c5bfbf169378ea94fa2b03108067d216c 100644 --- a/paddle/fluid/lite/kernels/arm/conv_compute.cc +++ b/paddle/fluid/lite/kernels/arm/conv_compute.cc @@ -127,7 +127,7 @@ void ConvComputeInt8::PrepareForRun() { VLOG(3) << "DepthwiseConv Int8"; } else if (param.groups == 1 && kw == 3 && (sw == 1 || sw == 2) && kps_equal && no_dilation) { - // impl_ = new lite::arm::math::DirectConv; + impl_ = new lite::arm::math::DirectConvInt8; } else { VLOG(3) << "GemmLikeConvInt8"; impl_ = new lite::arm::math::GemmLikeConvInt8;