[RELEASE ONLY] Use torchao 0.14.0 (#15071)

GregoryComer · web-flow · commit a728efa565df · 2025-10-13T17:55:01.000-07:00
### Summary AO has officially released 0.14.0 (see https://pypi.org/project/torchao/#history). Update the pyproject.toml and pins to use 0.14. I also bumped the submodule to the latest release/0.14.0 head, though this only picks up a minor GH actions workflow change. Fixes #14419. ### Test plan CI
diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh
@@ -17,7 +17,7 @@ install_domains() {
 }
 
 install_pytorch_and_domains() {
-  pip_install torch==2.9.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cpu
+  pip_install torch==2.9.0 torchvision torchaudio torchao==0.14.0 --index-url https://download.pytorch.org/whl/test/cpu
 }
 
 install_pytorch_and_domains
diff --git a/pyproject.toml b/pyproject.toml
@@ -78,6 +78,7 @@ dependencies=[
   "scikit-learn==1.7.1",
   "hydra-core>=1.3.0",
   "omegaconf>=2.3.0",
+  "torchao==0.14.0",
 ]
 
 [project.urls]
diff --git a/third-party/ao b/third-party/ao
@@ -1 +1 @@
-Subproject commit c40417e1996a560a17001d663c36ed622007b52e
+Subproject commit 02941240fd182fc3f37ffcb47a64084904428d1d

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ install_domains() {`
`17`	`17`	`}`
`18`	`18`
`19`	`19`	`install_pytorch_and_domains() {`
`20`		`- pip_install torch==2.9.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cpu`
	`20`	`+ pip_install torch==2.9.0 torchvision torchaudio torchao==0.14.0 --index-url https://download.pytorch.org/whl/test/cpu`
`21`	`21`	`}`
`22`	`22`
`23`	`23`	`install_pytorch_and_domains`
Original file line number	Diff line number	Diff line change
`@@ -78,6 +78,7 @@ dependencies=[`
`78`	`78`	`"scikit-learn==1.7.1",`
`79`	`79`	`"hydra-core>=1.3.0",`
`80`	`80`	`"omegaconf>=2.3.0",`
	`81`	`+ "torchao==0.14.0",`
`81`	`82`	`]`
`82`	`83`
`83`	`84`	`[project.urls]`