add help desc

neuralmagic · Aug 5, 2024 · 1ebec11 · 1ebec11
1 parent cd81bcc
commit 1ebec11
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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",