test/Transforms/SROA/preserve-nonnull.ll

   1 ; RUN: opt < %s -sroa -S | FileCheck %s
   2 ;
   3 ; Make sure that SROA doesn't lose nonnull metadata
   4 ; on loads from allocas that get optimized out.
   5
   6 declare void @llvm.memcpy.p0i8.p0i8.i64(i8* nocapture writeonly, i8* nocapture readonly, i64, i1)
   7
   8 ; Check that we do basic propagation of nonnull when rewriting.
   9 define i8* @propagate_nonnull(i32* %v) {
  10 ; CHECK-LABEL: define i8* @propagate_nonnull(
  11 ; CHECK-NEXT:  entry:
  12 ; CHECK-NEXT:    %[[A:.*]] = alloca i8*
  13 ; CHECK-NEXT:    %[[V_CAST:.*]] = bitcast i32* %v to i8*
  14 ; CHECK-NEXT:    store i8* %[[V_CAST]], i8** %[[A]]
  15 ; CHECK-NEXT:    %[[LOAD:.*]] = load volatile i8*, i8** %[[A]], !nonnull !0
  16 ; CHECK-NEXT:    ret i8* %[[LOAD]]
  17 entry:
  18   %a = alloca [2 x i8*]
  19   %a.gep0 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 0
  20   %a.gep1 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 1
  21   %a.gep0.cast = bitcast i8** %a.gep0 to i32**
  22   %a.gep1.cast = bitcast i8** %a.gep1 to i32**
  23   store i32* %v, i32** %a.gep1.cast
  24   store i32* null, i32** %a.gep0.cast
  25   %load = load volatile i8*, i8** %a.gep1, !nonnull !0
  26   ret i8* %load
  27 }
  28
  29 define float* @turn_nonnull_into_assume(float** %arg) {
  30 ; CHECK-LABEL: define float* @turn_nonnull_into_assume(
  31 ; CHECK-NEXT:  entry:
  32 ; CHECK-NEXT:    %[[RETURN:.*]] = load float*, float** %arg, align 8
  33 ; CHECK-NEXT:    %[[ASSUME:.*]] = icmp ne float* %[[RETURN]], null
  34 ; CHECK-NEXT:    call void @llvm.assume(i1 %[[ASSUME]])
  35 ; CHECK-NEXT:    ret float* %[[RETURN]]
  36 entry:
  37   %buf = alloca float*
  38   %_arg_i8 = bitcast float** %arg to i8*
  39   %_buf_i8 = bitcast float** %buf to i8*
  40   call void @llvm.memcpy.p0i8.p0i8.i64(i8* align 8 %_buf_i8, i8* align 8 %_arg_i8, i64 8, i1 false)
  41   %ret = load float*, float** %buf, align 8, !nonnull !0
  42   ret float* %ret
  43 }
  44
  45 ; Make sure we properly handle the !nonnull attribute when we convert
  46 ; a pointer load to an integer load.
  47 ; FIXME: While this doesn't do anythnig actively harmful today, it really
  48 ; should propagate the !nonnull metadata to range metadata. The irony is, it
  49 ; *does* initially, but then we lose that !range metadata before we finish
  50 ; SROA.
  51 define i8* @propagate_nonnull_to_int() {
  52 ; CHECK-LABEL: define i8* @propagate_nonnull_to_int(
  53 ; CHECK-NEXT:  entry:
  54 ; CHECK-NEXT:    %[[A:.*]] = alloca i64
  55 ; CHECK-NEXT:    store i64 42, i64* %[[A]]
  56 ; CHECK-NEXT:    %[[LOAD:.*]] = load volatile i64, i64* %[[A]]
  57 ; CHECK-NEXT:    %[[CAST:.*]] = inttoptr i64 %[[LOAD]] to i8*
  58 ; CHECK-NEXT:    ret i8* %[[CAST]]
  59 entry:
  60   %a = alloca [2 x i8*]
  61   %a.gep0 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 0
  62   %a.gep1 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 1
  63   %a.gep0.cast = bitcast i8** %a.gep0 to i64*
  64   %a.gep1.cast = bitcast i8** %a.gep1 to i64*
  65   store i64 42, i64* %a.gep1.cast
  66   store i64 0, i64* %a.gep0.cast
  67   %load = load volatile i8*, i8** %a.gep1, !nonnull !0
  68   ret i8* %load
  69 }
  70
  71 ; Make sure we properly handle the !nonnull attribute when we convert
  72 ; a pointer load to an integer load and immediately promote it to an SSA
  73 ; register. This can fail in interesting ways due to the rewrite iteration of
  74 ; SROA, resulting in PR32902.
  75 define i8* @propagate_nonnull_to_int_and_promote() {
  76 ; CHECK-LABEL: define i8* @propagate_nonnull_to_int_and_promote(
  77 ; CHECK-NEXT:  entry:
  78 ; CHECK-NEXT:    %[[PROMOTED_VALUE:.*]] = inttoptr i64 42 to i8*
  79 ; CHECK-NEXT:    ret i8* %[[PROMOTED_VALUE]]
  80 entry:
  81   %a = alloca [2 x i8*], align 8
  82   %a.gep0 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 0
  83   %a.gep1 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 1
  84   %a.gep0.cast = bitcast i8** %a.gep0 to i64*
  85   %a.gep1.cast = bitcast i8** %a.gep1 to i64*
  86   store i64 42, i64* %a.gep1.cast
  87   store i64 0, i64* %a.gep0.cast
  88   %load = load i8*, i8** %a.gep1, align 8, !nonnull !0
  89   ret i8* %load
  90 }
  91
  92 !0 = !{}