From 73bb6b7ad80801e56633ad4ea12b0404b586a979 Mon Sep 17 00:00:00 2001 From: Fadi Arafeh Date: Thu, 6 Oct 2022 16:20:14 +0000 Subject: ONCPUML-1072: Tuned MWS values (for N1, V1) for binary operators used by oneDNN Added approximate values for MWS for the following binary operators: Add, Sub, Mul, Min, Max, Div Change-Id: I5c4c75511129982a3f44c038ee272f09598469de Signed-off-by: Fadi Arafeh Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/c/VisualCompute/ComputeLibrary/+/459609 Tested-by: bsgcomp Reviewed-by: Viet-Hoa Do Comments-Addressed: bsgcomp Signed-off-by: fadara01 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8392 Reviewed-by: Gunes Bayir Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Benchmark: Arm Jenkins --- src/cpu/kernels/CpuMulKernel.h | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/cpu/kernels/CpuMulKernel.h') diff --git a/src/cpu/kernels/CpuMulKernel.h b/src/cpu/kernels/CpuMulKernel.h index c92e1efdf4..73ffc0dd2b 100644 --- a/src/cpu/kernels/CpuMulKernel.h +++ b/src/cpu/kernels/CpuMulKernel.h @@ -81,6 +81,15 @@ public: const char *name() const override; size_t get_mws(const CPUInfo &platform, size_t thread_count) const override; + /** Return minimum workload size of the relevant kernel + * + * @param[in] platform The CPU platform used to create the context. + * @param[in] thread_count Number of threads in the execution. + * + * @return[out] mws Minimum workload size for requested configuration. + */ + size_t get_mws(const CPUInfo &platform, size_t thread_count) const override; + /** Get the preferred dimension in which the scheduler splits the work into multiple jobs. * * @return The split dimension hint. -- cgit v1.2.1