Initial commit

1 files changed, 73 insertions, 0 deletions

diff --git a/src/ExecuteKernelU8S8.h b/src/ExecuteKernelU8S8.h
new file mode 100644
index 0000000..0bd7fc5
--- /dev/null
+++ b/src/ExecuteKernelU8S8.h
@@ -0,0 +1,73 @@
+/*
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ * All rights reserved.
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+#pragma once
+#include "ExecuteKernel.h"
+
+namespace fbgemm2 {
+
+/**
+ * @brief Execute Engine of uint 8 and int8 matrix
+ * multiplication for the macro-kernel and output processing. ExecuteKernel is a
+ * derived class of CodeGenBase.
+ */
+template <typename packingAMatrix, typename cT, typename processOutputType>
+class ExecuteKernel<
+    packingAMatrix,
+    PackBMatrix<int8_t, typename packingAMatrix::accType>,
+    cT,
+    processOutputType>
+    : public CodeGenBase<
+          uint8_t,
+          int8_t,
+          int32_t,
+          typename packingAMatrix::accType> {
+ public:
+  using BaseType =
+      CodeGenBase<uint8_t, int8_t, int32_t, typename packingAMatrix::accType>;
+  /**
+   * @brief Constructor for initializing the parameters for macro-kernel and
+   * output processing type.
+   */
+  ExecuteKernel(
+      PackMatrix<packingAMatrix, uint8_t, typename packingAMatrix::accType>&
+          packA,
+      PackMatrix<
+          PackBMatrix<int8_t, typename packingAMatrix::accType>,
+          int8_t,
+          typename packingAMatrix::accType>& packB,
+      int32_t kBlock,
+      cT* matC,
+      int32_t* C_buffer,
+      int32_t ldc,
+      const processOutputType& outputProcess);
+  void execute(int kBlock);
+
+  ~ExecuteKernel() {
+    delete[] C_tile_;
+  }
+
+ private:
+  PackMatrix<packingAMatrix, uint8_t, typename packingAMatrix::accType>&
+      packedA_; ///< Packed uint8 block of matrix A.
+  PackMatrix<
+      PackBMatrix<int8_t, typename packingAMatrix::accType>,
+      int8_t,
+      typename packingAMatrix::accType>&
+      packedB_; ///< Packed int8 matrix B.
+  int32_t kBlock_; ///< Block ID in the k dimension.
+  cT* matC_; ///< Output for matrix C.
+  int32_t* C_buffer_; ///< the accumulation buffer for matrix C.
+  int32_t ldc_; ///< the leading dimension of matrix C.
+  const processOutputType& outputProcess_; ///< output processing function for
+                                           ///< matrix C in the macro-kernel.
+  int32_t* C_tile_; ///< buffer for the last N block when NCB is not an exact
+                    ///< multiple of N.
+  int mbSize_; ///< block size in the m dimension.
+  int nbSize_; ///< block size in the n dimension.
+};
+
+} // namespace fbgemm2