aboutsummaryrefslogtreecommitdiff
path: root/src/cpu/kernels/CpuActivationKernel.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/cpu/kernels/CpuActivationKernel.cpp')
-rw-r--r--src/cpu/kernels/CpuActivationKernel.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuActivationKernel.cpp b/src/cpu/kernels/CpuActivationKernel.cpp
index 4a6468d022..aed73d1fec 100644
--- a/src/cpu/kernels/CpuActivationKernel.cpp
+++ b/src/cpu/kernels/CpuActivationKernel.cpp
@@ -238,10 +238,14 @@ size_t CpuActivationKernel::get_mws(const CPUInfo &platform, size_t thread_count
{
return 10240;
}
- else
+ else if (platform.get_cpu_model() == CPUModel::A76)
{
return 9216;
}
+ else
+ {
+ return ICPPKernel::default_mws;
+ }
}
void CpuActivationKernel::run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info)