schedule_tree-inl.h

Go to the documentation of this file.

#pragma once

namespace tc {
namespace polyhedral {
namespace detail {
template <typename MappingIdType>
inline ScheduleTreeUPtr ScheduleTree::makeMappingFilter(
    isl::union_set filter,
    const std::unordered_set<MappingIdType, typename MappingIdType::Hash>&
        mappingIds,
    std::vector<ScheduleTreeUPtr>&& children) {
  // slicing may happen below if not careful
  std::unordered_set<mapping::MappingId, typename mapping::MappingId::Hash> ids;
  for (auto id : mappingIds) {
    CHECK_EQ(1, mappingIds.count(id)) << "id: " << id << " mapped != 1 times";
    ids.insert(id);
  }
  isl::ctx ctx(filter.get_ctx());
  ScheduleTreeUPtr res(new ScheduleTree(ctx));
  res->elem_ = std::unique_ptr<ScheduleTreeElemMappingFilter>(
      new ScheduleTreeElemMappingFilter(filter, ids));
  res->type_ = ScheduleTreeType::MappingFilter;
  res->appendChildren(std::move(children));
  return res;
}
} // namespace detail
} // namespace polyhedral
} // namespace tc