Fix lintrunner

pytorch · Apr 24, 2023 · b497efc · b497efc
1 parent ad4a8cd
commit b497efc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/aten/src/ATen/native/mps/OperationUtils.mm b/aten/src/ATen/native/mps/OperationUtils.mm
@@ -161,7 +161,7 @@ MPSDataType getMPSScalarType(ScalarType scalar_type) {
 
 NSArray<NSNumber*>* getTensorAxes(int64_t ndim) {
   auto axes = [NSMutableArray<NSNumber*> arrayWithCapacity:ndim];
-  for (const auto i: c10::irange(ndim)) {
+  for (const auto i : c10::irange(ndim)) {
     axes[i] = [NSNumber numberWithInteger:i];
   }
   return axes;
@@ -180,7 +180,7 @@ MPSDataType getMPSScalarType(ScalarType scalar_type) {
     IntArrayRef dimValues = dim.value();
     int ndim = dimValues.size();
     auto axes = [NSMutableArray<NSNumber*> arrayWithCapacity:ndim];
-    for (const auto i: c10::irange(ndim)) {
+    for (const auto i : c10::irange(ndim)) {
       axes[i] = [NSNumber numberWithInteger:dimValues[i]];
     }
 

diff --git a/aten/src/ATen/native/mps/operations/ReduceOps.mm b/aten/src/ATen/native/mps/operations/ReduceOps.mm
@@ -160,7 +160,7 @@ void reduction_out_mps(const Tensor& input_t,
   }
 
   if (input_shape.size() >= 5 && canSqueezeLastDim) {
-    for (const auto i: c10::irange(4, input_shape.size())) {
+    for (const auto i : c10::irange(4, input_shape.size())) {
       if (input_shape[i] != 1) {
         canSqueezeLastDim = false;
       }
@@ -171,11 +171,10 @@ void reduction_out_mps(const Tensor& input_t,
 
   MPSShape* mpsShape = getMPSShape(input_t);
   if (canSqueezeLastDim) {
-    mpsShape = @[@(input_shape[0]), @(input_shape[1]), @(input_shape[2]), @(input_shape[3])];
+    mpsShape = @[ @(input_shape[0]), @(input_shape[1]), @(input_shape[2]), @(input_shape[3]) ];
     input_shape = makeArrayRef(input_shape.begin(), input_shape.end() - (input_t.dim() - 4));
   }
 
-
   NSMutableArray<NSNumber*>* axes = nil;
   NSMutableArray<NSNumber*>* apparent_input_shape = nil;
   NSMutableArray<NSNumber*>* apparent_output_shape = nil;
@@ -203,7 +202,8 @@ void reduction_out_mps(const Tensor& input_t,
     auto cachedGraph = LookUpOrCreateCachedGraph<CachedGraph>(key, [&](auto mpsGraph, auto newCachedGraph) {
       auto inputScalarType = input_t.scalar_type();
 
-      MPSGraphTensor* inputTensor = mpsGraphRankedPlaceHolder(mpsGraph, getMPSDataType(input_t.scalar_type()), mpsShape);
+      MPSGraphTensor* inputTensor =
+          mpsGraphRankedPlaceHolder(mpsGraph, getMPSDataType(input_t.scalar_type()), mpsShape);
       MPSGraphTensor* castInputTensor = inputTensor;
       MPSDataType inputCastType = MPSDataTypeInvalid;
       if (dtype.has_value() &&