diff options
| author | dskhudia <dskhudia@fb.com> | 2018-11-06 00:17:52 +0300 |
|---|---|---|
| committer | dskhudia <dskhudia@fb.com> | 2018-11-06 00:17:52 +0300 |
| commit | b96bc0bf311f7abdc83ffd3af0a485b4aef53f7c (patch) | |
| tree | 2a6c276d20753abe94c526aab7b109305e3d1d78 /src/Fbgemm.cc | |
| parent | 14adee1ac506e067489406af689ae9b73fb581bd (diff) | |
generalized conv_param_t and download third party libraries in build dir
Diffstat (limited to 'src/Fbgemm.cc')
| -rw-r--r-- | src/Fbgemm.cc | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/src/Fbgemm.cc b/src/Fbgemm.cc index f3bac97..9195a05 100644 --- a/src/Fbgemm.cc +++ b/src/Fbgemm.cc @@ -236,6 +236,16 @@ template void fbgemmPacked( int num_threads); template void fbgemmPacked( + PackMatrix<PackAWithIm2Col<uint8_t, int32_t, 3>, uint8_t, int32_t>& packA, + PackMatrix<PackBMatrix<int8_t, int32_t>, int8_t, int32_t>& packB, + int32_t* C, + int32_t* C_buffer, + uint32_t ldc, + const memCopy<>& outProcess, + int thread_id, + int num_threads); + +template void fbgemmPacked( PackMatrix<PackAWithQuantRowOffset<uint8_t, int32_t>, uint8_t, int32_t>& packA, PackMatrix<PackBMatrix<int8_t, int32_t>, int8_t, int32_t>& packB, @@ -341,6 +351,16 @@ template void fbgemmPacked( int num_threads); template void fbgemmPacked( + PackMatrix<PackAWithIm2Col<uint8_t, int16_t, 3>, uint8_t, int16_t>& packA, + PackMatrix<PackBMatrix<int8_t, int16_t>, int8_t, int16_t>& packB, + int32_t* C, + int32_t* C_buffer, + uint32_t ldc, + const memCopy<>& outProcess, + int thread_id, + int num_threads); + +template void fbgemmPacked( PackMatrix<PackAMatrix<uint8_t, int16_t>, uint8_t, int16_t>& packA, PackMatrix<PackBMatrix<int8_t, int16_t>, int8_t, int16_t>& packB, int32_t* C, |