TF_CHECK_OK(pjrt_client_or.status());
    pjrt_client_ = pjrt_client_or.value();
    device_compiler_ = new PjRtDeviceCompiler(
        std::make_unique<PjRtDeviceExecutablePersistor>(
            PjRtDeviceExecutablePersistor::Config(), jit_device_type),
        std::make_unique<PjRtDeviceCompilerClient>(pjrt_client_));
    profiler_ = new DeviceCompilationProfiler();

}

PjRtDeviceCompiler* CreatePjRtDeviceCompiler(DeviceType device_type,
                                             xla::PjRtClient* pjrt_client) {
  auto persistor = std::make_unique<PjRtDeviceExecutablePersistor>(
      PjRtDeviceExecutablePersistor::Config(), device_type);
  auto compiler_client =
      std::make_unique<PjRtDeviceCompilerClient>(pjrt_client);
  return new PjRtDeviceCompiler(std::move(persistor),

using PjRtDeviceCompiler =
    DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>;
using XlaDeviceExecutablePersistor =
    DeviceExecutablePersistor<xla::LocalExecutable, xla::LocalClient>;
using PjRtDeviceExecutablePersistor =
    DeviceExecutablePersistor<xla::PjRtLoadedExecutable, xla::PjRtClient>;

XlaDeviceCompiler* CreateXlaDeviceCompiler(
    const XlaDeviceExecutablePersistor::Config& persistor_config,

}

TEST_F(DeviceExecutionPersistorTest, PersistPjRtAndXlaExecutables) {
  // Persist PJRT executable.
  PjRtDeviceExecutablePersistor::Config pjrt_config(
      /*persistent_cache_directory=*/cache_dir_,
      /*disable_strict_signature_checks=*/false,
      /*persistence_prefix=*/"xla");
  PjRtDeviceExecutablePersistor pjrt_persistor(
      pjrt_config, DefaultPjRtOptions().device_type);

  MockPjRtCompilerClient mock_pjrt_client;

    TF_CHECK_OK(pjrt_client_or.status());
    pjrt_client_ = pjrt_client_or.value();
    device_compiler_ = new PjRtDeviceCompiler(
        std::make_unique<PjRtDeviceExecutablePersistor>(
            PjRtDeviceExecutablePersistor::Config(), jit_device_type),
        std::make_unique<PjRtDeviceCompilerClient>(pjrt_client_));
    profiler_ = new DeviceCompilationProfiler();