diff options
Diffstat (limited to 'src/core')
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/gemm_quint8.cpp | 4 | ||||
-rw-r--r-- | src/core/NEON/kernels/arm_gemm/quantized.cpp | 3 |
2 files changed, 7 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/gemm_quint8.cpp b/src/core/NEON/kernels/arm_gemm/gemm_quint8.cpp index 2c4af7a2b3..a402332552 100644 --- a/src/core/NEON/kernels/arm_gemm/gemm_quint8.cpp +++ b/src/core/NEON/kernels/arm_gemm/gemm_quint8.cpp @@ -21,6 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#ifdef __aarch64__ + #include "arm_gemm.hpp" #include "kernels/a64_hybrid_u8u32_dot_16x4.hpp" @@ -99,3 +101,5 @@ template KernelDescription get_gemm_method<uint8_t, uint8_t, ARequantizeLayer32> template std::vector<KernelDescription> get_compatible_kernels<uint8_t, uint8_t, ARequantizeLayer32>(const GemmArgs<uint8_t> &args, const ARequantizeLayer32 &os); } // namespace arm_gemm + +#endif // __aarch64__ diff --git a/src/core/NEON/kernels/arm_gemm/quantized.cpp b/src/core/NEON/kernels/arm_gemm/quantized.cpp index dd4eb31ea3..a51ed3f82a 100644 --- a/src/core/NEON/kernels/arm_gemm/quantized.cpp +++ b/src/core/NEON/kernels/arm_gemm/quantized.cpp @@ -21,6 +21,7 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#ifdef __aarch64__ #include "arm_gemm.hpp" @@ -767,3 +768,5 @@ template void compute_col_sums(const ARequantizeLayer32 &qp, unsigned int width, template void compute_col_sums(const ARequantizeLayer32 &qp, unsigned int width, unsigned int height, const uint8_t *input, unsigned int in_stride, int32_t *col_bias, unsigned int depth, unsigned int first_col); } // namespace arm_gemm + +#endif // __aarch64__ |