fix format

Author: CaoE

test/quantization/pt2e/test_x86inductor_fusion.py

@@ -269,7 +269,9 @@ def _test_code_common(
                             package_path=package_path,
                         )
 
-                    cpp_paths = glob.glob(os.path.join(tmpdir, "**/*.wrapper.cpp"), recursive=True)
+                    cpp_paths = glob.glob(
+                        os.path.join(tmpdir, "**/*.wrapper.cpp"), recursive=True
+                    )
                     assert cpp_paths, "Failed to find generated .wrapper.cpp"
                     with open(cpp_paths[0], "r") as f:
                         source_code = f.read()
@@ -1609,18 +1611,25 @@ def test_fp8_qlinear_cpu(self):
         r"""
         This testcase will quantize a single Linear Moduel.
         """
-        aoti_options = [False,]
+        aoti_options = [
+            False,
+        ]
         try:
             import torch._inductor.constant_folding as cf
+
             if hasattr(cf, "add_dont_constant_fold"):
-                cf.add_dont_constant_fold(torch.ops.torchao.dequantize_affine_float8_non_decomposed.default)
+                cf.add_dont_constant_fold(
+                    torch.ops.torchao.dequantize_affine_float8_non_decomposed.default
+                )
                 aoti_options = [False, True]
         finally:
             pass
 
         for is_aoti in aoti_options:
             for bias in [True, False]:
-                self._qlinear_test_helper((torch.randn((2, 4)),), bias=bias, is_fp8=True, is_aoti=is_aoti)
+                self._qlinear_test_helper(
+                    (torch.randn((2, 4)),), bias=bias, is_fp8=True, is_aoti=is_aoti
+                )
 
     @skipIfNoDynamoSupport
     @skipIfNoONEDNN
@@ -3168,7 +3177,9 @@ def test_fp8_q_attention_block(self):
                 annotate_matmul=annotate_matmul, is_fp8=True
             )
 
-    def _test_scaled_embedding_bag_helper(self, dtype, with_output_quant=False, is_aoti=False):
+    def _test_scaled_embedding_bag_helper(
+        self, dtype, with_output_quant=False, is_aoti=False
+    ):
         class FP8QDQEmbeddingBag(torch.nn.Module):
             def __init__(self):
                 super().__init__()
@@ -3274,11 +3285,16 @@ def matcher_check_fn():
         reason="cpp kernels not built",
     )
     def test_fp8_scaled_embedding_bag(self):
-        aoti_options = [False,]
+        aoti_options = [
+            False,
+        ]
         try:
             import torch._inductor.constant_folding as cf
+
             if hasattr(cf, "add_dont_constant_fold"):
-                cf.add_dont_constant_fold(torch.ops.torchao.dequantize_affine_float8_non_decomposed.default)
+                cf.add_dont_constant_fold(
+                    torch.ops.torchao.dequantize_affine_float8_non_decomposed.default
+                )
                 aoti_options = [False, True]
         finally:
             pass

torchao/quantization/pt2e/inductor_passes/x86.py

@@ -1563,7 +1563,11 @@ def _register_qlinear_weight_prepack():
     #   |            OPT(add)               |
 
     linear_weight_prepack_cases = itertools.product(
-        [torch.float32, torch.bfloat16], [True, False], [True, False], [True, False], [1, 2]
+        [torch.float32, torch.bfloat16],
+        [True, False],
+        [True, False],
+        [True, False],
+        [1, 2],
     )
 
     # Step 1: register patterns from mm and addmm