Lines Matching refs:async

1 // RUN: mlir-opt %s -async-runtime-policy-based-ref-counting | FileCheck %s
4 // CHECK:         %[[TOKEN:.*]]: !async.token
5 func.func @token_await(%arg0: !async.token) {
6   // CHECK: async.runtime.await %[[TOKEN]]
7   // CHECK-NOT: async.runtime.drop_ref
8   async.runtime.await %arg0 : !async.token
13 // CHECK:         %[[GROUP:.*]]: !async.group
14 func.func @group_await(%arg0: !async.group) {
15   // CHECK: async.runtime.await %[[GROUP]]
16   // CHECK-NOT: async.runtime.drop_ref
17   async.runtime.await %arg0 : !async.group
22 // CHECK:         %[[GROUP:.*]]: !async.group
23 // CHECK:         %[[TOKEN:.*]]: !async.token
24 func.func @add_token_to_group(%arg0: !async.group, %arg1: !async.token) {
25   // CHECK: async.runtime.add_to_group %[[TOKEN]], %[[GROUP]]
26   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
27   async.runtime.add_to_group %arg1, %arg0 : !async.token
32 // CHECK:         %[[VALUE:.*]]: !async.value<f32>
33 func.func @value_load(%arg0: !async.value<f32>) {
34   // CHECK: async.runtime.load %[[VALUE]]
35   // CHECK: async.runtime.drop_ref %[[VALUE]] {count = 1 : i64}
36   %0 = async.runtime.load %arg0 : !async.value<f32>
41 // CHECK:         %[[TOKEN:.*]]: !async.token
42 func.func @error_check(%arg0: !async.token) {
43   // CHECK: async.runtime.is_error %[[TOKEN]]
44   // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
45   %0 = async.runtime.is_error %arg0 : !async.token