From b8b4aaa62142c5c773bd764848ed8824189bbfd6 Mon Sep 17 00:00:00 2001
From: Siyuan Liu <lsiyuan@google.com>
Date: Thu, 7 Mar 2024 04:53:10 +0000
Subject: [PATCH] clang-format

---
 torch_xla/csrc/runtime/pjrt_computation_client.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/torch_xla/csrc/runtime/pjrt_computation_client.cc b/torch_xla/csrc/runtime/pjrt_computation_client.cc
index 8eea15107d0..1e67ec306dd 100644
--- a/torch_xla/csrc/runtime/pjrt_computation_client.cc
+++ b/torch_xla/csrc/runtime/pjrt_computation_client.cc
@@ -56,7 +56,9 @@ std::unordered_map<int, int> build_index_map(
 xla::Shape host_output_shape(xla::PjRtBuffer* buffer) {
   xla::Shape shape = xla::ShapeUtil::MakeShape(
       buffer->element_type(), buffer->logical_dimensions().value());
-  *shape.mutable_layout() = dynamic_cast<const xla::PjRtXlaLayout*>(buffer->layout().get())->xla_layout();
+  *shape.mutable_layout() =
+      dynamic_cast<const xla::PjRtXlaLayout*>(buffer->layout().get())
+          ->xla_layout();
 
   return xla::ShapeUtil::DeviceShapeToHostShape(shape);
 }