llvm/test/Transforms/InstCombine/assume-align.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -S -passes=instcombine,simplifycfg < %s 2>&1 | FileCheck %s
   3
   4 declare void @llvm.assume(i1 noundef)
   5
   6 define void @f1(ptr %a) {
   7 ; CHECK-LABEL: @f1(
   8 ; CHECK-NEXT:  entry:
   9 ; CHECK-NEXT:    [[PTR:%.*]] = getelementptr inbounds i8, ptr [[A:%.*]], i64 4
  10 ; CHECK-NEXT:    [[TMP0:%.*]] = ptrtoint ptr [[PTR]] to i64
  11 ; CHECK-NEXT:    [[TMP1:%.*]] = and i64 [[TMP0]], 3
  12 ; CHECK-NEXT:    [[TMP2:%.*]] = icmp eq i64 [[TMP1]], 0
  13 ; CHECK-NEXT:    br i1 [[TMP2]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
  14 ; CHECK:       if.then:
  15 ; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "align"(ptr [[PTR]], i64 4) ]
  16 ; CHECK-NEXT:    store i32 4, ptr [[PTR]], align 4
  17 ; CHECK-NEXT:    br label [[IF_END]]
  18 ; CHECK:       if.end:
  19 ; CHECK-NEXT:    ret void
  20 ;
  21 entry:
  22   %ptr = getelementptr inbounds i8, ptr %a, i64 4
  23   %0 = ptrtoint ptr %ptr to i64
  24   %1 = and i64 %0, 3
  25   %2 = icmp eq i64 %1, 0
  26   br i1 %2, label %if.then, label %if.end
  27
  28 if.then:                                          ; preds = %entry
  29   call void @llvm.assume(i1 true) [ "align"(ptr %ptr, i64 4) ]
  30   %3 = ptrtoint ptr %ptr to i64
  31   %4 = and i64 %3, 3
  32   %5 = icmp eq i64 %4, 0
  33   br i1 %5, label %if.then1, label %if.else1
  34
  35 if.then1:                                         ; preds = %if.then
  36   store i32 4, ptr %ptr, align 4
  37   br label %if.end
  38
  39 if.else1:                                         ; preds = %if.then
  40   store i8 1, ptr %ptr, align 1
  41   br label %if.end
  42
  43 if.end:                                           ; preds = %if.then1, %if.else1, %entry
  44   ret void
  45 }
  46
  47 ; TODO: We could fold away the branch "br i1 %3, ..." by either using a GEP or make getKnowledgeValidInContext aware the alignment bundle offset, and the improvement of value tracking of GEP.
  48
  49 define void @f2(ptr %a) {
  50 ; CHECK-LABEL: @f2(
  51 ; CHECK-NEXT:  entry:
  52 ; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "align"(ptr [[A:%.*]], i64 32, i32 24) ]
  53 ; CHECK-NEXT:    [[TMP0:%.*]] = getelementptr inbounds i8, ptr [[A]], i64 8
  54 ; CHECK-NEXT:    [[TMP1:%.*]] = ptrtoint ptr [[TMP0]] to i64
  55 ; CHECK-NEXT:    [[TMP2:%.*]] = and i64 [[TMP1]], 8
  56 ; CHECK-NEXT:    [[TMP3:%.*]] = icmp eq i64 [[TMP2]], 0
  57 ; CHECK-NEXT:    br i1 [[TMP3]], label [[IF_THEN:%.*]], label [[IF_ELSE:%.*]]
  58 ; CHECK:       if.then:
  59 ; CHECK-NEXT:    store i64 16, ptr [[TMP0]], align 4
  60 ; CHECK-NEXT:    br label [[IF_END:%.*]]
  61 ; CHECK:       if.else:
  62 ; CHECK-NEXT:    store i8 1, ptr [[TMP0]], align 1
  63 ; CHECK-NEXT:    br label [[IF_END]]
  64 ; CHECK:       if.end:
  65 ; CHECK-NEXT:    ret void
  66 ;
  67 entry:
  68   call void @llvm.assume(i1 true) [ "align"(ptr %a, i64 32, i32 24) ]
  69   %0 = getelementptr inbounds i8, ptr %a, i64 8
  70   %1 = ptrtoint ptr %0 to i64
  71   %2 = and i64 %1, 15
  72   %3 = icmp eq i64 %2, 0
  73   br i1 %3, label %if.then, label %if.else
  74
  75 if.then:                                          ; preds = %entry
  76   store i64 16, ptr %0, align 4
  77   br label %if.end
  78
  79 if.else:                                          ; preds = %entry
  80   store i8 1, ptr %0, align 1
  81   br label %if.end
  82
  83 if.end:                                           ; preds = %if.else, %if.then
  84   ret void
  85 }
  86
  87 define void @f3(i64 %a, ptr %b) {
  88 ; CHECK-LABEL: @f3(
  89 ; CHECK-NEXT:    [[C:%.*]] = ptrtoint ptr [[B:%.*]] to i64
  90 ; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "align"(ptr [[B]], i64 4294967296) ]
  91 ; CHECK-NEXT:    [[D:%.*]] = add i64 [[C]], [[A:%.*]]
  92 ; CHECK-NEXT:    call void @g(i64 [[D]])
  93 ; CHECK-NEXT:    ret void
  94 ;
  95   %c = ptrtoint ptr %b to i64
  96   call void @llvm.assume(i1 true) [ "align"(ptr %b, i64 4294967296) ]
  97   %d = add i64 %a, %c
  98   call void @g(i64 %d)
  99   ret void
 100 }
 101
 102 declare void @g(i64)
 103
 104 define i8 @assume_align_zero(ptr %p) {
 105 ; CHECK-LABEL: @assume_align_zero(
 106 ; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "align"(ptr [[P:%.*]], i64 0) ]
 107 ; CHECK-NEXT:    [[V:%.*]] = load i8, ptr [[P]], align 1
 108 ; CHECK-NEXT:    ret i8 [[V]]
 109 ;
 110   call void @llvm.assume(i1 true) [ "align"(ptr %p, i64 0) ]
 111   %v = load i8, ptr %p
 112   ret i8 %v
 113 }
 114
 115 define i8 @assume_align_non_pow2(ptr %p) {
 116 ; CHECK-LABEL: @assume_align_non_pow2(
 117 ; CHECK-NEXT:    call void @llvm.assume(i1 true) [ "align"(ptr [[P:%.*]], i64 123) ]
 118 ; CHECK-NEXT:    [[V:%.*]] = load i8, ptr [[P]], align 1
 119 ; CHECK-NEXT:    ret i8 [[V]]
 120 ;
 121   call void @llvm.assume(i1 true) [ "align"(ptr %p, i64 123) ]
 122   %v = load i8, ptr %p
 123   ret i8 %v
 124 }