diff options
Diffstat (limited to 'avx2_gemm.h')
-rw-r--r-- | avx2_gemm.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/avx2_gemm.h b/avx2_gemm.h index 23faa2a..31db7f6 100644 --- a/avx2_gemm.h +++ b/avx2_gemm.h @@ -20,6 +20,8 @@ struct AVX2_16bit { static void PrepareB(const float *input, int16_t *output, float quant_mult, int rows, int cols); + static void SelectColumnsB(const int16_t *input, int16_t *output, int rows, const int *cols_begin, const int *cols_end); + static void Multiply(const int16_t *A, const int16_t *B, float *C, float unquant_mult, int A_rows, int width, int B_cols); static const char *const kName; @@ -43,6 +45,8 @@ struct AVX2_8bit { static void PrepareB(const float *input, int8_t *output, float quant_mult, int rows, int cols); + static void SelectColumnsB(const int8_t *input, int8_t *output, int rows, const int *cols_begin, const int *cols_end); + static void Multiply(const int8_t *A, const int8_t *B, float *C, float unquant_mult, int A_rows, int width, int B_cols); static const char *const kName; |