From c5b37d260ee8cbd368a3e722bc40d000ec0230a2 Mon Sep 17 00:00:00 2001 From: Gregory James Comer Date: Mon, 13 Oct 2025 14:29:08 -0700 Subject: [PATCH 1/4] [RELEASE ONLY] Use torchao 0.14.0 --- .ci/docker/common/install_pytorch.sh | 2 +- install_requirements.py | 1 - pyproject.toml | 1 + third-party/ao | 2 +- 4 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh index 9af0e07c6da..af80396eda5 100755 --- 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/install_requirements.py b/install_requirements.py index 7cd47c07668..60393937aba 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -111,7 +111,6 @@ def install_requirements(use_pytorch_nightly): ) LOCAL_REQUIREMENTS = [ - "third-party/ao", # We need the latest kernels for fast iteration, so not relying on pypi. ] + ( [ "extension/llm/tokenizers", # TODO(larryliu0820): Setup a pypi package for this. diff --git a/pyproject.toml b/pyproject.toml index 393f8578c8e..a4f883c7925 100644 --- 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 index c40417e1996..02941240fd1 160000 --- a/third-party/ao +++ b/third-party/ao @@ -1 +1 @@ -Subproject commit c40417e1996a560a17001d663c36ed622007b52e +Subproject commit 02941240fd182fc3f37ffcb47a64084904428d1d From 268e1f4fb014707bfdb76120ee1d05f93405e269 Mon Sep 17 00:00:00 2001 From: Gregory Comer Date: Mon, 13 Oct 2025 14:49:32 -0700 Subject: [PATCH 2/4] Update pyproject.toml --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index a4f883c7925..ce5a1fae444 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -78,7 +78,7 @@ dependencies=[ "scikit-learn==1.7.1", "hydra-core>=1.3.0", "omegaconf>=2.3.0", - "torchao=0.14.0", + "torchao==0.14.0", ] [project.urls] From d51ecea736adab3a9b9c0dd4cb8672a4d2a62d80 Mon Sep 17 00:00:00 2001 From: Gregory Comer Date: Mon, 13 Oct 2025 14:49:49 -0700 Subject: [PATCH 3/4] Update install_pytorch.sh --- .ci/docker/common/install_pytorch.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh index af80396eda5..e375adf2220 100755 --- 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 torchao=0.14.0 --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 From 3f9989481df6e45589e022f5061e95c1d938d5c4 Mon Sep 17 00:00:00 2001 From: Gregory Comer Date: Mon, 13 Oct 2025 14:57:52 -0700 Subject: [PATCH 4/4] Update install_requirements.py --- install_requirements.py | 1 + 1 file changed, 1 insertion(+) diff --git a/install_requirements.py b/install_requirements.py index 60393937aba..7cd47c07668 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -111,6 +111,7 @@ def install_requirements(use_pytorch_nightly): ) LOCAL_REQUIREMENTS = [ + "third-party/ao", # We need the latest kernels for fast iteration, so not relying on pypi. ] + ( [ "extension/llm/tokenizers", # TODO(larryliu0820): Setup a pypi package for this.