| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,79 +1,74 @@ | ||
| ; Tests that variables of different type in a Corotuine whose lifetime range is not overlapping each other | ||
| ; re-use the same slot in Coroutine frame. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split<reuse-storage>),simplifycfg,early-cse' -S | FileCheck %s | ||
| %"struct.task::promise_type" = type { i8 } | ||
| %struct.awaitable = type { i8 } | ||
| %struct.big_structure = type { [500 x i8] } | ||
| %struct.big_structure.2 = type { [300 x i8] } | ||
| declare ptr @malloc(i64) | ||
| declare void @consume(ptr) | ||
| declare void @consume.2(ptr) | ||
| define void @a(i1 zeroext %cond) presplitcoroutine { | ||
| entry: | ||
| %__promise = alloca %"struct.task::promise_type", align 1 | ||
| %a = alloca %struct.big_structure, align 1 | ||
| %ref.tmp7 = alloca %struct.awaitable, align 1 | ||
| %b = alloca %struct.big_structure.2, align 1 | ||
| %ref.tmp18 = alloca %struct.awaitable, align 1 | ||
| %0 = call token @llvm.coro.id(i32 16, ptr nonnull %__promise, ptr @a, ptr null) | ||
| br label %init.ready | ||
| init.ready: | ||
| %1 = call noalias nonnull ptr @llvm.coro.begin(token %0, ptr null) | ||
| call void @llvm.lifetime.start.p0(i64 1, ptr nonnull %__promise) | ||
| br i1 %cond, label %if.then, label %if.else | ||
| if.then: | ||
| call void @llvm.lifetime.start.p0(i64 500, ptr nonnull %a) | ||
| call void @consume(ptr nonnull %a) | ||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %coro.ret [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %cleanup1 | ||
| ] | ||
| await.ready: | ||
| call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a) | ||
| br label %cleanup1 | ||
| if.else: | ||
| call void @llvm.lifetime.start.p0(i64 300, ptr nonnull %b) | ||
| call void @consume.2(ptr nonnull %b) | ||
| %save2 = call token @llvm.coro.save(ptr null) | ||
| %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false) | ||
| switch i8 %suspend2, label %coro.ret [ | ||
| i8 0, label %await2.ready | ||
| i8 1, label %cleanup2 | ||
| ] | ||
| await2.ready: | ||
| call void @llvm.lifetime.end.p0(i64 300, ptr nonnull %b) | ||
| br label %cleanup2 | ||
| cleanup1: | ||
| call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a) | ||
| br label %cleanup | ||
| cleanup2: | ||
| call void @llvm.lifetime.end.p0(i64 300, ptr nonnull %b) | ||
| br label %cleanup | ||
| cleanup: | ||
| call ptr @llvm.coro.free(token %0, ptr %1) | ||
| br label %coro.ret | ||
| coro.ret: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
| ; CHECK: %a.Frame = type { ptr, ptr, %"struct.task::promise_type", %struct.big_structure, i1 } | ||
| ; CHECK-LABEL: @a.resume( | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) | ||
| declare i1 @llvm.coro.alloc(token) #3 | ||
| declare i64 @llvm.coro.size.i64() #5 | ||
| declare ptr @llvm.coro.begin(token, ptr writeonly) #3 | ||
| declare token @llvm.coro.save(ptr) #3 | ||
| declare ptr @llvm.coro.frame() #5 | ||
| declare i8 @llvm.coro.suspend(token, i1) #3 | ||
| declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2 | ||
| declare i1 @llvm.coro.end(ptr, i1) #3 | ||
| declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4 | ||
| declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4 |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,79 +1,74 @@ | ||
| ; Tests that variables of different type with incompatible alignment in a Corotuine whose lifetime | ||
| ; range is not overlapping each other should not re-use the same slot in Coroutine frame. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split<reuse-storage>),simplifycfg,early-cse' -S | FileCheck %s | ||
| %"struct.task::promise_type" = type { i8 } | ||
| %struct.awaitable = type { i8 } | ||
| %struct.big_structure = type { [500 x i8] } | ||
| %struct.big_structure.2 = type { [300 x i8] } | ||
| declare ptr @malloc(i64) | ||
| declare void @consume(ptr) | ||
| declare void @consume.2(ptr) | ||
| define void @a(i1 zeroext %cond) presplitcoroutine { | ||
| entry: | ||
| %__promise = alloca %"struct.task::promise_type", align 1 | ||
| %a = alloca %struct.big_structure, align 1 | ||
| %ref.tmp7 = alloca %struct.awaitable, align 1 | ||
| %b = alloca %struct.big_structure.2, align 32 | ||
| %ref.tmp18 = alloca %struct.awaitable, align 1 | ||
| %0 = call token @llvm.coro.id(i32 16, ptr nonnull %__promise, ptr @a, ptr null) | ||
| br label %init.ready | ||
| init.ready: | ||
| %1 = call noalias nonnull ptr @llvm.coro.begin(token %0, ptr null) | ||
| call void @llvm.lifetime.start.p0(i64 1, ptr nonnull %__promise) | ||
| br i1 %cond, label %if.then, label %if.else | ||
| if.then: | ||
| call void @llvm.lifetime.start.p0(i64 500, ptr nonnull %a) | ||
| call void @consume(ptr nonnull %a) | ||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %coro.ret [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %cleanup1 | ||
| ] | ||
| await.ready: | ||
| call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a) | ||
| br label %cleanup1 | ||
| if.else: | ||
| call void @llvm.lifetime.start.p0(i64 300, ptr nonnull %b) | ||
| call void @consume.2(ptr nonnull %b) | ||
| %save2 = call token @llvm.coro.save(ptr null) | ||
| %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false) | ||
| switch i8 %suspend2, label %coro.ret [ | ||
| i8 0, label %await2.ready | ||
| i8 1, label %cleanup2 | ||
| ] | ||
| await2.ready: | ||
| call void @llvm.lifetime.end.p0(i64 300, ptr nonnull %b) | ||
| br label %cleanup2 | ||
| cleanup1: | ||
| call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a) | ||
| br label %cleanup | ||
| cleanup2: | ||
| call void @llvm.lifetime.end.p0(i64 300, ptr nonnull %b) | ||
| br label %cleanup | ||
| cleanup: | ||
| call ptr @llvm.coro.free(token %0, ptr %1) | ||
| br label %coro.ret | ||
| coro.ret: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
| ; CHECK: %a.Frame = type { ptr, ptr, %"struct.task::promise_type", %struct.big_structure, i1, [26 x i8], %struct.big_structure.2 } | ||
| ; CHECK-LABEL: @a.resume( | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) | ||
| declare i1 @llvm.coro.alloc(token) #3 | ||
| declare i64 @llvm.coro.size.i64() #5 | ||
| declare ptr @llvm.coro.begin(token, ptr writeonly) #3 | ||
| declare token @llvm.coro.save(ptr) #3 | ||
| declare ptr @llvm.coro.frame() #5 | ||
| declare i8 @llvm.coro.suspend(token, i1) #3 | ||
| declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2 | ||
| declare i1 @llvm.coro.end(ptr, i1) #3 | ||
| declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4 | ||
| declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4 |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,79 +1,74 @@ | ||
| ; Tests that variables of different type with incompatible alignment in a Corotuine whose | ||
| ; lifetime range is not overlapping each other re-use the same slot in CorotuineFrame. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split<reuse-storage>),simplifycfg,early-cse' -S | FileCheck %s | ||
| %"struct.task::promise_type" = type { i8 } | ||
| %struct.awaitable = type { i8 } | ||
| %struct.big_structure = type { [500 x i8] } | ||
| %struct.big_structure.2 = type { [400 x i8] } | ||
| declare ptr @malloc(i64) | ||
| declare void @consume(ptr) | ||
| declare void @consume.2(ptr) | ||
| define void @a(i1 zeroext %cond) presplitcoroutine { | ||
| entry: | ||
| %__promise = alloca %"struct.task::promise_type", align 1 | ||
| %a = alloca %struct.big_structure, align 32 | ||
| %ref.tmp7 = alloca %struct.awaitable, align 1 | ||
| %b = alloca %struct.big_structure.2, align 16 | ||
| %ref.tmp18 = alloca %struct.awaitable, align 1 | ||
| %0 = call token @llvm.coro.id(i32 16, ptr nonnull %__promise, ptr @a, ptr null) | ||
| br label %init.ready | ||
| init.ready: | ||
| %1 = call noalias nonnull ptr @llvm.coro.begin(token %0, ptr null) | ||
| call void @llvm.lifetime.start.p0(i64 1, ptr nonnull %__promise) | ||
| br i1 %cond, label %if.then, label %if.else | ||
| if.then: | ||
| call void @llvm.lifetime.start.p0(i64 500, ptr nonnull %a) | ||
| call void @consume(ptr nonnull %a) | ||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %coro.ret [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %cleanup1 | ||
| ] | ||
| await.ready: | ||
| call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a) | ||
| br label %cleanup1 | ||
| if.else: | ||
| call void @llvm.lifetime.start.p0(i64 400, ptr nonnull %b) | ||
| call void @consume.2(ptr nonnull %b) | ||
| %save2 = call token @llvm.coro.save(ptr null) | ||
| %suspend2 = call i8 @llvm.coro.suspend(token %save2, i1 false) | ||
| switch i8 %suspend2, label %coro.ret [ | ||
| i8 0, label %await2.ready | ||
| i8 1, label %cleanup2 | ||
| ] | ||
| await2.ready: | ||
| call void @llvm.lifetime.end.p0(i64 400, ptr nonnull %b) | ||
| br label %cleanup2 | ||
| cleanup1: | ||
| call void @llvm.lifetime.end.p0(i64 500, ptr nonnull %a) | ||
| br label %cleanup | ||
| cleanup2: | ||
| call void @llvm.lifetime.end.p0(i64 400, ptr nonnull %b) | ||
| br label %cleanup | ||
| cleanup: | ||
| call ptr @llvm.coro.free(token %0, ptr %1) | ||
| br label %coro.ret | ||
| coro.ret: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
| ; CHECK: %a.Frame = type { ptr, ptr, %"struct.task::promise_type", i1, [14 x i8], %struct.big_structure } | ||
| ; CHECK-LABEL: @a.resume( | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) | ||
| declare i1 @llvm.coro.alloc(token) #3 | ||
| declare i64 @llvm.coro.size.i64() #5 | ||
| declare ptr @llvm.coro.begin(token, ptr writeonly) #3 | ||
| declare token @llvm.coro.save(ptr) #3 | ||
| declare ptr @llvm.coro.frame() #5 | ||
| declare i8 @llvm.coro.suspend(token, i1) #3 | ||
| declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2 | ||
| declare i1 @llvm.coro.end(ptr, i1) #3 | ||
| declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4 | ||
| declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4 |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,61 +1,61 @@ | ||
| ; Check that we will insert the correct padding if natural alignment of the | ||
| ; spilled data does not match the alignment specified in alloca instruction. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s | ||
|
|
||
| %PackedStruct = type <{ i64 }> | ||
|
|
||
| declare void @consume(ptr) | ||
|
|
||
| define ptr @f() presplitcoroutine { | ||
| entry: | ||
| %data = alloca %PackedStruct, align 32 | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %size = call i32 @llvm.coro.size.i32() | ||
| %alloc = call ptr @malloc(i32 %size) | ||
| %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
| call void @consume(ptr %data) | ||
| %0 = call i8 @llvm.coro.suspend(token none, i1 false) | ||
| switch i8 %0, label %suspend [i8 0, label %resume | ||
| i8 1, label %cleanup] | ||
| resume: | ||
| call void @consume(ptr %data) | ||
| br label %cleanup | ||
|
|
||
| cleanup: | ||
| %mem = call ptr @llvm.coro.free(token %id, ptr %hdl) | ||
| call void @free(ptr %mem) | ||
| br label %suspend | ||
| suspend: | ||
| call i1 @llvm.coro.end(ptr %hdl, i1 0) | ||
| ret ptr %hdl | ||
| } | ||
|
|
||
| ; See if the padding was inserted before PackedStruct | ||
| ; CHECK-LABEL: %f.Frame = type { ptr, ptr, i1, [15 x i8], %PackedStruct } | ||
|
|
||
| ; See if we used correct index to access packed struct (padding is field 3) | ||
| ; CHECK-LABEL: @f( | ||
| ; CHECK: %[[DATA:.+]] = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 4 | ||
| ; CHECK-NEXT: call void @consume(ptr %[[DATA]]) | ||
| ; CHECK: ret ptr | ||
|
|
||
| ; See if we used correct index to access packed struct (padding is field 3) | ||
| ; CHECK-LABEL: @f.resume( | ||
| ; CHECK: %[[DATA:.+]] = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 4 | ||
| ; CHECK-NEXT: call void @consume(ptr %[[DATA]]) | ||
| ; CHECK: ret void | ||
|
|
||
| declare ptr @llvm.coro.free(token, ptr) | ||
| declare i32 @llvm.coro.size.i32() | ||
| declare i8 @llvm.coro.suspend(token, i1) | ||
| declare void @llvm.coro.resume(ptr) | ||
| declare void @llvm.coro.destroy(ptr) | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr, ptr, ptr) | ||
| declare i1 @llvm.coro.alloc(token) | ||
| declare ptr @llvm.coro.begin(token, ptr) | ||
| declare i1 @llvm.coro.end(ptr, i1) | ||
|
|
||
| declare noalias ptr @malloc(i32) | ||
| declare double @print(double) | ||
| declare void @free(ptr) |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,96 +1,93 @@ | ||
| ; Check that we create copy the data from the alloca into the coroutine | ||
| ; frame slot if it was written to. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s | ||
|
|
||
| define ptr @f() presplitcoroutine { | ||
| entry: | ||
| %a.addr = alloca i64 ; read-only before coro.begin | ||
| %a = load i64, ptr %a.addr ; cannot modify the value, don't need to copy | ||
|
|
||
| %x.addr = alloca i64 | ||
| call void @use(ptr %x.addr) ; uses %x.addr before coro.begin | ||
|
|
||
| %y.addr = alloca i64 | ||
|
|
||
| %z.addr = alloca i64 | ||
| %flag = call i1 @check() | ||
| br i1 %flag, label %flag_true, label %flag_merge | ||
|
|
||
| flag_true: | ||
| call void @use(ptr %z.addr) ; conditionally used %z.addr | ||
| br label %flag_merge | ||
|
|
||
| flag_merge: | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %size = call i32 @llvm.coro.size.i32() | ||
| %alloc = call ptr @myAlloc(i32 %size) | ||
| %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
| call void @llvm.memset.p0.i32(ptr %y.addr, i8 1, i32 4, i1 false) | ||
| %0 = call i8 @llvm.coro.suspend(token none, i1 false) | ||
| switch i8 %0, label %suspend [i8 0, label %resume | ||
| i8 1, label %cleanup] | ||
| resume: | ||
| call void @use(ptr %a.addr) | ||
| call void @use(ptr %x.addr) | ||
| call void @use(ptr %y.addr) | ||
| call void @use(ptr %z.addr) | ||
| br label %cleanup | ||
|
|
||
| cleanup: | ||
| %mem = call ptr @llvm.coro.free(token %id, ptr %hdl) | ||
| call void @free(ptr %mem) | ||
| br label %suspend | ||
| suspend: | ||
| call i1 @llvm.coro.end(ptr %hdl, i1 0) | ||
| ret ptr %hdl | ||
| } | ||
|
|
||
| ; See that we added both x and y to the frame. | ||
| ; CHECK: %f.Frame = type { ptr, ptr, i64, i64, i64, i64, i1 } | ||
|
|
||
| ; See that all of the uses prior to coro-begin stays put. | ||
| ; CHECK-LABEL: define ptr @f() { | ||
| ; CHECK-NEXT: entry: | ||
| ; CHECK-NEXT: %a.addr = alloca i64 | ||
| ; CHECK-NEXT: %x.addr = alloca i64 | ||
| ; CHECK-NEXT: call void @use(ptr %x.addr) | ||
| ; CHECK-NEXT: %z.addr = alloca i64 | ||
|
|
||
| ; See that we only copy the x as y was not modified prior to coro.begin. | ||
| ; CHECK: store ptr @f.destroy, ptr %destroy.addr | ||
| ; The next 3 instructions are to copy data in %x.addr from stack to frame. | ||
| ; CHECK-NEXT: %0 = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 3 | ||
| ; CHECK-NEXT: %1 = load i64, ptr %x.addr, align 4 | ||
| ; CHECK-NEXT: store i64 %1, ptr %0, align 4 | ||
| ; The next 3 instructions are to copy data in %z.addr from stack to frame. | ||
| ; CHECK-NEXT: [[T2:%.+]] = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 5 | ||
| ; CHECK-NEXT: [[T3:%.+]] = load i64, ptr %z.addr, align 4 | ||
| ; CHECK-NEXT: store i64 [[T3]], ptr [[T2]], align 4 | ||
| ; The next instruction is to recreate %y.cast in the original IR. | ||
| ; CHECK-NEXT: %y.addr.reload.addr = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 4 | ||
| ; CHECK-NEXT: call void @llvm.memset.p0.i32(ptr %y.addr.reload.addr, i8 1, i32 4, i1 false) | ||
| ; CHECK-NEXT: %index.addr1 = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 6 | ||
| ; CHECK-NEXT: store i1 false, ptr %index.addr1, align 1 | ||
| ; CHECK-NEXT: ret ptr %hdl | ||
|
|
||
|
|
||
| declare ptr @llvm.coro.free(token, ptr) | ||
| declare i32 @llvm.coro.size.i32() | ||
| declare i8 @llvm.coro.suspend(token, i1) | ||
| declare void @llvm.coro.resume(ptr) | ||
| declare void @llvm.coro.destroy(ptr) | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr, ptr, ptr) | ||
| declare i1 @llvm.coro.alloc(token) | ||
| declare ptr @llvm.coro.begin(token, ptr) | ||
| declare i1 @llvm.coro.end(ptr, i1) | ||
|
|
||
| declare void @llvm.memset.p0.i32(ptr, i8, i32, i1) | ||
|
|
||
| declare noalias ptr @myAlloc(i32) | ||
| declare void @use(ptr) | ||
| declare void @free(ptr) | ||
| declare i1 @check() |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,63 +1,56 @@ | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s | ||
|
|
||
| target datalayout = "p:64:64:64" | ||
|
|
||
| declare void @prototype_f(ptr, i1) | ||
|
|
||
| declare noalias ptr @allocate(i32 %size) | ||
| declare void @deallocate(ptr %ptr) | ||
| declare void @init(ptr %ptr) | ||
| declare void @use(ptr %ptr) | ||
| declare void @use_addr_val(i64 %val, ptr %addr) | ||
|
|
||
| define { ptr, ptr } @f(ptr %buffer) presplitcoroutine { | ||
| entry: | ||
| %tmp = alloca { i64, i64 }, align 8 | ||
| %proj.1 = getelementptr inbounds { i64, i64 }, ptr %tmp, i64 0, i32 0 | ||
| %proj.2 = getelementptr inbounds { i64, i64 }, ptr %tmp, i64 0, i32 1 | ||
| store i64 0, ptr %proj.1, align 8 | ||
| store i64 0, ptr %proj.2, align 8 | ||
| %escape_addr = ptrtoint ptr %tmp to i64 | ||
| %id = call token @llvm.coro.id.retcon.once(i32 32, i32 8, ptr %buffer, ptr @prototype_f, ptr @allocate, ptr @deallocate) | ||
| %hdl = call ptr @llvm.coro.begin(token %id, ptr null) | ||
| %proj.2.2 = getelementptr inbounds { i64, i64 }, ptr %tmp, i64 0, i32 1 | ||
| call void @init(ptr %proj.1) | ||
| call void @init(ptr %proj.2.2) | ||
| call void @use_addr_val(i64 %escape_addr, ptr %tmp) | ||
| %abort = call i1 (...) @llvm.coro.suspend.retcon.i1(ptr %tmp) | ||
| br i1 %abort, label %end, label %resume | ||
|
|
||
| resume: | ||
| call void @use(ptr %tmp) | ||
| br label %end | ||
|
|
||
| end: | ||
| call i1 @llvm.coro.end(ptr %hdl, i1 0) | ||
| unreachable | ||
| } | ||
| ; Make sure we don't lose writes to the frame. | ||
| ; CHECK-LABEL: define { ptr, ptr } @f(ptr %buffer) { | ||
| ; CHECK: [[PROJ2:%.*]] = getelementptr inbounds { i64, i64 }, ptr %buffer, i64 0, i32 1 | ||
| ; CHECK: store i64 0, ptr %buffer | ||
| ; CHECK: store i64 0, ptr [[PROJ2]] | ||
| ; CHECK: [[ESCAPED_ADDR:%.*]] = ptrtoint ptr %buffer to i64 | ||
| ; CHECK: call void @init(ptr %buffer) | ||
| ; CHECK: call void @init(ptr [[PROJ2]]) | ||
| ; CHECK: call void @use_addr_val(i64 [[ESCAPED_ADDR]], ptr %buffer) | ||
|
|
||
| ; CHECK-LABEL: define internal void @f.resume.0(ptr {{.*}} %0, i1 %1) { | ||
| ; CHECK: resume: | ||
| ; CHECK: call void @use(ptr %0) | ||
|
|
||
| declare token @llvm.coro.id.retcon.once(i32, i32, ptr, ptr, ptr, ptr) | ||
| declare ptr @llvm.coro.begin(token, ptr) | ||
| declare i1 @llvm.coro.suspend.retcon.i1(...) | ||
| declare i1 @llvm.coro.end(ptr, i1) | ||
|
|
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,71 +1,69 @@ | ||
| ; Check that we can spills coro.begin from an inlined inner coroutine. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s | ||
|
|
||
| %g.Frame = type { ptr, ptr, i32, i1, i32 } | ||
|
|
||
| @g.resumers = private constant [3 x ptr] [ptr @g.dummy, ptr @g.dummy, ptr @g.dummy] | ||
|
|
||
| declare void @g.dummy(ptr) | ||
|
|
||
| declare ptr @g() | ||
|
|
||
| define ptr @f() presplitcoroutine { | ||
| entry: | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %size = call i32 @llvm.coro.size.i32() | ||
| %alloc = call ptr @malloc(i32 %size) | ||
| %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
|
|
||
| %innerid = call token @llvm.coro.id(i32 0, ptr null, ptr @g, ptr @g.resumers) | ||
| %innerhdl = call noalias nonnull ptr @llvm.coro.begin(token %innerid, ptr null) | ||
|
|
||
| %tok = call i8 @llvm.coro.suspend(token none, i1 false) | ||
| switch i8 %tok, label %suspend [i8 0, label %resume | ||
| i8 1, label %cleanup] | ||
| resume: | ||
| %gvar.addr = getelementptr inbounds %g.Frame, ptr %innerhdl, i32 0, i32 4 | ||
| %gvar = load i32, ptr %gvar.addr | ||
| call void @print.i32(i32 %gvar) | ||
| br label %cleanup | ||
|
|
||
| cleanup: | ||
| %mem = call ptr @llvm.coro.free(token %id, ptr %hdl) | ||
| call void @free(ptr %mem) | ||
| br label %suspend | ||
| suspend: | ||
| call i1 @llvm.coro.end(ptr %hdl, i1 0) | ||
| ret ptr %hdl | ||
| } | ||
|
|
||
| ; See if the i8* for coro.begin was added to f.Frame | ||
| ; CHECK-LABEL: %f.Frame = type { ptr, ptr, ptr, i1 } | ||
|
|
||
| ; See if the g's coro.begin was spilled into the frame | ||
| ; CHECK-LABEL: @f( | ||
| ; CHECK: %innerid = call token @llvm.coro.id(i32 0, ptr null, ptr @g, ptr @g.resumers) | ||
| ; CHECK: %innerhdl = call noalias nonnull ptr @llvm.coro.begin(token %innerid, ptr null) | ||
| ; CHECK: %[[spilladdr:.+]] = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 2 | ||
| ; CHECK: store ptr %innerhdl, ptr %[[spilladdr]] | ||
|
|
||
| ; See if the coro.begin was loaded from the frame | ||
| ; CHECK-LABEL: @f.resume( | ||
| ; CHECK: %[[innerhdlAddr:.+]] = getelementptr inbounds %f.Frame, ptr %{{.+}}, i32 0, i32 2 | ||
| ; CHECK: %[[innerhdl:.+]] = load ptr, ptr %[[innerhdlAddr]] | ||
| ; CHECK: %[[gvarAddr:.+]] = getelementptr inbounds %g.Frame, ptr %[[innerhdl]], i32 0, i32 4 | ||
| ; CHECK: %[[gvar:.+]] = load i32, ptr %[[gvarAddr]] | ||
| ; CHECK: call void @print.i32(i32 %[[gvar]]) | ||
|
|
||
| declare ptr @llvm.coro.free(token, ptr) | ||
| declare i32 @llvm.coro.size.i32() | ||
| declare i8 @llvm.coro.suspend(token, i1) | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr, ptr, ptr) | ||
| declare i1 @llvm.coro.alloc(token) | ||
| declare ptr @llvm.coro.begin(token, ptr) | ||
| declare i1 @llvm.coro.end(ptr, i1) | ||
|
|
||
| declare noalias ptr @malloc(i32) | ||
| declare void @print.i32(i32) | ||
| declare void @free(ptr) |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,56 +1,55 @@ | ||
| ; Check that promise object is reloaded from the correct index of the coro frame. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse' -S | FileCheck %s | ||
|
|
||
| %"class.task::promise_type" = type { [64 x i8] } | ||
|
|
||
| declare void @consume(ptr) | ||
| declare void @consume2(ptr) | ||
|
|
||
| define ptr @f() presplitcoroutine { | ||
| entry: | ||
| %data = alloca i32, align 4 | ||
| %__promise = alloca %"class.task::promise_type", align 64 | ||
| %id = call token @llvm.coro.id(i32 0, ptr %__promise, ptr null, ptr null) | ||
| %size = call i32 @llvm.coro.size.i32() | ||
| %alloc = call ptr @malloc(i32 %size) | ||
| %hdl = call ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
| call void @consume(ptr %data) | ||
| %0 = call i8 @llvm.coro.suspend(token none, i1 false) | ||
| switch i8 %0, label %suspend [i8 0, label %resume | ||
| i8 1, label %cleanup] | ||
| resume: | ||
| call void @consume(ptr %data) | ||
| call void @consume2(ptr %__promise) | ||
| br label %cleanup | ||
|
|
||
| cleanup: | ||
| %mem = call ptr @llvm.coro.free(token %id, ptr %hdl) | ||
| call void @free(ptr %mem) | ||
| br label %suspend | ||
| suspend: | ||
| call i1 @llvm.coro.end(ptr %hdl, i1 0) | ||
| ret ptr %hdl | ||
| } | ||
|
|
||
| ; CHECK-LABEL: %f.Frame = type { ptr, ptr, i32, i1, [43 x i8], %"class.task::promise_type" } | ||
|
|
||
| ; CHECK-LABEL: @f.resume( | ||
| ; CHECK: %[[DATA:.+]] = getelementptr inbounds %f.Frame, ptr %hdl, i32 0, i32 5 | ||
| ; CHECK: call void @consume2(ptr %[[DATA]]) | ||
| ; CHECK: ret void | ||
|
|
||
| declare ptr @llvm.coro.free(token, ptr) | ||
| declare i32 @llvm.coro.size.i32() | ||
| declare i8 @llvm.coro.suspend(token, i1) | ||
| declare void @llvm.coro.resume(ptr) | ||
| declare void @llvm.coro.destroy(ptr) | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr, ptr, ptr) | ||
| declare i1 @llvm.coro.alloc(token) | ||
| declare ptr @llvm.coro.begin(token, ptr) | ||
| declare i1 @llvm.coro.end(ptr, i1) | ||
|
|
||
| declare noalias ptr @malloc(i32) | ||
| declare double @print(double) | ||
| declare void @free(ptr) |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,70 +1,65 @@ | ||
| ; Tests that coro-split can handle the case when a code after coro.suspend uses | ||
| ; a value produces between coro.save and coro.suspend (%Result.i19) | ||
| ; and checks whether stray coro.saves are properly removed | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse,simplifycfg' -S | FileCheck %s | ||
|
|
||
| %"struct.std::coroutine_handle" = type { ptr } | ||
| %"struct.std::coroutine_handle.0" = type { %"struct.std::coroutine_handle" } | ||
| %"struct.lean_future<int>::Awaiter" = type { i32, %"struct.std::coroutine_handle.0" } | ||
|
|
||
| declare ptr @malloc(i64) | ||
| declare void @print(i32) | ||
|
|
||
| define void @a() presplitcoroutine { | ||
| entry: | ||
| %ref.tmp7 = alloca %"struct.lean_future<int>::Awaiter", align 8 | ||
| %testval = alloca i32 | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %alloc = call ptr @malloc(i64 16) #3 | ||
| %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
|
|
||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %exit [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %exit | ||
| ] | ||
| await.ready: | ||
| %StrayCoroSave = call token @llvm.coro.save(ptr null) | ||
| %val = load i32, ptr %ref.tmp7 | ||
| call void @llvm.lifetime.start.p0(i64 4, ptr %testval) | ||
| %test = load i32, ptr %testval | ||
| call void @print(i32 %test) | ||
| call void @llvm.lifetime.end.p0(i64 4, ptr %testval) | ||
| call void @print(i32 %val) | ||
| br label %exit | ||
| exit: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
|
|
||
| ; CHECK-LABEL: @a.resume( | ||
| ; CHECK: %testval = alloca i32 | ||
| ; CHECK-NOT: call token @llvm.coro.save(ptr null) | ||
| ; CHECK: %val = load i32, ptr %ref.tmp7 | ||
| ; CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 4, ptr %testval) | ||
| ; CHECK-NEXT: %test = load i32, ptr %testval | ||
| ; CHECK-NEXT: call void @print(i32 %test) | ||
| ; CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 4, ptr %testval) | ||
| ; CHECK-NEXT: call void @print(i32 %val) | ||
| ; CHECK-NEXT: ret void | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) | ||
| declare i1 @llvm.coro.alloc(token) #3 | ||
| declare noalias nonnull ptr @"\01??2@YAPEAX_K@Z"(i64) local_unnamed_addr | ||
| declare i64 @llvm.coro.size.i64() #5 | ||
| declare ptr @llvm.coro.begin(token, ptr writeonly) #3 | ||
| declare void @"\01?puts@@YAXZZ"(...) | ||
| declare token @llvm.coro.save(ptr) #3 | ||
| declare ptr @llvm.coro.frame() #5 | ||
| declare i8 @llvm.coro.suspend(token, i1) #3 | ||
| declare void @"\01??3@YAXPEAX@Z"(ptr) local_unnamed_addr #10 | ||
| declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2 | ||
| declare i1 @llvm.coro.end(ptr, i1) #3 | ||
| declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4 | ||
| declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4 |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,103 +1,96 @@ | ||
| ; Tests that coro-split will optimize the lifetime.start maker of each local variable, | ||
| ; sink them to the places after the suspend block. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse,simplifycfg' -S | FileCheck %s | ||
|
|
||
| ; CHECK: %a.Frame = type { ptr, ptr, i1 } | ||
| ; CHECK: %a_optnone.Frame = type { ptr, ptr, i32, i1 } | ||
|
|
||
| %"struct.std::coroutine_handle" = type { ptr } | ||
| %"struct.std::coroutine_handle.0" = type { %"struct.std::coroutine_handle" } | ||
| %"struct.lean_future<int>::Awaiter" = type { i32, %"struct.std::coroutine_handle.0" } | ||
|
|
||
| declare ptr @malloc(i64) | ||
| declare void @print(i32) | ||
|
|
||
| define void @a() presplitcoroutine { | ||
| entry: | ||
| %ref.tmp7 = alloca %"struct.lean_future<int>::Awaiter", align 8 | ||
| %testval = alloca i32 | ||
| ; lifetime of %testval starts here, but not used until await.ready. | ||
| call void @llvm.lifetime.start.p0(i64 4, ptr %testval) | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %alloc = call ptr @malloc(i64 16) #3 | ||
| %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
|
|
||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %exit [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %exit | ||
| ] | ||
| await.ready: | ||
| %StrayCoroSave = call token @llvm.coro.save(ptr null) | ||
| %val = load i32, ptr %ref.tmp7 | ||
| %test = load i32, ptr %testval | ||
| call void @print(i32 %test) | ||
| call void @llvm.lifetime.end.p0(i64 4, ptr %testval) | ||
| call void @print(i32 %val) | ||
| br label %exit | ||
| exit: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
|
|
||
| ; CHECK-LABEL: @a.resume( | ||
| ; CHECK: %testval = alloca i32, align 4 | ||
| ; CHECK: call void @llvm.lifetime.start.p0(i64 4, ptr %testval) | ||
| ; CHECK-NEXT: %val = load i32, ptr %ref.tmp7 | ||
| ; CHECK-NEXT: %test = load i32, ptr %testval | ||
| ; CHECK-NEXT: call void @print(i32 %test) | ||
| ; CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 4, ptr %testval) | ||
| ; CHECK-NEXT: call void @print(i32 %val) | ||
| ; CHECK-NEXT: ret void | ||
|
|
||
| define void @a_optnone() presplitcoroutine optnone noinline { | ||
| entry: | ||
| %ref.tmp7 = alloca %"struct.lean_future<int>::Awaiter", align 8 | ||
| %testval = alloca i32 | ||
| ; lifetime of %testval starts here, but not used until await.ready. | ||
| call void @llvm.lifetime.start.p0(i64 4, ptr %testval) | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %alloc = call ptr @malloc(i64 16) #3 | ||
| %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
|
|
||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %exit [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %exit | ||
| ] | ||
| await.ready: | ||
| %StrayCoroSave = call token @llvm.coro.save(ptr null) | ||
| %val = load i32, ptr %ref.tmp7 | ||
| %test = load i32, ptr %testval | ||
| call void @print(i32 %test) | ||
| call void @llvm.lifetime.end.p0(i64 4, ptr %testval) | ||
| call void @print(i32 %val) | ||
| br label %exit | ||
| exit: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) | ||
| declare i1 @llvm.coro.alloc(token) #3 | ||
| declare noalias nonnull ptr @"\01??2@YAPEAX_K@Z"(i64) local_unnamed_addr | ||
| declare i64 @llvm.coro.size.i64() #5 | ||
| declare ptr @llvm.coro.begin(token, ptr writeonly) #3 | ||
| declare void @"\01?puts@@YAXZZ"(...) | ||
| declare token @llvm.coro.save(ptr) #3 | ||
| declare ptr @llvm.coro.frame() #5 | ||
| declare i8 @llvm.coro.suspend(token, i1) #3 | ||
| declare void @"\01??3@YAXPEAX@Z"(ptr) local_unnamed_addr #10 | ||
| declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2 | ||
| declare i1 @llvm.coro.end(ptr, i1) #3 | ||
| declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4 | ||
| declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4 |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,70 +1,64 @@ | ||
| ; Corresponding to coro-split-sink-lifetime-01.ll. This file tests that whether the CoroFrame | ||
| ; pass knows the operand of lifetime.start intrinsic may be GEP as well. | ||
| ; RUN: opt < %s -passes='cgscc(coro-split),simplifycfg,early-cse,simplifycfg' -S | FileCheck %s | ||
|
|
||
| %"struct.std::coroutine_handle" = type { ptr } | ||
| %"struct.std::coroutine_handle.0" = type { %"struct.std::coroutine_handle" } | ||
| %"struct.lean_future<int>::Awaiter" = type { i32, %"struct.std::coroutine_handle.0" } | ||
|
|
||
| declare ptr @malloc(i64) | ||
| declare void @print(i32) | ||
|
|
||
| %i8.array = type { [100 x i8] } | ||
| declare void @consume.i8.array(ptr) | ||
|
|
||
| define void @a.gep() presplitcoroutine { | ||
| entry: | ||
| %ref.tmp7 = alloca %"struct.lean_future<int>::Awaiter", align 8 | ||
| %testval = alloca %i8.array | ||
| ; lifetime of %testval starts here, but not used until await.ready. | ||
| call void @llvm.lifetime.start.p0(i64 100, ptr %testval) | ||
| %id = call token @llvm.coro.id(i32 0, ptr null, ptr null, ptr null) | ||
| %alloc = call ptr @malloc(i64 16) #3 | ||
| %vFrame = call noalias nonnull ptr @llvm.coro.begin(token %id, ptr %alloc) | ||
|
|
||
| %save = call token @llvm.coro.save(ptr null) | ||
| %suspend = call i8 @llvm.coro.suspend(token %save, i1 false) | ||
| switch i8 %suspend, label %exit [ | ||
| i8 0, label %await.ready | ||
| i8 1, label %exit | ||
| ] | ||
| await.ready: | ||
| %StrayCoroSave = call token @llvm.coro.save(ptr null) | ||
| %val = load i32, ptr %ref.tmp7 | ||
| call void @consume.i8.array(ptr %testval) | ||
| call void @llvm.lifetime.end.p0(i64 100, ptr %testval) | ||
| call void @print(i32 %val) | ||
| br label %exit | ||
| exit: | ||
| call i1 @llvm.coro.end(ptr null, i1 false) | ||
| ret void | ||
| } | ||
| ; CHECK-LABEL: @a.gep.resume( | ||
| ; CHECK: %testval = alloca %i8.array | ||
| ; CHECK: call void @llvm.lifetime.start.p0(i64 100, ptr %testval) | ||
| ; CHECK-NEXT: %val = load i32, ptr %ref.tmp7 | ||
| ; CHECK-NEXT: call void @consume.i8.array(ptr %testval) | ||
| ; CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 100, ptr %testval) | ||
| ; CHECK-NEXT: call void @print(i32 %val) | ||
| ; CHECK-NEXT: ret void | ||
|
|
||
| declare token @llvm.coro.id(i32, ptr readnone, ptr nocapture readonly, ptr) | ||
| declare i1 @llvm.coro.alloc(token) #3 | ||
| declare noalias nonnull ptr @"\01??2@YAPEAX_K@Z"(i64) local_unnamed_addr | ||
| declare i64 @llvm.coro.size.i64() #5 | ||
| declare ptr @llvm.coro.begin(token, ptr writeonly) #3 | ||
| declare void @"\01?puts@@YAXZZ"(...) | ||
| declare token @llvm.coro.save(ptr) #3 | ||
| declare ptr @llvm.coro.frame() #5 | ||
| declare i8 @llvm.coro.suspend(token, i1) #3 | ||
| declare void @"\01??3@YAXPEAX@Z"(ptr) local_unnamed_addr #10 | ||
| declare ptr @llvm.coro.free(token, ptr nocapture readonly) #2 | ||
| declare i1 @llvm.coro.end(ptr, i1) #3 | ||
| declare void @llvm.lifetime.start.p0(i64, ptr nocapture) #4 | ||
| declare void @llvm.lifetime.end.p0(i64, ptr nocapture) #4 |