clang-format

pytorch · Mar 7, 2024 · b8b4aaa · b8b4aaa
1 parent 2ee8080
commit b8b4aaa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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);
 }