diff options
Diffstat (limited to 'arm_compute/core/CPP/ICPPKernel.h')
-rw-r--r-- | arm_compute/core/CPP/ICPPKernel.h | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/arm_compute/core/CPP/ICPPKernel.h b/arm_compute/core/CPP/ICPPKernel.h index ab369ffe1d..af4a896a6c 100644 --- a/arm_compute/core/CPP/ICPPKernel.h +++ b/arm_compute/core/CPP/ICPPKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016-2020 Arm Limited. + * Copyright (c) 2016-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -38,6 +38,9 @@ class ITensor; class ICPPKernel : public IKernel { public: + static constexpr size_t default_mws = 128; /* Default minimum workload size value */ + static constexpr size_t small_network_mws = 256; /* Default Minimum workload size value for small networks */ + /** Default destructor */ virtual ~ICPPKernel() = default; @@ -88,6 +91,20 @@ public: ARM_COMPUTE_UNUSED(tensors, window, info); } + /** 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 requsted configuration. + */ + virtual size_t get_mws(const CPUInfo &platform, size_t thread_count) const + { + ARM_COMPUTE_UNUSED(platform, thread_count); + + return default_mws; + } + /** Name of the kernel * * @return Kernel name |