From b0756c35b643faa7487123e0bdb5862645d7b49b Mon Sep 17 00:00:00 2001
From: Manuel Candales <mcandales@meta.com>
Date: Thu, 19 Mar 2026 22:59:25 -0400
Subject: [PATCH] [RELEASE TEST] Use torchao 0.16.0

---
 .ci/docker/common/install_pytorch.sh | 4 ++--
 pyproject.toml                       | 1 +
 third-party/ao                       | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh
index 0635a09a2de..8e17cd8cc25 100755
--- a/.ci/docker/common/install_pytorch.sh
+++ b/.ci/docker/common/install_pytorch.sh
@@ -12,11 +12,11 @@ source "$(dirname "${BASH_SOURCE[0]}")/utils.sh"
 
 install_domains() {
   echo "Install torchvision and torchaudio"
-  pip_install --force-reinstall --no-cache-dir torchvision==0.26.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/test/cpu
+  pip_install --force-reinstall --no-cache-dir torchvision==0.26.0 torchaudio==2.11.0 torchao==0.16.0 --index-url https://download.pytorch.org/whl/test/cpu
 }
 
 install_pytorch_and_domains() {
-  pip_install --force-reinstall --no-cache-dir torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0 --index-url https://download.pytorch.org/whl/test/cpu
+  pip_install --force-reinstall --no-cache-dir torch==2.11.0 torchvision==0.26.0 torchaudio==2.11.0 torchao==0.16.0 --index-url https://download.pytorch.org/whl/test/cpu
 }
 
 install_pytorch_and_domains
diff --git a/pyproject.toml b/pyproject.toml
index 27d503f9597..49bc6d00f32 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -71,6 +71,7 @@ dependencies=[
   "sympy",
   "tabulate",
   # See also third-party/TARGETS for buck's typing-extensions version.
+  "torchao==0.16.0",
   "typing-extensions>=4.10.0",
   # Keep this version in sync with: ./backends/apple/coreml/scripts/install_requirements.sh
   "coremltools==9.0; platform_system == 'Darwin' or platform_system == 'Linux'",
diff --git a/third-party/ao b/third-party/ao
index 42bcdc49190..3c1065ca6ff 160000
--- a/third-party/ao
+++ b/third-party/ao
@@ -1 +1 @@
-Subproject commit 42bcdc4919040aa97d814e59726ae33d49c1a692
+Subproject commit 3c1065ca6ff2eb8a48262bd7c31ded2fffe0d15e