1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
// RUN: mlir-opt %s | FileCheck %s --dump-input=always
// CHECK-LABEL: @coro_id
func @coro_id() -> !async.coro.id {
// CHECK: %0 = async.coro.id
// CHECK: return %0 : !async.coro.id
%0 = async.coro.id
return %0 : !async.coro.id
}
// CHECK-LABEL: @coro_handle
func @coro_handle(%arg0: !async.coro.id) -> !async.coro.handle {
// CHECK: %0 = async.coro.begin %arg0
// CHECK: return %0 : !async.coro.handle
%0 = async.coro.begin %arg0
return %0 : !async.coro.handle
}
// CHECK-LABEL: @coro_free
func @coro_free(%arg0: !async.coro.id, %arg1: !async.coro.handle) {
// CHECK: async.coro.free %arg0, %arg1
async.coro.free %arg0, %arg1
return
}
// CHECK-LABEL: @coro_end
func @coro_end(%arg0: !async.coro.handle) {
// CHECK: async.coro.end %arg0
async.coro.end %arg0
return
}
// CHECK-LABEL: @coro_save
func @coro_save(%arg0: !async.coro.handle) -> !async.coro.state {
// CHECK: %0 = async.coro.save %arg0
%0 = async.coro.save %arg0
// CHECK: return %0 : !async.coro.state
return %0 : !async.coro.state
}
// CHECK-LABEL: @coro_suspend
func @coro_suspend(%arg0: !async.coro.state) {
// CHECK: async.coro.suspend %arg0
// CHECK-SAME: ^[[SUSPEND:.*]], ^[[RESUME:.*]], ^[[CLEANUP:.*]]
async.coro.suspend %arg0, ^suspend, ^resume, ^cleanup
^resume:
// CHECK: ^[[RESUME]]
// CHECK: return {coro.resume}
return { coro.resume }
^cleanup:
// CHECK: ^[[CLEANUP]]
// CHECK: return {coro.cleanup}
return { coro.cleanup }
^suspend:
// CHECK: ^[[SUSPEND]]
// CHECK: return {coro.suspend}
return { coro.suspend }
}
|
