Skip to content

GitLab

Unverified Commit 1fda0e85 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix atol in resize_image kernel test (#8133)

parent ec7f80be
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_transforms_v2.py
View file @ 1fda0e85
...@@ -599,7 +599,7 @@ class TestResize: ...@@ -599,7 +599,7 @@ class TestResize:
# In contrast to CPU, there is no native `InterpolationMode.BICUBIC` implementation for uint8 images on CUDA. # In contrast to CPU, there is no native `InterpolationMode.BICUBIC` implementation for uint8 images on CUDA.
# Internally, it uses the float path. Thus, we need to test with an enormous tolerance here to account for that. # Internally, it uses the float path. Thus, we need to test with an enormous tolerance here to account for that.
atol = 30 if transforms.InterpolationMode.BICUBIC and dtype is torch.uint8 else 1 atol = 30 if (interpolation is transforms.InterpolationMode.BICUBIC and dtype is torch.uint8) else 1
check_cuda_vs_cpu_tolerances = dict(rtol=0, atol=atol / 255 if dtype.is_floating_point else atol) check_cuda_vs_cpu_tolerances = dict(rtol=0, atol=atol / 255 if dtype.is_floating_point else atol)
check_kernel( check_kernel(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment