Push values from pi_ls benchmark too

microsoft · Jun 11, 2024 · 507e098 · 507e098
1 parent 8c07653
commit 507e098
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 7 deletions.
diff --git a/.github/workflows/bencher.yml b/.github/workflows/bencher.yml
@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - main
+      - bencher_improvements
   workflow_dispatch:
 
 jobs:

diff --git a/tests/infra/basicperf.py b/tests/infra/basicperf.py
@@ -378,7 +378,6 @@ def run(args):
                 for remote_client in clients:
                     remote_client.stop()
 
-                metrics = []
                 if not args.stop_primary_after_s:
                     primary, _ = network.find_primary()
                     with primary.client() as nc:
@@ -590,12 +589,10 @@ def table():
                     )
 
                 bf = infra.bencher.Bencher()
-                metrics.append(infra.bencher.Throughput(round(throughput, 1)))
-                for metric in metrics:
-                    bf.set(
-                        args.label,
-                        metric,
-                    )
+                bf.set(
+                    args.label,
+                    infra.bencher.Throughput(round(throughput, 1)),
+                )
 
             except Exception as e:
                 LOG.error(f"Stopping clients due to exception: {e}")

diff --git a/tests/infra/piccolo_driver.py b/tests/infra/piccolo_driver.py
@@ -210,6 +210,14 @@ def run(get_command, args):
                     perf_result = round(len(df_sends.index) / time_spent, 1)
                     LOG.success(f"{args.label}/{remote_client.name}: {perf_result}")
 
+                    # Throughput from only one client, preserved for legacy reason
+                    # see basicperf.py for a better, cross-client approach.
+                    bf = infra.bencher.Bencher()
+                    bf.set(
+                        args.label,
+                        infra.bencher.Throughput(perf_result),
+                    )
+
                 primary, _ = network.find_primary()
                 with primary.client() as nc:
                     r = nc.get("/node/memory")