Skip to content

Commit faa392b

Browse files
williamwen42williamwen42
committed
Update
[ghstack-poisoned]
2 parents 22a99c0 + d5ad9b8 commit faa392b

File tree

547 files changed

+12871
-3694
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

547 files changed

+12871
-3694
lines changed

‎.ci/docker/common/install_conda.sh‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,12 +49,20 @@ if [ -n "$ANACONDA_PYTHON_VERSION" ]; then
4949
export SYSROOT_DEP="sysroot_linux-64=2.17"
5050
fi
5151

52+
# Install correct Python version
53+
# Also ensure sysroot is using a modern GLIBC to match system compilers
54+
if [ "$ANACONDA_PYTHON_VERSION" = "3.14" ]; then
55+
as_jenkins conda create -n py_$ANACONDA_PYTHON_VERSION -y\
56+
python="3.14.0" \
57+
${SYSROOT_DEP} \
58+
-c conda-forge
59+
else
5260
# Install correct Python version
5361
# Also ensure sysroot is using a modern GLIBC to match system compilers
5462
as_jenkins conda create -n py_$ANACONDA_PYTHON_VERSION -y\
5563
python="$ANACONDA_PYTHON_VERSION" \
5664
${SYSROOT_DEP}
57-
65+
fi
5866
# libstdcxx from conda default channels are too old, we need GLIBCXX_3.4.30
5967
# which is provided in libstdcxx 12 and up.
6068
conda_install libstdcxx-ng=12.3.0 --update-deps -c conda-forge

‎.ci/docker/common/install_cuda.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ else
1010
arch_path='sbsa'
1111
fi
1212

13-
NVSHMEM_VERSION=3.3.24
13+
NVSHMEM_VERSION=3.4.5
1414

