diff --git a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py index eff3f421de03d49e2f0636175380b7d276a4a9d2..e2471ffe4f104d65dec25e37f85a7ecea9922957 100644 --- a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py +++ b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/profiler/run_generator.py @@ -145,11 +145,11 @@ class RunGenerator(object): @staticmethod def _check_overlap_data(title): - # csv: step / compute time / communication_not_overlap / overlap / communication / free time + # csv: step / compute time / communication_not_overlap / overlap / communication / other time length = len(title) if length < 5: return - key = ["compute time", "overlapped time", "communication time not overlapped", "free time"] + key = ["compute time", "overlapped time", "communication time not overlapped", "other time"] get_key = list() for j in key: for i in range(length): @@ -1150,7 +1150,7 @@ class DistributedRunGenerator(object): steps = data.step_to_overlap for k, v in steps.items(): steps_to_overlap.setdefault(k, OrderedDict()) - # v: computation / overlap / communication_not_overlap / free time + # v: computation / overlap / communication_not_overlap / other # steps_to_overlap: computation_not_overlap / overlap / communication_not_overlap / other steps_to_overlap[k][data.worker] = list([v[0] - v[1], v[1], v[2], v[3]]) steps_to_overlap['all'][data.worker] = [