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 846c6db42940f3e94f1b19203a2d25bb8e5d22af..333af12b04029804b3c77ecdab2821d3108331fb 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 @@ -467,7 +467,6 @@ class RunGenerator(object): if ls[tag_type_idx] == 'PTA+GE': process_data.setdefault(device_type, {}).setdefault('Allocated', []).append( [time_column, float(ls[allocated_idx])]) - elif ls[tag_type_idx] == 'APP': process_data.setdefault(device_type, {}).setdefault('Reserved', []).append( [time_column, float(ls[reserved_idx])]) elif ls[tag_type_idx] in ('PTA', 'GE'):