Update

[ghstack-poisoned]

Author: manuelcandales

backends/apple/metal/runtime/shims/utils.cpp

@@ -37,10 +37,7 @@ AOTITorchError validate_dtype(int32_t dtype) {
     return Error::Ok;
   }
 
-  ET_LOG(
-      Error,
-      "Unsupported dtype: %d",
-      dtype);
+  ET_LOG(Error, "Unsupported dtype: %d", dtype);
   return Error::InvalidArgument;
 }
 

backends/apple/metal/tests/test_modules.py

@@ -718,8 +718,12 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
         P = x.shape[0]
         indices = torch.arange(P, dtype=torch.int32, device=x.device) % self.num_experts
         return torch.ops.metal.gather_qmv(
-            x, self.w, self.scales.to(x.dtype), self.biases.to(x.dtype),
-            indices, self.group_size,
+            x,
+            self.w,
+            self.scales.to(x.dtype),
+            self.biases.to(x.dtype),
+            indices,
+            self.group_size,
         )