From a2f5d22cf2358995cababbea185a74cde506cc70 Mon Sep 17 00:00:00 2001
From: zhupengyang <zhupengyang@baidu.com>
Date: Tue, 25 Jun 2019 10:05:49 +0000
Subject: [PATCH] add conv_direct_int8 impl

---
 paddle/fluid/lite/arm/math/CMakeLists.txt     | 2 ++
 paddle/fluid/lite/kernels/arm/conv_compute.cc | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddle/fluid/lite/arm/math/CMakeLists.txt b/paddle/fluid/lite/arm/math/CMakeLists.txt
index 32f367f703..8f4fa4d9ec 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 811a4341f1..93bafd8c5b 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<Ptype_out>::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<Ptype_out>;
+    impl_ = new lite::arm::math::DirectConvInt8<Ptype_out>;
   } else {
     VLOG(3) << "GemmLikeConvInt8";
     impl_ = new lite::arm::math::GemmLikeConvInt8<Ptype_out>;
-- 
GitLab