llvm/test/Transforms/Attributor/ArgumentPromotion/alignment.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-attributes --check-globals
   2 ; RUN: opt -aa-pipeline=basic-aa -passes=attributor -attributor-manifest-internal  -attributor-annotate-decl-cs  -S < %s | FileCheck %s --check-prefixes=CHECK,TUNIT
   3 ; RUN: opt -aa-pipeline=basic-aa -passes=attributor-cgscc -attributor-manifest-internal  -attributor-annotate-decl-cs -S < %s | FileCheck %s --check-prefixes=CHECK,CGSCC
   4
   5 define void @f() {
   6 ; TUNIT-LABEL: define {{[^@]+}}@f() {
   7 ; TUNIT-NEXT:  entry:
   8 ; TUNIT-NEXT:    [[A:%.*]] = alloca i32, align 1
   9 ; TUNIT-NEXT:    [[TMP0:%.*]] = load i32, ptr [[A]], align 1
  10 ; TUNIT-NEXT:    call void @g(i32 [[TMP0]])
  11 ; TUNIT-NEXT:    ret void
  12 ;
  13 ; CGSCC-LABEL: define {{[^@]+}}@f() {
  14 ; CGSCC-NEXT:  entry:
  15 ; CGSCC-NEXT:    call void @g(i32 undef)
  16 ; CGSCC-NEXT:    ret void
  17 ;
  18 entry:
  19   %a = alloca i32, align 1
  20   call void @g(ptr %a)
  21   ret void
  22 }
  23
  24 define internal void @g(ptr %a) {
  25 ; CHECK: Function Attrs: memory(readwrite, argmem: none)
  26 ; CHECK-LABEL: define {{[^@]+}}@g
  27 ; CHECK-SAME: (i32 [[TMP0:%.*]]) #[[ATTR0:[0-9]+]] {
  28 ; CHECK-NEXT:    [[A_PRIV:%.*]] = alloca i32, align 4
  29 ; CHECK-NEXT:    store i32 [[TMP0]], ptr [[A_PRIV]], align 4
  30 ; CHECK-NEXT:    [[AA:%.*]] = load i32, ptr [[A_PRIV]], align 1
  31 ; CHECK-NEXT:    call void @z(i32 [[AA]])
  32 ; CHECK-NEXT:    ret void
  33 ;
  34   %aa = load i32, ptr %a, align 1
  35   call void @z(i32 %aa)
  36   ret void
  37 }
  38
  39 declare void @z(i32)
  40
  41 ; Test2
  42 ; Different alignemnt privatizable arguments
  43 define internal i32 @test(ptr %X, ptr %Y) {
  44 ; CGSCC: Function Attrs: mustprogress nofree norecurse nosync nounwind willreturn memory(argmem: read)
  45 ; CGSCC-LABEL: define {{[^@]+}}@test
  46 ; CGSCC-SAME: (ptr nocapture nofree noundef nonnull readonly align 4 dereferenceable(4) [[X:%.*]], i64 [[TMP0:%.*]]) #[[ATTR1:[0-9]+]] {
  47 ; CGSCC-NEXT:    [[Y_PRIV:%.*]] = alloca i64, align 8
  48 ; CGSCC-NEXT:    store i64 [[TMP0]], ptr [[Y_PRIV]], align 4
  49 ; CGSCC-NEXT:    [[A:%.*]] = load i32, ptr [[X]], align 4
  50 ; CGSCC-NEXT:    [[B:%.*]] = load i64, ptr [[Y_PRIV]], align 8
  51 ; CGSCC-NEXT:    [[C:%.*]] = add i32 [[A]], 1
  52 ; CGSCC-NEXT:    [[D:%.*]] = add i64 [[B]], 1
  53 ; CGSCC-NEXT:    [[COND:%.*]] = icmp sgt i64 [[D]], -1
  54 ; CGSCC-NEXT:    br i1 [[COND]], label [[RETURN1:%.*]], label [[RETURN2:%.*]]
  55 ; CGSCC:       Return1:
  56 ; CGSCC-NEXT:    ret i32 [[C]]
  57 ; CGSCC:       Return2:
  58 ; CGSCC-NEXT:    ret i32 [[A]]
  59 ;
  60   %A = load i32, ptr %X
  61   %B = load i64, ptr %Y
  62   %C = add i32 %A, 1
  63   %D = add i64 %B, 1
  64   %cond = icmp sgt i64 %D, -1
  65   br i1 %cond, label %Return1, label %Return2
  66 Return1:
  67   ret i32 %C
  68 Return2:
  69   ret i32 %A
  70 }
  71
  72 define internal i32 @caller(ptr %A) {
  73 ; CGSCC: Function Attrs: mustprogress nofree nosync nounwind willreturn memory(none)
  74 ; CGSCC-LABEL: define {{[^@]+}}@caller
  75 ; CGSCC-SAME: (i32 [[TMP0:%.*]]) #[[ATTR2:[0-9]+]] {
  76 ; CGSCC-NEXT:    [[A_PRIV:%.*]] = alloca i32, align 4
  77 ; CGSCC-NEXT:    store i32 [[TMP0]], ptr [[A_PRIV]], align 4
  78 ; CGSCC-NEXT:    [[C:%.*]] = call i32 @test(ptr noalias nocapture nofree noundef nonnull readonly align 4 dereferenceable(4) [[A_PRIV]], i64 noundef 1) #[[ATTR3:[0-9]+]]
  79 ; CGSCC-NEXT:    ret i32 [[C]]
  80 ;
  81   %B = alloca i64
  82   store i64 1, ptr %B
  83   %C = call i32 @test(ptr %A, ptr %B)
  84   ret i32 %C
  85 }
  86
  87 define i32 @callercaller() {
  88 ; TUNIT: Function Attrs: mustprogress nofree norecurse nosync nounwind willreturn memory(none)
  89 ; TUNIT-LABEL: define {{[^@]+}}@callercaller
  90 ; TUNIT-SAME: () #[[ATTR1:[0-9]+]] {
  91 ; TUNIT-NEXT:    [[B:%.*]] = alloca i32, align 4
  92 ; TUNIT-NEXT:    ret i32 3
  93 ;
  94 ; CGSCC: Function Attrs: mustprogress nofree nosync nounwind willreturn memory(none)
  95 ; CGSCC-LABEL: define {{[^@]+}}@callercaller
  96 ; CGSCC-SAME: () #[[ATTR2]] {
  97 ; CGSCC-NEXT:    [[X:%.*]] = call i32 @caller(i32 noundef 2) #[[ATTR4:[0-9]+]]
  98 ; CGSCC-NEXT:    ret i32 [[X]]
  99 ;
 100   %B = alloca i32
 101   store i32 2, ptr %B
 102   %X = call i32 @caller(ptr %B)
 103   ret i32 %X
 104 }
 105 ;.
 106 ; TUNIT: attributes #[[ATTR0]] = { memory(readwrite, argmem: none) }
 107 ; TUNIT: attributes #[[ATTR1]] = { mustprogress nofree norecurse nosync nounwind willreturn memory(none) }
 108 ;.
 109 ; CGSCC: attributes #[[ATTR0]] = { memory(readwrite, argmem: none) }
 110 ; CGSCC: attributes #[[ATTR1]] = { mustprogress nofree norecurse nosync nounwind willreturn memory(argmem: read) }
 111 ; CGSCC: attributes #[[ATTR2]] = { mustprogress nofree nosync nounwind willreturn memory(none) }
 112 ; CGSCC: attributes #[[ATTR3]] = { nofree willreturn memory(read) }
 113 ; CGSCC: attributes #[[ATTR4]] = { nofree nosync willreturn }
 114 ;.