diff --git a/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/consts.py b/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/consts.py index 460f508190223790ce056973c910166ea887b6f0..14ca1a8b0f6563b08a82ef90fdcf9b147c5008ff 100644 --- a/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/consts.py +++ b/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/consts.py @@ -29,7 +29,7 @@ WORKER_PATTERN = re.compile(r"""^(.*?) # worker name (?:\.gz)?$""", re.X) # optional .gz extension TRACE_PATTERN = re.compile(r"""^trace_view\.json(\.gz)?$""") -WORKER_SPAN_PATTERN = re.compile(r"""([^\\]*)_(\d+)_ascend_pt$""") +WORKER_SPAN_PATTERN = re.compile(r"""([^\\]*)_(\d+(\.\d+)?)_ascend_pt$""") NODE_PROCESS_PATTERN = re.compile(r"""^(.*)_(\d+)""") MONITOR_RUN_REFRESH_INTERNAL_IN_SECONDS = 10