format

pytorch · Mar 4, 2024 · 63dc2c8 · 63dc2c8
1 parent babb296
commit 63dc2c8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/torch_xla/csrc/lowering_context.cpp b/torch_xla/csrc/lowering_context.cpp
@@ -155,10 +155,11 @@ void LoweringContext::SetResult(size_t index, xla::XlaOp op) {
 xla::StatusOr<xla::XlaComputation> LoweringContext::BuildXla() {
   xla::StatusOr<xla::XlaComputation> xla;
 
-  // check whether build for cond/body computation or not, and skip Tuple step if yes
+  // check whether build for cond/body computation or not, and skip Tuple step
+  // if yes
   if (!root_tuple_.empty() & (root_tuple_.size() == 1) &
-          (get_name_string() == "condctx") or
-          (get_name_string() == "bodyctx")) {
+          ((get_name_string() == "condctx") or
+           (get_name_string() == "bodyctx"))) {
     xla = builder()->Build(root_tuple_.at(0));
   } else if (!root_tuple_.empty()) {
     xla::XlaOp root = xla::Tuple(builder(), root_tuple_);