Fix bug in setup_tolerance

k-artem · pytorchmergebot · commit 861ca4881625 · 2025-10-21T18:32:13.000Z
Bug introduced by ROCm@03c7da0 Signed-off-by: Artem Kuzmitckii <artem.kuzmitckii@amd.com>
diff --git a/test/inductor/test_decompose_mem_bound_mm.py b/test/inductor/test_decompose_mem_bound_mm.py
@@ -82,7 +82,9 @@ def setup_tolerance(self, rtol=None, atol=None):
         if rtol is None:
             rtol = self.rtol
         if atol is None:
-            atol = self.rtol
+            atol = self.atol
+        self.rtol = rtol
+        self.atol = atol
 
     def compare_dict_tensors(self, ref_dict, res_dict, rtol=None, atol=None):
         self.setup_tolerance(rtol, atol)
