genetic_autotuner_aten.h

Go to the documentation of this file.

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

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

namespace tc {
namespace autotune {

class GeneticAutotunerATen {
 public:
  GeneticAutotunerATen(const std::string tc);

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

  llvm::Optional<MappingOptions> tune(
      const std::string& cacheFileName,
      const std::string& tcName,
      const std::vector<at::Tensor>& inputs,
      MappingOptions baseMapping,
      std::vector<MappingOptions> startingPoints,
      const TuningParameterFixer& fixedParams = {});

 private:
  std::string tc_;
  std::unique_ptr<detail::GeneticAutotuner> geneticAutotuner_;
};

} // namespace autotune
} // namespace tc