NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-05-12 01:10:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add --no-color flag to nvbench_compare.py which can be used for github issues and PRs

Browse Source
This commit is contained in:
Nader Al Awar
2026-04-01 18:27:54 -04:00
committed by GitHub
parent 836a6c12f4 7a68e53df0
commit 488173a242
1 changed files with 32 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
python/scripts/nvbench_compare.py
View File

@@ -285,6 +285,7 @@ def compare_benches(
dark,
axis_filters,
benchmark_filters,
no_color,
):
if plot_along:
import matplotlib.pyplot as plt
@@ -452,19 +453,31 @@ def compare_benches(
if not min_noise:
unknown_count += 1
status_label = "????"
status = Fore.YELLOW + status_label + Fore.RESET
if no_color:
status = f"\U0001f7e1 {status_label}"
else:
status = f"{Fore.YELLOW}{status_label}{Fore.RESET}"
elif abs(frac_diff) <= min_noise:
pass_count += 1
status_label = "SAME"
status = Fore.BLUE + status_label + Fore.RESET
if no_color:
status = f"\U0001f535 {status_label}"
else:
status = f"{Fore.BLUE}{status_label}{Fore.RESET}"
elif diff < 0:
failure_count += 1
status_label = "FAST"
status = Fore.GREEN + status_label + Fore.RESET
if no_color:
status = f"\U0001f7e2 {status_label}"
else:
status = f"{Fore.GREEN}{status_label}{Fore.RESET}"
else:
failure_count += 1
status_label = "SLOW"
status = Fore.RED + status_label + Fore.RESET
if no_color:
status = f"\U0001f534 {status_label}"
else:
status = f"{Fore.RED}{status_label}{Fore.RESET}"
if abs(frac_diff) >= threshold:
row.append(format_duration(ref_time))
@@ -594,6 +607,12 @@ def main():
action="store_true",
help="Use dark theme (black background, white text)",
)
parser.add_argument(
"--no-color",
dest="no_color",
action="store_true",
help="Use emoji instead of ANSI color codes (useful for GitHub issues/PRs)",
)
parser.add_argument(
"-a",
"--axis",
@@ -650,11 +669,14 @@ def main():
all_cmp_devices = cmp_root["devices"]
if ref_root["devices"] != cmp_root["devices"]:
print(
(Fore.YELLOW if args.ignore_devices else Fore.RED)
+ "Device sections do not match:"
+ Fore.RESET
)
if args.no_color:
print("Device sections do not match:")
else:
print(
(Fore.YELLOW if args.ignore_devices else Fore.RED)
+ "Device sections do not match:"
+ Fore.RESET
)
print(
jsondiff.diff(
ref_root["devices"], cmp_root["devices"], syntax="symmetric"
@@ -672,6 +694,7 @@ def main():
args.dark,
axis_filters,
args.benchmark,
args.no_color,
)
print("# Summary\n")