1515
function install_cuda {
1616
version=$1

‎.ci/docker/common/install_rocm.sh‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,11 +40,7 @@ EOF
4040

4141
# Default url values
4242
rocm_baseurl="http://repo.radeon.com/rocm/apt/${ROCM_VERSION}"
43-
amdgpu_baseurl="https://repo.radeon.com/amdgpu/${ROCM_VERSION}/ubuntu"
44-
45-
# Add amdgpu repository
4643
UBUNTU_VERSION_NAME=`cat /etc/os-release | grep UBUNTU_CODENAME | awk -F= '{print $2}'`
47-
echo "deb [arch=amd64] ${amdgpu_baseurl} ${UBUNTU_VERSION_NAME} main" > /etc/apt/sources.list.d/amdgpu.list
4844

4945
# Add rocm repository
5046
wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -

‎.ci/docker/requirements-ci.txt‎

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,12 @@ numba==0.60.0 ; python_version == "3.12" and platform_machine != "s390x"
138138
#test_binary_ufuncs.py
139139
numpy==1.22.4; python_version == "3.10"
140140
numpy==1.26.2; python_version == "3.11" or python_version == "3.12"
141-
numpy==2.1.2; python_version >= "3.13"
141+
numpy==2.1.2; python_version >= "3.13" and python_version < "3.14"
142+
numpy==2.3.4; python_version >= "3.14"
142143

143144
pandas==2.0.3; python_version < "3.13"
144-
pandas==2.2.3; python_version >= "3.13"
145+
pandas==2.2.3; python_version >= "3.13" and python_version < "3.14"
146+
pandas==2.3.3; python_version >= "3.14"
145147

146148
#onnxruntime
147149
#Description: scoring engine for Open Neural Network Exchange (ONNX) models
@@ -153,7 +155,8 @@ opt-einsum==3.3
153155
#Pinned versions: 3.3
154156
#test that import: test_linalg.py
155157

156-
optree==0.13.0
158+
optree==0.13.0 ; python_version < "3.14"
159+
optree==0.17.0 ; python_version >= "3.14"
157160
#Description: A library for tree manipulation
158161
#Pinned versions: 0.13.0
159162
#test that import: test_vmap.py, test_aotdispatch.py, test_dynamic_shapes.py,
@@ -252,7 +255,8 @@ scikit-image==0.22.0
252255
#test that import:
253256

254257
scipy==1.10.1 ; python_version <= "3.11"
255-
scipy==1.14.1 ; python_version >= "3.12"
258+
scipy==1.14.1 ; python_version > "3.11" and python_version < "3.14"
259+
scipy==1.16.2 ; python_version >= "3.14"
256260
# Pin SciPy because of failing distribution tests (see #60347)
257261
#Description: scientific python
258262
#Pinned versions: 1.10.1
@@ -324,7 +328,8 @@ pywavelets==1.7.0 ; python_version >= "3.12"
324328
#Pinned versions: 1.4.1
325329
#test that import:
326330

327-
lxml==5.3.0
331+
lxml==5.3.0 ; python_version < "3.14"
332+
lxml==6.0.2 ; python_version >= "3.14"
328333
#Description: This is a requirement of unittest-xml-reporting
329334

330335
PyGithub==2.3.0
@@ -334,7 +339,9 @@ sympy==1.13.3
334339
#Pinned versions:
335340
#test that import:
336341

337-
onnx==1.19.1
342+
onnx==1.19.1 ; python_version < "3.14"
343+
# Unpin once Python 3.14 is supported. See onnxruntime issue 26309.
344+
onnx==1.18.0 ; python_version == "3.14"
338345
#Description: Required by onnx tests, and mypy and test_public_bindings.py when checking torch.onnx._internal
339346
#Pinned versions:
340347
#test that import:
@@ -359,7 +366,7 @@ pwlf==2.2.1
359366
#test that import: test_sac_estimator.py
360367

361368
# To build PyTorch itself
362-
pyyaml==6.0.2
369+
pyyaml==6.0.3
363370
pyzstd
364371
setuptools==78.1.1
365372
packaging==23.1

‎.ci/lumen_cli/pyproject.toml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ dependencies = [
66
"GitPython==3.1.45",
77
"docker==7.1.0",
88
"pytest==7.3.2",
9-
"uv==0.9.5"
9+
"uv==0.9.6"
1010
]
1111

1212
[tool.setuptools]

‎.github/actions/diskspace-cleanup/action.yml‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,9 @@ runs:
2727
docker system prune -af
2828
diskspace_new=$(df -H --output=pcent ${docker_root_dir} | sed -n 2p | sed 's/%//' | sed 's/ //')
2929
if [[ "$diskspace_new" -gt "$diskspace_cutoff" ]] ; then
30-
echo "Error: Available diskspace is less than $diskspace_cutoff percent. Not enough diskspace."
30+
diskspace_cutoff_int=$((diskspace_cutoff + 0))
31+
difference=$((100 - diskspace_cutoff_int))
32+
echo "Error: Available diskspace is less than $difference percent. Not enough diskspace."
3133
echo "$msg"
3234
exit 1
3335
else

‎.github/ci_commit_pins/audio.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
69bbe7363897764f9e758d851cd0340147d27f94
1+
3b0e7a6f192ca2715e7e6cbe5db007aea7165fe2

‎.github/ci_commit_pins/vision.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
1752fe6809b74921644866275ab80244b96e80bc
1+
218d2ab791d437309f91e0486eb9fa7f00badc17

‎.github/merge_rules.yaml‎

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -540,6 +540,26 @@
540540
- Lint
541541
- pull
542542

543+
- name: PrivateUse1
544+
patterns:
545+
- torch/accelerator/**
546+
- torch/utils/backend_registration.py
547+
- torch/csrc/acc/**
548+
- torch/csrc/DeviceAccelerator.*
549+
- torch/csrc/profiler/standalone/privateuse1_observer.*
550+
- aten/src/ATen/DeviceAccelerator.*
551+
- aten/src/ATen/core/GeneratorForPrivateuseone.*
552+
- aten/src/ATen/detail/PrivateUse1HooksInterface.*
553+
- docs/source/accelerator/**
554+
- test/cpp_extensions/open_registration_extension/torch_openreg/**
555+
approved_by:
556+
- albanD
557+
- fffrog
558+
mandatory_checks_name:
559+
- EasyCLA
560+
- Lint
561+
- pull
562+
543563
- name: superuser
544564
patterns:
545565
- '*'

‎.github/pytorch-probot.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ ciflow_push_tags:
2626
- ciflow/nightly
2727
- ciflow/op-benchmark
2828
- ciflow/periodic
29+
- ciflow/periodic-rocm-mi200
2930
- ciflow/periodic-rocm-mi300
3031
- ciflow/pull
3132
- ciflow/quantization-periodic

0 commit comments

Comments
 (0)