24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_OMPSCHEDULER_H
25 #define ARM_COMPUTE_OMPSCHEDULER_H
78 void run_workloads(std::vector<Workload> &workloads)
override;
81 unsigned int _num_threads;
void set_num_threads(unsigned int num_threads) override
Sets the number of threads the scheduler will use to run the kernels.
unsigned int num_threads() const override
Returns the number of threads that the OMPScheduler has in its pool.
void schedule_op(ICPPKernel *kernel, const Hints &hints, const Window &window, ITensorPack &tensors) override
Multithread the execution of the passed kernel if possible.
Common interface for all kernels implemented in C++.
Scheduler interface to run kernels.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
void schedule(ICPPKernel *kernel, const Hints &hints) override
Multithread the execution of the passed kernel if possible.
Pool of threads to automatically split a kernel's execution among several threads.
OMPScheduler()
Constructor.