probe.h

Go to the documentation of this file.

 
#pragma once
 
#include <functional>
#include <typeindex>
#include <utility>
 
namespace tvm::diagnostic::internal
{
template<typename T>
inline void hash_combine(std::size_t & seed, const T & v) noexcept
{
  std::hash<T> hasher;
  seed ^= hasher(v) + 0x9e3779b9 + (seed << 6) + (seed >> 2);
}
 
struct PairHasher
{
  template<typename T, typename U>
  std::size_t operator()(const std::pair<T, U> & p) const noexcept
  {
    std::size_t hash = 0;
    hash_combine(hash, p.first);
    hash_combine(hash, p.second);
    return hash;
  }
};
 
} // namespace tvm::diagnostic::internal
 
// Specialization
#include <tvm/graph/internal/Log.h>
 
namespace std
{
#ifdef _MSC_VER
template<>
struct hash<std::size_t>
{
  // identity function for this type
  std::size_t operator()(const std::size_t & i) const noexcept { return i; }
};
#endif
 
template<>
struct hash<tvm::graph::internal::Log::EnumValue>
{
  std::size_t operator()(const tvm::graph::internal::Log::EnumValue & e) const noexcept
  {
    std::size_t hash = 0;
    tvm::diagnostic::internal::hash_combine(hash, e.type);
    tvm::diagnostic::internal::hash_combine(hash, e.value);
    return hash;
  }
};
} // namespace std