pytorch/test/nn
Eddie Yan 846b4e614b [TF32][cuDNN][Convolution] Add some missing TF32 decorators (#138768)
Newer cuDNN versions seem to be able to dispatch to cuDNN kernels

Pull Request resolved: https://github.com/pytorch/pytorch/pull/138768
Approved by: https://github.com/Skylion007
2024-10-25 19:03:42 +00:00
..
test_convolution.py [TF32][cuDNN][Convolution] Add some missing TF32 decorators (#138768) 2024-10-25 19:03:42 +00:00
test_dropout.py Enable additional tests for MPS CI runs (#134356) 2024-10-04 21:52:38 +00:00
test_embedding.py Remove dtype check on meta device (#136774) 2024-10-12 05:45:21 +00:00
test_init.py [BE][Easy][8/19] enforce style for empty lines in import segments in test/[k-p]*/ (#129759) 2024-07-31 02:09:20 +00:00
test_lazy_modules.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_load_state_dict.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_module_hooks.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
test_multihead_attention.py [ROCm] Update to AOTriton 0.7b (#134498) 2024-09-11 20:34:01 +00:00
test_packed_sequence.py Add padding_side to pad_sequence with "left" and "right" options ("right" as default) (#131884) 2024-08-07 15:53:07 +00:00
test_parametrization.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_pooling.py Add type check for dilation in torch.quantized_max_pool3d() (#137845) 2024-10-21 16:15:57 +00:00
test_pruning.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00