[Transforms] Enhance CorrelatedValuePropagation to handle both values of select
authorDmitry Vassiliev <dvassiliev@accesssoftek.com>
Tue, 22 Feb 2022 20:11:20 +0000 (00:11 +0400)
committerDmitry Vassiliev <dvassiliev@accesssoftek.com>
Tue, 22 Feb 2022 20:11:20 +0000 (00:11 +0400)
commit90a3b310917031c802bb930626e85f067d53ea5b
treeb9e50888d05c8100f566847b589a01a63e3e95e5
parent0b302be023388e7cec2daf680a3ea6718c6af53f
[Transforms] Enhance CorrelatedValuePropagation to handle both values of select

The "Correlated Value Propagation" pass was missing a case when handling select instructions. It was only handling the "false" constant value, while in NVPTX the select may have the condition (and thus the branches) inverted, for example:
```
loop:
%phi = phi i32* [ %sel, %loop ], [ %x, %entry ]
%f = tail call i32* @f(i32* %phi)
%cmp1 = icmp ne i32* %f, %y
%sel = select i1 %cmp1, i32* %f, i32* null
%cmp2 = icmp eq i32* %sel, null
br i1 %cmp2, label %return, label %loop
```
But the select condition can be inverted:
```
%cmp1 = icmp eq i32* %f, %y
%sel = select i1 %cmp1, i32* null, i32* %f
```
The fix is to enhance "Correlated Value Propagation" to handle both branches of the select instruction.

Reviewed By: nikic, lebedev.ri

Differential Revision: https://reviews.llvm.org/D119643
llvm/lib/Transforms/Scalar/CorrelatedValuePropagation.cpp
llvm/test/Transforms/CorrelatedValuePropagation/basic.ll