8000 Update · pytorch/pytorch@453077c · GitHub
[go: up one dir, main page]
Skip to content

Commit 453077c

Browse files
anijain2305anijain2305
committed
Update
[ghstack-poisoned]
2 parents 87a7274 + 49339d9 commit 453077c

File tree

342 files changed

+8013
-2642
lines changed

Some content is hidden

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

342 files changed

+8013
-2642
lines changed

‎.ci/docker/README.md‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,6 @@ If your new Docker image needs a library installed from a specific pinned commit
104104
```bash
105105
pytorch-linux-jammy-cuda12.8-cudnn9-py3.12-gcc11-new1)
106106
CUDA_VERSION=12.8.1
107-
CUDNN_VERSION=9
108107
ANACONDA_PYTHON_VERSION=3.12
109108
GCC_VERSION=11
110109
VISION=yes

‎.ci/docker/build.sh‎

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,6 @@ tag=$(echo $image | awk -F':' '{print $2}')
9393
case "$tag" in
9494
pytorch-linux-jammy-cuda12.4-cudnn9-py3-gcc11)
9595
CUDA_VERSION=12.4
96-
CUDNN_VERSION=9
9796
ANACONDA_PYTHON_VERSION=3.10
9897
GCC_VERSION=11
9998
VISION=yes
@@ -104,7 +103,6 @@ case "$tag" in
104103
;;
105104
pytorch-linux-jammy-cuda12.8-cudnn9-py3-gcc11)
106105
CUDA_VERSION=12.8.1
107-
CUDNN_VERSION=9
108106
ANACONDA_PYTHON_VERSION=3.10
109107
GCC_VERSION=11
110108
VISION=yes
@@ -115,7 +113,6 @@ case "$tag" in
115113
;;
116114
pytorch-linux-jammy-cuda12.8-cudnn9-py3-gcc9-inductor-benchmarks)
117115
CUDA_VERSION=12.8.1
118-
CUDNN_VERSION=9
119116
ANACONDA_PYTHON_VERSION=3.10
120117
GCC_VERSION=9
121118
VISION=yes
@@ -127,7 +124,6 @@ case "$tag" in
127124
;;
128125
pytorch-linux-jammy-cuda12.8-cudnn9-py3.12-gcc9-inductor-benchmarks)
129126
CUDA_VERSION=12.8.1
130-
CUDNN_VERSION=9
131127
ANACONDA_PYTHON_VERSION=3.12
132128
GCC_VERSION=9
133129
VISION=yes
@@ -139,7 +135,6 @@ case "$tag" in
139135
;;
140136
pytorch-linux-jammy-cuda12.8-cudnn9-py3.13-gcc9-inductor-benchmarks)
141137
CUDA_VERSION=12.8.1
142-
CUDNN_VERSION=9
143138
ANACONDA_PYTHON_VERSION=3.13
144139
GCC_VERSION=9
145140
VISION=yes
@@ -151,7 +146,6 @@ case "$tag" in
151146
;;
152147
pytorch-linux-jammy-cuda12.6-cudnn9-py3-gcc9)
153148
CUDA_VERSION=12.6.3
154-
CUDNN_VERSION=9
155149
ANACONDA_PYTHON_VERSION=3.10
156150
GCC_VERSION=9
157151
VISION=yes
@@ -162,7 +156,6 @@ case "$tag" in
162156
;;
163157
pytorch-linux-jammy-cuda12.8-cudnn9-py3.12-gcc11-vllm)
164158
CUDA_VERSION=12.8.1
165-
CUDNN_VERSION=9
166159
ANACONDA_PYTHON_VERSION=3.12
167160
GCC_VERSION=11
168161
VISION=yes
@@ -173,7 +166,6 @@ case "$tag" in
173166
;;
174167
pytorch-linux-jammy-cuda12.6-cudnn9-py3-gcc9-inductor-benchmarks)
175168
CUDA_VERSION=12.6
176-
CUDNN_VERSION=9
177169
ANACONDA_PYTHON_VERSION=3.10
178170
GCC_VERSION=9
179171
VISION=yes
@@ -185,7 +177,6 @@ case "$tag" in
185177
;;
186178
pytorch-linux-jammy-cuda12.6-cudnn9-py3.12-gcc9-inductor-benchmarks)
187179
CUDA_VERSION=12.6
188-
CUDNN_VERSION=9
189180
ANACONDA_PYTHON_VERSION=3.12
190181
GCC_VERSION=9
191182
VISION=yes
@@ -197,7 +188,6 @@ case "$tag" in
197188
;;
198189
pytorch-linux-jammy-cuda12.6-cudnn9-py3.13-gcc9-inductor-benchmarks)
199190
CUDA_VERSION=12.6
200-
CUDNN_VERSION=9
201191
ANACONDA_PYTHON_VERSION=3.13
202192
GCC_VERSION=9
203193
VISION=yes
@@ -209,7 +199,6 @@ case "$tag" in
209199
;;
210200
pytorch-linux-jammy-cuda12.8-cudnn9-py3-gcc9)
211201
CUDA_VERSION=12.8.1
212-
CUDNN_VERSION=9
213202
ANACONDA_PYTHON_VERSION=3.10
214203
GCC_VERSION=9
215204
VISION=yes
@@ -299,7 +288,6 @@ case "$tag" in
299288
pytorch-linux-jammy-cuda12.8-cudnn9-py3.9-clang12)
300289
ANACONDA_PYTHON_VERSION=3.9
301290
CUDA_VERSION=12.8.1
302-
CUDNN_VERSION=9
303291
CLANG_VERSION=12
304292
VISION=yes
305293
TRITON=yes
@@ -378,7 +366,6 @@ case "$tag" in
378366
fi
379367
if [[ "$image" == *cuda* ]]; then
380368
extract_version_from_image_name cuda CUDA_VERSION
381-
extract_version_from_image_name cudnn CUDNN_VERSION
382369
fi
383370
if [[ "$image" == *rocm* ]]; then
384371
extract_version_from_image_name rocm ROCM_VERSION
@@ -430,9 +417,6 @@ docker build \
430417
--build-arg "PYTHON_VERSION=${PYTHON_VERSION}" \
431418
--build-arg "GCC_VERSION=${GCC_VERSION}" \
432419
--build-arg "CUDA_VERSION=${CUDA_VERSION}" \
433-
--build-arg "CUDNN_VERSION=${CUDNN_VERSION}" \
434-
--build-arg "TENSORRT_VERSION=${TENSORRT_VERSION}" \
435-
--build-arg "GRADLE_VERSION=${GRADLE_VERSION}" \
436420
--build-arg "NINJA_VERSION=${NINJA_VERSION:-}" \
437421
--build-arg "KATEX=${KATEX: 99C -}" \
438422
--build-arg "ROCM_VERSION=${ROCM_VERSION:-}" \

‎.ci/docker/common/install_cudnn.sh‎

Lines changed: 0 additions & 26 deletions
This file was deleted.

‎.ci/docker/common/install_triton.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,5 +103,5 @@ fi
103103
# It depends on torch and triton. We don't want to install
104104
# triton and torch from production on Docker CI images
105105
if [[ "$ANACONDA_PYTHON_VERSION" != 3.9* ]]; then
106-
pip_install helion --no-deps
106+
pip_install helion==0.0.10 --no-deps
107107
fi

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,13 +50,12 @@ IPython==8.12.0
5050
#Pinned versions: 8.12.0
5151

5252
myst-nb==0.17.2
53-
#Description: This is used to generate PyTorch functorch docs
54-
#Pinned versions: 0.13.2
53+
#Description: This is used to generate PyTorch functorch and torch.compile docs
54+
#Pinned versions: 0.17.2
5555

5656
# The following are required to build torch.distributed.elastic.rendezvous.etcd* docs
5757
python-etcd==0.4.5
5858
sphinx-copybutton==0.5.0
5959
sphinx-design==0.4.0
6060
sphinxcontrib-mermaid==1.0.0
6161
myst-parser==0.18.1
62-
myst-nb

‎.ci/pytorch/test.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@ test_inductor_aoti() {
462462
# rebuild with the build cache with `BUILD_AOT_INDUCTOR_TEST` enabled
463463
/usr/bin/env CMAKE_FRESH=1 BUILD_AOT_INDUCTOR_TEST=1 "${BUILD_COMMAND[@]}"
464464

465-
/usr/bin/env "${TEST_ENVS[@]}" python test/run_test.py --cpp --verbose -i cpp/test_aoti_abi_check cpp/test_aoti_inference -dist=loadfile
465+
/usr/bin/env "${TEST_ENVS[@]}" python test/run_test.py --cpp --verbose -i cpp/test_aoti_abi_check cpp/test_aoti_inference cpp/test_vec_half_AVX2 -dist=loadfile
466466
}
467467

468468
test_inductor_cpp_wrapper_shard() {

‎.github/actionlint.yaml‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -57,12 +57,8 @@ self-hosted-runner:
5757
- linux.rocm.gpu.mi300.2
5858
- linux.rocm.gpu.mi300.4
5959
- rocm-docker
60-
# Repo-specific Apple hosted runners
61-
- macos-m1-ultra
62-
- macos-m2-14
6360
# Org wise AWS `mac2.metal` runners (2020 Mac mini hardware powered by Apple silicon M1 processors)
6461
- macos-m1-stable
65-
- macos-m1-13
6662
- macos-m1-14
6763
# GitHub-hosted MacOS runners
6864
- macos-latest-xlarge

‎.github/requirements/pip-requirements-macOS.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ boto3==1.35.42
22
cmake==3.27.*
33
expecttest==0.3.0
44
fbscribelogger==0.1.7
5-
filelock==3.13.1
5+
filelock==3.18.0
66
hypothesis==6.56.4
77
librosa>=0.6.2
88
mpmath==1.3.0

‎.github/workflows/build-triton-wheel.yml‎

Lines changed: 7 additions & 1 deletion
< 99C /tr>
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ jobs:
5050
strategy:
5151
fail-fast: false
5252
matrix:
53-
py_vers: [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.13t" ]
53+
py_vers: [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.13t", "3.14", "3.14t" ]
5454
device: ["cuda", "rocm", "xpu", "aarch64"]
5555
docker-image: ["pytorch/manylinux2_28-builder:cpu"]
5656
include:
@@ -126,6 +126,12 @@ jobs:
126126
3.13t)
127127
PYTHON_EXECUTABLE=/opt/python/cp313-cp313t/bin/python
128128
;;
129+
3.14)
130+
PYTHON_EXECUTABLE=/opt/python/cp314-cp314/bin/python
131+
;;
132+
3.14t)
133+
PYTHON_EXECUTABLE=/opt/python/cp314-cp314t/bin/python
134+
;;
129135
*)
130136
echo "Unsupported python version ${PY_VERS}"
131137
exit 1

‎.github/workflows/mac-mps.yml‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@ jobs:
2828
# than our AWS macos-m1-14 runners
2929
test-matrix: |
3030
{ include: [
31-
{ config: "test_mps", shard: 1, num_shards: 1, runner: "macos-m1-13" },
3231
{ config: "test_mps", shard: 1, num_shards: 1, runner: "macos-m1-14" },
3332
{ config: "test_mps", shard: 1, num_shards: 1, runner: "macos-m2-15" },
3433
]}

0 commit comments

Comments
 (0)
0