MPIParallelizedStrategy.h

Go to the documentation of this file.

 
#pragma once
 
#include <cstddef>
#include <random>
#include <type_traits>
#include <vector>
 
#include "autopas/containers/ParticleContainerInterface.h"
#include "autopas/options/ContainerOption.h"
#include "autopas/options/DataLayoutOption.h"
#include "autopas/options/LoadEstimatorOption.h"
#include "autopas/options/Newton3Option.h"
#include "autopas/options/TraversalOption.h"
#include "autopas/options/TuningStrategyOption.h"
#include "autopas/tuning/Configuration.h"
#include "autopas/tuning/searchSpace/EvidenceCollection.h"
#include "autopas/tuning/tuningStrategy/TuningStrategyInterface.h"
#include "autopas/utils/AutoPasConfigurationCommunicator.h"
#include "autopas/utils/ConfigurationAndRankIteratorHandler.h"
#include "autopas/utils/ExceptionHandler.h"
#include "autopas/utils/NumberSet.h"
#include "autopas/utils/WrapMPI.h"
 
namespace autopas {
 
class MPIParallelizedStrategy : public TuningStrategyInterface {
 public:
  MPIParallelizedStrategy(const Configuration &fallbackConfiguration, const AutoPas_MPI_Comm &comm,
                          double mpiTuningMaxDifferenceForBucket, double mpiTuningWeightForMaxDensity);
 
  TuningStrategyOption getOptionType() const override;
 
  bool optimizeSuggestions(std::vector<Configuration> &configQueue,
                           const EvidenceCollection &evidenceCollection) override;
 
  void receiveSmoothedHomogeneityAndMaxDensity(double homogeneity, double maxDensity) override;
 
  bool reset(size_t iteration, size_t tuningPhase, std::vector<Configuration> &configQueue,
             const autopas::EvidenceCollection &evidenceCollection) override;
 
  [[nodiscard]] inline bool needsSmoothedHomogeneityAndMaxDensity() const override;
 
  void rejectConfiguration(const Configuration &configuration, bool indefinitely) override;
 
  const AutoPas_MPI_Comm &getBucket() const;
 
  static Configuration createFallBackConfiguration(const std::set<Configuration> &searchSpace,
                                                   const InteractionTypeOption &interactionType);
 
 private:
  double _smoothedHomogeneity{-1.};
 
  double _maxDensity{-1.};
 
  std::unique_ptr<TuningStrategyInterface> _tuningStrategy;
 
  AutoPas_MPI_Comm _comm;
  AutoPas_MPI_Comm _bucket{AUTOPAS_MPI_COMM_NULL};
  Configuration _fallbackConfiguration;
  bool _usingFallbackConfig{false};
  std::set<Configuration> _rejectedConfigurations{};
  double _mpiTuningMaxDifferenceForBucket;
  double _mpiTuningWeightForMaxDensity;
  std::mt19937 _rng{std::random_device()()};
};
}  // namespace autopas