1// RUN: mlir-opt %s -async-runtime-policy-based-ref-counting | FileCheck %s 2 3// CHECK-LABEL: @token_await 4// CHECK: %[[TOKEN:.*]]: !async.token 5func.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 9 return 10} 11 12// CHECK-LABEL: @group_await 13// CHECK: %[[GROUP:.*]]: !async.group 14func.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 18 return 19} 20 21// CHECK-LABEL: @add_token_to_group 22// CHECK: %[[GROUP:.*]]: !async.group 23// CHECK: %[[TOKEN:.*]]: !async.token 24func.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 28 return 29} 30 31// CHECK-LABEL: @value_load 32// CHECK: %[[VALUE:.*]]: !async.value<f32> 33func.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> 37 return 38} 39 40// CHECK-LABEL: @error_check 41// CHECK: %[[TOKEN:.*]]: !async.token 42func.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 46 return 47} 48