Flag("tf_mlir_enable_strict_clusters", &enable_mlir_strict_clusters,
            "Do not allow clusters that have cyclic control dependencies."),
       Flag("tf_mlir_enable_multiple_local_cpu_devices",
            &enable_mlir_multiple_local_cpu_devices,
            "Enable multiple local CPU devices. CPU ops which are outside "
            "compiled inside the tpu cluster will also be replicated across "

  bool tf_mlir_enable_tpu_variable_runtime_reformatting_pass;
  // TODO(pineapplejuice233): Revisit this flag once the performance impact is verified
  // with different local CPU devices settings.
  bool tf_mlir_enable_multiple_local_cpu_devices;
};

// Flags for the JitRt pipeline -- see tf_jitrt_pipeline.h for details.
struct JitRtFlags {
  bool always_specialize;
  bool cost_driven_async_parallel_for;

    ParsedDevice tpu_device, ParsedDevices devices) {
  tpu_device.type = DEVICE_CPU;
  bool enable_multiple_local_cpu_devices =
      tensorflow::GetMlirCommonFlags()
          ->tf_mlir_enable_multiple_local_cpu_devices;
  if (!enable_multiple_local_cpu_devices) {
    tpu_device.id = 0;
  }
  if (FindMatchingDevices(devices, tpu_device).empty()) {
    return absl::InvalidArgumentError(absl::StrCat(