From ee7644932b213e0de5ec8a8a8b210bdc2ccc7fed Mon Sep 17 00:00:00 2001 From: qazal <77887910+Qazalin@users.noreply.github.com> Date: Thu, 23 Apr 2026 22:13:16 +0300 Subject: [PATCH] viz/cli: -t default number (#15894) * viz/cli: accept one path argument * -t default * hm * only the -t change --- tinygrad/viz/cli.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tinygrad/viz/cli.py b/tinygrad/viz/cli.py index b8078714e4..e678349e98 100755 --- a/tinygrad/viz/cli.py +++ b/tinygrad/viz/cli.py @@ -203,8 +203,7 @@ def get_arg_parser() -> argparse.ArgumentParser: g_opts = parser.add_argument_group("optional args") g_opts.add_argument("-s", "--src", type=str, default=None, metavar="NAME", help="Select a data source (default: list all sources)") g_opts.add_argument("-i", "--item", type=str, default=None, metavar="NAME", help="Select an item within the source (default: list all items)") - g_opts.add_argument("-t", "--top", type=int, default=None, metavar="COUNT", - help="Number of top kernels to aggregate (default: do not aggregate, set -1 to aggregate all)") + g_opts.add_argument("-t", "--top", nargs="?", type=int, const=20, metavar="COUNT", help="Aggregate top kernels (optional count, default 20)") g_opts.add_argument("--profile-path", type=pathlib.Path, metavar="PATH", help="Optional path to profile.pkl (default: latest profile)", default=pathlib.Path(temp("profile.pkl", append_user=True))) g_opts.add_argument("--rewrites-path", type=pathlib.Path, metavar="PATH", help="Optional path to rewrites.pkl (default: latest rewrites)",