[Release/2.9] pin requirements

.ci/docker/requirements-ci.txt

@@ -10,7 +10,7 @@ boto3==1.35.42
 #Pinned versions: 1.19.12, 1.16.34
 #test that import:
 
-click
+click==8.3.0
 #Description: Command Line Interface Creation Kit
 #Pinned versions:
 #test that import:
@@ -63,7 +63,7 @@ lark==0.12.0
 #Pinned versions: 0.12.0
 #test that import:
 
-librosa>=0.6.2 ; python_version < "3.11" and platform_machine != "s390x"
+librosa==0.11.0 ; python_version < "3.11" and platform_machine != "s390x"
 librosa==0.10.2 ; python_version == "3.12" and platform_machine != "s390x"
 #Description: A python package for music and audio analysis
 #Pinned versions: >=0.6.2
@@ -192,7 +192,7 @@ pytest-flakefinder==1.1.0
 #Pinned versions: 1.1.0
 #test that import:
 
-pytest-rerunfailures>=10.3
+pytest-rerunfailures==14.0
 #Description: plugin for rerunning failure tests in pytest
 #Pinned versions:
 #test that import:
@@ -263,7 +263,7 @@ scipy==1.14.1 ; python_version > "3.9"
 #test that import:
 
 # needed by torchgen utils
-typing-extensions>=4.10.0
+typing_extensions==4.15.0
 #Description: type hints for python
 #Pinned versions:
 #test that import:
@@ -273,7 +273,7 @@ typing-extensions>=4.10.0
 #Pinned versions:
 #test that import:
 
-unittest-xml-reporting<=3.2.0,>=2.0.0
+unittest-xml-reporting==3.2.0
 #Description: saves unit test results to xml
 #Pinned versions:
 #test that import:
@@ -284,7 +284,7 @@ lintrunner==0.12.7
 #Pinned versions: 0.12.7
 #test that import:
 
-redis>=4.0.0
+redis==6.4.0
 #Description: redis database
 #test that import: anything that tests OSS caching/mocking (inductor/test_codecache.py, inductor/test_max_autotune.py)
 
@@ -359,10 +359,10 @@ pwlf==2.2.1
 #test that import: test_sac_estimator.py
 
 # To build PyTorch itself
-pyyaml
-pyzstd
-setuptools>=70.1.0
-six
+PyYAML==6.0.3
+pyzstd==0.18.0
+setuptools==79.0.1
+six==1.17.0
 
 scons==4.5.2 ; platform_machine == "aarch64"
 

aten/src/ATen/CMakeLists.txt

@@ -1,5 +1,15 @@
 cmake_minimum_required(VERSION 3.27 FATAL_ERROR)
 set(CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake ${CMAKE_MODULE_PATH})
+list(APPEND CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}/cmake/public")
+if(USE_ROCM)
+    include(LoadHIP OPTIONAL RESULT_VARIABLE _had_loadhip)
+    if(_had_loadhip)
+        # Exposed by LoadHIP.cmake, e.g. "7.1.2" or "7.2.0"
+        message(STATUS "LoadHIP loaded: ROCM_VERSION_DEV='${ROCM_VERSION_DEV}'")
+    else()
+        message(WARNING "LoadHIP.cmake not found; ROCM_VERSION_DEV unavailable")
+    endif()
+endif()
 
 if(NOT MSVC)
   string(APPEND CMAKE_CXX_FLAGS " -Wno-ignored-qualifiers")
@@ -301,13 +311,14 @@ IF(USE_FBGEMM_GENAI)
 
       # Add additional HIPCC compiler flags for performance
       set(FBGEMM_GENAI_EXTRA_HIPCC_FLAGS
-        -mllvm
-        -amdgpu-coerce-illegal-types=1
         -mllvm
         -enable-post-misched=0
         -mllvm
         -greedy-reverse-local-assignment=1
         -fhip-new-launch-api)
+      if(DEFINED ROCM_VERSION_DEV AND ROCM_VERSION_DEV VERSION_LESS "7.2.0")
+        list(PREPEND FBGEMM_GENAI_EXTRA_HIPCC_FLAGS -mllvm -amdgpu-coerce-illegal-types=1)
+      endif()
 
       hip_add_library(
         fbgemm_genai STATIC

requirements-build.txt

@@ -1,10 +1,10 @@
 # Build System requirements
-setuptools>=70.1.0,<80.0  # setuptools develop deprecated on 80.0
-cmake>=3.27
-ninja
-numpy
-packaging
-pyyaml
-requests
-six  # dependency chain: NNPACK -> PeachPy -> six
-typing-extensions>=4.10.0
+setuptools==79.0.1
+cmake==4.0.0
+ninja==1.11.1.3
+numpy==2.1.2
+packaging==25.0
+pyyaml==6.0.3
+requests==2.32.5
+six==1.17.0  # dependency chain: NNPACK -> PeachPy -> six
+typing_extensions==4.15.0

requirements.txt

@@ -5,18 +5,18 @@
 
 # Install / Development extra requirements
 build[uv]  # for building sdist and wheel
-expecttest>=0.3.0
-filelock
-fsspec>=0.8.5
-hypothesis
-jinja2
-lintrunner ; platform_machine != "s390x" and platform_machine != "riscv64"
-networkx>=2.5.1
-ninja
+expecttest==0.3.0
+filelock==3.20.0
+fsspec==2025.9.0
+hypothesis==5.35.1
+Jinja2==3.1.6
+lintrunner==0.12.7 ; platform_machine != "s390x" and platform_machine != "riscv64"
+networkx==2.8.8
+ninja==1.11.1.3
 numpy==2.0.2 ; python_version == "3.9"
 numpy==2.1.2 ; python_version > "3.9"
-optree>=0.13.0
-psutil
-sympy>=1.13.3
-typing-extensions>=4.13.2
+optree==0.13.0
+psutil==7.1.0
+sympy==1.13.3
+typing_extensions==4.15.0
 wheel