diff options
Diffstat (limited to 'src/cpu/kernels')
-rw-r--r-- | src/cpu/kernels/CpuActivationKernel.cpp | 6 | ||||
-rw-r--r-- | src/cpu/kernels/CpuAddKernel.cpp | 6 | ||||
-rw-r--r-- | src/cpu/kernels/CpuIm2ColKernel.cpp | 6 | ||||
-rw-r--r-- | src/cpu/kernels/CpuReshapeKernel.cpp | 6 | ||||
-rw-r--r-- | src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h | 6 | ||||
-rw-r--r-- | src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp | 6 |
6 files changed, 30 insertions, 6 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) diff --git a/src/cpu/kernels/CpuAddKernel.cpp b/src/cpu/kernels/CpuAddKernel.cpp index 0c3540f0d4..f3ee032ec5 100644 --- a/src/cpu/kernels/CpuAddKernel.cpp +++ b/src/cpu/kernels/CpuAddKernel.cpp @@ -298,10 +298,14 @@ size_t CpuAddKernel::get_mws(const CPUInfo &platform, size_t thread_count) const { return 10240; } - else + else if (platform.get_cpu_model() == CPUModel::A76) { return 9216; } + else + { + return ICPPKernel::default_mws; + } } } // namespace kernels diff --git a/src/cpu/kernels/CpuIm2ColKernel.cpp b/src/cpu/kernels/CpuIm2ColKernel.cpp index ecd1748a44..58890fe5a2 100644 --- a/src/cpu/kernels/CpuIm2ColKernel.cpp +++ b/src/cpu/kernels/CpuIm2ColKernel.cpp @@ -452,10 +452,14 @@ size_t CpuIm2ColKernel::get_mws(const CPUInfo &platform, size_t thread_count) co { return 10240; } - else + else if (platform.get_cpu_model() == CPUModel::A76) { return 9216; } + else + { + return ICPPKernel::default_mws; + } } } // namespace kernels } // namespace cpu diff --git a/src/cpu/kernels/CpuReshapeKernel.cpp b/src/cpu/kernels/CpuReshapeKernel.cpp index e19707dd0c..49cb91b330 100644 --- a/src/cpu/kernels/CpuReshapeKernel.cpp +++ b/src/cpu/kernels/CpuReshapeKernel.cpp @@ -143,10 +143,14 @@ size_t CpuReshapeKernel::get_mws(const CPUInfo &platform, size_t thread_count) c { return 10240; } - else + else if (platform.get_cpu_model() == CPUModel::A76) { return 9216; } + else + { + return ICPPKernel::default_mws; + } } } // namespace kernels diff --git a/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h b/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h index 47548b2538..212fd79306 100644 --- a/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h +++ b/src/cpu/kernels/assembly/CpuGemmAssemblyWrapperKernel.h @@ -130,10 +130,14 @@ public: { return 3072; } - else + else if (platform.get_cpu_model() == CPUModel::A76) { return 4096; } + else + { + return ICPPKernel::default_mws; + } } private: diff --git a/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp b/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp index 934e38b054..6c235df59f 100644 --- a/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp +++ b/src/cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp @@ -363,10 +363,14 @@ size_t CpuDepthwiseConv2dAssemblyWrapperKernel::get_mws(const CPUInfo &platform, { return 10240; } - else + else if (platform.get_cpu_model() == CPUModel::A76) { return 9216; } + else + { + return ICPPKernel::default_mws; + } } } // namespace kernels } // namespace cpu |