Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 12 additions & 12 deletions .ci/scripts/tests/test_cu134_dependencies.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 = (
Expand Down Expand Up @@ -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))
Expand All @@ -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)

Expand Down Expand Up @@ -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 (
Expand Down
11 changes: 5 additions & 6 deletions install_requirements.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
]


Expand Down
2 changes: 1 addition & 1 deletion third-party/ao
Submodule ao updated 421 files
Loading