[llvm] [AArch64] Replace AND with LSL#2 for LDR target (#34101) (PR #89531)

[David Green via llvm-commits]

================
@@ -16918,6 +16918,24 @@ bool AArch64TargetLowering::shouldFoldConstantShiftPairToMask(
     return (!C1 || !C2 || C1->getZExtValue() >= C2->getZExtValue());
   }
 
+  // We do not need to fold when this shifting used in specific load case:
+  // (ldr x, (add x, (shl (srl x, c1) 2)))
+  if (N->getOpcode() == ISD::SHL) {
+    auto C2 = dyn_cast_or_null<ConstantSDNode>(N->getOperand(1));
+    if (C2 && C2->getZExtValue() == 2) {
----------------
davemgreen wrote:

Yeah sounds good. A few other types sizes: https://godbolt.org/z/GqPT3KbbT. The shift in the addressing mode needs to match the loaded size.

https://github.com/llvm/llvm-project/pull/89531