llvm/test/Transforms/Inline/arg-attr-propagation.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature
   2 ; RUN: opt -inline -S < %s | FileCheck --check-prefixes=CHECK,NO_ASSUME %s
   3 ; RUN: opt -inline -S --enable-knowledge-retention < %s | FileCheck %s --check-prefixes=CHECK,USE_ASSUME
   4
   5 ; The callee guarantees that the pointer argument is nonnull and dereferenceable.
   6 ; That information should transfer to the caller.
   7
   8 define i32 @callee(i32* dereferenceable(32) %t1) {
   9 ; CHECK-LABEL: define {{[^@]+}}@callee
  10 ; CHECK-SAME: (i32* dereferenceable(32) [[T1:%.*]]) {
  11 ; CHECK-NEXT:    [[T2:%.*]] = load i32, i32* [[T1]], align 4
  12 ; CHECK-NEXT:    ret i32 [[T2]]
  13 ;
  14   %t2 = load i32, i32* %t1
  15   ret i32 %t2
  16 }
  17
  18 define i32 @callee2(i32* dereferenceable(32) %t1, i32 noundef %t2) {
  19 ; CHECK-LABEL: define {{[^@]+}}@callee2
  20 ; CHECK-SAME: (i32* dereferenceable(32) [[T1:%.*]], i32 noundef [[T2:%.*]]) {
  21 ; CHECK-NEXT:    [[V:%.*]] = load i32, i32* [[T1]], align 4
  22 ; CHECK-NEXT:    ret i32 [[V]]
  23 ;
  24   %v = load i32, i32* %t1
  25   ret i32 %v
  26 }
  27
  28 ; FIXME: All dereferenceability information is lost.
  29 ; The caller argument could be known nonnull and dereferenceable(32).
  30
  31 define i32 @caller1(i32* %t1, i32 %t2) {
  32 ; NO_ASSUME-LABEL: define {{[^@]+}}@caller1
  33 ; NO_ASSUME-SAME: (i32* [[T1:%.*]], i32 [[T2:%.*]]) {
  34 ; NO_ASSUME-NEXT:    [[V_I:%.*]] = load i32, i32* [[T1]], align 4
  35 ; NO_ASSUME-NEXT:    ret i32 [[V_I]]
  36 ;
  37 ; USE_ASSUME-LABEL: define {{[^@]+}}@caller1
  38 ; USE_ASSUME-SAME: (i32* [[T1:%.*]], i32 [[T2:%.*]]) {
  39 ; USE_ASSUME-NEXT:    call void @llvm.assume(i1 true) [ "dereferenceable"(i32* [[T1]], i64 32), "noundef"(i32 [[T2]]) ]
  40 ; USE_ASSUME-NEXT:    [[V_I:%.*]] = load i32, i32* [[T1]], align 4
  41 ; USE_ASSUME-NEXT:    ret i32 [[V_I]]
  42 ;
  43   %v = tail call i32 @callee2(i32* dereferenceable(32) %t1, i32 noundef %t2)
  44   ret i32 %v
  45 }
  46
  47 ; The caller argument is nonnull, but that can be explicit.
  48 ; The dereferenceable amount could be increased.
  49
  50 define i32 @caller2(i32* dereferenceable(31) %t1) {
  51 ; NO_ASSUME-LABEL: define {{[^@]+}}@caller2
  52 ; NO_ASSUME-SAME: (i32* dereferenceable(31) [[T1:%.*]])
  53 ; NO_ASSUME-NEXT:    [[T2_I:%.*]] = load i32, i32* [[T1]]
  54 ; NO_ASSUME-NEXT:    ret i32 [[T2_I]]
  55 ;
  56 ; USE_ASSUME-LABEL: define {{[^@]+}}@caller2
  57 ; USE_ASSUME-SAME: (i32* dereferenceable(31) [[T1:%.*]])
  58 ; USE_ASSUME-NEXT:    call void @llvm.assume(i1 true) [ "dereferenceable"(i32* [[T1]], i64 32) ]
  59 ; USE_ASSUME-NEXT:    [[T2_I:%.*]] = load i32, i32* [[T1]]
  60 ; USE_ASSUME-NEXT:    ret i32 [[T2_I]]
  61 ;
  62   %t2 = tail call i32 @callee(i32* dereferenceable(32) %t1)
  63   ret i32 %t2
  64 }
  65
  66 ; The caller argument is nonnull, but that can be explicit.
  67 ; Make sure that we don't propagate a smaller dereferenceable amount.
  68
  69 define i32 @caller3(i32* dereferenceable(33) %t1) {
  70 ; CHECK-LABEL: define {{[^@]+}}@caller3
  71 ; CHECK-SAME: (i32* dereferenceable(33) [[T1:%.*]])
  72 ; CHECK-NEXT:    [[T2_I:%.*]] = load i32, i32* [[T1]]
  73 ; CHECK-NEXT:    ret i32 [[T2_I]]
  74 ;
  75   %t2 = tail call i32 @callee(i32* dereferenceable(32) %t1)
  76   ret i32 %t2
  77 }
  78