[CodeGen][MachinePipeliner] Limit register pressure when scheduling (#74807)
commit7556626dcff15c8cc5160078a4d6ed2469eed81b
authorRyotaro KASUGA <kasuga.ryotaro@fujitsu.com>
Mon, 22 Jan 2024 08:06:37 +0000 (22 17:06 +0900)
committerGitHub <noreply@github.com>
Mon, 22 Jan 2024 08:06:37 +0000 (22 17:06 +0900)
tree7fff63c4a944c619b796fbd001d45e370fed13eb
parent5ab2d9c0d9d4ea381cc2f5cbb047c1b1847d0b21
[CodeGen][MachinePipeliner] Limit register pressure when scheduling (#74807)

In software pipelining, when searching for the Initiation Interval (II),
`MachinePipeliner` tries to reduce register pressure, but doesn't check
how many variables can actually be alive at the same time. As a result,
a lot of register spills/fills can be generated after register
allocation, which might cause performance degradation. To prevent such
cases, this patch adds a check phase that calculates the maximum
register pressure of the scheduled loop and reject it if the pressure is
too high. This can be enabled this by specifying
`pipeliner-register-pressure`. Additionally, an II search range is
currently fixed at 10, which is too small to find a schedule when the
above algorithm is applied. Therefore this patch also adds a new option
`pipeliner-ii-search-range` to specify the length of the range to
search. There is one more new option
`pipeliner-register-pressure-margin`, which can be used to estimate a
register pressure limit less than actual for conservative analysis.

Discourse thread:
https://discourse.llvm.org/t/considering-register-pressure-when-deciding-initiation-interval-in-machinepipeliner/74725
llvm/include/llvm/CodeGen/MachinePipeliner.h
llvm/lib/CodeGen/MachinePipeliner.cpp
llvm/test/CodeGen/PowerPC/sms-regpress.mir [new file with mode: 0644]