diff --git a/.ci/docker/requirements-ci.txt b/.ci/docker/requirements-ci.txt index 248ee8409036a..d3da8e93c639a 100644 --- a/.ci/docker/requirements-ci.txt +++ b/.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" diff --git a/aten/src/ATen/CMakeLists.txt b/aten/src/ATen/CMakeLists.txt index 6c095680733fe..38b383c2bb31e 100644 --- a/aten/src/ATen/CMakeLists.txt +++ b/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 diff --git a/requirements-build.txt b/requirements-build.txt index be19d987f73db..3d21094159d79 100644 --- a/requirements-build.txt +++ b/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 diff --git a/requirements.txt b/requirements.txt index f6dc86a0aa462..824ca112602a0 100644 --- a/requirements.txt +++ b/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