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

Rename flag from markdown to no-color

Browse Source
This commit is contained in:
Nader Al Awar
2026-04-01 17:01:29 -05:00
parent 7e5e784855
commit 7a68e53df0
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
python/scripts/nvbench_compare.py
View File

@@ -285,7 +285,7 @@ def compare_benches(
dark,
axis_filters,
benchmark_filters,
markdown=False,
no_color,
):
if plot_along:
import matplotlib.pyplot as plt
@@ -453,28 +453,28 @@ def compare_benches(
if not min_noise:
unknown_count += 1
status_label = "????"
if markdown:
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"
if markdown:
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"
if markdown:
if no_color:
status = f"\U0001f7e2 {status_label}"
else:
status = f"{Fore.GREEN}{status_label}{Fore.RESET}"
else:
failure_count += 1
status_label = "SLOW"
if markdown:
if no_color:
status = f"\U0001f534 {status_label}"
else:
status = f"{Fore.RED}{status_label}{Fore.RESET}"
@@ -608,9 +608,8 @@ def main():
help="Use dark theme (black background, white text)",
)
parser.add_argument(
"--markdown",
dest="markdown",
default=False,
"--no-color",
dest="no_color",
action="store_true",
help="Use emoji instead of ANSI color codes (useful for GitHub issues/PRs)",
)
@@ -670,7 +669,7 @@ def main():
all_cmp_devices = cmp_root["devices"]
if ref_root["devices"] != cmp_root["devices"]:
if args.markdown:
if args.no_color:
print("Device sections do not match:")
else:
print(
@@ -695,7 +694,7 @@ def main():
args.dark,
axis_filters,
args.benchmark,
markdown=args.markdown,
args.no_color,
)
print("# Summary\n")