From 21ce5f4769542419989864bc2f246ce3d4831590 Mon Sep 17 00:00:00 2001 From: Scott Roy Date: Mon, 28 Sep 2026 10:01:04 -0700 Subject: [PATCH] bump pins --- .ci/scripts/tests/test_cu134_dependencies.py | 24 ++++++++++---------- install_requirements.py | 11 ++++----- third-party/ao | 2 +- 3 files changed, 18 insertions(+), 19 deletions(-) diff --git a/.ci/scripts/tests/test_cu134_dependencies.py b/.ci/scripts/tests/test_cu134_dependencies.py index 663cdb3b3bf..33acd45cc97 100644 --- a/.ci/scripts/tests/test_cu134_dependencies.py +++ b/.ci/scripts/tests/test_cu134_dependencies.py @@ -58,10 +58,10 @@ def test_all_install_steps_preserve_exact_cu134_selection(self): commands = self.install_commands((13, 4), machine) self.assertEqual(len(commands), 4) expected = { - "torch==2.14.0.dev20260810+cu134", - "torchvision==0.29.0.dev20260811+cu134", - "torchaudio==2.11.0.dev20260811+cu134", - f"torchao==0.19.0.dev20260811+{ao_variant}", + "torch==2.15.0.dev20260913+cu134", + "torchvision==0.30.0.dev20260913+cu134", + "torchaudio==2.11.0.dev20260913+cu134", + f"torchao==0.19.0.dev20260913+{ao_variant}", } for index, command in enumerate(commands): required = ( @@ -101,7 +101,7 @@ def test_other_cuda_trains_keep_existing_pins(self): cuda, machine ) self.assertIn("torch==2.14.0", core) - self.assertIn("torchao==0.18.0.dev20260729", core) + self.assertIn("torchao==0.19.0.dev20260913", core) self.assertIn("torchvision==0.29.0", domains) self.assertIn("torchaudio==2.11.0", domains) self.assertFalse(any("==" in arg for arg in local)) @@ -112,14 +112,14 @@ def test_source_pinned_torch_is_not_replaced(self): with self.subTest(cuda=cuda): core, _, domains, _ = self.install_commands(cuda, nightly=False) self.assertIn("torch", core) - self.assertNotIn("torch==2.14.0.dev20260810+cu134", core) + self.assertNotIn("torch==2.15.0.dev20260913+cu134", core) self.assertIn("torchvision", domains) self.assertIn("torchaudio", domains) def test_no_cuda_keeps_default_pins(self): core, _, domains, _ = self.install_commands(None) self.assertIn("torch==2.14.0", core) - self.assertIn("torchao==0.18.0.dev20260729", core) + self.assertIn("torchao==0.19.0.dev20260913", core) self.assertIn("torchvision==0.29.0", domains) self.assertIn("https://download.pytorch.org/whl/test/cpu", core) @@ -238,14 +238,14 @@ def test_cu134_keeps_explicit_torchao_source_build(self): self.assertFalse( any(arg.startswith("torchao==") for arg in command) ) - self.assertIn("torch==2.14.0.dev20260810+cu134", commands[-1]) - self.assertIn("0.18.0+git03ca489", metadata.specifier) + self.assertIn("torch==2.15.0.dev20260913+cu134", commands[-1]) + self.assertIn("0.19.0+git9e5ea7f", metadata.specifier) def test_wheel_torchao_bound_matches_selected_train(self): for cuda, expected in ( - ((13, 4), "torchao>=0.19.0.dev20260811,<0.20"), - ((13, 2), "torchao>=0.18.0.dev20260729,<0.19"), - (None, "torchao>=0.18.0.dev20260729,<0.19"), + ((13, 4), "torchao>=0.19.0.dev20260913,<0.20"), + ((13, 2), "torchao>=0.19.0.dev20260913,<0.20"), + (None, "torchao>=0.19.0.dev20260913,<0.20"), ): self.utils.determine_torch_url.cache_clear() with ( diff --git a/install_requirements.py b/install_requirements.py index 3429f06dd8c..ad66f41d38b 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -17,13 +17,12 @@ # 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.dev20260729" -CU134_TORCHAO_NIGHTLY_VERSION = "0.19.0.dev20260811" -# These wheels' metadata pairs August 11 domain libraries with August 10 torch. +TORCHAO_NIGHTLY_VERSION = "0.19.0.dev20260913" +CU134_TORCHAO_NIGHTLY_VERSION = "0.19.0.dev20260913" CU134_TORCH_PACKAGES = [ - "torch==2.14.0.dev20260810+cu134", - "torchvision==0.29.0.dev20260811+cu134", - "torchaudio==2.11.0.dev20260811+cu134", + "torch==2.15.0.dev20260913+cu134", + "torchvision==0.30.0.dev20260913+cu134", + "torchaudio==2.11.0.dev20260913+cu134", ] diff --git a/third-party/ao b/third-party/ao index 03ca489dc62..9e5ea7fde71 160000 --- a/third-party/ao +++ b/third-party/ao @@ -1 +1 @@ -Subproject commit 03ca489dc62776955c13999eb6b9d66453f4de7a +Subproject commit 9e5ea7fde7132df0c09b2acb12c2defad90aecb9