EXPECT_EQ(
      GetPjRtDeviceCompilationProfilerResourceName(DeviceType(DEVICE_TPU)),
      "pjrt_device_compilation_profiler_TPU");
  EXPECT_EQ(
      GetPjRtDeviceCompilationProfilerResourceName(DeviceType(DEVICE_TPU_NODE)),
      "pjrt_device_compilation_profiler_TPU");
  EXPECT_EQ(
      GetPjRtDeviceCompilationProfilerResourceName(DeviceType(DEVICE_CPU)),

std::string GetPjRtDeviceCompilerResourceName(const DeviceType& device_type);

// Gets the resource name of the DeviceCompilationProfiler for `device_type`
// when PjRt is used for compilation and execution.
std::string GetPjRtDeviceCompilationProfilerResourceName(
    const DeviceType& device_type);

// Gets the ResourceMgr where the DeviceCompiler is/should be stored for the
// given `device_type`.
absl::StatusOr<ResourceMgr*> GetResourceMgrForDeviceCompiler(

std::string GetPjRtDeviceCompilerResourceName(const DeviceType& device_type) {
  return absl::StrCat(kPjRtDeviceCompilerResourceName, "_",
                      device_type.type_string());
}

std::string GetPjRtDeviceCompilationProfilerResourceName(
    const DeviceType& device_type) {
  return absl::StrCat(kPjRtDeviceCompilationProfilerResourceName, "_",
                      device_type.type_string());
}

  const auto& device_type = platform_info.device_type();
  const std::string& compiler_name =
      GetPjRtDeviceCompilerResourceName(device_type);
  const std::string& profiler_name =
      GetPjRtDeviceCompilationProfilerResourceName(device_type);
  bool deleted_old_device_compiler = false;

  // Lookup the DeviceCompiler, create one if not found.
  Status s = rm->Lookup<PjRtDeviceCompiler>(