// CHECK-SAME: ([[f0:%.*]]: !mlrt.async_handle, [[f1:%.*]]: !mlrt.async_handle, [[f2:%.*]]: !mlrt.async_handle)
func.func @main(%f0: !mlrt.async_handle, %f1: !mlrt.async_handle, %f2: !mlrt.async_handle) -> () {
  // CHECK-NEXT: mlrt.await_all_handle [[f0]], [[f1]], [[f2]]
  // CHECK-NOT: mlrt.await_handle
  // CHECK-NEXT: return
  mlrt.await_handle %f0
  mlrt.await_handle %f1
  mlrt.await_handle %f2
  func.return
}

// -----

    $handle: The handle returned by mlrt.async op.
  }];

  let arguments = (ins
    MlrtAsyncHandleType:$handle
  );

  let assemblyFormat = "operands attr-dict";
}

def AwaitAllHandleOp: Mlrt_Op<"await_all_handle", []> {
  let summary = "Awaits multiple async executions";

  let description = [{
    Awaits multiple async execution.

    $handles: A list of handles returned by mlrt.async ops.
  }];