Skip to content

Commit af2a7cd

Browse files
authored
Update microbenchmarks README and output format (#54)
* Update README to reflect the most result benchmark result. * Update the HBM test output format for easier parsing.
1 parent 155beed commit af2a7cd

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

microbenchmarks/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ python benchmark_hbm.py \
5555

5656
Example output:
5757
```
58-
Tensor size: 32.0 MBs, time taken (median): 0.0461 ms, bandwidth: 1454.54 GBps
58+
Tensor size (bytes): 33554432, time taken (ms, median): 0.049359414, bandwidth (GBps, median): 1359.5960438266143
5959
```
6060

6161
Run `python benchmark_hbm.py -h` to view the how to set the arguments.

microbenchmarks/benchmark_hbm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,8 @@ def main():
125125
bw_gbps = (tensor_size * 2) / result.time_median / 1e9 # read + write = 2
126126

127127
print(
128-
f"Tensor size: {tensor_size / 1024**2} MB, time taken (median):"
129-
f" {result.time_median * 1000:.4f} ms, bandwidth: {bw_gbps:.2f} GBps"
128+
f"Tensor size (bytes): {tensor_size}, time taken (ms, median):"
129+
f" {result.time_median * 1000}, bandwidth (GBps, median): {bw_gbps} "
130130
)
131131

132132

0 commit comments

Comments
 (0)