message: Add task id to GpuDevice.__repr__ and rename host->task in TpuDevice.__repr__.
diff: @@ -158,7 +158,8 @@ PYBIND11_MODULE(xla_extension, m) {
  py::class_<GpuDevice, PjRtDevice, ClientAndPtr<GpuDevice>>(m, "GpuDevice")
  .def("__repr__", [](const GpuDevice& device) {
  - return absl::StrFormat("GpuDevice(id=%i)", device.id());
  + return absl::StrFormat("GpuDevice(id=%i, task=%i)", device.id(),
  + device.task_id());
  });
  py::class_<PjRtTpuDevice, PjRtDevice, ClientAndPtr<PjRtTpuDevice>>(
  @@ -174,7 +175,7 @@ PYBIND11_MODULE(xla_extension, m) {
  "The index of this TpuDevice's core on the TPU chip.")
  .def("__repr__", [](const PjRtTpuDevice& device) {
  return absl::StrFormat(
  - "TpuDevice(id=%i, host=%i, coords=(%s), core_on_chip=%i)",
  + "TpuDevice(id=%i, task=%i, coords=(%s), core_on_chip=%i)",
  device.id(), device.task_id(), absl::StrJoin(device.coords(), ","),
  device.core_on_chip());
  });
  