130 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 22; Verifies that LICM is disabled for loops that contains coro.suspend.3; RUN: opt -S < %s -passes=licm | FileCheck %s4 5define i64 @licm(i64 %n) #0 {6; CHECK-LABEL: define i64 @licm7; CHECK-SAME: (i64 [[N:%.*]]) {8; CHECK-NEXT: entry:9; CHECK-NEXT: [[P:%.*]] = alloca i64, align 810; CHECK-NEXT: br label [[BB0:%.*]]11; CHECK: bb0:12; CHECK-NEXT: br label [[LOOP:%.*]]13; CHECK: loop:14; CHECK-NEXT: [[I:%.*]] = phi i64 [ 0, [[BB0]] ], [ [[T5:%.*]], [[AWAIT_READY:%.*]] ]15; CHECK-NEXT: [[T5]] = add i64 [[I]], 116; CHECK-NEXT: [[SUSPEND:%.*]] = call i8 @llvm.coro.suspend(token none, i1 false)17; CHECK-NEXT: switch i8 [[SUSPEND]], label [[BB2:%.*]] [18; CHECK-NEXT: i8 0, label [[AWAIT_READY]]19; CHECK-NEXT: ]20; CHECK: await.ready:21; CHECK-NEXT: store i64 1, ptr [[P]], align 422; CHECK-NEXT: [[T6:%.*]] = icmp ult i64 [[T5]], [[N]]23; CHECK-NEXT: br i1 [[T6]], label [[LOOP]], label [[BB2]]24; CHECK: bb2:25; CHECK-NEXT: call void @llvm.coro.end(ptr null, i1 false, token none)26; CHECK-NEXT: ret i64 027;28entry:29 %p = alloca i6430 br label %bb031 32bb0:33 br label %loop34 35loop:36 %i = phi i64 [ 0, %bb0 ], [ %t5, %await.ready ]37 %t5 = add i64 %i, 138 %suspend = call i8 @llvm.coro.suspend(token none, i1 false)39 switch i8 %suspend, label %bb2 [40 i8 0, label %await.ready41 ]42 43await.ready:44 store i64 1, ptr %p45 %t6 = icmp ult i64 %t5, %n46 br i1 %t6, label %loop, label %bb247 48bb2:49 call void @llvm.coro.end(ptr null, i1 false, token none)50 ret i64 051}52 53@tls = thread_local global i32 054 55define i64 @hoist_threadlocal() presplitcoroutine {56; CHECK-LABEL: define i64 @hoist_threadlocal57; CHECK-SAME: () #[[ATTR0:[0-9]+]] {58; CHECK-NEXT: entry:59; CHECK-NEXT: [[P:%.*]] = alloca i64, align 860; CHECK-NEXT: br label [[LOOP_PREHEADER:%.*]]61; CHECK: loop.preheader:62; CHECK-NEXT: br label [[FOR_BODY:%.*]]63; CHECK: for.body:64; CHECK-NEXT: [[I:%.*]] = phi i64 [ 0, [[LOOP_PREHEADER]] ], [ [[I_NEXT:%.*]], [[LOOP_END:%.*]] ]65; CHECK-NEXT: [[I_NEXT]] = add i64 [[I]], 166; CHECK-NEXT: [[THREAD_LOCAL_0:%.*]] = call ptr @llvm.threadlocal.address.p0(ptr @tls)67; CHECK-NEXT: [[READONLY_0:%.*]] = call ptr @readonly_funcs()68; CHECK-NEXT: [[SUSPEND:%.*]] = call i8 @llvm.coro.suspend(token none, i1 false)69; CHECK-NEXT: switch i8 [[SUSPEND]], label [[EXIT:%.*]] [70; CHECK-NEXT: i8 0, label [[AWAIT_READY:%.*]]71; CHECK-NEXT: ]72; CHECK: await.ready:73; CHECK-NEXT: [[THREAD_LOCAL_1:%.*]] = call ptr @llvm.threadlocal.address.p0(ptr @tls)74; CHECK-NEXT: [[READONLY_1:%.*]] = call ptr @readonly_funcs()75; CHECK-NEXT: [[CMP_0:%.*]] = icmp eq ptr [[THREAD_LOCAL_0]], [[THREAD_LOCAL_1]]76; CHECK-NEXT: [[CMP_1:%.*]] = icmp eq ptr [[READONLY_0]], [[READONLY_1]]77; CHECK-NEXT: [[CMP:%.*]] = and i1 [[CMP_0]], [[CMP_1]]78; CHECK-NEXT: br i1 [[CMP]], label [[NOT_REACHABLE:%.*]], label [[LOOP_END]]79; CHECK: not.reachable:80; CHECK-NEXT: call void @not.reachable()81; CHECK-NEXT: br label [[LOOP_END]]82; CHECK: loop.end:83; CHECK-NEXT: br i1 [[CMP]], label [[EXIT]], label [[FOR_BODY]]84; CHECK: exit:85; CHECK-NEXT: call void @llvm.coro.end(ptr null, i1 false, token none)86; CHECK-NEXT: ret i64 087;88entry:89 %p = alloca i6490 br label %loop.preheader91 92loop.preheader:93 br label %for.body94 95for.body:96 %i = phi i64 [ 0, %loop.preheader ], [ %i.next, %loop.end ]97 %i.next = add i64 %i, 198 %thread_local.0 = call ptr @llvm.threadlocal.address(ptr @tls)99 %readonly.0 = call ptr @readonly_funcs()100 %suspend = call i8 @llvm.coro.suspend(token none, i1 false)101 switch i8 %suspend, label %exit [102 i8 0, label %await.ready103 ]104 105await.ready:106 %thread_local.1 = call ptr @llvm.threadlocal.address(ptr @tls)107 %readonly.1 = call ptr @readonly_funcs()108 %cmp.0 = icmp eq ptr %thread_local.0, %thread_local.1109 %cmp.1 = icmp eq ptr %readonly.0, %readonly.1110 %cmp = and i1 %cmp.0, %cmp.1111 br i1 %cmp, label %not.reachable, label %loop.end112 113not.reachable:114 call void @not.reachable()115 br label %loop.end116 117loop.end:118 %loop.end.cond = icmp ugt i64 %i.next, 5119 br i1 %cmp, label %exit, label %for.body120 121exit:122 call void @llvm.coro.end(ptr null, i1 false, token none)123 ret i64 0124}125 126declare i8 @llvm.coro.suspend(token, i1)127declare nonnull ptr @readonly_funcs() readonly128declare nonnull ptr @llvm.threadlocal.address(ptr nonnull) nounwind readnone willreturn129declare void @not.reachable()130