[TF32] Account for TF32 in test_conv_double_backward (pytorch#135716)

Pull Request resolved: pytorch#135716
Approved by: https://github.com/Skylion007

Author: eqy

test/functorch/test_vmap.py

@@ -50,6 +50,7 @@
     PLATFORM_SUPPORTS_CUDNN_ATTENTION,
     PLATFORM_SUPPORTS_FLASH_ATTENTION,
     PLATFORM_SUPPORTS_MEM_EFF_ATTENTION,
+    tf32_on_and_off,
     with_tf32_off,
 )
 from torch.testing._internal.common_device_type import (
@@ -4757,6 +4758,7 @@ def test():
 
         check_vmap_fallback(self, test, Tensor.fill_)
 
+    @tf32_on_and_off(0.005)
     def test_conv_double_backward(self, device):
         images = torch.randn(2, 1, 5, 5, device=device)
         weight = torch.randn(2, 1, 2, 2, device=device)