TaskHandleRepository.h

Go to the documentation of this file.

// This file is part of the Peano project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org.
#pragma once
 
#include "config.h"
 
 
#if defined(SharedTBBExtension) or defined(SharedTBB)
 
#include <functional>
#include <set>
#include <tbb/task_arena.h>
 
#include <oneapi/tbb/task_group.h>
#include <oneapi/tbb/task_arena.h>
#include <oneapi/tbb/concurrent_hash_map.h>
#include <oneapi/tbb/concurrent_queue.h>
 
 
#if !defined(SharedTBBExtension) and !defined(TBB_USE_TASK_GROUP_PREVIEW)
// In the new TBB version, all the handle and dynamic task graph stuff is
// merged into the core tasking. No additional includes required.
#else
#include "tbb/dynamic_task_graph.h"
#include "tbb/dynamic_task_graph_node.h"
#include "tbb/dynamic_task_graph_utils.h"
#endif
 
 
namespace tarch {
  namespace multicore {
    namespace tbb {
      class TaskHandleRepository;
    }
  }
}
 
class tarch::multicore::tbb::TaskHandleRepository {
  public:
    using TaskNumber = int;
 
    TaskHandleRepository() = default;
    ~TaskHandleRepository();
 
    #if !defined(SharedTBBExtension) and !defined(TBB_USE_TASK_GROUP_PREVIEW)
    using Handle = ::oneapi::tbb::task_handle;
    #else
    using Handle = ::tbb::dynamic_task_graph_node;
    #endif
 
 
 
    void registerTask(
      Handle*      newTask,
      TaskNumber   number
    );
 
    void addDependency( Handle*      newTask, const TaskNumber&  in_dependency);
 
    void addDependencies( Handle*    newTask, const std::set< TaskNumber >&  in_dependencies );
 
  private:
    #if !defined(SharedTBBExtension) and !defined(TBB_USE_TASK_GROUP_PREVIEW)
    using HashMap = oneapi::tbb::concurrent_hash_map<TaskNumber, oneapi::tbb::task_handle*>;
    #else
    using HashMap = oneapi::tbb::concurrent_hash_map<TaskNumber, ::tbb::dynamic_task_graph_node*>;
    #endif
 
    HashMap  _taskHandleContainer;
};
 
#endif