Skip to content
This repository was archived by the owner on Apr 28, 2023. It is now read-only.

Commit e5f0096

Browse files
author
Sven Verdoolaege
committed
ScheduleTreeElemBand::drop: drop elements from list
The function isl_multi_union_pw_aff_drop_dims will not be exposed in the mainline C++ interface. A similar function that drops base expression in the multi expression may get added later and this function could then be used here. For now, use slightly more verbose code to perform the dropping of elements on a list extracted from the multi expression.
1 parent dc8ac72 commit e5f0096

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tc/core/polyhedral/schedule_tree_elem.cc

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,11 @@ void ScheduleTreeElemBand::drop(size_t pos, size_t n) {
159159
TC_CHECK_GE(nMember(), pos + n) << "range out of bounds";
160160
auto nBegin = nMember();
161161

162-
mupa_ = mupa_.drop_dims(isl::dim_type::set, pos, n);
162+
auto list = mupa_.get_union_pw_aff_list();
163+
auto space = mupa_.get_space().domain();
164+
list = list.drop(pos, n);
165+
space = addRange(space, list.size());
166+
mupa_ = isl::multi_union_pw_aff(space, list);
163167

164168
std::copy(
165169
coincident_.begin() + pos + n,

0 commit comments

Comments
 (0)