From 29facdc78ac6df57b0ac917756c6982ee660e26b Mon Sep 17 00:00:00 2001 From: Antoine Simoulin Date: Tue, 4 Nov 2025 06:19:07 -0800 Subject: [PATCH] Relax tolerance for TestAffine::test_kernel_bounding_boxes (#9255) 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. Reviewed By: scotts Differential Revision: D85890420 --- test/test_transforms_v2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py index 62077be1db5..487018b7c69 100644 --- 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(