提交 644d97f5 编写于 作者: C chenjiaoAngel

fic format. test=develop

上级 e5066279
......@@ -4333,8 +4333,19 @@ void gemm_prepack_int8(const int8_t* A_packed,
ctx);
}
#else
gemm_prepack_oth_int8<int8_t>(
A_packed, B, bias, C, M, N, K, is_bias, flag_act, is_transB, scale, alpha, ctx);
gemm_prepack_oth_int8<int8_t>(A_packed,
B,
bias,
C,
M,
N,
K,
is_bias,
flag_act,
is_transB,
scale,
alpha,
ctx);
#endif
}
......
......@@ -65,9 +65,6 @@ inline void write_gemv_out(const int* in,
vout1 = vmaxq_f32(vout1, vzero);
vout0 = vminq_f32(vout0, vsix);
vout1 = vminq_f32(vout1, vsix);
}
vout0 = vmaxq_f32(vout0, vzero);
vout1 = vmaxq_f32(vout1, vzero);
} else if (act == lite_api::ActivationType::kLeakyRelu) {
float32x4_t valpha = vdupq_n_f32(alpha);
uint32x4_t maska = vcgeq_f32(vout0, vzero);
......@@ -77,6 +74,7 @@ inline void write_gemv_out(const int* in,
vout0 = vbslq_f32(maska, vout0, suma);
vout1 = vbslq_f32(maskb, vout1, sumb);
}
}
vst1q_f32(out, vout0);
vst1q_f32(out + 4, vout1);
bias += 8;
......
......@@ -23,6 +23,7 @@
#include "lite/core/profile/timer.h"
#include "lite/core/tensor.h"
#include "lite/tests/utils/tensor_utils.h"
#include "lite/operators/op_params.h"
typedef paddle::lite::Tensor Tensor;
using paddle::lite::profile::Timer;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册