AllOnGPU.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 "Strategy.h"
 
namespace tarch {
  namespace multicore {
    namespace orchestration {
      class AllOnGPU;
    } // namespace orchestration
  }   // namespace multicore
} // namespace tarch
 
 
class tarch::multicore::orchestration::AllOnGPU: public tarch::multicore::orchestration::Strategy {
private:
  const int   _device;
  int         _deviceToUse;
  bool        _isInBspSection;
  const FusionImplementation _fusionImplementation;
 
  void toggleDevice();
public:
  static int constexpr RoundRobin = -1;
 
  AllOnGPU(int device, FusionImplementation FusionImplementation = FusionImplementation::LazyEvaluation);
 
  virtual void startBSPSection(int nestedParallelismLevel) override;
  virtual void endBSPSection(int nestedParallelismLevel) override;
 
  virtual FuseInstruction fuse(int taskType) override;
 
  virtual ExecutionPolicy paralleliseForkJoinSection(int nestedParallelismLevel, int numberOfTasks, int taskType)
    override;
 
  virtual void updateFuseStatistics(
    int    taskType,
    int    queueSize,
    double averageNumberOfPendingReadyTasksPerProcessTaskExecution,
    double probabilityThatTaskIsCompleted
  ) override;
 
  virtual FusionImplementation getFusionImplementation() const override;
};