Lines Matching full:token

3 // CHECK-LABEL: @token
4 func.func private @token() -> !async.token
10 func.func private @take_token(%arg0: !async.token)
13 // CHECK: %[[TOKEN:.*]]: !async.token
14 func.func @token_arg_no_uses(%arg0: !async.token) {
15   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
21   // CHECK: %[[TOKEN:.*]] = async.runtime.create : !async.token
22   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
23   %0 = async.runtime.create : !async.token
29   // CHECK: %[[TOKEN:.*]] = call @token
30   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
31   %0 = call @token() : () -> !async.token
36 // CHECK: %[[TOKEN:.*]]: !async.token
37 func.func @token_arg_to_func(%arg0: !async.token) {
38   // CHECK: async.runtime.add_ref %[[TOKEN]] {count = 1 : i64} : !async.token
39   call @take_token(%arg0): (!async.token) -> ()
40   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64} : !async.token
46   // CHECK: %[[TOKEN:.*]] = async.runtime.create : !async.token
47   %0 = async.runtime.create : !async.token
48   // CHECK: async.runtime.add_ref %[[TOKEN]] {count = 1 : i64} : !async.token
49   call @take_token(%0): (!async.token) -> ()
50   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
55 // CHECK: %[[TOKEN:.*]]: !async.token
56 func.func @token_arg_cond_br_await_with_fallthough(%arg0: !async.token, %arg1: i1) {
66   // CHECK:   async.runtime.await %[[TOKEN]]
67   // CHECK:   async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
68   async.runtime.await %arg0 : !async.token
73 func.func @token_simple_return() -> !async.token {
74   // CHECK: %[[TOKEN:.*]] = async.runtime.create : !async.token
75   %token = async.runtime.create : !async.token
76   // CHECK: return %[[TOKEN]]
77   return %token : !async.token
83 func.func @token_coro_return() -> !async.token {
84   %token = async.runtime.create : !async.token
97   return %token : !async.token
101 // CHECK: %[[TOKEN:.*]]: !async.token
102 func.func @token_coro_await_and_resume(%arg0: !async.token) -> !async.token {
103   %token = async.runtime.create : !async.token
107   // CHECK: async.runtime.await_and_resume %[[TOKEN]]
108   async.runtime.await_and_resume %arg0, %hdl : !async.token
109   // CHECK-NEXT: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
118   return %token : !async.token
123 func.func @value_coro_await_and_resume(%arg0: !async.value<f32>) -> !async.token {
124   %token = async.runtime.create : !async.token
146   return %token : !async.token
150 // CHECK: %[[TOKEN:.*]]: !async.token
151 func.func private @outlined_async_execute(%arg0: !async.token) -> !async.token {
152   %0 = async.runtime.create : !async.token
162   async.runtime.await_and_resume %arg0, %2 : !async.token
163   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
169   async.runtime.set_available %0 : !async.token
180   return %0 : !async.token
186   // CHECK: %[[TOKEN:.*]] = call @token()
187   %token = call @token() : () -> !async.token
190     // CHECK: async.runtime.await %[[TOKEN]]
191     async.runtime.await %token : !async.token
194   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
204   // CHECK:   %[[TOKEN:.*]] = call @token()
205   %token = call @token() : () -> !async.token
206   // CHECK:   async.runtime.await %[[TOKEN]]
207   // CHECK:   async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
208   async.runtime.await %token : !async.token
219   // CHECK: %[[TOKEN:.*]] = call @token()
220   %token = call @token() : () -> !async.token
225   // CHECK:   async.runtime.await %[[TOKEN]]
226   // CHECK:   async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
228   async.runtime.await %token : !async.token
231   // CHECK:   async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
241   // CHECK: %[[TOKEN:.*]] = call @token()
242   %token = call @token() : () -> !async.token
247   // CHECK:   async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
252   // CHECK:   async.runtime.await %[[TOKEN]]
253   // CHECK:   async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
255   async.runtime.await %token : !async.token
265   // CHECK: %[[TOKEN0:.*]] = call @token()
266   // CHECK: %[[TOKEN1:.*]] = call @token()
267   %token0 = call @token() : () -> !async.token
268   %token1 = call @token() : () -> !async.token
277   async.runtime.await %token0 : !async.token
285   async.runtime.await %token1 : !async.token