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. */