未验证 提交 43e41d51 编写于 作者: J Jiaying Zhao 提交者: GitHub

open arm64 EXEC_GEMM_INT8 exec mode (#1710)

上级 abb30cba
...@@ -30,10 +30,10 @@ bool ConvKernel<CPU, float>::Init(ConvParam<CPU> *param) { ...@@ -30,10 +30,10 @@ bool ConvKernel<CPU, float>::Init(ConvParam<CPU> *param) {
template <> template <>
void ConvKernel<CPU, float>::Compute(const ConvParam<CPU> &param) { void ConvKernel<CPU, float>::Compute(const ConvParam<CPU> &param) {
switch (param.ExecMode()) { switch (param.ExecMode()) {
#ifndef __aarch64__
case ConvParam<CPU>::EXEC_GEMM_INT8: case ConvParam<CPU>::EXEC_GEMM_INT8:
GemmConv<int8_t, int32_t>(param); GemmConv<int8_t, int32_t>(param);
break; break;
#ifndef __aarch64__
case ConvParam<CPU>::EXEC_DEPTHWISE3x3_INT8: case ConvParam<CPU>::EXEC_DEPTHWISE3x3_INT8:
DepthwiseConv3x3<int8_t, int32_t>(param); DepthwiseConv3x3<int8_t, int32_t>(param);
break; break;
......
...@@ -263,8 +263,8 @@ template void DepthwiseConv3x3<float, float>(const ConvParam<CPU> &param); ...@@ -263,8 +263,8 @@ template void DepthwiseConv3x3<float, float>(const ConvParam<CPU> &param);
template void DepthwiseConv5x5<float, float>(const ConvParam<CPU> &param); template void DepthwiseConv5x5<float, float>(const ConvParam<CPU> &param);
template void SlidingwindowConv3x3<float, float>(const ConvParam<CPU> &param); template void SlidingwindowConv3x3<float, float>(const ConvParam<CPU> &param);
#ifndef __aarch64__
template void GemmConv<int8_t, int32_t>(const ConvParam<CPU> &param); template void GemmConv<int8_t, int32_t>(const ConvParam<CPU> &param);
#ifndef __aarch64__
template void DepthwiseConv3x3<int8_t, int32_t>(const ConvParam<CPU> &param); template void DepthwiseConv3x3<int8_t, int32_t>(const ConvParam<CPU> &param);
template void DepthwiseConv5x5<int8_t, int32_t>(const ConvParam<CPU> &param); template void DepthwiseConv5x5<int8_t, int32_t>(const ConvParam<CPU> &param);
#endif #endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册