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

..

caffe2

[BE][CI] change references from .jenkins to .ci (#92624)

2023-01-30 22:50:07 +00:00

docker

[ROCm] Update to AOTriton 0.7b (#134498)

2024-09-11 20:34:01 +00:00

onnx

Workaround dind-rootless bind mount permissions (#123641)

2024-04-10 22:44:57 +00:00

pytorch

[aoti] Add cpp loader (#135374)

2024-09-11 03:00:01 +00:00