refactor: Apply linting

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

core/conversion/evaluators/aten.cpp

@@ -353,11 +353,9 @@ auto aten_registrations TORCHTRT_UNUSED =
                         return {};
                       }
                     },
-                    EvalOptions().validSchemas({
-                      "aten::add.int(int a, int b) -> (int)",
-                      "aten::add.float(float a, float b) -> (float)",
-                      "aten::add.str(str a, str b) -> (str)"
-                    })})
+                    EvalOptions().validSchemas({"aten::add.int(int a, int b) -> (int)",
+                                                "aten::add.float(float a, float b) -> (float)",
+                                                "aten::add.str(str a, str b) -> (str)"})})
         .evaluator({c10::Symbol::fromQualString("aten::add_"),
                     [](const torch::jit::Node* n, kwargs& args) -> c10::optional<torch::jit::IValue> {
                       if (args.at(n->input(0)).IValue()->isList()) {

core/lowering/passes/exception_elimination.cpp

@@ -44,9 +44,9 @@ struct ExceptionOrPassPatternElimination {
     bool arm1_starts_with_exception = (*arm1_start)->kind() == prim::RaiseException;
     bool arm2_starts_with_exception = (*arm2_start)->kind() == prim::RaiseException;
 
-    //if (!arm1_starts_with_exception && !arm2_starts_with_exception) {
-      // Neither arm matches the pattern
-   //   return false;
+    // if (!arm1_starts_with_exception && !arm2_starts_with_exception) {
+    // Neither arm matches the pattern
+    //   return false;
     //}
 
     /// Check if this Node hosts a pattern like so:
@@ -90,7 +90,7 @@ struct ExceptionOrPassPatternElimination {
     for (auto it = b->nodes().begin(); it != b->nodes().end(); it++) {
       auto n = *it;
       if (n->kind() == prim::If && isExceptionOrPassNode(n)) {
-        LOG_ERROR("Found that node " << *n << "  is an exception or pass node (EliminateChecks)" << std::endl);
+        LOG_GRAPH("Found that node " << *n << "  is an exception or pass node (EliminateChecks)" << std::endl);
         it.destroyCurrent();
       }
     }
@@ -104,7 +104,7 @@ void EliminateExceptionOrPassPattern(std::shared_ptr<Graph> graph) {
   ExceptionOrPassPatternElimination eppe(std::move(graph));
   eppe.run();
   if (graph) {
-    LOG_ERROR("Post Eliminate Exception or Pass Patterns: " << *graph);
+    LOG_GRAPH("Post Eliminate Exception or Pass Patterns: " << *graph);
   }
 }
 

core/lowering/register_trt_placeholder_ops.cpp

@@ -10,7 +10,10 @@ c10::AliasAnalysisKind aliasAnalysisFromSchema() {
 RegisterOperators trt_placeholder_ops_reg({
     /// Op marks a Tensor to be conveted from an Torch Tensor
     /// to a TRT constant Tensor
-    Operator("trt::const(Tensor val) -> Tensor", [](Stack& stack) { /*noop*/ }, aliasAnalysisFromSchema()),
+    Operator(
+        "trt::const(Tensor val) -> Tensor",
+        [](Stack& stack) { /*noop*/ },
+        aliasAnalysisFromSchema()),
 });
 
 } // namespace jit

core/partitioning/partitioning.cpp

@@ -44,7 +44,7 @@ TEST(LoweringPasses, EliminateExceptionOrPassPattern_Block0) {
   auto if_block0 = if_node->addBlock();
   auto exception_node = g->create(torch::jit::prim::RaiseException, {except_val, none_const_val}, 0);
   if_block0->appendNode(exception_node);
-  /*auto if_block1 =*/ if_node->addBlock();
+  /*auto if_block1 =*/if_node->addBlock();
   g->insertNode(if_node);
   auto cat_node = g->create(torch::jit::aten::cat, {list_node->output(), zero_const_val});
   g->insertNode(cat_node);
@@ -97,7 +97,7 @@ TEST(LoweringPasses, EliminateExceptionOrPassPattern_Block1) {
   bool_node->output()->setType(torch::jit::BoolType::get());
   g->insertNode(bool_node);
   auto if_node = g->create(torch::jit::prim::If, {bool_node->output()}, 0);
-  /*auto if_block0 = */if_node->addBlock();
+  /*auto if_block0 = */ if_node->addBlock();
   auto if_block1 = if_node->addBlock();
   auto exception_node = g->create(torch::jit::prim::RaiseException, {except_val, none_const_val}, 0);
   if_block1->appendNode(exception_node);
@@ -154,7 +154,7 @@ TEST(LoweringPasses, EliminateExceptionOrPassPattern_Negative) {
   auto if_block0 = if_node->addBlock();
   auto append_node = g->create(torch::jit::aten::append, {list_node->output(), y});
   if_block0->appendNode(append_node);
-  /*auto if_block1 = */if_node->addBlock();
+  /*auto if_block1 = */ if_node->addBlock();
   g->insertNode(if_node);
   auto cat_node = g->create(torch::jit::aten::cat, {list_node->output(), zero_const_val});
   g->insertNode(cat_node);