diff options
author | Kenneth Heafield <github@kheafield.com> | 2018-06-28 02:54:55 +0300 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2018-06-28 02:54:55 +0300 |
commit | dd41e65676a5fc00aa396cf721068d5f8cd58ae9 (patch) | |
tree | c8605011a45a89644265c19d335553f956403d6b /sse2_gemm.cc | |
parent | d4459a2901c1df058e2cdd53be37c3e1dec8caca (diff) |
Untested SelectColumnsB function
Diffstat (limited to 'sse2_gemm.cc')
-rw-r--r-- | sse2_gemm.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/sse2_gemm.cc b/sse2_gemm.cc index 05a4080..3e0a0d8 100644 --- a/sse2_gemm.cc +++ b/sse2_gemm.cc @@ -61,6 +61,10 @@ void SSE2_16bit::PrepareB(const float *input, int16_t *output, float quant_mult, PrepareBFor16(input, output, QuantizeTile16(quant_mult), rows, cols); } +void SSE2_16bit::SelectColumnsB(const int16_t *input, int16_t *output, int rows, const int *cols_begin, const int *cols_end) { + SelectColumnsOfB((const __m128i*)input, (__m128i*)output, rows * 2, cols_begin, cols_end); +} + void SSE2_16bit::Multiply(const int16_t *A, const int16_t *B, float *C, float unquant_mult, int A_rows, int width, int B_cols) { Multiply16<__m128i, __m128>(A, B, C, unquant_mult, A_rows, width, B_cols); } |