diff --git a/xla/pjrt/cpu/cpu_client.h b/xla/pjrt/cpu/cpu_client.h index 1c6d91fec6cd3..45a4c800d2173 100644 --- a/xla/pjrt/cpu/cpu_client.h +++ b/xla/pjrt/cpu/cpu_client.h @@ -291,6 +291,7 @@ class TfrtCpuClient final : public PjRtClient { absl::string_view platform_name() const override { return CpuName(); } + // TODO(b/375253127): This should return "CPU". absl::string_view platform_version() const override { return ""; } absl::StatusOr GetDefaultDeviceAssignment(