Public Member Functions
	ParallelScheduler (const BackendSet &backends)
	Constructs ParallelScheduler object.

void	assign (std::unique_ptr< IFunction > &&fn, const backend::Backend *backend)
	Assign a task to the given backend.

void	finish ()
	Block until all jobs are finished.

Detailed Description

Definition at line 32 of file ParallelScheduler.h.

Constructor & Destructor Documentation

onert::exec::ParallelScheduler::ParallelScheduler ( const BackendSet & backends )

Constructs ParallelScheduler object.

Parameters

backends Backend set

Definition at line 29 of file ParallelScheduler.cc.

{
  assert(!backends.empty());
 
  for (auto &&backend : backends)
  {
    _thread_pools[backend] = std::make_unique<ThreadPool>();
  }
}

void onert::exec::ParallelScheduler::assign	(	std::unique_ptr< IFunction > &&	fn,
		const backend::Backend *	backend
	)

Assign a task to the given backend.

Parameters

[in]	fn	Function to be assigned
[in]	fn	Target backend

Definition at line 39 of file ParallelScheduler.cc.

{
  assert(!_thread_pools.empty());
 
  _thread_pools.at(backend)->enqueue(std::move(fn));
}

void onert::exec::ParallelScheduler::finish ( )

Block until all jobs are finished.

Definition at line 46 of file ParallelScheduler.cc.

{
  for (auto &&itr : _thread_pools)
  {
    itr.second->finish();
  }
}

The documentation for this class was generated from the following files: