Increase atol for test_matmul_split_k (#155)

jansel · web-flow · commit 2729f4378377 · 2025-06-11T20:00:53.000-07:00
diff --git a/test/test_matmul.py b/test/test_matmul.py
@@ -672,7 +672,7 @@ def matmul_split_k(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
             indexing="block_ptr",
         )
         expected = x @ y
-        torch.testing.assert_close(result, expected, rtol=0.01, atol=0.01)
+        torch.testing.assert_close(result, expected, atol=1e-1, rtol=1e-2)
         self.assertExpectedInline(
             code,
             """\

Original file line number	Diff line number	Diff line change
`@@ -672,7 +672,7 @@ def matmul_split_k(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:`
`672`	`672`	`indexing="block_ptr",`
`673`	`673`	`)`
`674`	`674`	`expected = x @ y`
`675`		`- torch.testing.assert_close(result, expected, rtol=0.01, atol=0.01)`
	`675`	`+ torch.testing.assert_close(result, expected, atol=1e-1, rtol=1e-2)`
`676`	`676`	`self.assertExpectedInline(`
`677`	`677`	`code,`
`678`	`678`	`"""\`