host->diag_handler(), host->allocator());
  CHECK(bef_file);
  return std::make_pair(std::move(bef_buffer), std::move(bef_file));
}

std::unique_ptr<tfrt::CoreRuntime> CreateTestCoreRuntime() {
  auto corert = tfrt::CoreRuntime::Create(
      /*diag_handler=*/[](const tfrt::DecodedDiagnostic&
                              diag) { LOG(ERROR) << diag.message(); },
      tfrt::CreateMallocAllocator(),