_sgemm.hpp

Go to the documentation of this file.

/*************************************************************************
 *
 *  Project
 *                         _____ _____  __  __ _____
 *                        / ____|  __ \|  \/  |  __ \
 *  ___  _ __   ___ _ __ | |  __| |__) | \  / | |__) |
 * / _ \| '_ \ / _ \ '_ \| | |_ |  ___/| |\/| |  ___/
 *| (_) | |_) |  __/ | | | |__| | |    | |  | | |
 * \___/| .__/ \___|_| |_|\_____|_|    |_|  |_|_|
 *      | |
 *      |_|
 *
 * Copyright (C) Akiel Aries, <akiel@akiel.org>, et al.
 *
 * This software is licensed as described in the file LICENSE, which
 * you should have received as part of this distribution. The terms
 * among other details are referenced in the official documentation
 * seen here : https://akielaries.github.io/openGPMP/ along with
 * important files seen in this project.
 *
 * You may opt to use, copy, modify, merge, publish, distribute
 * and/or sell copies of the Software, and permit persons to whom
 * the Software is furnished to do so, under the terms of the
 * LICENSE file. As this is an Open Source effort, all implementations
 * must be of the same methodology.
 *
 *
 *
 * This software is distributed on an AS IS basis, WITHOUT
 * WARRANTY OF ANY KIND, either express or implied.
 *
 ************************************************************************/
#ifndef _SGEMM_HPP
#define _SGEMM_HPP
 
#define BLOCK_SZ_M 384  
#define BLOCK_SZ_K 384  
#define BLOCK_SZ_N 4096 
#define BLOCK_SZ_MR 4   
#define BLOCK_SZ_NR 4   
namespace gpmp {
namespace linalg {
 
class SGEMM {
  public:
    static float SGEMM_BUFF_A[BLOCK_SZ_M * BLOCK_SZ_K];
    static float SGEMM_BUFF_B[BLOCK_SZ_K * BLOCK_SZ_N];
    static float SGEMM_BUFF_C[BLOCK_SZ_MR * BLOCK_SZ_NR];
 
    void pack_micro_A(int k,
                      const float *A,
                      int incRowA,
                      int incColA,
                      float *buffer);
 
    void pack_buffer_A(int mc,
                       int kc,
                       const float *A,
                       int incRowA,
                       int incColA,
                       float *buffer);
 
    void pack_micro_B(int k,
                      const float *B,
                      int incRowB,
                      int incColB,
                      float *buffer);
 
    void pack_buffer_B(int kc,
                       int nc,
                       const float *B,
                       int incRowB,
                       int incColB,
                       float *buffer);
 
    void sgemm_micro_kernel(int kc,
                            float alpha,
                            const float *A,
                            const float *B,
                            float beta,
                            float *C,
                            int incRowC,
                            int incColC);
 
    void sgeaxpy(int m,
                 int n,
                 float alpha,
                 const float *X,
                 int incRowX,
                 int incColX,
                 float *Y,
                 int incRowY,
                 int incColY);
 
    void sgescal(int m, int n, float alpha, float *X, int incRowX, int incColX);
 
    void sgemm_macro_kernel(int mc,
                            int nc,
                            int kc,
                            float alpha,
                            float beta,
                            float *C,
                            int incRowC,
                            int incColC);
 
    void sgemm_nn(int m,
                  int n,
                  int k,
                  float alpha,
                  const float *A,
                  int incRowA,
                  int incColA,
                  const float *B,
                  int incRowB,
                  int incColB,
                  float beta,
                  float *C,
                  int incRowC,
                  int incColC);
};
 
} // namespace linalg
} // namespace gpmp
 
#endif