update

Author: shink

.ci/benchmark.py

@@ -1,14 +1,23 @@
 import argparse
 import os
-import sys
+
 from src.benchmark.utils import read_metrics, to_markdown_table
 
 
 def parse_args():
     parser = argparse.ArgumentParser()
-    parser.add_argument("--path", type=str, required=True, help="Report path.")
-    parser.add_argument("--write-gh-job-summary", action="store_true", help="Write to GitHub job summary.")
-    parser.add_argument("--update-readme", action="store_true", help="Update statistics report in README.md.")
+    parser.add_argument(
+        "--path", type=str, required=True, help="The path of benchmark report."
+    )
+    parser.add_argument(
+        "--output", type=str, required=False, help="The output path of the Markdown report."
+    )
+    parser.add_argument(
+        "--write-gh-job-summary", action="store_true", help="Write to GitHub job summary."
+    )
+    parser.add_argument(
+        "--update-readme", action="store_true", help="Update statistics report in README.md."
+    )
     return parser.parse_args()
 
 
@@ -18,6 +27,11 @@ def generate_report(path: str):
     return html_table
 
 
+def save_output_report(path: str, report):
+    with open(path, "w") as f:
+        f.write(report)
+
+
 def write_job_summary(report):
     summary_path = os.environ["GITHUB_STEP_SUMMARY"]
     with open(summary_path, "a") as f:
@@ -55,6 +69,12 @@ def update_readme(report):
     # Generate statistics report
     report = generate_report(args.path)
 
+    # Output to markdown report
+    if args.output:
+        save_output_report(args.output, report)
+    else:
+        print(report)
+
     # Write to workflow job summary
     if args.write_gh_job_summary:
         write_job_summary(report)

.github/workflows/_ascend_npu_benchmark.yml

@@ -32,7 +32,7 @@ defaults:
     shell: bash -el {0}
 
 jobs:
-  test:
+  benchmark:
     name: run benchmarks for torch_npu
     runs-on: ${{ inputs.runner }}
     env: