168 lines · plain
1// RUN: mlir-opt %s | FileCheck %s --dump-input=always2 3// CHECK-LABEL: @create_token4func.func @create_token() -> !async.token {5 // CHECK: %0 = async.runtime.create : !async.token6 %0 = async.runtime.create : !async.token7 // CHECK: return %0 : !async.token8 return %0 : !async.token9}10 11// CHECK-LABEL: @create_value12func.func @create_value() -> !async.value<f32> {13 // CHECK: %0 = async.runtime.create : !async.value<f32>14 %0 = async.runtime.create : !async.value<f32>15 // CHECK: return %0 : !async.value<f32>16 return %0 : !async.value<f32>17}18 19// CHECK-LABEL: @create_group20func.func @create_group() -> !async.group {21 // CHECK: %[[C:.*]] = arith.constant 10 : index22 %c = arith.constant 10 : index23 // CHECK: %[[V:.*]] = async.runtime.create_group %[[C]] : !async.group24 %0 = async.runtime.create_group %c : !async.group25 // CHECK: return %[[V]] : !async.group26 return %0 : !async.group27}28 29// CHECK-LABEL: @set_token_available30func.func @set_token_available(%arg0: !async.token) {31 // CHECK: async.runtime.set_available %arg0 : !async.token32 async.runtime.set_available %arg0 : !async.token33 return34}35 36// CHECK-LABEL: @set_value_available37func.func @set_value_available(%arg0: !async.value<f32>) {38 // CHECK: async.runtime.set_available %arg0 : !async.value<f32>39 async.runtime.set_available %arg0 : !async.value<f32>40 return41}42 43// CHECK-LABEL: @set_token_error44func.func @set_token_error(%arg0: !async.token) {45 // CHECK: async.runtime.set_error %arg0 : !async.token46 async.runtime.set_error %arg0 : !async.token47 return48}49 50// CHECK-LABEL: @set_value_error51func.func @set_value_error(%arg0: !async.value<f32>) {52 // CHECK: async.runtime.set_error %arg0 : !async.value<f32>53 async.runtime.set_error %arg0 : !async.value<f32>54 return55}56 57// CHECK-LABEL: @is_token_error58func.func @is_token_error(%arg0: !async.token) -> i1 {59 // CHECK: %[[ERR:.*]] = async.runtime.is_error %arg0 : !async.token60 %0 = async.runtime.is_error %arg0 : !async.token61 return %0 : i162}63 64// CHECK-LABEL: @is_value_error65func.func @is_value_error(%arg0: !async.value<f32>) -> i1 {66 // CHECK: %[[ERR:.*]] = async.runtime.is_error %arg0 : !async.value<f32>67 %0 = async.runtime.is_error %arg0 : !async.value<f32>68 return %0 : i169}70 71// CHECK-LABEL: @is_group_error72func.func @is_group_error(%arg0: !async.group) -> i1 {73 // CHECK: %[[ERR:.*]] = async.runtime.is_error %arg0 : !async.group74 %0 = async.runtime.is_error %arg0 : !async.group75 return %0 : i176}77 78// CHECK-LABEL: @await_token79func.func @await_token(%arg0: !async.token) {80 // CHECK: async.runtime.await %arg0 : !async.token81 async.runtime.await %arg0 : !async.token82 return83}84 85// CHECK-LABEL: @await_value86func.func @await_value(%arg0: !async.value<f32>) {87 // CHECK: async.runtime.await %arg0 : !async.value<f32>88 async.runtime.await %arg0 : !async.value<f32>89 return90}91 92// CHECK-LABEL: @await_group93func.func @await_group(%arg0: !async.group) {94 // CHECK: async.runtime.await %arg0 : !async.group95 async.runtime.await %arg0 : !async.group96 return97}98 99// CHECK-LABEL: @await_and_resume_token100func.func @await_and_resume_token(%arg0: !async.token,101 %arg1: !async.coro.handle) {102 // CHECK: async.runtime.await_and_resume %arg0, %arg1 : !async.token103 async.runtime.await_and_resume %arg0, %arg1 : !async.token104 return105}106 107// CHECK-LABEL: @await_and_resume_value108func.func @await_and_resume_value(%arg0: !async.value<f32>,109 %arg1: !async.coro.handle) {110 // CHECK: async.runtime.await_and_resume %arg0, %arg1 : !async.value<f32>111 async.runtime.await_and_resume %arg0, %arg1 : !async.value<f32>112 return113}114 115// CHECK-LABEL: @await_and_resume_group116func.func @await_and_resume_group(%arg0: !async.group,117 %arg1: !async.coro.handle) {118 // CHECK: async.runtime.await_and_resume %arg0, %arg1 : !async.group119 async.runtime.await_and_resume %arg0, %arg1 : !async.group120 return121}122 123// CHECK-LABEL: @resume124func.func @resume(%arg0: !async.coro.handle) {125 // CHECK: async.runtime.resume %arg0126 async.runtime.resume %arg0127 return128}129 130// CHECK-LABEL: @store131func.func @store(%arg0: f32, %arg1: !async.value<f32>) {132 // CHECK: async.runtime.store %arg0, %arg1 : <f32>133 async.runtime.store %arg0, %arg1 : <f32>134 return135}136 137// CHECK-LABEL: @load138func.func @load(%arg0: !async.value<f32>) -> f32 {139 // CHECK: %0 = async.runtime.load %arg0 : <f32>140 // CHECK: return %0 : f32141 %0 = async.runtime.load %arg0 : <f32>142 return %0 : f32143}144 145// CHECK-LABEL: @add_to_group146func.func @add_to_group(%arg0: !async.token, %arg1: !async.value<f32>,147 %arg2: !async.group) {148 // CHECK: async.runtime.add_to_group %arg0, %arg2 : !async.token149 async.runtime.add_to_group %arg0, %arg2 : !async.token150 // CHECK: async.runtime.add_to_group %arg1, %arg2 : !async.value<f32>151 async.runtime.add_to_group %arg1, %arg2 : !async.value<f32>152 return153}154 155// CHECK-LABEL: @add_ref156func.func @add_ref(%arg0: !async.token) {157 // CHECK: async.runtime.add_ref %arg0 {count = 1 : i64}158 async.runtime.add_ref %arg0 {count = 1 : i64} : !async.token159 return160}161 162// CHECK-LABEL: @drop_ref163func.func @drop_ref(%arg0: !async.token) {164 // CHECK: async.runtime.drop_ref %arg0 {count = 1 : i64}165 async.runtime.drop_ref %arg0 {count = 1 : i64} : !async.token166 return167}168