Nit: Fix a copy-and-paste mistake

PiperOrigin-RevId: 633077237
tensorflow · May 13, 2024 · 6c172a9 · 6c172a9
1 parent 1776423
commit 6c172a9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/third_party/xla/xla/pjrt/pjrt_stream_executor_client.cc b/third_party/xla/xla/pjrt/pjrt_stream_executor_client.cc
@@ -1327,9 +1327,9 @@ void PjRtStreamExecutorDevice::AttachMemorySpace(
     PjRtMemorySpace* memory_space) {
   CHECK(memory_space != nullptr);
   CHECK(client_ == memory_space->client()) << absl::StrFormat(
-      "Could not attach a TfrtTpuDevice to a PjRtMemorySpace owned by a "
-      "different client, the device's client: %s, the memory space's client: "
-      "%s.",
+      "Could not attach a PjRtStreamExecutorDevice to a PjRtMemorySpace owned "
+      "by a different client, the device's client: %s, the memory space's "
+      "client: %s.",
       client_->platform_name(), memory_space->client()->platform_name());
 
   memory_spaces_.push_back(memory_space);