From 1ebec11f21a0d0de4b8ea1a055636fa43218ada8 Mon Sep 17 00:00:00 2001 From: Varun Sundar Rabindranath Date: Mon, 5 Aug 2024 13:19:21 +0000 Subject: [PATCH] add help desc --- neuralmagic/tools/profiler/visualize_trace.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/neuralmagic/tools/profiler/visualize_trace.py b/neuralmagic/tools/profiler/visualize_trace.py index 8d9ff24916b8a..f4b1449281f69 100644 --- a/neuralmagic/tools/profiler/visualize_trace.py +++ b/neuralmagic/tools/profiler/visualize_trace.py @@ -397,7 +397,10 @@ def make_plot_title_suffix(profile_json: dict) -> str: help="Only graph the top `top_k` entries by time.") parser.add_argument("--fold-json-node", nargs='+', - default=['Sampler', 'LogitsProcessor']) + default=['Sampler', 'LogitsProcessor'], + help='Do not plot the children of these nodes. Let, \ + the node represent the aggregate of all its \ + children') parser.add_argument("--plot-metric", type=str, default="cuda_time_ms",