workload_generator/analysis_pytorch_trace.py (1 line): - line 61: # TODO: set group in default dp_group , get group in trace info