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 e32f556503de840c4fe49627866c80ea33629922..b4c7680214b38038576a7f8b0940a8e41c74fb73 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 @@ -369,7 +369,7 @@ class RunGenerator(object): self.name_idx = idx elif column == 'Duration(us)': self.duration_idx = idx - elif column == 'Type': + elif column == 'Accelerator Core': self.core_type_idx = idx if column in display_columns: