diff --git a/.ci/docker/ci_commit_pins/pytorch.txt b/.ci/docker/ci_commit_pins/pytorch.txt index a846b87c198..ee800549518 100644 --- a/.ci/docker/ci_commit_pins/pytorch.txt +++ b/.ci/docker/ci_commit_pins/pytorch.txt @@ -1 +1 @@ -7ae0ce6360b6e4f944906502d20da24c04debee5 +59d5cf083b4f860dea76fe8936076177f9367f10 diff --git a/backends/arm/test/models/test_conformer.py b/backends/arm/test/models/test_conformer.py index dc5ecc7ca97..b293555e66b 100644 --- a/backends/arm/test/models/test_conformer.py +++ b/backends/arm/test/models/test_conformer.py @@ -31,7 +31,7 @@ class TestConformer(unittest.TestCase): # .to_executorch step, i.e. after Arm partitioner. ops_after_partitioner = { "executorch_exir_dialects_edge__ops_aten_max_default": 1, - "torch.ops.aten._assert_scalar.default": 10, + "torch.ops.aten._assert_scalar.default": 7, "torch.ops.aten._local_scalar_dense.default": 1, } diff --git a/install_requirements.py b/install_requirements.py index 6c3b4186697..4c468b979d1 100644 --- a/install_requirements.py +++ b/install_requirements.py @@ -71,7 +71,7 @@ def python_is_compatible(): # # NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt # by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/ -NIGHTLY_VERSION = "dev20250310" +NIGHTLY_VERSION = "dev20250325" def install_requirements(use_pytorch_nightly): @@ -80,7 +80,7 @@ def install_requirements(use_pytorch_nightly): # Setting use_pytorch_nightly to false to test the pinned PyTorch commit. Note # that we don't need to set any version number there because they have already # been installed on CI before this step, so pip won't reinstall them - f"torch==2.7.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torch", + f"torch==2.8.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torch", ( f"torchvision==0.22.0.{NIGHTLY_VERSION}" if use_pytorch_nightly