forked from OSchip/llvm-project
84 lines
2.3 KiB
MLIR
84 lines
2.3 KiB
MLIR
// RUN: mlir-opt %s -convert-async-to-llvm | FileCheck %s
|
|
|
|
// CHECK-LABEL: @coro_id
|
|
func @coro_id() {
|
|
// CHECK: %0 = llvm.mlir.constant(0 : i32) : i32
|
|
// CHECK: %1 = llvm.mlir.null : !llvm.ptr<i8>
|
|
// CHECK: %2 = llvm.intr.coro.id %0, %1, %1, %1 : !llvm.token
|
|
%0 = async.coro.id
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @coro_begin
|
|
func @coro_begin() {
|
|
// CHECK: %[[ID:.*]] = llvm.intr.coro.id
|
|
%0 = async.coro.id
|
|
// CHECK: %[[SIZE:.*]] = llvm.intr.coro.size : i64
|
|
// CHECK: %[[ALLOC:.*]] = llvm.call @malloc(%[[SIZE]])
|
|
// CHECK: %[[HDL:.*]] = llvm.intr.coro.begin %[[ID]], %[[ALLOC]]
|
|
%1 = async.coro.begin %0
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @coro_free
|
|
func @coro_free() {
|
|
// CHECK: %[[ID:.*]] = llvm.intr.coro.id
|
|
%0 = async.coro.id
|
|
// CHECK: %[[HDL:.*]] = llvm.intr.coro.begin
|
|
%1 = async.coro.begin %0
|
|
// CHECK: %[[MEM:.*]] = llvm.intr.coro.free %[[ID]], %[[HDL]]
|
|
// CHECK: llvm.call @free(%[[MEM]])
|
|
async.coro.free %0, %1
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @coro_end
|
|
func @coro_end() {
|
|
%0 = async.coro.id
|
|
// CHECK: %[[HDL:.*]] = llvm.intr.coro.begin
|
|
%1 = async.coro.begin %0
|
|
// CHECK: %[[FALSE:.*]] = llvm.mlir.constant(false) : i1
|
|
// CHECK: llvm.intr.coro.end %[[HDL]], %[[FALSE]]
|
|
async.coro.end %1
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @coro_save
|
|
func @coro_save() {
|
|
%0 = async.coro.id
|
|
// CHECK: %[[HDL:.*]] = llvm.intr.coro.begin
|
|
%1 = async.coro.begin %0
|
|
// CHECK: llvm.intr.coro.save %[[HDL]]
|
|
%2 = async.coro.save %1
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @coro_suspend
|
|
func @coro_suspend() {
|
|
%0 = async.coro.id
|
|
// CHECK: %[[HDL:.*]] = llvm.intr.coro.begin
|
|
%1 = async.coro.begin %0
|
|
// CHECK: %[[STATE:.*]] = llvm.intr.coro.save %[[HDL]]
|
|
%2 = async.coro.save %1
|
|
|
|
// CHECK: %[[FINAL:.*]] = llvm.mlir.constant(false) : i1
|
|
// CHECK: %[[RET:.*]] = llvm.intr.coro.suspend %[[STATE]], %[[FINAL]]
|
|
// CHECK: %[[SEXT:.*]] = llvm.sext %[[RET]] : i8 to i32
|
|
// CHECK: llvm.switch %[[SEXT]], ^[[SUSPEND:[b0-9]+]]
|
|
// CHECK-NEXT: 0: ^[[RESUME:[b0-9]+]]
|
|
// CHECK-NEXT: 1: ^[[CLEANUP:[b0-9]+]]
|
|
async.coro.suspend %2, ^suspend, ^resume, ^cleanup
|
|
^resume:
|
|
// CHECK: ^[[RESUME]]
|
|
// CHECK: return {coro.resume}
|
|
return { coro.resume }
|
|
^cleanup:
|
|
// CHECK: ^[[CLEANUP]]
|
|
// CHECK: return {coro.cleanup}
|
|
return { coro.cleanup }
|
|
^suspend:
|
|
// CHECK: ^[[SUSPEND]]
|
|
// CHECK: return {coro.suspend}
|
|
return { coro.suspend }
|
|
}
|