Skip to content

Commit f8ba103

Browse files
committed
final
Signed-off-by: Yang Wang <[email protected]>
1 parent e27779e commit f8ba103

File tree

2 files changed

+22
-10
lines changed

2 files changed

+22
-10
lines changed

.ci/scripts/benchmark_tooling/README.md

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,21 +30,24 @@ python3 .ci/scripts/benchmark_tooling/get_benchmark_analysis_data.py \
3030
- `--startTime`: Start time in ISO format (e.g., "2025-06-11T00:00:00") (required)
3131
- `--endTime`: End time in ISO format (e.g., "2025-06-17T18:00:00") (required)
3232
- `--env`: Choose environment ("local" or "prod", default: "prod")
33-
- `--not-silent`: Show processing logs (default: only show results & minimum logging)
33+
- `--no-silent`: Show processing logs (default: only show results & minimum logging)
34+
- `print-all-table-info`: show all cleaned table infos, this helps user to pick the correct format of filters
3435

3536
##### Output Options:
3637
- `--outputType`: Choose output format (default: "print")
3738
- `print`: Display results in console
3839
- `json`: Generate JSON file
3940
- `df`: Display results in DataFrame format
40-
- `excel`: Generate Excel files with multiple sheets
41-
- `csv`: Generate CSV files in separate folders
41+
- `excel`: Generate Excel files with multiple sheets, the field in first row and first column contains the json string of the raw metadata
42+
- `csv`: Generate CSV files in separate folders, the field in first row and first column contains the json string of the raw metadata
4243
- `--outputDir`: Directory to save output files (default: current directory)
4344

4445
##### Filtering Options:
46+
Notice, the filter needs full name matchings with correct format, to see all the options of the filter choices, please run the script with `--print-all-table-info`, and pay attention to section `Full list of table info from HUD API` with the field 'info', which contains normalized data we use to filter records from the original metadata 'groupInfo'.
47+
4548
- `--devices`: Filter by specific device names (e.g., "samsung-galaxy-s22-5g", "samsung-galaxy-s22plus-5g")
46-
- `--backends`: Filter by specific backend names
47-
- `--models`: Filter by specific model names
49+
- `--backends`: Filter by specific backend names (e.g., "qnn-q8" , ""llama3-spinquan)
50+
- `--models`: Filter by specific model names (e.g "mv3" "meta-llama-llama-3.2-1b-instruct-qlora-int4-eo8")
4851

4952
#### Working with Output Files
5053

.ci/scripts/benchmark_tooling/get_benchmark_analysis_data.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,7 @@ def _get_base_url(self) -> str:
456456
}
457457
return base_urls[self.env]
458458

459-
def print_all_groups_info(self) -> None:
459+
def print_all_table_info(self) -> None:
460460
"""
461461
Print all benchmark table group info found in the data.
462462
Separates results by category and displays counts.
@@ -670,6 +670,13 @@ def argparsers():
670670
default=True,
671671
help="Allow output (disable silent mode)",
672672
)
673+
674+
parser.add_argument(
675+
"--print-all-table-info",
676+
action="store_true",
677+
help="Print all table info for debugging",
678+
)
679+
673680
# Options for generate_data
674681
parser.add_argument(
675682
"--outputType",
@@ -689,9 +696,11 @@ def argparsers():
689696
parser.add_argument(
690697
"--devices",
691698
nargs="+",
692-
help="Filter results by device names (e.g. --devices samsung-galaxy-s22-5g)(OR logic)",
699+
help="Filter results by one or more device names (e.g. --devices samsung-galaxy-s22-5g)(OR logic)",
700+
)
701+
parser.add_argument(
702+
"--models", nargs="+", help="Filter by one or more models (OR logic)"
693703
)
694-
parser.add_argument("--models", nargs="+", help="Filter by models (OR logic)")
695704
return parser.parse_args()
696705

697706

@@ -705,6 +714,6 @@ def argparsers():
705714
models=args.models, backends=args.backends, devices=args.devices
706715
),
707716
)
708-
if not args.silent:
709-
fetcher.print_all_groups_info()
717+
if args.print_all_table_info:
718+
fetcher.print_all_table_info()
710719
fetcher.output_data(args.outputType, args.outputDir)

0 commit comments

Comments
 (0)