[llvm] dfd27eb - [MemCpyOpt] Add test for memcpy in loop (NFC)

[Nikita Popov via llvm-commits]

Author: Nikita Popov
Date: 2021-03-12T22:54:24+01:00
New Revision: dfd27ebbd0eb137c9a439b7c537bb87ba903efd3

URL: https://github.com/llvm/llvm-project/commit/dfd27ebbd0eb137c9a439b7c537bb87ba903efd3
DIFF: https://github.com/llvm/llvm-project/commit/dfd27ebbd0eb137c9a439b7c537bb87ba903efd3.diff

LOG: [MemCpyOpt] Add test for memcpy in loop (NFC)

This is currently not being optimized.

Added: 
    llvm/test/Transforms/MemCpyOpt/memcpy-in-loop.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/MemCpyOpt/memcpy-in-loop.ll b/llvm/test/Transforms/MemCpyOpt/memcpy-in-loop.ll
new file mode 100644
index 000000000000..c9786987fa21
--- /dev/null
+++ b/llvm/test/Transforms/MemCpyOpt/memcpy-in-loop.ll
@@ -0,0 +1,81 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -S -memcpyopt < %s | FileCheck %s
+
+define void @test_copy_uninit([1000 x [1000 x i32]]* %arg) {
+; CHECK-LABEL: @test_copy_uninit(
+; CHECK-NEXT:  start:
+; CHECK-NEXT:    [[ALLOCA:%.*]] = alloca [1000 x i32], align 4
+; CHECK-NEXT:    [[ALLOCA_I8:%.*]] = bitcast [1000 x i32]* [[ALLOCA]] to i8*
+; CHECK-NEXT:    [[BEGIN:%.*]] = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* [[ARG:%.*]], i64 0, i64 0
+; CHECK-NEXT:    [[END:%.*]] = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* [[ARG]], i64 0, i64 1000
+; CHECK-NEXT:    br label [[LOOP:%.*]]
+; CHECK:       loop:
+; CHECK-NEXT:    [[CURRENT:%.*]] = phi [1000 x i32]* [ [[BEGIN]], [[START:%.*]] ], [ [[NEXT:%.*]], [[LOOP]] ]
+; CHECK-NEXT:    [[CURRENT_I8:%.*]] = bitcast [1000 x i32]* [[CURRENT]] to i8*
+; CHECK-NEXT:    call void @llvm.memcpy.p0i8.p0i8.i64(i8* nonnull align 4 dereferenceable(4000) [[CURRENT_I8]], i8* nonnull align 4 dereferenceable(4000) [[ALLOCA_I8]], i64 4000, i1 false)
+; CHECK-NEXT:    [[NEXT]] = getelementptr inbounds [1000 x i32], [1000 x i32]* [[CURRENT]], i64 1
+; CHECK-NEXT:    [[COND:%.*]] = icmp eq [1000 x i32]* [[NEXT]], [[END]]
+; CHECK-NEXT:    br i1 [[COND]], label [[EXIT:%.*]], label [[LOOP]]
+; CHECK:       exit:
+; CHECK-NEXT:    ret void
+;
+start:
+  %alloca = alloca [1000 x i32], align 4
+  %alloca.i8 = bitcast [1000 x i32]* %alloca to i8*
+  %begin = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* %arg, i64 0, i64 0
+  %end = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* %arg, i64 0, i64 1000
+  br label %loop
+
+loop:                                             ; preds = %loop, %start
+  %current = phi [1000 x i32]* [ %begin, %start ], [ %next, %loop ]
+  %current.i8 = bitcast [1000 x i32]* %current to i8*
+  call void @llvm.memcpy.p0i8.p0i8.i64(i8* nonnull align 4 dereferenceable(4000) %current.i8, i8* nonnull align 4 dereferenceable(4000) %alloca.i8, i64 4000, i1 false)
+  %next = getelementptr inbounds [1000 x i32], [1000 x i32]* %current, i64 1
+  %cond = icmp eq [1000 x i32]* %next, %end
+  br i1 %cond, label %exit, label %loop
+
+exit:                                             ; preds = %loop
+  ret void
+}
+
+define void @test_copy_zero([1000 x [1000 x i32]]* %arg) {
+; CHECK-LABEL: @test_copy_zero(
+; CHECK-NEXT:  start:
+; CHECK-NEXT:    [[ALLOCA:%.*]] = alloca [1000 x i32], align 4
+; CHECK-NEXT:    [[ALLOCA_I8:%.*]] = bitcast [1000 x i32]* [[ALLOCA]] to i8*
+; CHECK-NEXT:    call void @llvm.memset.p0i8.i64(i8* nonnull align 4 dereferenceable(4000) [[ALLOCA_I8]], i8 0, i64 4000, i1 false)
+; CHECK-NEXT:    [[BEGIN:%.*]] = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* [[ARG:%.*]], i64 0, i64 0
+; CHECK-NEXT:    [[END:%.*]] = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* [[ARG]], i64 0, i64 1000
+; CHECK-NEXT:    br label [[LOOP:%.*]]
+; CHECK:       loop:
+; CHECK-NEXT:    [[CURRENT:%.*]] = phi [1000 x i32]* [ [[BEGIN]], [[START:%.*]] ], [ [[NEXT:%.*]], [[LOOP]] ]
+; CHECK-NEXT:    [[CURRENT_I8:%.*]] = bitcast [1000 x i32]* [[CURRENT]] to i8*
+; CHECK-NEXT:    call void @llvm.memcpy.p0i8.p0i8.i64(i8* nonnull align 4 dereferenceable(4000) [[CURRENT_I8]], i8* nonnull align 4 dereferenceable(4000) [[ALLOCA_I8]], i64 4000, i1 false)
+; CHECK-NEXT:    [[NEXT]] = getelementptr inbounds [1000 x i32], [1000 x i32]* [[CURRENT]], i64 1
+; CHECK-NEXT:    [[COND:%.*]] = icmp eq [1000 x i32]* [[NEXT]], [[END]]
+; CHECK-NEXT:    br i1 [[COND]], label [[EXIT:%.*]], label [[LOOP]]
+; CHECK:       exit:
+; CHECK-NEXT:    ret void
+;
+start:
+  %alloca = alloca [1000 x i32], align 4
+  %alloca.i8 = bitcast [1000 x i32]* %alloca to i8*
+  call void @llvm.memset.p0i8.i64(i8* nonnull align 4 dereferenceable(4000) %alloca.i8, i8 0, i64 4000, i1 false)
+  %begin = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* %arg, i64 0, i64 0
+  %end = getelementptr inbounds [1000 x [1000 x i32]], [1000 x [1000 x i32]]* %arg, i64 0, i64 1000
+  br label %loop
+
+loop:                                             ; preds = %loop, %start
+  %current = phi [1000 x i32]* [ %begin, %start ], [ %next, %loop ]
+  %current.i8 = bitcast [1000 x i32]* %current to i8*
+  call void @llvm.memcpy.p0i8.p0i8.i64(i8* nonnull align 4 dereferenceable(4000) %current.i8, i8* nonnull align 4 dereferenceable(4000) %alloca.i8, i64 4000, i1 false)
+  %next = getelementptr inbounds [1000 x i32], [1000 x i32]* %current, i64 1
+  %cond = icmp eq [1000 x i32]* %next, %end
+  br i1 %cond, label %exit, label %loop
+
+exit:                                             ; preds = %loop
+  ret void
+}
+
+declare void @llvm.memset.p0i8.i64(i8* nocapture writeonly, i8, i64, i1 immarg)
+declare void @llvm.memcpy.p0i8.p0i8.i64(i8* noalias nocapture writeonly, i8* noalias nocapture readonly, i64, i1 immarg)