Skip to content

Commit ad25616

Browse files
committed
run_benchmarks.py script and folder
Signed-off-by: adir <adir@il.ibm.com>
1 parent 0aca3c5 commit ad25616

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

cmd/benchmarking/run_benchmarks.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -115,14 +115,14 @@ def run_and_parse_non_parallel_metrics(benchName, params, folder=transfer_benchm
115115
f"{name} allocs": allocs,
116116
}
117117

118-
def run_and_parse_parallel_metrics(test, params, folder=transfer_benchmarks_folder) -> dict:
118+
def run_and_parse_parallel_metrics(benchName, params, folder=transfer_benchmarks_folder) -> dict:
119119
if folder == "":
120120
folder = transfer_benchmarks_folder
121121

122122
global I
123123
global output_folder_path
124124

125-
cmd = f"go test {folder} -test.run={test} -test.v -test.timeout 0 -bits='32' -num_inputs='2' -num_outputs='2' -workers='NumCPU' -duration='10s' -setup_samples=128 {params}"
125+
cmd = f"go test {folder} -test.run={benchName} -test.v -test.timeout 0 -bits='32' -num_inputs='2' -num_outputs='2' -workers='NumCPU' -duration='10s' -setup_samples=128 {params}"
126126
print(f"{I} Running: {cmd}")
127127
I = I+1
128128

@@ -135,7 +135,7 @@ def run_and_parse_parallel_metrics(test, params, folder=transfer_benchmarks_fold
135135
check=True
136136
)
137137

138-
log_file_path = os.path.join(output_folder_path, test+".log")
138+
log_file_path = os.path.join(output_folder_path, benchName+".log")
139139
if not os.path.exists(log_file_path):
140140
with open(log_file_path, "w", encoding="utf-8") as f:
141141
f.write(result.stdout)
@@ -151,8 +151,6 @@ def run_and_parse_parallel_metrics(test, params, folder=transfer_benchmarks_fold
151151
if not name_match:
152152
raise ValueError("Could not extract test name")
153153

154-
name = name_match.group(1)
155-
156154
# --- Unit conversion ---
157155
time_mult = {
158156
"ns": 1,
@@ -181,6 +179,7 @@ def to_ns(value: float, unit: str) -> int:
181179
avg_val, avg_unit = avg_lat_re.search(output).groups()
182180
max_val, max_unit = max_lat_re.search(output).groups()
183181

182+
name = benchName
184183
return {
185184
f"{name} real throughput": real_tp,
186185
f"{name} pure throughput": pure_tp,
@@ -243,4 +242,3 @@ def append_dict_as_row(filename: str, data: dict):
243242
dst = os.path.join(output_folder_path, "benchmark_results.csv")
244243
if os.path.exists(src) and not os.path.exists(dst):
245244
shutil.copy(src, dst)
246-

0 commit comments

Comments
 (0)