Added opt.

Author: egorshamshura

llvm/lib/Transforms/Scalar/InferAlignment.cpp

@@ -12,19 +12,26 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Transforms/Scalar/InferAlignment.h"
+#include "llvm/ADT/STLFunctionalExtras.h"
 #include "llvm/Analysis/AssumptionCache.h"
 #include "llvm/Analysis/ValueTracking.h"
+#include "llvm/IR/Instruction.h"
 #include "llvm/IR/Instructions.h"
 #include "llvm/IR/IntrinsicInst.h"
+#include "llvm/IR/PatternMatch.h"
+#include "llvm/IR/Value.h"
+#include "llvm/Support/Alignment.h"
 #include "llvm/Support/KnownBits.h"
 #include "llvm/Transforms/Scalar.h"
 #include "llvm/Transforms/Utils/Local.h"
 
 using namespace llvm;
+using namespace llvm::PatternMatch;
 
 static bool tryToImproveAlign(
     const DataLayout &DL, Instruction *I,
-    function_ref<Align(Value *PtrOp, Align OldAlign, Align PrefAlign)> Fn) {
+    function_ref<Align(Value *PtrOp, Align OldAlign, Align PrefAlign)> Fn,
+    function_ref<Align(Instruction &I, Value *PtrOp)> ActualAlignFn) {
 
   if (auto *PtrOp = getLoadStorePointerOperand(I)) {
     Align OldAlign = getLoadStoreAlignment(I);
@@ -37,6 +44,16 @@ static bool tryToImproveAlign(
     }
   }
 
+  Value *PtrOp;
+  ConstantInt *Const;
+  if (match(I, m_c_And(m_PtrToIntOrAddr(m_Value(PtrOp)), m_ConstantInt(Const)))) {
+    Align ActualAlign = ActualAlignFn(*I, PtrOp);
+    if (Const->getValue().ult(ActualAlign.value())) {
+      I->replaceAllUsesWith(Constant::getNullValue(I->getType()));
+      return true;
+    }
+  }
+
   IntrinsicInst *II = dyn_cast<IntrinsicInst>(I);
   if (!II)
     return false;
@@ -80,6 +97,8 @@ bool inferAlignment(Function &F, AssumptionCache &AC, DominatorTree &DT) {
               return std::max(OldAlign,
                               tryEnforceAlignment(PtrOp, PrefAlign, DL));
             return OldAlign;
+          }, [&](Instruction&, Value *PtrOp) {
+            return PtrOp->getPointerAlignment(DL);
           });
     }
   }
@@ -120,6 +139,17 @@ bool inferAlignment(Function &F, AssumptionCache &AC, DominatorTree &DT) {
     return LoadStoreAlign;
   };
 
+  auto ActualAlginFn = [&](Instruction &I, Value *PtrOp) {
+    Align KnownAlign = InferFromKnownBits(I, PtrOp);
+    Align BaseAlign = PtrOp->getPointerAlignment(DL);
+    if (auto It = BestBasePointerAligns.find(PtrOp);
+        It != BestBasePointerAligns.end()) {
+      BaseAlign = std::max(BaseAlign, It->second);
+    }
+    Align ActualAlign = std::max(KnownAlign, BaseAlign);
+    return ActualAlign;
+  };
+
   for (BasicBlock &BB : F) {
     // We need to reset the map for each block because alignment information
     // can only be propagated from instruction A to B if A dominates B.
@@ -134,7 +164,7 @@ bool inferAlignment(Function &F, AssumptionCache &AC, DominatorTree &DT) {
           DL, &I, [&](Value *PtrOp, Align OldAlign, Align PrefAlign) {
             return std::max(InferFromKnownBits(I, PtrOp),
                             InferFromBasePointer(PtrOp, OldAlign));
-          });
+          }, ActualAlginFn);
     }
   }
 

llvm/test/Transforms/InferAlignment/ptrtoint.ll

@@ -7,7 +7,7 @@ define i64 @base(ptr %0) {
 ; CHECK-NEXT:    [[V1:%.*]] = load i32, ptr [[TMP0]], align 4
 ; CHECK-NEXT:    [[V3:%.*]] = ptrtoint ptr [[TMP0]] to i64
 ; CHECK-NEXT:    [[V5:%.*]] = and i64 [[V3]], 2
-; CHECK-NEXT:    ret i64 [[V5]]
+; CHECK-NEXT:    ret i64 0
 ;
   %v1 = load i32, ptr %0, align 4
   %v3 = ptrtoint ptr %0 to i64
@@ -22,7 +22,7 @@ define i64 @best_alignment(ptr %0) {
 ; CHECK-NEXT:    [[V1_3:%.*]] = load i32, ptr [[TMP0]], align 16
 ; CHECK-NEXT:    [[V3:%.*]] = ptrtoint ptr [[TMP0]] to i64
 ; CHECK-NEXT:    [[V5:%.*]] = and i64 [[V3]], 15
-; CHECK-NEXT:    ret i64 [[V5]]
+; CHECK-NEXT:    ret i64 0
 ;
   %v1 = load i32, ptr %0, align 8
   %v2 = load i32, ptr %0, align 16
@@ -55,7 +55,7 @@ define i64 @ptrtoaddr(ptr %0) {
 ; CHECK-NEXT:    [[V3:%.*]] = ptrtoaddr ptr [[TMP0]] to i64
 ; CHECK-NEXT:    [[V1:%.*]] = load i32, ptr [[TMP0]], align 4
 ; CHECK-NEXT:    [[V5:%.*]] = and i64 2, [[V3]]
-; CHECK-NEXT:    ret i64 [[V5]]
+; CHECK-NEXT:    ret i64 0
 ;
   %v3 = ptrtoaddr ptr %0 to i64
   %v1 = load i32, ptr %0, align 4