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 0fc67e71215978674b0afe0477ab982145afded7..05c51fb2ffe02c41f001372ca4972c7f6b961c05 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 ca0c1f7d4aafbd78228f3a5e70653d48903a97d4..4f6afdb7ec6dc48fbaeccfece6b9983bbceeaf30 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 [], []