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

Commit 7571d68

Browse files
author
Sven Verdoolaege
committed
addSingletonReferenceGroup: use templated isl types
1 parent f9ebd3d commit 7571d68

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

tc/core/polyhedral/memory_promotion.cc

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -282,13 +282,12 @@ void joinOverlappingWrites(
282282
void addSingletonReferenceGroup(
283283
TensorGroups& tensorGroups,
284284
isl::id targetTensor,
285-
isl::union_map schedule,
286-
isl::map access,
285+
isl::UnionMap<Statement, Prefix> schedule,
286+
isl::Map<isl::Pair<Statement, Tag>, Tensor> access,
287287
AccessType type) {
288-
auto scopedUnionAccess = isl::union_map(access.curry());
289-
scopedUnionAccess = scopedUnionAccess.apply_domain(schedule);
290-
auto scopedAccess = isl::map::from(scopedUnionAccess);
291-
scopedAccess = scopedAccess.uncurry();
288+
auto unionAccess = access.curry().asUnionMap();
289+
auto scopedUnionAccess = unionAccess.apply_domain(schedule);
290+
auto scopedAccess = scopedUnionAccess.toMap().uncurry();
292291
tensorGroups[targetTensor].push_back(
293292
TensorReferenceGroup::makeSingleton(access, scopedAccess, type));
294293
}

0 commit comments

Comments
 (0)