From a921884a72cc2fdf4241b2463a3b39a2a493e801 Mon Sep 17 00:00:00 2001 From: wuyulong11 Date: Mon, 10 Jul 2023 09:59:37 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BF=AE=E6=94=B9=E8=AF=B4=E6=98=8E?= =?UTF-8?q?=E3=80=91=20=E6=95=B0=E6=8D=AE=E7=9B=AE=E5=BD=95=E5=91=BD?= =?UTF-8?q?=E5=90=8D=E6=B3=A8=E9=87=8A=E4=BF=AE=E6=94=B9=20=E3=80=90?= =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BA=BA=E3=80=91=20wuyulong=2030031080?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../profiling/tb_plugin/torch_tb_profiler/plugin.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/plugin.py b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/plugin.py index 777d778ca8..fcfaf7860d 100644 --- a/tb_plugins/profiling/tb_plugin/torch_tb_profiler/plugin.py +++ b/tb_plugins/profiling/tb_plugin/torch_tb_profiler/plugin.py @@ -507,7 +507,7 @@ class TorchProfilerPlugin(base_plugin.TBPlugin): /run2 /[worker1].pt.trace.json A directory is considered to be an ascend run if it satisfies the following two conditions: - 1.At least one subdirectory with the name in this format: {worker_span}. + 1.At least one subdirectory with the name in this format: {worker}_{span}_ascend_pt. 2.The subdirectory in condition 1 has a 'ASCEND_PROFILER_OUTPUT' subdirectory which contains at least one of these 4 kind of files: [ @@ -518,16 +518,19 @@ class TorchProfilerPlugin(base_plugin.TBPlugin): ] E.g. there are 2 runs: run1, run2 /run1 - /[worker1]_[span1] + /[worker1]_[span1]_ascend_pt /ASCEND_PROFILER_OUTPUT /trace_view.json /kernel_details.csv - /[worker2]_[span1] + /operator_details.csv + /operator_memory.csv + /memory_record.csv + /[worker2]_[span1]_ascend_pt /ASCEND_PROFILER_OUTPUT /trace_view.json /operator_details.csv /run2 - /[worker1]_[span1] + /[worker1]_[span1]_ascend_pt /ASCEND_PROFILER_OUTPUT /memory_record.csv /operator_memory.csv -- Gitee