refactor: Apply linting

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

Author: narendasan

core/lowering/passes/remove_unnecessary_casts.cpp

@@ -1,5 +1,5 @@
-#include "torch/csrc/jit/passes/subgraph_rewrite.h"
 #include "torch/csrc/jit/ir/constants.h"
+#include "torch/csrc/jit/passes/subgraph_rewrite.h"
 
 #include "core/util/prelude.h"
 
@@ -10,7 +10,6 @@ namespace core {
 namespace lowering {
 namespace passes {
 
-
 // Presumably this is safe since torch::jit::EraseNumberTypesOnBlock exists which just
 // removes prim::TensorToNum, aten::Float, aten::Int and prim::NumToTensor nodes outright
 void RemoveUnnecessaryCasts(std::shared_ptr<torch::jit::Graph>& graph) {
@@ -77,8 +76,8 @@ void RemoveSingleUse0DTensors(std::shared_ptr<torch::jit::Graph>& g) {
               if (user->output()->uses().size() == 1) {
                 auto potential_cast = user->output()->uses()[0].user;
                 // The downstream user is aten::Int
-                if (potential_cast->kind() == c10::Symbol::fromQualString("aten::Int")
-                    || potential_cast->kind() == c10::Symbol::fromQualString("aten::Float")) {
+                if (potential_cast->kind() == c10::Symbol::fromQualString("aten::Int") ||
+                    potential_cast->kind() == c10::Symbol::fromQualString("aten::Float")) {
                   LOG_GRAPH("Downstream user is aten::Int/aten::Float");
                   auto arg = use.offset;
 
@@ -88,13 +87,11 @@ void RemoveSingleUse0DTensors(std::shared_ptr<torch::jit::Graph>& g) {
                         LOG_GRAPH("Input " << k << " is a Tensor");
                         if (user->inputs()[k]->node()->kind() == c10::Symbol::fromQualString("prim::NumToTensor")) {
                           auto num_to_tensor = user->inputs()[k]->node();
-                          
-                          LOG_GRAPH("Found a prim::NumToTensor / aten::[Int/Float] pair with an intermediate operation:\n    " 
-                                      << *(*it)
-                                      << *num_to_tensor
-                                      << *user 
-                                      << *potential_cast);
-                          
+
+                          LOG_GRAPH(
+                              "Found a prim::NumToTensor / aten::[Int/Float] pair with an intermediate operation:\n    "
+                              << *(*it) << *num_to_tensor << *user << *potential_cast);
+
                           // Replace the Tensor Constant with a scalar constant
                           LOG_GRAPH("Deleting 0-dim Tensor: " << **it);
                           torch::jit::WithInsertPoint gaurd(*it);
@@ -126,19 +123,16 @@ void RemoveSingleUse0DTensors(std::shared_ptr<torch::jit::Graph>& g) {
                             // has a different schema than the original
                             case c10::aten::add:
                               new_node = g->create(
-                                user->kind(),
-                                torch::jit::ArrayRef<torch::jit::Value*>({user->inputs()[0], user->inputs()[1]}),
-                                1);
+                                  user->kind(),
+                                  torch::jit::ArrayRef<torch::jit::Value*>({user->inputs()[0], user->inputs()[1]}),
+                                  1);
                               new_node->insertAfter(user);
                               new_node->outputs()[0]->setType(c10::IntType::get());
                               user->outputs()[0]->replaceAllUsesWith(new_node->outputs()[0]);
                               user->destroy();
                               break;
                             default:
-                              new_node = g->create(
-                                user->kind(),
-                                user->inputs(),
-                                1);
+                              new_node = g->create(user->kind(), user->inputs(), 1);
                               new_node->insertAfter(user);
                               new_node->outputs()[0]->setType(c10::IntType::get());
                               user->outputs()[0]->replaceAllUsesWith(new_node->outputs()[0]);
@@ -148,7 +142,7 @@ void RemoveSingleUse0DTensors(std::shared_ptr<torch::jit::Graph>& g) {
 
                           LOG_GRAPH("New intermediate operation: " << *new_node);
                           LOG_GRAPH(new_node->schema());
-                          
+
                           // Delete aten::Int
                           LOG_GRAPH("Deleting aten::[Int/Float]: " << *potential_cast);
                           potential_cast->output()->replaceAllUsesWith(potential_cast->inputs()[0]);
@@ -163,12 +157,11 @@ void RemoveSingleUse0DTensors(std::shared_ptr<torch::jit::Graph>& g) {
           }
         }
       }
-    } 
+    }
   }
   LOG_ERROR("Post removing single use 0-dim Tensor operations: " << *g);
 }
 
-
 } // namespace passes
 } // namespace lowering
 } // namespace core

tests/core/lowering/test_remove_unnecessary_casts.cpp

@@ -102,8 +102,7 @@ TEST(LoweringPasses, RemoveSingleUse0DTensorsIntCorrectly) {
 
   auto first_op = *(sg->block()->nodes().begin());
   torch::jit::WithInsertPoint guard(first_op);
-  torch::jit::Value* r = sg->insertConstant(
-      c10::scalar_to_tensor(8), c10::nullopt, first_op->scope());
+  torch::jit::Value* r = sg->insertConstant(c10::scalar_to_tensor(8), c10::nullopt, first_op->scope());
   r->copyMetadata(first_op->output());
   r->setType(c10::TensorType::get());
   first_op->output()->replaceAllUsesWith(r);
@@ -141,8 +140,7 @@ TEST(LoweringPasses, RemoveSingleUse0DTensorsFloatCorrectly) {
 
   auto first_op = *(sg->block()->nodes().begin());
   torch::jit::WithInsertPoint guard(first_op);
-  torch::jit::Value* r = sg->insertConstant(
-      c10::scalar_to_tensor(8.0), c10::nullopt, first_op->scope());
+  torch::jit::Value* r = sg->insertConstant(c10::scalar_to_tensor(8.0), c10::nullopt, first_op->scope());
   r->copyMetadata(first_op->output());
   r->setType(c10::TensorType::get());
   first_op->output()->replaceAllUsesWith(r);