Skip to content

Commit ee8c4e9

Browse files
ethanwee1jithunnair-amd
authored andcommitted
[rocm7.0_internal_testing] upgrading numpy (#2256)
Relates to: ROCm/builder#82 Validation: http://rocm-ci.amd.com/job/mainline-pytorch_internal-manylinux-wheels/98/ Using `registry-sc-harbor.amd.com/framework/compute-rocm-dkms-no-npi-hipclang:16180_ubuntu24.04_py3.12_pytorch_lw_rocm7.0_IT_upgrade_numpy_452f3df6`: ``` root@d92befdbb2a6:/# pip list | egrep "numpy|pandas" numpy 2.1.2 pandas 2.2.3 root@d92befdbb2a6:/# python3 Python 3.12.3 (main, Feb 4 2025, 14:48:35) [GCC 13.3.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import pandas >>> import torch >>> import numpy >>> exit() root@d92befdbb2a6:/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.11369450092315674 Throughput [img/sec] : 562.9120096428937 ``` --------- Co-authored-by: Jithun Nair <[email protected]> (cherry picked from commit cf32479)
1 parent 9237b26 commit ee8c4e9

File tree

2 files changed

+7
-9
lines changed

2 files changed

+7
-9
lines changed

.ci/docker/requirements-ci.txt

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -117,10 +117,10 @@ ninja==1.11.1.4
117117
#Pinned versions: 1.11.1.4
118118
#test that import: run_test.py, test_cpp_extensions_aot.py,test_determination.py
119119

120-
numba==0.55.2 ; python_version == "3.10" and platform_machine != "s390x"
121-
numba==0.60.0 ; python_version == "3.12" and platform_machine != "s390x"
120+
numba==0.60.0 ; python_version == "3.9"
121+
numba==0.61.2 ; python_version > "3.9"
122122
#Description: Just-In-Time Compiler for Numerical Functions
123-
#Pinned versions: 0.55.2, 0.60.0
123+
#Pinned versions: 0.61.2, 0.60.0
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,12 +136,10 @@ numba==0.60.0 ; python_version == "3.12" and platform_machine != "s390x"
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==1.22.4; python_version == "3.10"
140-
numpy==1.26.2; python_version == "3.11" or python_version == "3.12"
141-
numpy==2.1.2; python_version >= "3.13"
139+
numpy==2.0.2; python_version == "3.9"
140+
numpy==2.1.2; python_version >= "3.9"
142141

143-
pandas==2.0.3; python_version < "3.13"
144-
pandas==2.2.3; python_version >= "3.13"
142+
pandas==2.2.3
145143

146144
#onnxruntime
147145
#Description: scoring engine for Open Neural Network Exchange (ONNX) models

requirements-build.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
setuptools>=70.1.0
33
cmake>=3.27
44
ninja
5-
numpy
5+
numpy==2.1.2
66
packaging
77
pyyaml
88
requests

0 commit comments

Comments
 (0)