48 lines · plain
1// RUN: mlir-opt %s -async-runtime-policy-based-ref-counting | FileCheck %s2 3// CHECK-LABEL: @token_await4// CHECK: %[[TOKEN:.*]]: !async.token5func.func @token_await(%arg0: !async.token) {6 // CHECK: async.runtime.await %[[TOKEN]]7 // CHECK-NOT: async.runtime.drop_ref8 async.runtime.await %arg0 : !async.token9 return10}11 12// CHECK-LABEL: @group_await13// CHECK: %[[GROUP:.*]]: !async.group14func.func @group_await(%arg0: !async.group) {15 // CHECK: async.runtime.await %[[GROUP]]16 // CHECK-NOT: async.runtime.drop_ref17 async.runtime.await %arg0 : !async.group18 return19}20 21// CHECK-LABEL: @add_token_to_group22// CHECK: %[[GROUP:.*]]: !async.group23// CHECK: %[[TOKEN:.*]]: !async.token24func.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.token28 return29}30 31// CHECK-LABEL: @value_load32// 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 return38}39 40// CHECK-LABEL: @error_check41// CHECK: %[[TOKEN:.*]]: !async.token42func.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.token46 return47}48