[single-node-performance] Add runner information in the output

aptos-labs · Oct 10, 2024 · f7e2acb · f7e2acb
1 parent 918b643
commit f7e2acb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/.github/workflows/workflow-run-execution-performance.yaml b/.github/workflows/workflow-run-execution-performance.yaml
@@ -98,7 +98,7 @@ jobs:
 
       - name: Run single node execution benchmark in performance build mode
         shell: bash
-        run: TABULATE_INSTALL=lib-only pip install tabulate && FLOW=${{ inputs.FLOW }} SOURCE=${{ inputs.SOURCE }} testsuite/single_node_performance.py
+        run: TABULATE_INSTALL=lib-only pip install tabulate && FLOW="${{ inputs.FLOW }}" SOURCE="${{ inputs.SOURCE }}" RUNNER_NAME="${{ inputs.RUNNER_NAME }}" testsuite/single_node_performance.py
         if: ${{ (inputs.IGNORE_TARGET_DETERMINATION || needs.test-target-determinator.outputs.run_execution_performance_test == 'true') }}
 
       - run: echo "Skipping single node execution performance! Unrelated changes detected."

diff --git a/testsuite/single_node_performance.py b/testsuite/single_node_performance.py
@@ -45,6 +45,8 @@ class Flow(Flag):
     print(f"Unrecogznied source {SOURCE}")
     exit(1)
 
+RUNNER_NAME = os.environ.get("RUNNER_NAME", default="none")
+
 DEFAULT_NUM_INIT_ACCOUNTS = (
     "100000000" if SELECTED_FLOW == Flow.MAINNET_LARGE_DB else "2000000"
 )
@@ -726,13 +728,9 @@ def print_table(
         print(
             json.dumps(
                 {
-                    "grep": "grep_json_single_node_perf"
-                    if SOURCE == "CI"
-                    else (
-                        "grep_json_single_node_perf_adhoc"
-                        if SOURCE == "ADHOC"
-                        else "grep_json_single_node_perf_local"
-                    ),
+                    "grep": "grep_json_single_node_perf",
+                    "source": SOURCE,
+                    "runner_name": RUNNER_NAME, 
                     "transaction_type": test.key.transaction_type,
                     "module_working_set_size": test.key.module_working_set_size,
                     "executor_type": test.key.executor_type,