NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-07-12 18:17:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move lazy load after cheap CLI arg validation

Browse Source 
Move `load_nvbench_compare_tooling()` after:
  - config resolution / --dump-config
  - filter parsing
  - device-filter parsing
  - input arity validation
This commit is contained in:
Oleksandr Pavlyk
2026-06-29 12:22:10 -05:00
parent d682a8b0e0
commit 68d638ca85
2 changed files with 44 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/scripts/nvbench_compare.py
View File

@@ -3349,12 +3349,6 @@ def main() -> int:
print(dump_comparison_config(comparison_preset, comparison_thresholds), end="")
return 0
try:
load_nvbench_compare_tooling()
except MissingToolingDependencyError as exc:
print(str(exc), file=sys.stderr)
return 1
try:
filter_plan = build_benchmark_filter_plan(args.filter_actions)
reference_device_filter = parse_device_filter(
@@ -3371,6 +3365,12 @@ def main() -> int:
parser.print_help()
return 1
try:
load_nvbench_compare_tooling()
except MissingToolingDependencyError as exc:
print(str(exc), file=sys.stderr)
return 1
bulk_debug_output = (
None
if args.bulk_debug_python is None