diff --git a/backends/arm/test/models/test_llama.py b/backends/arm/test/models/test_llama.py index 6f770e89652..ee2f4bb48fc 100644 --- a/backends/arm/test/models/test_llama.py +++ b/backends/arm/test/models/test_llama.py @@ -229,6 +229,11 @@ def test_llama_tosa_INT(): pipeline.run() +@pytest.mark.xfail( + reason="index_put into a preserved fp32 mutable buffer (torchao pytorch/ao#4466) is " + "not delegatable by the INT backend, so the KV-cache round-trip forms a partition " + "dependency cycle. Same root cause as the xfailed static-cache tests: MLETORCH-1971." +) def test_llama_tosa_INT_static(): llama_model, llama_inputs, _ = TestLlama().prepare_model_hf_static() if llama_model is None or llama_inputs is None: diff --git a/install_requirements.py b/install_requirements.py index 7986cfacff9..1aedcf6f0f8 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -16,7 +16,7 @@ # This will be dynamically set based on CUDA availability and CUDA backend enabled/disabled. TORCH_URL_BASE = "https://download.pytorch.org/whl/test" TORCHAO_URL_BASE = "https://download.pytorch.org/whl/nightly" -TORCHAO_NIGHTLY_VERSION = "0.18.0.dev20260715" +TORCHAO_NIGHTLY_VERSION = "0.18.0.dev20260729" # Since ExecuTorch often uses main-branch features of pytorch, only the nightly # pip versions will have the required features. diff --git a/third-party/ao b/third-party/ao index 4aa81011346..03ca489dc62 160000 --- a/third-party/ao +++ b/third-party/ao @@ -1 +1 @@ -Subproject commit 4aa810113463571cabb7342d1a00a1f3de9c54f2 +Subproject commit 03ca489dc62776955c13999eb6b9d66453f4de7a