From d35382f4fb20a1062e0d725e7887a81c57e4cffb Mon Sep 17 00:00:00 2001 From: wuyulong11 Date: Tue, 5 Sep 2023 22:09:58 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BF=AE=E6=94=B9=E4=BF=A1=E6=81=AF?= =?UTF-8?q?=E3=80=91=20=E3=80=90tbplugin=E3=80=91=E3=80=90=E9=97=AE?= =?UTF-8?q?=E9=A2=98=E5=8D=95=E3=80=91=E4=BF=AE=E6=94=B9Diff=20View?= =?UTF-8?q?=E4=B8=AD=E8=A1=A8=E6=A0=BC=E5=86=85=E5=B0=8F=E6=95=B0=E4=BD=8D?= =?UTF-8?q?=E8=BF=87=E9=95=BF=E9=97=AE=E9=A2=98=20=E3=80=90=E4=BF=AE?= =?UTF-8?q?=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 --- .../tb_plugin/fe/src/components/DiffOverview.tsx | 8 ++++---- .../torch_tb_profiler/profiler/diffrun/operator.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/plugins/tensorboard-plugins/tb_plugin/fe/src/components/DiffOverview.tsx b/plugins/tensorboard-plugins/tb_plugin/fe/src/components/DiffOverview.tsx index 0fc67e7121..05c51fb2ff 100644 --- a/plugins/tensorboard-plugins/tb_plugin/fe/src/components/DiffOverview.tsx +++ b/plugins/tensorboard-plugins/tb_plugin/fe/src/components/DiffOverview.tsx @@ -566,7 +566,7 @@ export const DiffOverview: React.FC = (props: IProps) => { baselineHostDuration: left.host_duration, expHostDuration: right.host_duration, - deltaHostDuration: right.host_duration - left.host_duration, + deltaHostDuration: parseFloat((right.host_duration - left.host_duration).toFixed(3)), deltaHostDurationPercentNumber: deltaHostDurationPercentNumber, deltaHostDurationPercent: toPercentString( deltaHostDurationPercentNumber @@ -575,7 +575,7 @@ export const DiffOverview: React.FC = (props: IProps) => { baselineSelfHostDuration: left.self_host_duration, expSelfHostDuration: right.self_host_duration, deltaSelfHostDuration: - right.self_host_duration - left.self_host_duration, + parseFloat((right.self_host_duration - left.self_host_duration).toFixed(3)), deltaSelfHostDurationPercentNumber: deltaSelfHostDurationPercentNumber, deltaSelfHostDurationPercent: toPercentString( deltaSelfHostDurationPercentNumber @@ -583,7 +583,7 @@ export const DiffOverview: React.FC = (props: IProps) => { baselineDeviceDuration: left.device_duration, expDeviceDuration: right.device_duration, - deltaDeviceDuration: right.device_duration - left.device_duration, + deltaDeviceDuration: parseFloat((right.device_duration - left.device_duration).toFixed(3)), deltaDeviceDurationPercentNumber: deltaDeviceDurationPercentNumber, deltaDeviceDurationPercent: toPercentString( deltaDeviceDurationPercentNumber @@ -592,7 +592,7 @@ export const DiffOverview: React.FC = (props: IProps) => { baselineSelfDeviceDuration: left.self_device_duration, expSelfDeviceDuration: right.self_device_duration, deltaSelfDeviceDuration: - right.self_device_duration - left.self_device_duration, + parseFloat((right.self_device_duration - left.self_device_duration).toFixed(3)), deltaSelfDeviceDurationPercentNumber: deltaSelfDeviceDurationPercentNumber, deltaSelfDeviceDurationPercent: toPercentString( deltaSelfDeviceDurationPercentNumber diff --git a/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/diffrun/operator.py b/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/diffrun/operator.py index ca0c1f7d4a..4f6afdb7ec 100644 --- a/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/diffrun/operator.py +++ b/plugins/tensorboard-plugins/tb_plugin/torch_tb_profiler/profiler/diffrun/operator.py @@ -35,10 +35,10 @@ class Operator(metaclass=ABCMeta): yield OpAgg( agg.name, agg.calls, - agg.host_duration, - agg.device_duration, - agg.self_host_duration, - agg.self_device_duration) + round(agg.host_duration, 3), + round(agg.device_duration, 3), + round(agg.self_host_duration, 3), + round(agg.self_device_duration, 3)) def get_operators_and_kernels(self) -> Tuple[List[OperatorNode], List[DeviceNode]]: return [], [] -- Gitee