Relax tolerance for TestAffine::test_kernel_bounding_boxes

AntoineSimoulin · facebook-github-bot · commit 6c00915951cf · 2025-10-31T06:25:54.000-07:00
Summary: The following test "TestAffine::test_kernel_bounding_boxes" produce approximation errors between GPU and CPU and could fail for some random values. We are relaxing the tolerance with values inline with what is set for other tests (e.g. `test_transform_bounding_boxes_correctness`) to ensure the test is not flaky.

Differential Revision: D85890420
diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py
@@ -1439,6 +1439,7 @@ def test_kernel_bounding_boxes(self, param, value, format, dtype, device):
             canvas_size=bounding_boxes.canvas_size,
             **{param: value},
             check_scripted_vs_eager=not (param == "shear" and isinstance(value, (int, float))),
+            check_cuda_vs_cpu=dict(atol=1e-5, rtol=1e-5),
         )
 
     @param_value_parametrization(

Original file line number	Diff line number	Diff line change
`@@ -1439,6 +1439,7 @@ def test_kernel_bounding_boxes(self, param, value, format, dtype, device):`
`1439`	`1439`	`canvas_size=bounding_boxes.canvas_size,`
`1440`	`1440`	`**{param: value},`
`1441`	`1441`	`check_scripted_vs_eager=not (param == "shear" and isinstance(value, (int, float))),`
	`1442`	`+ check_cuda_vs_cpu=dict(atol=1e-5, rtol=1e-5),`
`1442`	`1443`	`)`
`1443`	`1444`
`1444`	`1445`	`@param_value_parametrization(`