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