| win_64_channel_targetsconda-forge_maincuda_compiler_version12.8is_rcFalse |
diff --git a/recipe/bld.bat b/recipe/bld.bat
index 965a1178..e5e215ed 100644
--- a/recipe/bld.bat
+++ b/recipe/bld.bat
@@ -46,11 +46,11 @@ if "%PKG_NAME%" == "pytorch" (
if %ERRORLEVEL% neq 0 exit 1
@REM Replace version string v3.12.8() with ie v3.11.11()
- sed -i.bak -E "s/v3\.12\.[0-9]+/v%PY_VERSION_FULL%/g" build/CMakeCache.txt
+ sed -i.bak -E "s/v3\.12\.[0-9]+/v!PY_VERSION_FULL!/g" build/CMakeCache.txt
if %ERRORLEVEL% neq 0 exit 1
@REM Replace interpreter properties Python;3;12;8;64 with ie Python;3;11;11;64
- sed -i.bak -E "s/Python;3;12;[0-9]+;64/Python;%PY_VERSION_FULL:.=;%;64/g" build/CMakeCache.txt
+ sed -i.bak -E "s/Python;3;12;[0-9]+;64/Python;!PY_VERSION_FULL:.=;!;64/g" build/CMakeCache.txt
if %ERRORLEVEL% neq 0 exit 1
@REM Replace cp312-win_amd64 with ie cp311-win_amd64
diff --git a/recipe/build.sh b/recipe/build.sh
index d27a67c8..ab944468 100644
--- a/recipe/build.sh
+++ b/recipe/build.sh
@@ -98,6 +98,7 @@ export BUILD_CUSTOM_PROTOBUF=OFF
rm -rf $PREFIX/bin/protoc
export USE_SYSTEM_PYBIND11=1
export USE_SYSTEM_EIGEN_INSTALL=1
+export Python_ROOT_DIR=$PREFIX
# workaround to stop setup.py from trying to check whether we checked out
# all submodules (we don't use all of them)
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
index 87a4de6a..b98c86b0 100644
--- a/recipe/meta.yaml
+++ b/recipe/meta.yaml
@@ -1,6 +1,6 @@
# if you wish to build release candidate number X, append the version string with ".rcX"
-{% set version = "2.7.1" %}
-{% set build = 4 %}
+{% set version = "2.8.0" %}
+{% set build = 0 %}
# Use a higher build number for the CUDA variant, to ensure that it's
# preferred by conda's solver, and it's preferentially
@@ -16,7 +16,7 @@
# see https://github.com/pytorch/pytorch/blame/v{{ version }}/.ci/docker/ci_commit_pins/triton.txt
# pytorch and triton are released in tandem, see notes in their release process
# https://github.com/pytorch/pytorch/blob/main/RELEASE.md#triton-dependency-for-the-release
-{% set triton = "3.3.1" %}
+{% set triton = "3.4.0" %}
# TODO Temporary pin, remove me
{% set mkl = "<2025" %}
@@ -38,7 +38,7 @@ source:
{% else %}
# The "pytorch-v" tarballs contain submodules; the "pytorch-" ones don't.
- url: https://github.com/pytorch/pytorch/releases/download/v{{ version }}/pytorch-v{{ version }}.tar.gz
- sha256: 5befd2e540fd55ce4782d0ca7610ce5b572d756d7ea38090ef0f3c7c428fb20f
+ sha256: c70a2c9488f6f6e8af5982a10d1cc2c37b7df5e6506d839daa5d5e250953d7b5
{% endif %}
patches:
- patches/0001-Force-usage-of-python-3-and-error-without-numpy.patch
@@ -61,10 +61,6 @@ source:
- patches/0012-avoid-deprecated-find_package-CUDA-in-caffe2-CMake-m.patch
# backport https://github.com/pytorch/pytorch/pull/148668
- patches/0013-Fix-CUPTI-lookup-to-include-target-directory.patch
- # backport (practically speaking) https://github.com/pytorch/pytorch/pull/149861
- - patches/0014-Always-use-system-nvtx.patch
- # https://github.com/pytorch/pytorch/pull/152533
- - patches/0015-Do-not-check-out-nccl-when-not-building-it.patch
# skip a test that fails with numpy 2.3; can be dropped for pytorch>2.7
- patches/0016-skip-test_norm_matrix_degenerate_shapes-on-numpy-2.3.patch
# backport https://github.com/pytorch/pytorch/pull/127702
@@ -113,7 +109,7 @@ requirements:
- llvm-openmp # [unix]
- intel-openmp {{ mkl }} # [win]
- libuv # [win]
- - cmake <4
+ - cmake
- ninja
# Keep libprotobuf here so that a compatibile version
# of protobuf is installed between build and host
@@ -125,7 +121,9 @@ requirements:
- rsync # [unix]
host:
# GPU requirements
- - cudnn # [cuda_compiler_version != "None"]
+ # cudnn pinned over segfaults
+ # https://github.com/conda-forge/pytorch-cpu-feedstock/pull/411
+ - cudnn <9.11 # [cuda_compiler_version != "None"]
- nccl # [cuda_compiler_version != "None" and linux]
- magma # [cuda_compiler_version != "None"]
- cuda-version {{ cuda_compiler_version }} # [cuda_compiler_version != "None"]
@@ -171,11 +169,12 @@ requirements:
- pkg-config # [unix]
- typing_extensions
- pybind11
+ - pybind11-abi
- eigen
- zlib
run:
# GPU requirements without run_exports
- - {{ pin_compatible('cudnn') }} # [cuda_compiler_version != "None"]
+ - cudnn <9.11 # [cuda_compiler_version != "None"]
- intel-openmp {{ mkl }} # [win]
- libblas * *{{ blas_impl }} # [blas_impl == "mkl"]
run_constrained:
@@ -192,6 +191,7 @@ requirements:
# if using OpenBLAS, ensure that a version compatible with OpenMP is used
# otherwise, we get the following warnings:
# OpenBLAS Warning : Detect OpenMP Loop and this application may hang. Please rebuild the library with USE_OPENMP=1 option.
+ - libopenblas * openmp_* # [unix and blas_impl != "mkl"]
- openblas * openmp_* # [unix and blas_impl != "mkl"]
# these tests are for the libtorch output below, but due to
@@ -206,7 +206,7 @@ test:
- {{ compiler('cuda') }} # [cuda_compiler_version != "None"]
- cuda-nvrtc-dev # [cuda_compiler_version != "None"]
- nvtx-c # [cuda_compiler_version != "None"]
- - cmake <4
+ - cmake
- ninja
- pkg-config
files:
@@ -266,7 +266,7 @@ outputs:
- {{ compiler('cuda') }} # [cuda_compiler_version != "None"]
- llvm-openmp # [unix]
- intel-openmp {{ mkl }} # [win]
- - cmake <4
+ - cmake
- ninja
# Keep libprotobuf here so that a compatibile version
# of protobuf is installed between build and host
@@ -277,7 +277,7 @@ outputs:
host:
- {{ pin_subpackage('libtorch', exact=True) }}
# GPU requirements
- - cudnn # [cuda_compiler_version != "None"]
+ - cudnn <9.11 # [cuda_compiler_version != "None"]
- nccl # [cuda_compiler_version != "None" and linux]
- cuda-version {{ cuda_compiler_version }} # [cuda_compiler_version != "None"]
- nvtx-c # [cuda_compiler_version != "None"]
@@ -316,6 +316,7 @@ outputs:
- libabseil
- libprotobuf
- pybind11
+ - pybind11-abi
- eigen
- sleef
- libuv
@@ -331,8 +332,8 @@ outputs:
- libblas * *{{ blas_impl }} # [blas_impl == "mkl"]
- nomkl # [blas_impl != "mkl"]
# GPU requirements without run_exports
- - {{ pin_compatible('cudnn') }} # [cuda_compiler_version != "None"]
- - triton {{ triton }} # [cuda_compiler_version != "None" and not win]
+ - cudnn <9.11 # [cuda_compiler_version != "None"]
+ - triton =={{ triton }} # [cuda_compiler_version != "None" and not win]
# avoid that people without GPUs needlessly download ~0.5-1GB
- __cuda # [cuda_compiler_version != "None"]
- python
@@ -371,6 +372,7 @@ outputs:
- pip
- expecttest
- xmlrunner
+ - pyyaml
# Required by run_test.py
- pytest-flakefinder
- pytest-rerunfailures
@@ -380,7 +382,7 @@ outputs:
# https://github.com/pytorch/pytorch/blob/main/test/pytest_shard_custom.py
# - pytest-shard
# for cmake_test
- - cmake <4
+ - cmake
- cuda-nvrtc-dev # [cuda_compiler_version != "None"]
- nvtx-c # [cuda_compiler_version != "None"]
- pybind11
@@ -443,6 +445,7 @@ outputs:
# minor tolerance violations
{% set skips = skips ~ " or test_1_sized_with_0_strided_cpu_float32" %} # [osx]
{% set skips = skips ~ " or test_batchnorm_nhwc_cpu" %} # [unix]
+ {% set skips = skips ~ " or test_layer_norm_backwards_eps" %} # [unix]
# timeouts and failures on aarch, see https://github.com/conda-forge/pytorch-cpu-feedstock/pull/298#issuecomment-2555888508
{% set skips = skips ~ " or test_pynode_destruction_deadlock" %} # [aarch64]
{% set skips = skips ~ " or (TestLinalgCPU and test_cholesky_cpu_float32)" %} # [aarch64]
diff --git a/recipe/patches/0014-Always-use-system-nvtx.patch b/recipe/patches/0014-Always-use-system-nvtx.patch
deleted file mode 100644
index 0a0895d1..00000000
--- a/recipe/patches/0014-Always-use-system-nvtx.patch
+++ /dev/null
@@ -1,22 +0,0 @@
-From b3371dcb971658239ce8aa8f6fa0a4d7bcea6735 Mon Sep 17 00:00:00 2001
-From: Jeongseok Lee
-Date: Sat, 22 Mar 2025 22:50:49 -0700
-Subject: [PATCH 14/17] Always use system nvtx
-
----
- cmake/public/cuda.cmake | 2 +-
- 1 file changed, 1 insertion(+), 1 deletion(-)
-
-diff --git a/cmake/public/cuda.cmake b/cmake/public/cuda.cmake
-index e538fb5662b..b409c6c30e9 100644
---- a/cmake/public/cuda.cmake
-+++ b/cmake/public/cuda.cmake
-@@ -146,7 +146,7 @@ else()
- endif()
-
- # nvToolsExt
--if(USE_SYSTEM_NVTX)
-+if(TRUE)
- find_path(nvtx3_dir NAMES nvtx3 PATHS ${CUDA_INCLUDE_DIRS})
- else()
- find_path(nvtx3_dir NAMES nvtx3 PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)
diff --git a/recipe/patches/0015-Do-not-check-out-nccl-when-not-building-it.patch b/recipe/patches/0015-Do-not-check-out-nccl-when-not-building-it.patch
deleted file mode 100644
index 9bca436a..00000000
--- a/recipe/patches/0015-Do-not-check-out-nccl-when-not-building-it.patch
+++ /dev/null
@@ -1,45 +0,0 @@
-From 647a6da2006050b4983cab517167105d6b4badfe Mon Sep 17 00:00:00 2001
-From: =?UTF-8?q?Micha=C5=82=20G=C3=B3rny?=
-Date: Wed, 30 Apr 2025 17:11:56 +0200
-Subject: [PATCH 15/17] Do not check out nccl when not building it
-
-Add additional conditions to `build_pytorch_libs.py` to avoid fetching
-NCCL when `USE_CUDA` or `USE_NCCL` are disabled. While at it, adjust
-the existing condition for `USE_SYSTEM_NCCL` to use the utility
-function.
----
- tools/build_pytorch_libs.py | 14 ++++++++++++--
- 1 file changed, 12 insertions(+), 2 deletions(-)
-
-diff --git a/tools/build_pytorch_libs.py b/tools/build_pytorch_libs.py
-index 5dd5a221975..2b8b868eaa8 100644
---- a/tools/build_pytorch_libs.py
-+++ b/tools/build_pytorch_libs.py
-@@ -7,7 +7,12 @@ from glob import glob
- from pathlib import Path
-
- from .setup_helpers.cmake import CMake, USE_NINJA
--from .setup_helpers.env import check_negative_env_flag, IS_64BIT, IS_WINDOWS
-+from .setup_helpers.env import (
-+ check_env_flag,
-+ check_negative_env_flag,
-+ IS_64BIT,
-+ IS_WINDOWS,
-+)
-
-
- repo_root = Path(__file__).absolute().parent.parent
-@@ -119,7 +124,12 @@ def build_pytorch(
- cmake: CMake,
- ) -> None:
- my_env = _create_build_env()
-- checkout_nccl()
-+ if (
-+ not check_negative_env_flag("USE_CUDA")
-+ and not check_negative_env_flag("USE_NCCL")
-+ and not check_env_flag("USE_SYSTEM_NCCL")
-+ ):
-+ checkout_nccl()
- build_test = not check_negative_env_flag("BUILD_TEST")
- cmake.generate(
- version, cmake_python_library, build_python, build_test, my_env, rerun_cmake
|