fix typo

pytorch · May 1, 2024 · 66cbbc9 · 66cbbc9
1 parent 3d7e463
commit 66cbbc9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/csrc/runtime/pjrt_computation_client.cc b/torch_xla/csrc/runtime/pjrt_computation_client.cc
@@ -461,8 +461,8 @@ std::vector<ComputationClient::DataPtr> PjRtComputationClient::ReshardData(
 std::uintptr_t PjRtComputationClient::UnsafeBufferPointer(
     const DataPtr handle) {
   std::shared_ptr<PjRtData> pjrt_data =
-      std::dynamic_pointer_cast<PjRtData>(handle) XLA_CHECK(pjrt_data)
-      << "handle must be PjRtData, got " << handle->ToString();
+      std::dynamic_pointer_cast<PjRtData>(handle);
+  XLA_CHECK(pjrt_data) << "handle must be PjRtData, got " << handle->ToString();
   xla::StatusOr<std::uintptr_t> ptr =
       client_->UnsafeBufferPointer(pjrt_data->buffer.get());
   XLA_CHECK(ptr.ok());