mapping_types-inl.h

Go to the documentation of this file.

#pragma once

namespace tc {
namespace polyhedral {
namespace mapping {
bool MappingId::isBlockId() {
  return *this == BlockId::x() or *this == BlockId::y() or
      *this == BlockId::z();
}
BlockId* MappingId::asBlockId() {
  if (!isBlockId()) {
    return nullptr;
  }
  return static_cast<BlockId*>(this);
}
bool MappingId::isThreadId() {
  return *this == ThreadId::x() or *this == ThreadId::y() or
      *this == ThreadId::z();
}
ThreadId* MappingId::asThreadId() {
  if (!isThreadId()) {
    return nullptr;
  }
  return static_cast<ThreadId*>(this);
}

ThreadId ThreadId::makeId(size_t dim) {
  CHECK(dim < 3);
  if (dim == 0) {
    return ThreadId::x();
  }
  if (dim == 1) {
    return ThreadId::y();
  }
  return ThreadId::z();
}
template <unsigned char Dim>
ThreadId ThreadId::makeId(isl::id id) {
  static_assert(Dim < kMaxDim, "Incorrect Dim >= kMaxDim");
  return ThreadId(id, Dim);
}
ThreadId ThreadId::x() {
  static thread_local isl::id x(
      isl::with_exceptions::globalIslCtx(), std::string("t0"));
  return makeId<0>(x);
}
ThreadId ThreadId::y() {
  static thread_local isl::id y(
      isl::with_exceptions::globalIslCtx(), std::string("t1"));
  return makeId<1>(y);
}
ThreadId ThreadId::z() {
  static thread_local isl::id z(
      isl::with_exceptions::globalIslCtx(), std::string("t2"));
  return makeId<2>(z);
}

BlockId BlockId::makeId(size_t dim) {
  CHECK(dim < 3);
  if (dim == 0) {
    return BlockId::x();
  }
  if (dim == 1) {
    return BlockId::y();
  }
  return BlockId::z();
}
template <unsigned char Dim>
BlockId BlockId::makeId(isl::id id) {
  static_assert(Dim < kMaxDim, "Incorrect Dim >= kMaxDim");
  return BlockId(id, Dim);
}
BlockId BlockId::x() {
  static thread_local isl::id x(
      isl::with_exceptions::globalIslCtx(), std::string("b0"));
  return makeId<0>(x);
}
BlockId BlockId::y() {
  static thread_local isl::id y(
      isl::with_exceptions::globalIslCtx(), std::string("b1"));
  return makeId<1>(y);
}
BlockId BlockId::z() {
  static thread_local isl::id z(
      isl::with_exceptions::globalIslCtx(), std::string("b2"));
  return makeId<2>(z);
}
} // namespace mapping
} // namespace polyhedral
} // namespace tc