diff options
Diffstat (limited to 'src/cpu/kernels/CpuActivationKernel.cpp')
-rw-r--r-- | src/cpu/kernels/CpuActivationKernel.cpp | 6 |
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) |