format

neuralmagic · Aug 14, 2024 · 7af2fd8 · 7af2fd8
1 parent fd2b3d2
commit 7af2fd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/kernels/test_machete_gemm.py b/tests/kernels/test_machete_gemm.py
@@ -116,7 +116,7 @@ def test_machete_all_schedules(shape, atype: torch.dtype,
         )
 
         # Relax atol as our reduction dim becomes larger (more rounding error)
-        # Relax atol when we have zeropoints since the way machete applies 
+        # Relax atol when we have zeropoints since the way machete applies
         #  zeropoints (after scales) causes noise around 0
         atol = 1 if zero_points else min(5e-2 * math.sqrt(size_k), 1)
         torch.testing.assert_close(output, output_ref, rtol=1e-1, atol=atol),\
@@ -162,7 +162,7 @@ def test_machete_heuristic(shape, atype: torch.dtype,
     )
 
     # Relax atol as our reduction dim becomes larger (more rounding error)
-    # Relax atol when we have zeropoints since the way machete applies 
+    # Relax atol when we have zeropoints since the way machete applies
     #  zeropoints (after scales) causes noise around 0
     atol = 1 if zero_points else min(5e-2 * math.sqrt(size_k), 1)
     torch.testing.assert_close(output, output_ref, rtol=1e-1, atol=atol)