pytorch/torch/testing
Eddie Yan 3870e68644 TF32 threshold twiddling for tests (#60209)
Summary:
Following https://github.com/pytorch/pytorch/issues/59624 I observed some straggling failing tests on Ampere due to TF32 thresholds. This PR just twiddles some more thresholds to fix the (6) failing tests I saw on A100.

CC Flamefire ptrblck ngimel

Pull Request resolved: https://github.com/pytorch/pytorch/pull/60209

Reviewed By: gchanan

Differential Revision: D29220508

Pulled By: ngimel

fbshipit-source-id: 7c83187a246e1b3a24b181334117c0ccf2baf311
2021-06-18 11:41:33 -07:00
..
_internal TF32 threshold twiddling for tests (#60209) 2021-06-18 11:41:33 -07:00
__init__.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
_asserts.py enable support for nested containers in torch.testing.assert(equal|close) (#57270) 2021-05-12 15:37:42 -07:00
_check_kernel_launches.py T90561249: Enforce kernel launch checks (#58178) 2021-05-13 10:41:20 -07:00
_core.py [testing] OpInfoHelper tool (#58698) 2021-06-16 17:17:03 -07:00