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

Commit c133cb5

Browse files
committed
Call register promotion from MappedScop
1 parent 0b3379d commit c133cb5

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/core/polyhedral/mapped_scop.cc

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -689,10 +689,16 @@ std::unique_ptr<MappedScop> MappedScop::makeWithOuterBlockInnerThreadStrategy(
689689
}
690690
}
691691

692-
// 8. Insert mapping context
692+
// 8. Promote to registers below the loops mapped to threads.
693+
if (options.proto.use_private_memory()) {
694+
promoteToRegistersBelowThreads(
695+
mappedScop->scop(), mappedScop->threadIdxxScheduleDepthState, -1ull);
696+
}
697+
698+
// 9. Insert mapping context
693699
mappedScop->insertMappingContext();
694700

695-
// 9. Optionally insert reduction synchronizations
701+
// 10. Optionally insert reduction synchronizations
696702
for (auto bandUpdate : mappedScop->reductionBandUpdates_) {
697703
for (auto updateId : bandUpdate.second.ids) {
698704
scop->insertReductionSync1D(

0 commit comments

Comments
 (0)