Skip to content

Commit a016608

Browse files
[rocm7.0_internal_testing] Use different package versions for py3.9 (#2269)
Fixes SWDEV-536456 Fixes error post-#2256: ``` 00:12:44.248 #22 155.3 ERROR: Ignored the following versions that require a different python version: 0.52.0 Requires-Python >=3.6,<3.9; 0.52.0rc3 Requires-Python >=3.6,<3.9; 0.61.0 Requires-Python >=3.10; 0.61.0rc1 Requires-Python >=3.10; 0.61.0rc2 Requires-Python >=3.10; 0.61.1rc1 Requires-Python >=3.10; 0.61.2 Requires-Python >=3.10; 3.3 Requires-Python >=3.10; 3.3rc0 Requires-Python >=3.10; 3.4 Requires-Python >=3.10; 3.4.1 Requires-Python >=3.10; 3.4.2 Requires-Python >=3.10; 3.4rc0 Requires-Python >=3.10; 3.5 Requires-Python >=3.11; 3.5rc0 Requires-Python >=3.11; 8.2.0 Requires-Python >=3.10; 8.2.1 Requires-Python >=3.10 00:12:44.248 #22 155.3 ERROR: Could not find a version that satisfies the requirement numba==0.61.2 (from versions: 0.1, 0.2, 0.3, 0.5.0, 0.6.0, 0.7.0, 0.7.1, 0.7.2, 0.8.0, 0.8.1, 0.9.0, 0.10.0, 0.10.1, 0.11.0, 0.12.0, 0.12.1, 0.12.2, 0.13.0, 0.13.2, 0.13.3, 0.13.4, 0.14.0, 0.15.1, 0.16.0, 0.17.0, 0.18.1, 0.18.2, 0.19.1, 0.19.2, 0.20.0, 0.21.0, 0.22.0, 0.22.1, 0.23.0, 0.23.1, 0.24.0, 0.25.0, 0.26.0, 0.27.0, 0.28.1, 0.29.0, 0.30.0, 0.30.1, 0.31.0, 0.32.0, 0.33.0, 0.34.0, 0.35.0, 0.36.1, 0.36.2, 0.37.0, 0.38.0, 0.38.1, 0.39.0, 0.40.0, 0.40.1, 0.41.0, 0.42.0, 0.42.1, 0.43.0, 0.43.1, 0.44.0, 0.44.1, 0.45.0, 0.45.1, 0.46.0, 0.47.0, 0.48.0, 0.49.0, 0.49.1rc1, 0.49.1, 0.50.0rc1, 0.50.0, 0.50.1, 0.51.0rc1, 0.51.0, 0.51.1, 0.51.2, 0.52.0rc2, 0.53.0rc1.post1, 0.53.0rc2, 0.53.0rc3, 0.53.0, 0.53.1, 0.54.0rc2, 0.54.0rc3, 0.54.0, 0.54.1rc1, 0.54.1, 0.55.0rc1, 0.55.0, 0.55.1, 0.55.2, 0.56.0rc1, 0.56.0, 0.56.2, 0.56.3, 0.56.4, 0.57.0rc1, 0.57.0, 0.57.1rc1, 0.57.1, 0.58.0rc1, 0.58.0rc2, 0.58.0, 0.58.1, 0.59.0rc1, 0.59.0, 0.59.1, 0.60.0rc1, 0.60.0) 00:12:44.248 #22 155.3 ERROR: No matching distribution found for numba==0.61.2 ``` Validation: * Docker image: http://rocm-ci.amd.com/job/mainline-framework-pytorch-internal-cs9-ci/132 * Wheels: http://rocm-ci.amd.com/job/mainline-pytorch_internal-manylinux-wheels/102/ From `registry-sc-harbor.amd.com/framework/compute-rocm-dkms-no-npi-hipclang:16180_ubuntu22.04_py3.9_pytorch_lw_rocm7.0_IT_py3.9_a11d94ad`: ``` root@f43861a0a856:/# pip list | egrep "numpy|pandas" numpy 2.0.2 pandas 2.2.3 root@f43861a0a856:/# python Python 3.9.23 (main, Jun 4 2025, 08:55:38) [GCC 11.4.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> import numpy >>> import pandas root@f43861a0a856:/data/pytorch-micro-benchmarking# HIP_VISIBLE_DEVICES=1 python3 micro_benchmarking_pytorch.py --network resnet50 INFO: running forward and backward for warmup. INFO: running the benchmark.. OK: finished running benchmark.. --------------------SUMMARY-------------------------- Microbenchmark for network : resnet50 Num devices: 1 Dtype: FP32 Mini batch size [img] : 64 Time per mini-batch : 0.11354223489761353 Throughput [img/sec] : 563.6669038416574 ``` (cherry picked from commit a0a9d81)
1 parent b296de5 commit a016608

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

.ci/docker/requirements-ci.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ ninja==1.11.1.4
120120
numba==0.60.0 ; python_version == "3.9"
121121
numba==0.61.2 ; python_version > "3.9"
122122
#Description: Just-In-Time Compiler for Numerical Functions
123-
#Pinned versions: 0.61.2, 0.60.0
123+
#Pinned versions: 0.54.1, 0.49.0, <=0.49.1
124124
#test that import: test_numba_integration.py
125125
#Need release > 0.61.2 for s390x due to https://github.com/numba/numba/pull/10073
126126

@@ -136,8 +136,8 @@ numba==0.61.2 ; python_version > "3.9"
136136
#test_nn.py, test_namedtensor.py, test_linalg.py, test_jit_cuda_fuser.py,
137137
#test_jit.py, test_indexing.py, test_datapipe.py, test_dataloader.py,
138138
#test_binary_ufuncs.py
139-
numpy==2.0.2; python_version == "3.9"
140-
numpy==2.1.2; python_version >= "3.9"
139+
numpy==2.0.2 ; python_version == "3.9"
140+
numpy==2.1.2 ; python_version > "3.9"
141141

142142
pandas==2.2.3
143143

@@ -249,8 +249,8 @@ scikit-image==0.22.0
249249
#Pinned versions: 0.20.3
250250
#test that import:
251251

252-
scipy==1.10.1 ; python_version <= "3.11"
253-
scipy==1.14.1 ; python_version >= "3.12"
252+
scipy==1.13.1 ; python_version == "3.9"
253+
scipy==1.14.1 ; python_version > "3.9"
254254
# Pin SciPy because of failing distribution tests (see #60347)
255255
#Description: scientific python
256256
#Pinned versions: 1.10.1

requirements-build.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,8 @@
22
setuptools>=70.1.0
33
cmake>=3.27
44
ninja
5-
numpy==2.1.2
5+
numpy==2.0.2 ; python_version == "3.9""
6+
numpy==2.1.2 ; python_version > "3.9"
67
packaging
78
pyyaml
89
requests

0 commit comments

Comments
 (0)