utils.h

Go to the documentation of this file.

#pragma once
#include <vector>

#include <ATen/ATen.h>

#include "tc/core/cuda.h"
#include "tc/core/mapping_options.h"
#include "tc/core/utils/dlpack.h"

#include <llvm/ADT/Optional.h>

namespace tc {
namespace autotune {

struct OptionsWithMedianTime {
  MappingOptions options;
  Duration medianRuntime;
};

std::vector<std::size_t> powers2andCeilDivisors(std::size_t val);

template <typename Vector, typename... Vectors>
Vector mergeVectors(Vector&& v, Vectors&&... vs);

std::vector<OptionsWithMedianTime> getOptionsAndMedianRuntimes(
    const std::string& id,
    const std::vector<const DLTensor*>& inputs);

std::vector<MappingOptions> restoreCandidates(
    const std::string& id,
    const std::vector<const DLTensor*>& inputs,
    const std::vector<const DLTensor*>& outputs);

llvm::Optional<MappingOptions> getBestOptions(
    const std::string& id,
    const std::vector<const DLTensor*>& inputs,
    const std::vector<const DLTensor*>& outputs);

} // namespace autotune
} // namespace tc

#include "tc/autotuner/utils/utils-inl.h"