Skip to content

Commit cf32479

Browse files
[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]>
1 parent f1ad49a commit cf32479

File tree

2 files changed

+4
-14
lines changed

2 files changed

+4
-14
lines changed

.ci/docker/requirements-ci.txt

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -113,11 +113,7 @@ ninja==1.11.1.3
113113
#Pinned versions: 1.11.1.3
114114
#test that import: run_test.py, test_cpp_extensions_aot.py,test_determination.py
115115

116-
numba==0.49.0 ; python_version < "3.9"
117-
numba==0.55.2 ; python_version == "3.9"
118-
numba==0.55.2 ; python_version == "3.10"
119-
numba==0.60.0 ; python_version == "3.12"
120-
#Description: Just-In-Time Compiler for Numerical Functions
116+
numba==0.61.2
121117
#Pinned versions: 0.54.1, 0.49.0, <=0.49.1
122118
#test that import: test_numba_integration.py
123119
#For numba issue see https://github.com/pytorch/pytorch/issues/51511
@@ -134,13 +130,9 @@ numba==0.60.0 ; python_version == "3.12"
134130
#test_nn.py, test_namedtensor.py, test_linalg.py, test_jit_cuda_fuser.py,
135131
#test_jit.py, test_indexing.py, test_datapipe.py, test_dataloader.py,
136132
#test_binary_ufuncs.py
137-
numpy==1.22.4; python_version == "3.9" or python_version == "3.10"
138-
numpy==1.26.2; python_version == "3.11" or python_version == "3.12"
139-
numpy==2.1.2; python_version >= "3.13"
140-
141-
pandas==2.0.3; python_version < "3.13"
142-
pandas==2.2.3; python_version >= "3.13"
133+
numpy==2.1.2
143134

135+
pandas==2.2.3
144136
#onnxruntime
145137
#Description: scoring engine for Open Neural Network Exchange (ONNX) models
146138
#Pinned versions: 1.9.0

requirements.txt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,7 @@ jinja2
99
lintrunner ; platform_machine != "s390x"
1010
networkx
1111
ninja
12-
numpy==1.22.4; python_version == "3.9" or python_version == "3.10"
13-
numpy==1.26.2; python_version == "3.11" or python_version == "3.12"
14-
numpy==2.1.2; python_version >= "3.13"
12+
numpy==2.1.2
1513
optree>=0.13.0
1614
packaging
1715
psutil

0 commit comments

Comments
 (0)