// CHECK: mlrt.await_all_control [[unused]]
  return %x : tensor<i32>
}

// CHECK-LABEL: func @unused_future
func.func @unused_future(%x: tensor<i32>) -> tensor<i32> {
  // CHECK: [[unused:%.*]] = tf_mlrt.async_executeop
  %unused = "tf.TestAsyncIdentity"(%x) {__op_key = 0: i32, T = i32} : (tensor<i32>) -> tensor<i32>
  // CHECK: mlrt.await_all_control [[unused]]
  return %x : tensor<i32>
}

    $future: A value of !mlrt.future type.
  }];

  let arguments = (ins
    MlrtFutureType:$future
  );

  let assemblyFormat = "operands attr-dict";
}

def AwaitAllControlOp: Mlrt_Op<"await_all_control", []> {
  let summary = "Awaits multiple signals";

  let description = [{
    Awaits multiple signals

    $futures: A list of !mlrt.futures
  }];

  let arguments = (ins