saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/nn
History
Xinya Zhang 74fd1bf965 [ROCm] Update to AOTriton 0.7b (#134498) 
Notable changes:
1. Enable CudaGraph related tests
2. Fix UT problems
3. EXPERIMENTAL Navi31 support. User should enable Navi31 support with Env Var `TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1`

Know Problem:
1. `test/test_transformers.py` will massive failures and/or NaN outputs with `--use-pytest`
    + Update: Confirmed skip `class TestSDPAPrivateUse1Only` can fix the problem with `--use-pytest`

Note:
AOTriton 0.7b adds support to nestedtenosrs+SDPA but need more work (and consequently a separate PR) to enable it.

Fixes #133540

Pull Request resolved: https://github.com/pytorch/pytorch/pull/134498
Approved by: https://github.com/pruthvistony, https://github.com/jeffdaily, https://github.com/malfet
2024-09-11 20:34:01 +00:00
..
test_convolution.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00
test_dropout.py
test_embedding.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_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 [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_pruning.py Fix failures when default is flipped for weights_only (#127627) 2024-08-16 00:22:43 +00:00