genetic_autotuner.h

Go to the documentation of this file.

#pragma once
#include <chrono>
#include <iostream>
#include <memory>
#include <vector>

#include "tc/autotuner/genetic_tuning_harness.h"
#include "tc/autotuner/utils/utils.h"
#include "tc/lang/parser.h"

#include <llvm/ADT/Optional.h>

namespace tc {
namespace autotune {
namespace detail {

using Duration = std::chrono::high_resolution_clock::duration;

class GeneticAutotuner {
 public:
  explicit GeneticAutotuner(const std::string& tc);

  void storeCaches(const std::string& filename);

  std::vector<MappingOptions> load(
      const std::string& cacheFileName,
      const std::string& tcName,
      const std::vector<const DLTensor*>& inputs,
      const size_t numCandidates);

  llvm::Optional<MappingOptions> tune(
      const std::string& cacheFileName,
      const std::string& tcName,
      const std::unordered_map<size_t, std::vector<const DLTensor*>>& inputs,
      std::unordered_map<size_t, std::vector<DLTensor*>>& outputs,
      MappingOptions baseMapping,
      std::vector<MappingOptions> startingPoints,
      const TuningParameterFixer& fixedParams);

 private:
  std::string tc_;
  std::map<std::string, lang::TreeRef> tcNameMap_;
};

} // namespace detail
} // namespace autotune
} // namespace tc