diff options
Diffstat (limited to 'test/Transforms/SROA/preserve-nonnull.ll')
| -rw-r--r-- | test/Transforms/SROA/preserve-nonnull.ll | 94 |
1 files changed, 80 insertions, 14 deletions
diff --git a/test/Transforms/SROA/preserve-nonnull.ll b/test/Transforms/SROA/preserve-nonnull.ll index fc5ce6a445fa0..a29da6dc2c377 100644 --- a/test/Transforms/SROA/preserve-nonnull.ll +++ b/test/Transforms/SROA/preserve-nonnull.ll @@ -3,24 +3,90 @@ ; Make sure that SROA doesn't lose nonnull metadata ; on loads from allocas that get optimized out. -; CHECK-LABEL: define float* @yummy_nonnull -; CHECK: [[RETURN:%(.*)]] = load float*, float** %arg, align 8 -; CHECK: [[ASSUME:%(.*)]] = icmp ne float* {{.*}}[[RETURN]], null -; CHECK: call void @llvm.assume(i1 {{.*}}[[ASSUME]]) -; CHECK: ret float* {{.*}}[[RETURN]] +declare void @llvm.memcpy.p0i8.p0i8.i64(i8* nocapture writeonly, i8* nocapture readonly, i64, i32, i1) -define float* @yummy_nonnull(float** %arg) { -entry-block: - %buf = alloca float* +; Check that we do basic propagation of nonnull when rewriting. +define i8* @propagate_nonnull(i32* %v) { +; CHECK-LABEL: define i8* @propagate_nonnull( +; CHECK-NEXT: entry: +; CHECK-NEXT: %[[A:.*]] = alloca i8* +; CHECK-NEXT: %[[V_CAST:.*]] = bitcast i32* %v to i8* +; CHECK-NEXT: store i8* %[[V_CAST]], i8** %[[A]] +; CHECK-NEXT: %[[LOAD:.*]] = load volatile i8*, i8** %[[A]], !nonnull !0 +; CHECK-NEXT: ret i8* %[[LOAD]] +entry: + %a = alloca [2 x i8*] + %a.gep0 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 0 + %a.gep1 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 1 + %a.gep0.cast = bitcast i8** %a.gep0 to i32** + %a.gep1.cast = bitcast i8** %a.gep1 to i32** + store i32* %v, i32** %a.gep1.cast + store i32* null, i32** %a.gep0.cast + %load = load volatile i8*, i8** %a.gep1, !nonnull !0 + ret i8* %load +} - %_arg_i8 = bitcast float** %arg to i8* - %_buf_i8 = bitcast float** %buf to i8* - call void @llvm.memcpy.p0i8.p0i8.i64(i8* %_buf_i8, i8* %_arg_i8, i64 8, i32 8, i1 false) +define float* @turn_nonnull_into_assume(float** %arg) { +; CHECK-LABEL: define float* @turn_nonnull_into_assume( +; CHECK-NEXT: entry: +; CHECK-NEXT: %[[RETURN:.*]] = load float*, float** %arg, align 8 +; CHECK-NEXT: %[[ASSUME:.*]] = icmp ne float* %[[RETURN]], null +; CHECK-NEXT: call void @llvm.assume(i1 %[[ASSUME]]) +; CHECK-NEXT: ret float* %[[RETURN]] +entry: + %buf = alloca float* + %_arg_i8 = bitcast float** %arg to i8* + %_buf_i8 = bitcast float** %buf to i8* + call void @llvm.memcpy.p0i8.p0i8.i64(i8* %_buf_i8, i8* %_arg_i8, i64 8, i32 8, i1 false) + %ret = load float*, float** %buf, align 8, !nonnull !0 + ret float* %ret +} - %ret = load float*, float** %buf, align 8, !nonnull !0 - ret float* %ret +; Make sure we properly handle the !nonnull attribute when we convert +; a pointer load to an integer load. +; FIXME: While this doesn't do anythnig actively harmful today, it really +; should propagate the !nonnull metadata to range metadata. The irony is, it +; *does* initially, but then we lose that !range metadata before we finish +; SROA. +define i8* @propagate_nonnull_to_int() { +; CHECK-LABEL: define i8* @propagate_nonnull_to_int( +; CHECK-NEXT: entry: +; CHECK-NEXT: %[[A:.*]] = alloca i64 +; CHECK-NEXT: store i64 42, i64* %[[A]] +; CHECK-NEXT: %[[LOAD:.*]] = load volatile i64, i64* %[[A]] +; CHECK-NEXT: %[[CAST:.*]] = inttoptr i64 %[[LOAD]] to i8* +; CHECK-NEXT: ret i8* %[[CAST]] +entry: + %a = alloca [2 x i8*] + %a.gep0 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 0 + %a.gep1 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 1 + %a.gep0.cast = bitcast i8** %a.gep0 to i64* + %a.gep1.cast = bitcast i8** %a.gep1 to i64* + store i64 42, i64* %a.gep1.cast + store i64 0, i64* %a.gep0.cast + %load = load volatile i8*, i8** %a.gep1, !nonnull !0 + ret i8* %load } -declare void @llvm.memcpy.p0i8.p0i8.i64(i8* nocapture writeonly, i8* nocapture readonly, i64, i32, i1) +; Make sure we properly handle the !nonnull attribute when we convert +; a pointer load to an integer load and immediately promote it to an SSA +; register. This can fail in interesting ways due to the rewrite iteration of +; SROA, resulting in PR32902. +define i8* @propagate_nonnull_to_int_and_promote() { +; CHECK-LABEL: define i8* @propagate_nonnull_to_int_and_promote( +; CHECK-NEXT: entry: +; CHECK-NEXT: %[[PROMOTED_VALUE:.*]] = inttoptr i64 42 to i8* +; CHECK-NEXT: ret i8* %[[PROMOTED_VALUE]] +entry: + %a = alloca [2 x i8*], align 8 + %a.gep0 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 0 + %a.gep1 = getelementptr [2 x i8*], [2 x i8*]* %a, i32 0, i32 1 + %a.gep0.cast = bitcast i8** %a.gep0 to i64* + %a.gep1.cast = bitcast i8** %a.gep1 to i64* + store i64 42, i64* %a.gep1.cast + store i64 0, i64* %a.gep0.cast + %load = load i8*, i8** %a.gep1, align 8, !nonnull !0 + ret i8* %load +} !0 = !{} |