diff --git a/test/profiler/test_profiler.py b/test/profiler/test_profiler.py
index 38e83d448fd..ca0481922e4 100644
--- a/test/profiler/test_profiler.py
+++ b/test/profiler/test_profiler.py
@@ -1723,9 +1723,12 @@ assert KinetoStepTracker.current_step() == initial_step + 2 * niters
             gpu_value = traceEvent.get("args", {}).get("labels", None)
             if gpu_value and "GPU" in gpu_value:
                 gpu_dict[gpu_value] += 1
+                # Max PID offset is 5M, based from pytorch/kineto include header:
+                # https://github.com/pytorch/kineto/blob/8681ff11e1fa54da39023076c5c43eddd87b7a8a/libkineto/include/output_base.h#L35
+                kExceedMaxPid = 5000000
                 self.assertTrue(
                     traceEvents[i + 1]["args"]["sort_index"]
-                    == 0x1000000 + int(gpu_value.split()[1])
+                    == kExceedMaxPid + int(gpu_value.split()[1])
                 )
 
         # TODO add checking gpu count if cpuOnly_ is true or not
diff --git a/third_party/kineto b/third_party/kineto
index be1317644c6..8681ff11e1f 160000
--- a/third_party/kineto
+++ b/third_party/kineto
@@ -1 +1 @@
-Subproject commit be1317644c68b4bfc4646024a6b221066e430031
+Subproject commit 8681ff11e1fa54da39023076c5c43eddd87b7a8a