Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
utils.h File Reference
#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>
#include "tc/autotuner/utils/utils-inl.h"
Include dependency graph for utils.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  tc::autotune::OptionsWithMedianTime
 

Namespaces

 tc
 
 tc::autotune
 

Functions

std::vector< std::size_t > tc::autotune::powers2andCeilDivisors (std::size_t val)
 
template<typename Vector , typename... Vectors>
Vector tc::autotune::mergeVectors (Vector &&v, Vectors &&...vs)
 
std::vector
< OptionsWithMedianTime > 
tc::autotune::getOptionsAndMedianRuntimes (const std::string &id, const std::vector< const DLTensor * > &inputs)
 
std::vector< MappingOptions > tc::autotune::restoreCandidates (const std::string &id, const std::vector< const DLTensor * > &inputs, const std::vector< const DLTensor * > &outputs)
 
llvm::Optional< MappingOptions > tc::autotune::getBestOptions (const std::string &id, const std::vector< const DLTensor * > &inputs, const std::vector< const DLTensor * > &outputs)