lint

jeffdaily · jeffdaily · commit 37c138faf7d5 · 2025-10-21T18:45:16.000Z
diff --git a/test/inductor/test_decompose_mem_bound_mm.py b/test/inductor/test_decompose_mem_bound_mm.py
@@ -12,9 +12,9 @@
 from torch.testing import FileCheck
 from torch.testing._internal.common_utils import (
     instantiate_parametrized_tests,
-    patch_test_members,
     is_navi3_arch,
     parametrize,
+    patch_test_members,
     TEST_XPU,
 )
 from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CUDA_AND_TRITON
@@ -73,7 +73,7 @@ def forward(
 )
 @instantiate_parametrized_tests
 class TestDecomposeMemMM(TestCase):
-    def __init__(self, method_name='runTest', methodName='runTest'):
+    def __init__(self, method_name="runTest", methodName="runTest"):
         super().__init__(method_name, methodName)
         self.atol = 1e-3
         self.rtol = 1e-3
@@ -93,7 +93,9 @@ def compare_dict_tensors(self, ref_dict, res_dict, rtol=None, atol=None):
         for key1 in ref_dict.keys():
             key2 = "_orig_mod." + key1
             assert key2 in res_dict, f"{key1} does not exist in traced module"
-            if not torch.allclose(ref_dict[key1], res_dict[key2], rtol=self.rtol, atol=self.atol):
+            if not torch.allclose(
+                ref_dict[key1], res_dict[key2], rtol=self.rtol, atol=self.atol
+            ):
                 return False
         return True
 
@@ -107,14 +109,20 @@ def compare_parameters(self, module, traced, rtol=None, atol=None):
         self.setup_tolerance(rtol, atol)
         ref_params = dict(module.named_parameters())
         res_params = dict(traced.named_parameters())
-        self.assertTrue(self.compare_dict_tensors(ref_params, res_params, rtol=self.rtol, atol=self.atol))
+        self.assertTrue(
+            self.compare_dict_tensors(
+                ref_params, res_params, rtol=self.rtol, atol=self.atol
+            )
+        )
 
     def compare_gradients(self, module, traced, rtol=None, atol=None):
         self.setup_tolerance(rtol, atol)
         ref_grad = {key: param.grad for key, param in module.named_parameters()}
         res_grad = {key: param.grad for key, param in traced.named_parameters()}
         self.assertTrue(
-            self.compare_dict_tensors(ref_grad, res_grad, rtol=self.rtol, atol=self.atol)
+            self.compare_dict_tensors(
+                ref_grad, res_grad, rtol=self.rtol, atol=self.atol
+            )
         )
 
     @parametrize(
@@ -223,10 +231,12 @@ def test_decompose_linear(self, m, n, k, has_bias, should_decompose):
 
     # We have to increase tolerance for navi3 because all fp16, bf16
     # GEMMs operations have an accuracy issue caused by hardware limitation
-    @patch_test_members({
-        "atol": 2e-3 if is_navi3_arch() else 1e-3,
-        "rtol": 2e-3 if is_navi3_arch() else 1e-3
-    })
+    @patch_test_members(
+        {
+            "atol": 2e-3 if is_navi3_arch() else 1e-3,
+            "rtol": 2e-3 if is_navi3_arch() else 1e-3,
+        }
+    )
     @parametrize(
         "m,k,n, should_decompose",
         [(20480, 5, 2, True), (20480, 32, 2, False), (2048, 2, 2, False)],
@@ -337,10 +347,12 @@ def test_decompose_mm_cpu(self, m, n, k, should_decompose):
 
     # We have to increase tolerance for navi3 because all fp16, bf16
     # GEMMs operations have an accuracy issue caused by hardware limitation
-    @patch_test_members({
-        "atol": 3e-3 if is_navi3_arch() else 1e-3,
-        "rtol": 4e-3 if is_navi3_arch() else 1e-3
-    })
+    @patch_test_members(
+        {
+            "atol": 3e-3 if is_navi3_arch() else 1e-3,
+            "rtol": 4e-3 if is_navi3_arch() else 1e-3,
+        }
+    )
     @parametrize(
         "m,k,n, should_decompose",
         [(20480, 5, 2, True), (20480, 32, 2, False), (2048, 2, 2, False)],
