1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -basic-aa -memcpyopt -S %s -verify-memoryssa -opaque-pointers | FileCheck %s
3
4define void @test_memset_memcpy(ptr %src, i64 %src_size, ptr noalias %dst, i64 %dst_size, i8 %c) {
5; CHECK-LABEL: @test_memset_memcpy(
6; CHECK-NEXT:    [[TMP1:%.*]] = icmp ule i64 [[DST_SIZE:%.*]], [[SRC_SIZE:%.*]]
7; CHECK-NEXT:    [[TMP2:%.*]] = sub i64 [[DST_SIZE]], [[SRC_SIZE]]
8; CHECK-NEXT:    [[TMP3:%.*]] = select i1 [[TMP1]], i64 0, i64 [[TMP2]]
9; CHECK-NEXT:    [[TMP4:%.*]] = getelementptr i8, ptr [[DST:%.*]], i64 [[SRC_SIZE]]
10; CHECK-NEXT:    call void @llvm.memset.p0.i64(ptr align 1 [[TMP4]], i8 [[C:%.*]], i64 [[TMP3]], i1 false)
11; CHECK-NEXT:    call void @llvm.memcpy.p0.p0.i64(ptr [[DST]], ptr [[SRC:%.*]], i64 [[SRC_SIZE]], i1 false)
12; CHECK-NEXT:    ret void
13;
14  call void @llvm.memset.p0.i64(ptr %dst, i8 %c, i64 %dst_size, i1 false)
15  call void @llvm.memcpy.p0.p0.i64(ptr %dst, ptr %src, i64 %src_size, i1 false)
16  ret void
17}
18
19%a = type { i64, i64, i64 }
20%b = type { i32, i32, i32 }
21
22define void @test_different_gep_source_elements(ptr %src) {
23; CHECK-LABEL: @test_different_gep_source_elements(
24; CHECK-NEXT:    [[PB:%.*]] = getelementptr [[B:%.*]], ptr [[SRC:%.*]], i64 0, i32 1
25; CHECK-NEXT:    [[PA:%.*]] = getelementptr [[A:%.*]], ptr [[SRC]], i64 0, i32 1
26; CHECK-NEXT:    [[PA2:%.*]] = getelementptr [[A]], ptr [[SRC]], i64 0, i32 2
27; CHECK-NEXT:    call void @llvm.memset.p0.i64(ptr align 4 [[PB]], i8 0, i64 20, i1 false)
28; CHECK-NEXT:    ret void
29;
30  %pb = getelementptr %b, ptr %src, i64 0, i32 1
31  store i64 0, ptr %pb
32  %pa = getelementptr %a, ptr %src, i64 0, i32 1
33  store i64 0, ptr %pa
34  %pa2 = getelementptr %a, ptr %src, i64 0, i32 2
35  store i64 0, ptr %pa2
36  ret void
37}
38
39define void @test_gep_of_vscale_non_const_gep(ptr %p, i64 %idx) {
40; CHECK-LABEL: @test_gep_of_vscale_non_const_gep(
41; CHECK-NEXT:    [[G1:%.*]] = getelementptr <vscale x 16 x i8>, ptr [[P:%.*]], i64 [[IDX:%.*]], i32 1
42; CHECK-NEXT:    [[G2:%.*]] = getelementptr <vscale x 16 x i8>, ptr [[P]], i64 [[IDX]], i32 5
43; CHECK-NEXT:    [[H1:%.*]] = getelementptr i8, ptr [[G1]], i64 2
44; CHECK-NEXT:    [[H2:%.*]] = getelementptr i8, ptr [[G2]], i64 6
45; CHECK-NEXT:    call void @llvm.memset.p0.i64(ptr align 4 [[H1]], i8 0, i64 16, i1 false)
46; CHECK-NEXT:    ret void
47;
48  %g1 = getelementptr <vscale x 16 x i8>, ptr %p, i64 %idx, i32 1
49  %g2 = getelementptr <vscale x 16 x i8>, ptr %p, i64 %idx, i32 5
50  %h1 = getelementptr i8, ptr %g1, i64 2
51  %h2 = getelementptr i8, ptr %g2, i64 6
52  store i64 0, ptr %h1
53  store i64 0, ptr %h2
54  ret void
55}
56
57declare void @llvm.memset.p0.i64(ptr nocapture, i8, i64, i1)
58declare void @llvm.memcpy.p0.p0.i64(ptr nocapture, ptr nocapture readonly, i64, i1)
59