This repository was archived by the owner on Apr 28, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +12
-5
lines changed Expand file tree Collapse file tree 3 files changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -208,6 +208,14 @@ std::unique_ptr<ScheduleTree> ScheduleTree::makeBand(
208
208
return res;
209
209
}
210
210
211
+ ScheduleTreeUPtr ScheduleTree::makeEmptyBand (const ScheduleTree* root) {
212
+ auto domain = root->elemAs <ScheduleTreeElemDomain>();
213
+ CHECK (domain);
214
+ auto space = domain->domain_ .get_space ().set_from_params ();
215
+ auto zero = isl::multi_union_pw_aff::zero (space);
216
+ return ScheduleTree::makeBand (zero);
217
+ }
218
+
211
219
std::unique_ptr<ScheduleTree> ScheduleTree::makeDomain (
212
220
isl::union_set domain,
213
221
std::vector<ScheduleTreeUPtr>&& children) {
Original file line number Diff line number Diff line change @@ -285,6 +285,9 @@ struct ScheduleTree {
285
285
isl::multi_union_pw_aff mupa,
286
286
std::vector<ScheduleTreeUPtr>&& children = {});
287
287
288
+ // Return a zero-dimensional band for use in a tree with the given root.
289
+ static ScheduleTreeUPtr makeEmptyBand (const ScheduleTree* root);
290
+
288
291
static ScheduleTreeUPtr makeDomain (
289
292
isl::union_set domain,
290
293
std::vector<ScheduleTreeUPtr>&& children = {});
Original file line number Diff line number Diff line change @@ -488,11 +488,7 @@ detail::ScheduleTree* obtainOuterBand(detail::ScheduleTree* root) {
488
488
continue ;
489
489
}
490
490
491
- auto domain = root->elemAs <ScheduleTreeElemDomain>();
492
- CHECK (domain);
493
- auto space = domain->domain_ .get_space ().set_from_params ();
494
- auto zero = isl::multi_union_pw_aff::zero (space);
495
- auto band = ScheduleTree::makeBand (zero);
491
+ auto band = ScheduleTree::makeEmptyBand (root);
496
492
if (n == 0 ) {
497
493
return setPermutable (insertNodeBelow (tree, std::move (band)));
498
494
} else {
You can’t perform that action at this time.
0 commit comments