diff options
| author | Daya S Khudia <dskhudia@fb.com> | 2019-01-03 05:39:26 +0300 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-03 05:41:50 +0300 |
| commit | 79f1a3b5464352508141813633df168eb19b6fae (patch) | |
| tree | b09f76960d06d8d52807ce26b9db1762964cf7ea | |
| parent | 172048ae4c0ae44de941e2f42a4a87ac15690858 (diff) | |
fix shared lib build
Summary: Newly added benchmarks require certain functions to be exported
Reviewed By: jianyuh
Differential Revision: D13571151
fbshipit-source-id: c0785a5b8d254686cc2403c91c5379358fc2987b
| -rw-r--r-- | include/fbgemm/QuantUtilsAvx2.h | 2 | ||||
| -rw-r--r-- | src/OptimizedKernelsAvx2.h | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/include/fbgemm/QuantUtilsAvx2.h b/include/fbgemm/QuantUtilsAvx2.h index bd6e699..ec7c6a5 100644 --- a/include/fbgemm/QuantUtilsAvx2.h +++ b/include/fbgemm/QuantUtilsAvx2.h @@ -72,7 +72,7 @@ template < QuantizationGranularity Q_GRAN, bool HAS_BIAS, bool FUSE_RELU> -void requantizeOutputProcessingAvx2( +FBGEMM_API void requantizeOutputProcessingAvx2( std::uint8_t* out, const std::int32_t* inp, const block_type_t& block, diff --git a/src/OptimizedKernelsAvx2.h b/src/OptimizedKernelsAvx2.h index b74750e..36d7b2b 100644 --- a/src/OptimizedKernelsAvx2.h +++ b/src/OptimizedKernelsAvx2.h @@ -7,13 +7,14 @@ #pragma once #include <cstdint> // for std::int32_t +#include "fbgemm/FbgemmBuild.h" namespace fbgemm { /** * @brief Sum a given vector. */ -std::int32_t reduceAvx2(const std::uint8_t* A, int len); +FBGEMM_API std::int32_t reduceAvx2(const std::uint8_t* A, int len); /** * @brief Transpose 8 rows from source matrix. |