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

Commit 6644cb2

Browse files
author
Sven Verdoolaege
committed
schedule_isl_conversion.cc: extendChild: extract out insertChild
extendChild moves down in both the original tree and the result of the conversion. Sometimes, however, it can be useful to only move down in the original tree without moving in the converted tree, e.g., when no corresponding node is introduced in the converted tree. This will be used for the thread specific marker nodes added in the next commit.
1 parent 249b314 commit 6644cb2

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

tc/core/polyhedral/schedule_isl_conversion.cc

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@ constexpr std::initializer_list<detail::ScheduleTreeType>
5151

5252
namespace {
5353

54+
isl::schedule_node insertChild(isl::schedule_node node, const ScheduleTree* st);
5455
isl::schedule_node extendChild(isl::schedule_node node, const ScheduleTree* st);
5556

5657
/*
@@ -210,17 +211,27 @@ isl::schedule_node insert(isl::schedule_node node, const ScheduleTree* st) {
210211

211212
/*
212213
* Recursively add nodes corresponding to the descendants of "st"
213-
* underneath "node".
214+
* at "node".
214215
* If "st" does not have any children, then no descendants need to be added.
215216
*/
216-
isl::schedule_node extendChild(
217+
isl::schedule_node insertChild(
217218
isl::schedule_node node,
218219
const ScheduleTree* st) {
219220
if (st->numChildren() == 0) {
220221
return node;
221222
}
222223

223-
return insert(node.child(0), st->child({0})).parent();
224+
return insert(node, st->child({0}));
225+
}
226+
227+
/*
228+
* Recursively add nodes corresponding to the descendants of "st"
229+
* underneath "node".
230+
*/
231+
isl::schedule_node extendChild(
232+
isl::schedule_node node,
233+
const ScheduleTree* st) {
234+
return insertChild(node.child(0), st).parent();
224235
}
225236
} // namespace
226237

0 commit comments

Comments
 (0)