Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
88 commits
Select commit Hold shift + click to select a range
af5c4ec
[RELEASE 2.11] Release only changes (#175091)
atalman Feb 16, 2026
8b1d03b
Update inductor expected accuracy files (#175096)
pytorchbot Feb 16, 2026
148630b
Revert "[fix] DISABLED test_index (__main__.DistTensorOpsTest) (#1723…
jeffdaily Feb 16, 2026
1b0497e
Revert "[CI] Enable TIMM pretrained model caching on shared HF cache …
jeffdaily Feb 16, 2026
7afdbae
Fix macOS arm64 libtorch release upload failure (#175108)
pytorchbot Feb 17, 2026
2ceed74
[benchmark] Skip pytorch_CycleGAN_and_pix2pix from inductor benchmark…
pytorchbot Feb 19, 2026
d80a584
[CI] Move CUDA 12.8 GPU tests from per-commit trunk to periodic (#175…
pytorchbot Feb 19, 2026
c649f1a
[BE] Remove cuda 12.4 periodic tests (#175362)
pytorchbot Feb 19, 2026
4b8a514
[CI] Add CUDA 13 periodic tests (#175380)
pytorchbot Feb 20, 2026
5decbe0
[ROCm] forward fix #174087, take 4 (#175159)
pytorchbot Feb 20, 2026
a2813af
[release-only] Remove +ptx from cuda 13.0 builds (#175567)
atalman Feb 23, 2026
98e3502
[MPS] Fix 2-pass SDPA memory corruption by forcing float accumulators…
pytorchbot Feb 24, 2026
ef5c69d
Disable einops 0.8.2 check on PyTorch (#175442)
pytorchbot Feb 24, 2026
013fdc2
[CPUBLAS] Fix UB: use vector::resize() instead of reserve() before op…
pytorchbot Feb 24, 2026
d70d867
Remove python constraint on setuptools (#175627)
pytorchbot Feb 24, 2026
883a7e2
Supports custom empty tensor in InputObserver (#175581)
pytorchbot Feb 24, 2026
9bb65cc
Bump transformers version to 5.2.0 (#175661)
pytorchbot Feb 24, 2026
8fcce8e
[CI] Switch vLLM test and benchmark workflows to CUDA 13.0 (#175781)
pytorchbot Feb 25, 2026
6fe4bfb
Update vLLM pinned commit (#175238) (#175783)
huydhn Feb 25, 2026
57614ed
[ROCm][CI] Upgrade ROCm CI to 7.2 - 4/N (#175767)
pytorchbot Feb 25, 2026
c67ec25
[ROCm] Added CUDA check to test_pattern_matcher (#175766)
pytorchbot Feb 25, 2026
9e94d2d
[cuDNN][2.11] cuDNN upgrade / sync to 9.19 for linux/windows on 2.11 …
eqy Feb 25, 2026
195f9cd
Fix pep517 release handling (#175793)
pytorchbot Feb 26, 2026
f99ab99
[CI] Update inductor CI jobs to CUDA 13.0 (#175826) (#175955)
huydhn Feb 27, 2026
f95d7a4
update previous version 2.10 installation in get start xpu (#176408)
pytorchbot Mar 8, 2026
dc12b65
[inductor] Fix Identity comparability and evalf recursion (#176783)
pytorchbot Mar 8, 2026
63fcbe1
[XPU] Fix SyclExtension Windows build for oneAPI 2025.3+ breaking cha…
pytorchbot Mar 8, 2026
5d919bf
[Inductor] Reject non-contiguous subnode fusion in mix-order reductio…
pytorchbot Mar 9, 2026
3c40486
[inductor] avoid multi-stage for mix-order-red by default (#176228) (…
shunting314 Mar 9, 2026
f31baaa
Fix the torch.Stream context manager reentrance (#176603)
pytorchbot Mar 9, 2026
052ff5c
Windows override AMI pre-installed cudnn (#177094)
pytorchbot Mar 11, 2026
bac7b59
fix acc failure for vit_base_patch14_dinov2.lvd142m (#177142)
pytorchbot Mar 11, 2026
e04ddea
[Inductor] Don't unfuse addmm for bf16/fp16 to avoid precision loss (…
pytorchbot Mar 11, 2026
e2fa295
[CD] Unpin cuda-bindings dependencies (#177159)
pytorchbot Mar 11, 2026
41f8e3e
[CI] Stop using G3 runners (#177161)
pytorchbot Mar 11, 2026
036b25f
Let stable::from_blob accept a lambda as deleter (cherry-pick) (#176440)
mikaylagawarecki Mar 11, 2026
fa384de
[Inductor][MPS] Fix half-precision type mismatches in Metal shader co…
malfet Mar 11, 2026
76fd078
[release-only] Fix libtorch builds. Fix lint (#177299)
atalman Mar 12, 2026
7f2cdeb
[windows][smoke test] Add an option to install cuda if required cuda/…
pytorchbot Mar 13, 2026
483b55d
Update pytorch_sphinx_theme2 version to 0.4.6 (#177616)
pytorchbot Mar 17, 2026
db741c7
[MPS] fix compiling of SDPA producing nan results (#178009)
pytorchbot Mar 21, 2026
3e05c5a
[MPS] Properly handle conjugated tensors in bmm (#178010)
pytorchbot Mar 21, 2026
70d99e9
[release only] Increase timeout for rocm libtorch and manywheel build…
atalman Mar 21, 2026
4e45bcd
[release/2.8] Upgrade numpy versions; Use different package versions …
jithunnair-amd Jul 16, 2025
3482eaf
Clean up CUDA state between tests (#2335)
rraminen Jul 14, 2025
55881ef
reset per process memory fraction in test_cuda.py test_mempool_limite…
jeffdaily Nov 18, 2025
54750ed
[AUTOGENERATED] [release/2.8] [SWDEV-539215] - Autotune support for p…
okakarpa Aug 11, 2025
0bfe1e3
Update version to 2.11.0
chinmaydk99 Dec 17, 2025
03639f4
[AUTOGENERATED] [release/2.11] Move getenv to main thread to avoid NC…
rocm-repo-management-api[bot] Mar 10, 2026
41a0f89
[Release/2.11] No-fence in normalization kernel (#175286) (#3057)
anatoliylitv Mar 12, 2026
be321e4
Prefer cublas when TORCH_BLAS_PREFER_CUBLASLT is false (#https://gith…
mstankov-amd Mar 17, 2026
cc97152
Increase precision for golden solution in transformer tests. (#3087)
anatoliylitv Mar 20, 2026
dfff4e1
[ROCm][CI] Fix failing FP8 tests on RDNA4 (#174873) (#3090)
mstankov-amd Mar 20, 2026
83524a4
[CI][release/2.11]Pin all Python dependency versions in requirements …
ethanwee1 Mar 25, 2026
bb7978b
Update triton pin to tip of https://github.com/ROCm/triton/commits/re…
jithunnair-amd Mar 25, 2026
752cc24
Skip nccl_device.h header include for ROCm (causes build failures in …
jithunnair-amd Mar 25, 2026
2fea146
[ROCm] Reland: Enable expandable segments (#173330) (#177974) (#3106)
pragupta Mar 27, 2026
5f21bad
[release/2.11] Enable wheels (#3111)
ethanwee1 Mar 31, 2026
c8e635b
[release/2.11] Only skip linalg.eig assertion in test_torch_return_ty…
ethanwee1 Mar 31, 2026
8ecd5da
[ROCm] Optimize RadixSelect synchronization overhead (#174837)
apakbin Feb 27, 2026
e0fbdce
[ROCm] RadixSelect: Remove loop padding and make prefetching conditio…
apakbin Feb 24, 2026
ccba6b0
[ROCm] fix radixselect (#177149)
wangzhenict Mar 12, 2026
7a10d22
[ROCm] Reduce RadixSelect sync overhead by moving __syncthreads to fi…
apakbin Mar 31, 2026
f5a3aa1
[release/2.11] Use triton with updated pyproject.toml to use cmake 4 …
ethanwee1 Apr 3, 2026
a5c71cc
[release/2.11] [ROCm] Fix test/dynamo/test_repros.py::ReproTestsDevic…
xinyazhang Apr 3, 2026
8f4963d
[release/2.11][CI] Add related_commits file (#3131)
ethanwee1 Apr 3, 2026
4e32305
[ROCm] Require rocm_smi package (#175648)
naromero77amd Mar 24, 2026
0446f7b
[release/2.11] Fix numpy compatibility for Python 3.14 (#3100) (#3143)
WBobby Apr 9, 2026
8543095
[ROCm][CI][release/2.11] Backport checking existence of /etc/rocm_env…
CSkmd Apr 16, 2026
520641b
[release/2.11] Fix int4mm device memcpy error on Windows (#175410) (#…
jeffdaily Apr 20, 2026
141ba65
[UP][release/2.11] [ROCm][TunableOp] Support FP64 on hipBLASLt (#1781…
naromero77amd Apr 21, 2026
50bfde7
[release/2.11][ROCm][inductor] Additional GEMM, pointwise and reducti…
naromero77amd Apr 22, 2026
0320cc5
[release/2.11] Update Numba version constraints to support Python 3.1…
sohbodas Apr 23, 2026
3aaa914
[release/2.11] Fix MIOpen CTC loss crash on Windows (#179264) (#3181)
jeffdaily Apr 25, 2026
48211a7
[release/2.11] Fix Windows access violation in MIOpen CTC loss dispat…
tvukovic-amd Apr 27, 2026
e16e349
[release/2.11] Fix missing native header includes causing DLL export …
tvukovic-amd Apr 27, 2026
1a6ad28
[release/2.11] Windows specific test fixes (#176024) (#3182)
tvukovic-amd Apr 28, 2026
9413e9b
[AUTOGENERATED] [release/2.11] [UP][UT][ROCm][TunableOp] Fix test_cal…
rocm-repo-management-api-6[bot] Apr 28, 2026
7d37be2
[release/2.11] Update composable_kernel submodule with gfx1033 suppor…
harkgill-amd Apr 28, 2026
345ca6f
Cleanup custom op polluting global state for subsequent tests (#3170)
zichguan-amd Apr 28, 2026
443606e
[AUTOGENERATED] [release/2.11] Fix SIGSEGV on AMD RDNA due to reducti…
rocm-repo-management-api-6[bot] Apr 29, 2026
5223630
[Inductor] Fix ReinterpretView stride mismatch in TritonTemplateKerne…
tomjen12 May 7, 2026
96bfee1
Dynamo/fix ignore logging functions (#178506) (#3206)
jataylo May 12, 2026
0c210c5
Cleanup custom op polluting global state for subsequent tests (#18099…
zichguan-amd May 19, 2026
5f7b013
[release/2.11][UP][ROCm][inductor] Use hipModuleLoadData in StaticCud…
naromero77amd May 19, 2026
6c683dd
[Inductor] Fix flaky epilogue fusion tests by adding missing tearDown…
umechand-amd May 20, 2026
53679bf
[release/2.11] Free deferred record_stream blocks at graph capture en…
darren-amd May 21, 2026
f28bb50
Merge remote-tracking branch 'upstream/release/2.11' into release/2.1…
github-actions[bot] May 21, 2026
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
10 changes: 10 additions & 0 deletions .ci/docker/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,10 @@ case "$tag" in
GCC_VERSION=11
KATEX=yes
TRITON=yes
<<<<<<< HEAD
INSTALL_MINGW=yes
=======
>>>>>>> upstream/release/2.11
;;
pytorch-linux-jammy-cuda13.0-cudnn9-py3-gcc11-inductor-benchmarks)
CUDA_VERSION=13.0.2
Expand Down Expand Up @@ -159,8 +162,15 @@ case "$tag" in
else
ANACONDA_PYTHON_VERSION=3.12
fi
<<<<<<< HEAD
GCC_VERSION=13
ROCM_VERSION=7.2
=======
GCC_VERSION=11
VISION=yes
ROCM_VERSION=7.2
NINJA_VERSION=1.9.0
>>>>>>> upstream/release/2.11
TRITON=yes
KATEX=yes
PYTORCH_ROCM_ARCH="gfx90a;gfx942;gfx950;gfx1100"
Expand Down
4 changes: 4 additions & 0 deletions .ci/docker/ci_commit_pins/huggingface-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,6 @@
<<<<<<< HEAD
transformers==5.5.3
=======
transformers==5.2.0
>>>>>>> upstream/release/2.11
soxr==0.5.0
4 changes: 4 additions & 0 deletions .ci/docker/ci_commit_pins/triton.txt
Original file line number Diff line number Diff line change
@@ -1 +1,5 @@
<<<<<<< HEAD
b4e20bbe55617cc798b986c2555a2bc7b303c737
=======
4ed888920c5a0871957f1cf912e557bc79fbe56c
>>>>>>> upstream/release/2.11
10 changes: 10 additions & 0 deletions .ci/docker/common/install_cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -132,9 +132,14 @@ function install_129 {
}

function install_128 {
<<<<<<< HEAD
CUDNN_VERSION=9.20.0.48
CUSPARSELT_VERSION=0.7.1.0
echo "Installing CUDA 12.8.1 and cuDNN ${CUDNN_VERSION} and NVSHMEM and NCCL and cuSparseLt-${CUSPARSELT_VERSION}"
=======
CUDNN_VERSION=9.19.0.56
echo "Installing CUDA 12.8.1 and cuDNN ${CUDNN_VERSION} and NVSHMEM and NCCL and cuSparseLt-0.7.1"
>>>>>>> upstream/release/2.11
# install CUDA 12.8.1 in the same container
install_cuda 12.8.1 cuda_12.8.1_570.124.06_linux

Expand All @@ -151,9 +156,14 @@ function install_128 {
}

function install_130 {
<<<<<<< HEAD
CUDNN_VERSION=9.20.0.48
CUSPARSELT_VERSION=0.8.1.1
echo "Installing CUDA 13.0 and cuDNN ${CUDNN_VERSION} and NVSHMEM and NCCL and cuSparseLt-${CUSPARSELT_VERSION}"
=======
CUDNN_VERSION=9.19.0.56
echo "Installing CUDA 13.0 and cuDNN ${CUDNN_VERSION} and NVSHMEM and NCCL and cuSparseLt-0.7.1"
>>>>>>> upstream/release/2.11
# install CUDA 13.0 in the same container
install_cuda 13.0.2 cuda_13.0.2_580.95.05_linux

Expand Down
23 changes: 23 additions & 0 deletions .ci/docker/common/install_rocm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -203,6 +203,29 @@ EOF
fi
fi

# ROCm 7.2 needs a fix from procprof sdk that isn't available until 7.2.1
if [[ $(ver $ROCM_VERSION) -eq $(ver 7.2) ]]; then
git clone --no-checkout --filter=blob:none https://github.com/ROCm/rocm-systems.git
pushd rocm-systems/
git sparse-checkout init --cone
git sparse-checkout set projects/rocprofiler-sdk shared/rocprofiler-compute
git checkout develop
git checkout rocm-7.2.0
git config --global user.email "you@example.com"
git config --global user.name "Your Name"
git cherry-pick a71cc3cc88ed68b24c40cefec77d764053044862
sudo apt install -y cmake libdw-dev libsqlite3-dev
cmake \
-B rocprofiler-sdk-build \
-DCMAKE_INSTALL_PREFIX=/opt/rocm \
-DCMAKE_PREFIX_PATH=/opt/rocm \
-DGPU_TARGETS="${PYTORCH_ROCM_ARCH}" \
projects/rocprofiler-sdk
cmake --build rocprofiler-sdk-build --target all --parallel $(nproc)
cmake --build rocprofiler-sdk-build --target install
popd
fi

# ROCm 6.0 had a regression where journal_mode was enabled on the kdb files resulting in permission errors at runtime
for kdb in /opt/rocm/share/miopen/db/*.kdb
do
Expand Down
2 changes: 1 addition & 1 deletion .ci/docker/common/install_triton.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ elif [ -n "${TRITON_CPU}" ]; then
TRITON_REPO="https://github.com/triton-lang/triton-cpu"
TRITON_TEXT_FILE="triton-cpu"
else
TRITON_REPO="https://github.com/triton-lang/triton"
TRITON_REPO="https://github.com/ROCm/triton"
TRITON_TEXT_FILE="triton"
fi

Expand Down
22 changes: 21 additions & 1 deletion .ci/docker/requirements-ci.txt
Original file line number Diff line number Diff line change
Expand Up @@ -117,9 +117,16 @@ ninja==1.11.1.4
#Pinned versions: 1.11.1.4
#test that import: run_test.py, test_cpp_extensions_aot.py,test_determination.py

<<<<<<< HEAD
numba==0.61.2 ; python_version < "3.14" and platform_machine != "s390x"
numba==0.64.0 ; python_version >= "3.14" and platform_machine != "s390x"

=======
numba==0.49.0 ; python_version < "3.9" and platform_machine != "s390x"
numba==0.60.0 ; python_version == "3.9" and platform_machine != "s390x"
numba==0.61.2 ; python_version >= "3.10" and python_version < "3.14" and platform_machine != "s390x"
numba==0.64.0 ; python_version >= "3.14" and platform_machine != "s390x"
>>>>>>> upstream/release/2.11
#Description: Just-In-Time Compiler for Numerical Functions
#Pinned versions: 0.55.2, 0.60.0
#test that import: test_numba_integration.py
Expand All @@ -137,8 +144,13 @@ numba==0.64.0 ; python_version >= "3.14" and platform_machine != "s390x"
#test_nn.py, test_namedtensor.py, test_linalg.py, test_jit_cuda_fuser.py,
#test_jit.py, test_indexing.py, test_datapipe.py, test_dataloader.py,
#test_binary_ufuncs.py
<<<<<<< HEAD
numpy==2.1.2 ; python_version < "3.14"
numpy==2.3.4; python_version >= "3.14"
=======
numpy==2.0.2 ; python_version == "3.9"
numpy==2.1.2 ; python_version > "3.9"
>>>>>>> upstream/release/2.11

pandas==2.2.3; python_version < "3.14"
pandas==2.3.3; python_version >= "3.14"
Expand Down Expand Up @@ -252,7 +264,12 @@ scikit-image==0.22.0
#Pinned versions: 0.20.3
#test that import:

<<<<<<< HEAD
scipy==1.14.1 ; python_version < "3.14"
=======
scipy==1.13.1 ; python_version == "3.9"
scipy==1.14.1 ; python_version > "3.9" and python_version < "3.14"
>>>>>>> upstream/release/2.11
scipy==1.16.2 ; python_version >= "3.14"

# Pin SciPy because of failing distribution tests (see #60347)
Expand Down Expand Up @@ -289,11 +306,14 @@ lintrunner==0.12.11
#Pinned versions: 0.12.11
#test that import:

<<<<<<< HEAD
spin==0.17
#Description: developer CLI for common build/lint tasks
#Pinned versions: 0.17
#test that import:

=======
>>>>>>> upstream/release/2.11
redis==7.4.0
#Description: redis database
#test that import: anything that tests OSS caching/mocking (inductor/test_codecache.py, inductor/test_max_autotune.py)
Expand Down Expand Up @@ -390,7 +410,7 @@ dataclasses_json==0.6.7
#Pinned versions: 0.6.7
#test that import:

cmake==3.31.6
cmake==4.0.0
#Description: required for building

tlparse==0.4.0
Expand Down
6 changes: 6 additions & 0 deletions .ci/docker/requirements-docs.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,15 @@ sphinx==7.2.6
#Description: This is used to generate PyTorch docs
#Pinned versions: 7.2.6

<<<<<<< HEAD
pytorch_sphinx_theme2==0.4.9
#Description: This is needed to generate PyTorch docs
#Pinned versions: 0.4.9
=======
pytorch_sphinx_theme2==0.4.6
#Description: This is needed to generate PyTorch docs
#Pinned versions: 0.4.6
>>>>>>> upstream/release/2.11

sphinxcontrib.katex==0.9.11
#Description: This is used to generate PyTorch docs
Expand Down
5 changes: 5 additions & 0 deletions .ci/lumen_cli/cli/lib/core/vllm/vllm_test_library.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,12 @@ vllm_basic_models_test:
- pytest -v -s models/test_registry.py
- pytest -v -s models/test_utils.py
- pytest -v -s models/test_vision.py
<<<<<<< HEAD
- HF_DATASETS_OFFLINE=0 TRANSFORMERS_OFFLINE=0 pytest -v -s models/test_initialization.py
=======
- pytest -v -s models/test_initialization.py -k 'not voxtral'
- HF_DATASETS_OFFLINE=0 TRANSFORMERS_OFFLINE=0 pytest -v -s models/test_initialization.py -k voxtral
>>>>>>> upstream/release/2.11

vllm_entrypoints_test:
title: Entrypoints Test
Expand Down
4 changes: 4 additions & 0 deletions .ci/manywheel/build_cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,11 @@ case ${CUDA_VERSION} in
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST//8.6;/}" # Remove 8.6 for libtorch
fi
;;
<<<<<<< HEAD
13.0|13.2)
=======
13.0)
>>>>>>> upstream/release/2.11
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};$([[ "$ARCH" == "aarch64" ]] && echo "11.0;" || echo "")12.0"
export TORCH_NVCC_FLAGS="-compress-mode=size"
export BUILD_BUNDLE_PTXAS=1
Expand Down
7 changes: 6 additions & 1 deletion .ci/pytorch/binary_populate_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,9 @@ export TZ=UTC
tagged_version() {
GIT_DIR="${workdir}/pytorch/.git"
GIT_DESCRIBE="git --git-dir ${GIT_DIR} describe --tags --match v[0-9]*.[0-9]*.[0-9]*"
if [[ ! -d "${GIT_DIR}" ]]; then
if [[ -n "${CIRCLE_TAG:-}" ]]; then
echo "${CIRCLE_TAG}"
elif [[ ! -d "${GIT_DIR}" ]]; then
echo "Abort, abort! Git dir ${GIT_DIR} does not exists!"
kill $$
elif ${GIT_DESCRIBE} --exact >/dev/null; then
Expand Down Expand Up @@ -70,6 +72,8 @@ fi

export PYTORCH_BUILD_NUMBER=1

# This part is done in the builder scripts so commenting the duplicate code
: <<'BLOCK_COMMENT'
# Set triton version as part of PYTORCH_EXTRA_INSTALL_REQUIREMENTS
TRITON_VERSION=$(cat $PYTORCH_ROOT/.ci/docker/triton_version.txt)
TRITON_CONSTRAINT="platform_system == 'Linux'"
Expand Down Expand Up @@ -111,6 +115,7 @@ if [[ "$PACKAGE_TYPE" =~ .*wheel.* && -n "$PYTORCH_BUILD_VERSION" && "$PYTORCH_B
export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="${PYTORCH_EXTRA_INSTALL_REQUIREMENTS} | ${TRITON_REQUIREMENT}"
fi
fi
BLOCK_COMMENT

USE_GLOO_WITH_OPENSSL="OFF"
if [[ "$GPU_ARCH_TYPE" =~ .*aarch64.* ]]; then
Expand Down
4 changes: 4 additions & 0 deletions .ci/pytorch/common.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,11 @@
source "$(dirname "${BASH_SOURCE[0]}")/common_utils.sh"
set -ex -o pipefail

<<<<<<< HEAD
# Source ROCm environment variables (paths may vary between tarball/wheel installs)
=======
# for ROCm environment variables
>>>>>>> upstream/release/2.11
if [[ "${BUILD_ENVIRONMENT}" == *rocm* ]] && [[ -f /etc/rocm_env.sh ]]; then
# shellcheck disable=SC1091
source /etc/rocm_env.sh
Expand Down
4 changes: 4 additions & 0 deletions .ci/pytorch/common_utils.sh
Original file line number Diff line number Diff line change
Expand Up @@ -279,7 +279,11 @@ function install_torchrec_and_fbgemm() {

function clone_pytorch_xla() {
if [[ ! -d ./xla ]]; then
<<<<<<< HEAD
git clone --recursive -b r2.12 https://github.com/pytorch/xla.git
=======
git clone --recursive -b r2.11 https://github.com/pytorch/xla.git
>>>>>>> upstream/release/2.11
pushd xla
# pin the xla hash so that we don't get broken by changes to xla
git checkout "$(cat ../.github/ci_commit_pins/xla.txt)"
Expand Down
25 changes: 25 additions & 0 deletions .ci/pytorch/windows/internal/cuda_install.bat
Original file line number Diff line number Diff line change
Expand Up @@ -43,25 +43,37 @@ goto cuda_download
:cuda128
set CUDA_INSTALL_EXE=cuda_12.8.0_571.96_windows.exe
set "ARGS=cuda_profiler_api_12.8 thrust_12.8 nvcc_12.8 cuobjdump_12.8 nvprune_12.8 nvprof_12.8 cupti_12.8 cublas_12.8 cublas_dev_12.8 cudart_12.8 cufft_12.8 cufft_dev_12.8 curand_12.8 curand_dev_12.8 cusolver_12.8 cusolver_dev_12.8 cusparse_12.8 cusparse_dev_12.8 npp_12.8 npp_dev_12.8 nvrtc_12.8 nvrtc_dev_12.8 nvml_dev_12.8 nvjitlink_12.8 nvtx_12.8"
<<<<<<< HEAD
set CUDNN_FOLDER=cudnn-windows-x86_64-9.20.0.48_cuda12-archive
=======
set CUDNN_FOLDER=cudnn-windows-x86_64-9.19.0.56_cuda12-archive
>>>>>>> upstream/release/2.11
goto cuda_download

:cuda129
set CUDA_INSTALL_EXE=cuda_12.9.1_576.57_windows.exe
set "ARGS=cuda_profiler_api_12.9 thrust_12.9 nvcc_12.9 cuobjdump_12.9 nvprune_12.9 nvprof_12.9 cupti_12.9 cublas_12.9 cublas_dev_12.9 cudart_12.9 cufft_12.9 cufft_dev_12.9 curand_12.9 curand_dev_12.9 cusolver_12.9 cusolver_dev_12.9 cusparse_12.9 cusparse_dev_12.9 npp_12.9 npp_dev_12.9 nvrtc_12.9 nvrtc_dev_12.9 nvml_dev_12.9 nvjitlink_12.9 nvtx_12.9"
<<<<<<< HEAD
set CUDNN_FOLDER=cudnn-windows-x86_64-9.20.0.48_cuda12-archive
=======
set CUDNN_FOLDER=cudnn-windows-x86_64-9.17.1.4_cuda12-archive
>>>>>>> upstream/release/2.11
goto cuda_download

:cuda130
set CUDA_INSTALL_EXE=cuda_13.0.0_windows.exe
set "ARGS="
<<<<<<< HEAD
set CUDNN_FOLDER=cudnn-windows-x86_64-9.20.0.48_cuda13-archive
goto cuda_download

:cuda132
set CUDA_INSTALL_EXE=cuda_13.2.1_windows.exe
set "ARGS="
set CUDNN_FOLDER=cudnn-windows-x86_64-9.20.0.48_cuda13-archive
=======
set CUDNN_FOLDER=cudnn-windows-x86_64-9.19.0.56_cuda13-archive
>>>>>>> upstream/release/2.11
goto cuda_download

:: Common download logic for CUDA toolkit, cuDNN, and ZLIB
Expand Down Expand Up @@ -168,6 +180,7 @@ if %CUDA_VER% EQU 126 (
set EXPECTED_CUDNN_VERSION=9.10.2
)
if %CUDA_VER% EQU 128 (
<<<<<<< HEAD
set CUDNN_FOLDER=cudnn-windows-x86_64-9.20.0.48_cuda12-archive
set EXPECTED_CUDNN_VERSION=9.20.0
)
Expand All @@ -182,6 +195,18 @@ if %CUDA_VER% EQU 130 (
if %CUDA_VER% EQU 132 (
set CUDNN_FOLDER=cudnn-windows-x86_64-9.20.0.48_cuda13-archive
set EXPECTED_CUDNN_VERSION=9.20.0
=======
set CUDNN_FOLDER=cudnn-windows-x86_64-9.19.0.56_cuda12-archive
set EXPECTED_CUDNN_VERSION=9.19.0
)
if %CUDA_VER% EQU 129 (
set CUDNN_FOLDER=cudnn-windows-x86_64-9.17.1.4_cuda12-archive
set EXPECTED_CUDNN_VERSION=9.17.1
)
if %CUDA_VER% EQU 130 (
set CUDNN_FOLDER=cudnn-windows-x86_64-9.19.0.56_cuda13-archive
set EXPECTED_CUDNN_VERSION=9.19.0
>>>>>>> upstream/release/2.11
)
set "CUDNN_INSTALL_ZIP=%CUDNN_FOLDER%.zip"

Expand Down
4 changes: 4 additions & 0 deletions .github/ci_commit_pins/vllm.txt
Original file line number Diff line number Diff line change
@@ -1 +1,5 @@
<<<<<<< HEAD
ba4a78eb5d2ea30477b58a0bb8109b129f35c8b1
=======
a4047d4ea993fd52038433d87c16e603bee4f214
>>>>>>> upstream/release/2.11
4 changes: 4 additions & 0 deletions .github/ci_commit_pins/xla.txt
Original file line number Diff line number Diff line change
@@ -1 +1,5 @@
<<<<<<< HEAD
r2.12
=======
r2.11
>>>>>>> upstream/release/2.11
Loading