TuningManager.h

Go to the documentation of this file.

 
#pragma once
 
#include <limits>
#include <memory>
#include <set>
#include <unordered_map>
 
#include "autopas/options/ContainerOption.h"
#include "autopas/options/InteractionTypeOption.h"
#include "autopas/tuning/AutoTuner.h"
 
namespace autopas {
 
class TuningManager {
 public:
  explicit TuningManager(const AutoTunerInfo &autoTunerInfo);
 
  void addAutoTuner(std::unique_ptr<AutoTuner> tuner, InteractionTypeOption::Value interactionType);
 
  bool tune(size_t currentIteration, const LiveInfo &info);
 
  void addMeasurement(long sampleRebuild, long sampleTraverseParticles, bool neighborListRebuilt, size_t iteration,
                      InteractionTypeOption::Value interactionType) const;
 
  void setRebuildFrequency(double rebuildFrequency);
 
  Configuration rejectConfiguration(const Configuration &rejectedConfig, bool indefinitely,
                                    InteractionTypeOption::Value interactionType);
 
  void forceRetune();
 
  void logTuningResult(long tuningTime, size_t currentIteration, InteractionTypeOption::Value interactionType) const;
 
  bool requiresRebuilding(size_t currentIteration);
 
  bool needsLiveInfo(size_t currentIteration);
 
  bool allSearchSpacesAreTrivial() const;
 
  bool tuningPhaseJustFinished() const;
 
  bool isStartOfTuningPhase(size_t currentIteration) const;
 
  bool inFirstConfigurationLastSample(size_t currentIteration) const;
 
  const Configuration &getCurrentConfig(InteractionTypeOption::Value interactionType) const;
 
  const TuningMetricOption &getTuningMetric(InteractionTypeOption::Value interactionType) const;
 
  std::unordered_map<InteractionTypeOption::Value, std::unique_ptr<AutoTuner>> &getAutoTuners() { return _autoTuners; }
 
 private:
  void tuneConfigurations(size_t currentIteration);
 
  void setOptimalConfigurations();
 
  std::set<std::pair<ContainerOption, double>> getCommonContainerAndCellSizeFactors() const;
 
  bool tuningPhaseAboutToBegin(size_t currentIteration) const;
 
  std::unordered_map<InteractionTypeOption::Value, std::unique_ptr<AutoTuner>> _autoTuners;
 
  size_t _tuningPhase = 0;
 
  size_t _tuningInterval = 0;
 
  size_t _maxSamples = 0;
 
  bool _tuningFinished = true;
 
  size_t _lastTuningIteration{std::numeric_limits<size_t>::max()};
 
  bool _tuningFinishedLastIteration = false;
 
  bool _forceRetunePending{false};
};
 
}  // namespace autopas