Skip to content

Commit 66d9a10

Browse files
Gasoonjiafacebook-github-bot
authored andcommitted
pin bump with better architecture (#14957)
Summary: This diff bump torch pin to 1008. also expose cuda version into torch_pin.py for better update if needed; Reviewed By: JacobSzwejbka Differential Revision: D84280496
1 parent d4129b7 commit 66d9a10

File tree

3 files changed

+20
-24
lines changed

3 files changed

+20
-24
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
53a2908a10f414a2f85caa06703a26a40e873869
1+
7d06d03434c7c8de8fe1a432ee454b953f84ab0d

install_requirements.py

Lines changed: 1 addition & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -12,33 +12,12 @@
1212

1313
from install_utils import determine_torch_url, is_intel_mac_os, python_is_compatible
1414

15-
from torch_pin import NIGHTLY_VERSION, TORCH_VERSION
15+
from torch_pin import NIGHTLY_VERSION, SUPPORTED_CUDA_VERSIONS, TORCH_VERSION
1616

1717
# The pip repository that hosts nightly torch packages.
1818
# This will be dynamically set based on CUDA availability and CUDA backend enabled/disabled.
1919
TORCH_NIGHTLY_URL_BASE = "https://download.pytorch.org/whl/nightly"
2020

21-
# Supported CUDA versions - modify this to add/remove supported versions
22-
# Format: tuple of (major, minor) version numbers
23-
SUPPORTED_CUDA_VERSIONS = (
24-
(12, 6),
25-
(12, 8),
26-
(13, 0),
27-
)
28-
29-
# Since ExecuTorch often uses main-branch features of pytorch, only the nightly
30-
# pip versions will have the required features.
31-
#
32-
# NOTE: If a newly-fetched version of the executorch repo changes the value of
33-
# NIGHTLY_VERSION, you should re-run this script to install the necessary
34-
# package versions.
35-
#
36-
# NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt
37-
# by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/
38-
#
39-
# NOTE: If you're changing, make the corresponding supported CUDA versions in
40-
# SUPPORTED_CUDA_VERSIONS above if needed.
41-
4221

4322
def install_requirements(use_pytorch_nightly):
4423
# Skip pip install on Intel macOS if using nightly.

torch_pin.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,19 @@
1+
# Since ExecuTorch often uses main-branch features of pytorch, only the nightly
2+
# pip versions will have the required features.
3+
#
4+
# NOTE: If a newly-fetched version of the executorch repo changes the value of
5+
# NIGHTLY_VERSION, you should re-run install_executorch.sh script to install the necessary
6+
# package versions.
7+
#
8+
# NOTE: If you're changing, make the corresponding change in .ci/docker/ci_commit_pins/pytorch.txt
9+
# by picking the hash from the same date in https://hud.pytorch.org/hud/pytorch/pytorch/nightly/
10+
#
11+
# NOTE: If you're changing, make the corresponding supported CUDA versions in
12+
# SUPPORTED_CUDA_VERSIONS above if needed.
113
TORCH_VERSION = "2.10.0"
2-
NIGHTLY_VERSION = "dev20251003"
14+
NIGHTLY_VERSION = "dev20251008"
15+
SUPPORTED_CUDA_VERSIONS = (
16+
(12, 6),
17+
(12, 8),
18+
(13, 0),
19+
)

0 commit comments

Comments
 (0)