From e776dae3a3bf0f1d54b694ee37d0e6dcd05595ea Mon Sep 17 00:00:00 2001
From: Xuehai Pan <XuehaiPan@pku.edu.cn>
Date: Thu, 3 Aug 2023 21:27:28 +0800
Subject: [PATCH] fix(api/device): remove extra timestamp margin for GPU
 process utilization

---
 nvitop/api/device.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nvitop/api/device.py b/nvitop/api/device.py
index 98166fb..7491482 100644
--- a/nvitop/api/device.py
+++ b/nvitop/api/device.py
@@ -1703,8 +1703,8 @@ class Device:  # pylint: disable=too-many-instance-attributes,too-many-public-me
                 self._timestamp,
                 default=(),
             )
-            self._timestamp = max(min((s.timeStamp for s in samples), default=0) - 2_000_000, 0)
-            for s in samples:
+            self._timestamp = min((s.timeStamp for s in samples), default=self._timestamp)
+            for s in sorted(samples, key=lambda s: s.timeStamp):
                 try:
                     processes[s.pid].set_gpu_utilization(s.smUtil, s.memUtil, s.encUtil, s.decUtil)
                 except KeyError: