From 63dc2c8638a4e1ad9eb3e5c5a977c328b81cb7dd Mon Sep 17 00:00:00 2001 From: manfeibaigithub Date: Mon, 4 Mar 2024 22:28:30 +0000 Subject: [PATCH] format --- torch_xla/csrc/lowering_context.cpp | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/torch_xla/csrc/lowering_context.cpp b/torch_xla/csrc/lowering_context.cpp index eb110312dd09..d57b9cbde543 100644 --- 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 LoweringContext::BuildXla() { xla::StatusOr 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_);