Commit f309230
[release/2.6] Upgrade numpy (#2254)
Relates to: ROCm/builder#82
Validation:
http://rocm-ci.amd.com/job/mainline-pytorch2.6-manylinux-wheels/120/
Ran import torch, import numpy, import pandas and the microbenchmark in
image called
_**registry-sc-harbor.amd.com/framework/compute-rocm-dkms-no-npi-hipclang:16180_ubuntu24.04_py3.12_pytorch_lw_release2.6_upgrade_numpy_57e5941d**_
```
root@ubb4-rack-22:/var/lib/jenkins/pytorch-micro-benchmarking# pip list | grep pandas
pandas 2.2.3
root@ubb4-rack-22:/var/lib/jenkins/pytorch-micro-benchmarking# pip list | grep numpy
numpy 2.1.2
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 numpy
>>> import torch
>>> exit()
root@ubb4-rack-22:/var/lib/jenkins# cd pytorch-micro-benchmarking/
root@ubb4-rack-22:/var/lib/jenkins/pytorch-micro-benchmarking# python3 micro_benchmarking_pytorch.py --network resnet50
/opt/venv/lib/python3.12/site-packages/redis/connection.py:77: UserWarning: redis-py works best with hiredis. Please consider installing
warnings.warn(msg)
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.1023630142211914
Throughput [img/sec] : 625.2258248443663
root@ubb4-rack-22:/var/lib/jenkins/pytorch-micro-benchmarking#
```
---------
Co-authored-by: Jithun Nair <[email protected]>1 parent ce580d3 commit f309230
2 files changed
+5
-15
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
110 | 110 | | |
111 | 111 | | |
112 | 112 | | |
113 | | - | |
114 | | - | |
115 | | - | |
116 | | - | |
| 113 | + | |
117 | 114 | | |
118 | 115 | | |
119 | 116 | | |
| |||
131 | 128 | | |
132 | 129 | | |
133 | 130 | | |
134 | | - | |
135 | | - | |
136 | | - | |
137 | | - | |
138 | | - | |
139 | | - | |
| 131 | + | |
140 | 132 | | |
| 133 | + | |
141 | 134 | | |
142 | 135 | | |
143 | 136 | | |
| |||
247 | 240 | | |
248 | 241 | | |
249 | 242 | | |
250 | | - | |
251 | | - | |
| 243 | + | |
252 | 244 | | |
253 | 245 | | |
254 | 246 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
5 | | - | |
6 | | - | |
7 | | - | |
| 5 | + | |
8 | 6 | | |
9 | 7 | | |
10 | 8 | | |
| |||
0 commit comments