From 428a0b6cede232eb5c4e9c3bbd8e9d74d8e34500 Mon Sep 17 00:00:00 2001
From: Jianyu Huang <jianyuhuang@fb.com>
Date: Thu, 8 Nov 2018 11:09:04 -0800
Subject: Sync with internal copy: Asymmetric padding; fbgemm2 -> fbgemm

---
 src/ExecuteKernelU8S8.cc | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

(limited to 'src/ExecuteKernelU8S8.cc')

diff --git a/src/ExecuteKernelU8S8.cc b/src/ExecuteKernelU8S8.cc
index e091a87..b3f8c15 100644
--- a/src/ExecuteKernelU8S8.cc
+++ b/src/ExecuteKernelU8S8.cc
@@ -14,7 +14,7 @@ double kernel_time = 0.0;
 double postprocessing_time = 0.0;
 #endif
 
-namespace fbgemm2 {
+namespace fbgemm {
 
 template <typename packingAMatrix, typename cT, typename processOutputType>
 ExecuteKernel<
@@ -327,6 +327,18 @@ template class ExecuteKernel<
     int32_t,
     memCopy<>>;
 
+template class ExecuteKernel<
+    PackAWithIm2Col<uint8_t, int16_t>,
+    PackBMatrix<int8_t, int16_t>,
+    uint8_t,
+    ReQuantizeOutput<false>>;
+
+template class ExecuteKernel<
+    PackAWithIm2Col<uint8_t, int16_t, 3>,
+    PackBMatrix<int8_t, int16_t>,
+    uint8_t,
+    ReQuantizeOutput<false>>;
+
 template class ExecuteKernel<
     PackAWithRowOffset<uint8_t, int32_t>,
     PackBMatrix<int8_t, int32_t>,
@@ -345,6 +357,18 @@ template class ExecuteKernel<
     int32_t,
     memCopy<>>;
 
+template class ExecuteKernel<
+    PackAWithIm2Col<uint8_t, int32_t>,
+    PackBMatrix<int8_t, int32_t>,
+    uint8_t,
+    ReQuantizeOutput<false>>;
+
+template class ExecuteKernel<
+    PackAWithIm2Col<uint8_t, int32_t, 3>,
+    PackBMatrix<int8_t, int32_t>,
+    uint8_t,
+    ReQuantizeOutput<false>>;
+
 template class ExecuteKernel<
     PackAWithQuantRowOffset<uint8_t, int32_t>,
     PackBMatrix<int8_t, int32_t>,
@@ -363,4 +387,4 @@ template class ExecuteKernel<
     int32_t,
     DoNothing<int32_t, int32_t>>;
 
-} // namespace fbgemm2
+} // namespace fbgemm
-- 
cgit v1.2.3