diff options
| author | Nikolay Bogoychev <nheart@gmail.com> | 2019-04-17 02:03:20 +0300 |
|---|---|---|
| committer | Nikolay Bogoychev <nheart@gmail.com> | 2019-04-17 02:03:20 +0300 |
| commit | 4d719af5632ba5d5f5861389fab382c4bdf3056b (patch) | |
| tree | b4d1d7bde619b652b3daba9a78b046dd478e8942 /ssse3_gemm.h | |
| parent | 48533fa72adb77d551030d732512740cccf8c70b (diff) | |
SelectColumnOfB is now #define
Diffstat (limited to 'ssse3_gemm.h')
| -rw-r--r-- | ssse3_gemm.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ssse3_gemm.h b/ssse3_gemm.h index 66e6f5a..f0d76b7 100644 --- a/ssse3_gemm.h +++ b/ssse3_gemm.h @@ -17,6 +17,8 @@ SSSE3 inline __m128i QuantizerGrab(const float *input, const __m128 quant_mult_r return _mm_cvtps_epi32(_mm_mul_ps(*reinterpret_cast<const __m128*>(input), quant_mult_reg)); } +SELECT_COL_B_DEF(SSSE3, __m128i) + class QuantizeTile8 { public: typedef __m128i Integer; @@ -93,7 +95,7 @@ struct SSSE3_8bit { PREPARE_B_8_DEF(SSSE3, ssse3::QuantizeTile8) SSSE3 static void SelectColumnsB(const int8_t *input, int8_t *output, Index rows, const Index *cols_begin, const Index *cols_end) { - SelectColumnsOfB((const __m128i*)input, (__m128i*)output, rows, cols_begin, cols_end); + ssse3::SelectColumnsOfB((const __m128i*)input, (__m128i*)output, rows, cols_begin, cols_end); } SSSE3 static void Multiply(const int8_t *A, const int8_t *B, float *C, float unquant_mult, Index A_rows, Index width, Index B_cols) { |