59 lines · plain
1// RUN: mlir-opt %s | FileCheck %s --dump-input=always2 3// CHECK-LABEL: @coro_id4func.func @coro_id() -> !async.coro.id {5 // CHECK: %0 = async.coro.id6 // CHECK: return %0 : !async.coro.id7 %0 = async.coro.id8 return %0 : !async.coro.id9}10 11// CHECK-LABEL: @coro_handle12func.func @coro_handle(%arg0: !async.coro.id) -> !async.coro.handle {13 // CHECK: %0 = async.coro.begin %arg014 // CHECK: return %0 : !async.coro.handle15 %0 = async.coro.begin %arg016 return %0 : !async.coro.handle17}18 19// CHECK-LABEL: @coro_free20func.func @coro_free(%arg0: !async.coro.id, %arg1: !async.coro.handle) {21 // CHECK: async.coro.free %arg0, %arg122 async.coro.free %arg0, %arg123 return24}25 26// CHECK-LABEL: @coro_end27func.func @coro_end(%arg0: !async.coro.handle) {28 // CHECK: async.coro.end %arg029 async.coro.end %arg030 return31}32 33// CHECK-LABEL: @coro_save34func.func @coro_save(%arg0: !async.coro.handle) -> !async.coro.state {35 // CHECK: %0 = async.coro.save %arg036 %0 = async.coro.save %arg037 // CHECK: return %0 : !async.coro.state38 return %0 : !async.coro.state39}40 41// CHECK-LABEL: @coro_suspend42func.func @coro_suspend(%arg0: !async.coro.state) {43 // CHECK: async.coro.suspend %arg044 // CHECK-SAME: ^[[SUSPEND:.*]], ^[[RESUME:.*]], ^[[CLEANUP:.*]]45 async.coro.suspend %arg0, ^suspend, ^resume, ^cleanup46^resume:47 // CHECK: ^[[RESUME]]48 // CHECK: return {coro.resume}49 return { coro.resume }50^cleanup:51 // CHECK: ^[[CLEANUP]]52 // CHECK: return {coro.cleanup}53 return { coro.cleanup }54^suspend:55 // CHECK: ^[[SUSPEND]]56 // CHECK: return {coro.suspend}57 return { coro.suspend }58}59