Adjust rtol/atol for test_sum_keepdims (#14)

yf225 · web-flow · commit de2a3e5c4454 · 2025-04-29T00:12:17.000-04:00
diff --git a/test/test_reductions.py b/test/test_reductions.py
@@ -91,7 +91,7 @@ def test_sum_keepdims(self):
         code, output = code_and_output(
             sum_kernel_keepdims, args, block_size=16, indexing="block_ptr"
         )
-        torch.testing.assert_close(output, args[0].sum(0, keepdim=True))
+        torch.testing.assert_close(output, args[0].sum(0, keepdim=True), rtol=2e-05, atol=1e-05)
         self.assertExpectedInline(
             code,
             """\

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ def test_sum_keepdims(self):`
`91`	`91`	`code, output = code_and_output(`
`92`	`92`	`sum_kernel_keepdims, args, block_size=16, indexing="block_ptr"`
`93`	`93`	`)`
`94`		`- torch.testing.assert_close(output, args[0].sum(0, keepdim=True))`
	`94`	`+ torch.testing.assert_close(output, args[0].sum(0, keepdim=True), rtol=2e-05, atol=1e-05)`
`95`	`95`	`self.assertExpectedInline(`
`96`	`96`	`code,`
`97`	`97`	`"""\`