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)",