From b09e5a992a987a3ee971be30c4e399df7e5c7509 Mon Sep 17 00:00:00 2001
From: cabbage <chenxiaofeng45@huawei.com>
Date: Thu, 19 Oct 2023 21:21:39 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8D=95=E4=BD=8D=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../tb_plugin/torch_tb_profiler/profiler/run_generator.py   | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/run_generator.py b/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/run_generator.py
index ed9b099ccba..e9ce006f223 100644
--- a/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/run_generator.py
+++ b/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/run_generator.py
@@ -1265,11 +1265,11 @@ class DistributedRunGenerator(object):
             row = [
                 op,
                 stats[0],
-                stats[1],
+                stats[1] * 1024 * 1024,
                 round(stats[1] * 1024 * 1024 / stats[0]),  # 1MB = 1024 * 1024 bytes
-                stats[2],
+                stats[2] * 1000,
                 round(stats[2] * 1000 / stats[0]),  # 1ms = 1000us
-                stats[3],
+                stats[3] * 1000,
                 round(stats[3] * 1000 / stats[0])  # 1ms = 1000us
             ]
             table['rows'].append(row)
-- 
Gitee