未验证 提交 5a5794b3 编写于 作者: H HappyAngel 提交者: GitHub

[cherry-pick] fix xiaodu run kernnel in a53 problem. test=develop (#4445)

* fix xiaodu run kernnel in a53 problem. test=develop

* fix Mac build. test=develop
上级 d13c94d9
......@@ -355,7 +355,7 @@ void sgemm_prepack(bool is_transB,
(has_act == false) ||
(has_act == true && act_type == lite_api::ActivationType::kRelu);
bool has_beta = fabsf(beta) > 1e-8f ? true : false;
bool a53_sgemm = act_flag && !has_beta;
bool a53_sgemm = act_flag && !has_beta && ctx->has_a53_valid();
if (a53_sgemm) {
sgemm_prepacked_6x8_a53(is_transB,
M,
......
......@@ -217,6 +217,7 @@ class Context<TargetType::kARM> {
int llc_size() const { return DeviceInfo::Global().llc_size(); }
bool has_dot() const { return DeviceInfo::Global().has_dot(); }
bool has_fp16() const { return DeviceInfo::Global().has_fp16(); }
bool has_a53_valid() const { return DeviceInfo::Global().set_a53_valid(); }
template <typename T>
T* workspace_data() {
......
......@@ -1009,6 +1009,20 @@ void DeviceInfo::RequestPowerRandLowMode(int shift_num, int thread_num) {
}
}
bool DeviceInfo::set_a53_valid() {
std::string dev_name = "null";
#ifdef LITE_WITH_LINUX
dev_name = get_cpu_name();
#endif
// xiaodu device_name
if (dev_name.find("MT8765WA") != std::string::npos ||
dev_name.find("MT8167S") != std::string::npos) {
return false;
} else {
return true;
}
}
int DeviceInfo::Setup() {
core_num_ = get_cpu_num();
mem_size_ = get_mem_size();
......
......@@ -56,7 +56,7 @@ class DeviceInfo {
}
int Setup();
bool set_a53_valid();
void SetRunMode(lite_api::PowerMode mode, int thread_num);
void SetCache(int l1size, int l2size, int l3size);
void SetArch(ARMArch arch) { arch_ = arch; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册