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

Commit 608c798

Browse files
author
Sven Verdoolaege
committed
use templated isl types makeSingleton
1 parent a0ecd81 commit 608c798

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tc/core/polyhedral/memory_promotion.cc

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -84,13 +84,13 @@ ScopedFootprint outputRanges(isl::map access) {
8484
// Access has the shape :: [S -> ref] -> O
8585
// Extract the reference ID, store it separately and simplify the access.
8686
std::unique_ptr<TensorReferenceGroup> TensorReferenceGroup::makeSingleton(
87-
isl::map originalAccess,
88-
isl::map scopedAccess,
87+
isl::Map<isl::Pair<Statement, Tag>, Tensor> originalAccess,
88+
isl::Map<isl::Pair<Scope, Tag>, Tensor> scopedTaggedAccess,
8989
AccessType type) {
9090
auto ref = std::unique_ptr<TensorReference>(new TensorReference);
9191
auto refId =
92-
scopedAccess.get_space().domain().unwrap().get_map_range_tuple_id();
93-
scopedAccess = scopedAccess.domain_factor_domain();
92+
scopedTaggedAccess.get_space().domain().unwrap().get_map_range_tuple_id();
93+
auto scopedAccess = scopedTaggedAccess.domain_factor_domain();
9494
ref->originalAccess = originalAccess.domain_factor_domain();
9595
ref->scopedAccess = scopedAccess;
9696
ref->type = type;

tc/core/polyhedral/memory_promotion.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,8 @@ class TensorReferenceGroup {
156156
std::unique_ptr<TensorReferenceGroup>&& g1,
157157
std::unique_ptr<TensorReferenceGroup>&& g2);
158158
static std::unique_ptr<TensorReferenceGroup> makeSingleton(
159-
isl::map originalAccess,
160-
isl::map scopedAccess,
159+
isl::Map<isl::Pair<Statement, Tag>, Tensor> originalAccess,
160+
isl::Map<isl::Pair<Scope, Tag>, Tensor> scopedAccess,
161161
AccessType type);
162162

163163
public:

0 commit comments

Comments
 (0)