[DAG] try to convert multiply to shift via demanded bits
authorSanjay Patel <spatel@rotateright.com>
Wed, 23 Feb 2022 16:25:01 +0000 (11:25 -0500)
committerSanjay Patel <spatel@rotateright.com>
Wed, 23 Feb 2022 17:09:32 +0000 (12:09 -0500)
commit21d7c3bcc646f5db73bc3d21f9d1b1327b6a5ec0
treed1734a433f9bdf88c233c8cf2f7944e6873694ec
parent1fd980de04eda1bead75263a67e6e3bf8c4ebd00
[DAG] try to convert multiply to shift via demanded bits

This is a fix for a regression discussed in:
https://github.com/llvm/llvm-project/issues/53829

We cleared more high multiplier bits with 995d400,
but that can lead to worse codegen because we would fail
to recognize the now disguised multiplication by neg-power-of-2
as a shift-left. The problem exists independently of the IR
change in the case that the multiply already had cleared high
bits. We also convert shl+sub into mul+add in instcombine's
negator.

This patch fills in the high-bits to see the shift transform
opportunity. Alive2 attempt to show correctness:
https://alive2.llvm.org/ce/z/GgSKVX

The AArch64, RISCV, and MIPS diffs look like clear wins. The
x86 code requires an extra move register in the minimal examples,
but it's still an improvement to get rid of the multiply on all
CPUs that I am aware of (because multiply is never as fast as a
shift).

There's a potential follow-up noted by the TODO comment. We
should already convert that pattern into shl+add in IR, so
it's probably not common:
https://alive2.llvm.org/ce/z/7QY_Ga

Fixes #53829

Differential Revision: https://reviews.llvm.org/D120216
llvm/lib/CodeGen/SelectionDAG/TargetLowering.cpp
llvm/test/CodeGen/AArch64/mul_pow2.ll
llvm/test/CodeGen/Mips/urem-seteq-illegal-types.ll
llvm/test/CodeGen/RISCV/mul.ll
llvm/test/CodeGen/X86/mul-demand.ll