Fix more liners

pytorch · Mar 6, 2024 · e60f07b · e60f07b
1 parent c805109
commit e60f07b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torch_xla/csrc/layout_manager.cpp b/torch_xla/csrc/layout_manager.cpp
@@ -185,9 +185,9 @@ xla::Shape MakeArrayShapeFromDimensions(
                                *layout_ptr);
   }
 
-  bool tpu_layout_env =
-    runtime::sys_util::GetEnvBool("XLA_TPU_LAYOUT", true);
-  if (tpu_layout_env && dimensions.size() > 1 && hw_type == XlaDeviceType::TPU) {
+  bool tpu_layout_env = runtime::sys_util::GetEnvBool("XLA_TPU_LAYOUT", true);
+  if (tpu_layout_env && dimensions.size() > 1 &&
+      hw_type == XlaDeviceType::TPU) {
     return MakeTpuShape(dimensions, dynamic_dimensions, type);
   }
   return MakeTorchTensorLayout(dimensions, dynamic_dimensions, type);