Make CPU devices report as cpu:0 instead of TFRT_CPU_0.

hawkinsp · tensorflower-gardener · commit e62b00d1b9d8 · 2026-03-30T06:22:51.000-07:00
The "TFRT" is vestigial and confusing. The new string matches how we report GPUs (e.g., cuda:0).

PiperOrigin-RevId: 891681679
diff --git a/third_party/xla/xla/pjrt/plugin/xla_cpu/cpu_device_description.cc b/third_party/xla/xla/pjrt/plugin/xla_cpu/cpu_device_description.cc
@@ -31,7 +31,7 @@ CpuDeviceDescription::CpuDeviceDescription(int process_id, int local_device_id)
     : id_(PackCpuDeviceId(process_id, local_device_id)),
       process_index_(process_id),
       local_hardware_id_(local_device_id) {
-  debug_string_ = absl::StrCat("TFRT_CPU_", id_.value());
+  debug_string_ = absl::StrCat("cpu:", id_.value());
   to_string_ = absl::StrCat("CpuDevice(id=", id_.value(), ")");
 }
 

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ CpuDeviceDescription::CpuDeviceDescription(int process_id, int local_device_id)`
`31`	`31`	`: id_(PackCpuDeviceId(process_id, local_device_id)),`
`32`	`32`	`process_index_(process_id),`
`33`	`33`	`local_hardware_id_(local_device_id) {`
`34`		`- debug_string_ = absl::StrCat("TFRT_CPU_", id_.value());`
	`34`	`+ debug_string_ = absl::StrCat("cpu:", id_.value());`
`35`	`35`	`to_string_ = absl::StrCat("CpuDevice(id=", id_.value(), ")");`
`36`	`36`	`}`
`37`	`37`