diff options
Diffstat (limited to 'src/cpu/kernels/CpuMulKernel.h')
-rw-r--r-- | src/cpu/kernels/CpuMulKernel.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/cpu/kernels/CpuMulKernel.h b/src/cpu/kernels/CpuMulKernel.h index 85fcf88a96..5727b9d012 100644 --- a/src/cpu/kernels/CpuMulKernel.h +++ b/src/cpu/kernels/CpuMulKernel.h @@ -80,6 +80,15 @@ public: void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override; const char *name() const override; + /** Get the preferred dimension in which the scheduler splits the work into multiple jobs. + * + * @return The split dimension hint. + */ + size_t get_split_dimension_hint() const + { + return _split_dimension; + } + private: /** Common signature for all the specialised multiplication functions with integer scaling factor * @@ -115,6 +124,7 @@ private: MulFunctionQuantized *_func_quantized{ nullptr }; float _scale{ 0 }; int _scale_exponent{ 0 }; + size_t _split_dimension{ Window::DimY }; }; /** Interface for the complex pixelwise multiplication kernel. */ |