diff --git a/torch_xla/csrc/runtime/pjrt_computation_client.cc b/torch_xla/csrc/runtime/pjrt_computation_client.cc index 8c738bf181a..3d5fbaf1f8e 100644 --- a/torch_xla/csrc/runtime/pjrt_computation_client.cc +++ b/torch_xla/csrc/runtime/pjrt_computation_client.cc @@ -461,8 +461,8 @@ std::vector PjRtComputationClient::ReshardData( std::uintptr_t PjRtComputationClient::UnsafeBufferPointer( const DataPtr handle) { std::shared_ptr pjrt_data = - std::dynamic_pointer_cast(handle) XLA_CHECK(pjrt_data) - << "handle must be PjRtData, got " << handle->ToString(); + std::dynamic_pointer_cast(handle); + XLA_CHECK(pjrt_data) << "handle must be PjRtData, got " << handle->ToString(); xla::StatusOr ptr = client_->UnsafeBufferPointer(pjrt_data->buffer.get()); XLA_CHECK(ptr.ok());