pytorch icon indicating copy to clipboard operation
pytorch copied to clipboard

[cuDNN][functorch] Bump tolerances for `nn.functional.conv2d` in `test_vmap_autograd_grad`

Open eqy opened this issue 7 months ago • 4 comments

Newer versions of cuDNN can dispatch to a winograd kernel here on A100 which affects numerics a bit

cc @csarofeen @ptrblck @xwang233 @zou3519 @Chillee @samdow @kshitij12345 @janeyx99

eqy avatar Jun 28 '24 23:06 eqy