From 1b7445faa47e69c9c9c54ca9fb018d05eb24a7d4 Mon Sep 17 00:00:00 2001
From: Fei Zheng <44449748+feizheng10@users.noreply.github.com>
Date: Tue, 8 Jul 2025 06:52:01 -0600
Subject: [PATCH] PC Sampling: fix potential_end_offset (#792)

* fix potential_end_offset

* format code: unrelated
---
 src/rocprof_compute_soc/soc_base.py | 1 +
 src/utils/parser.py                 | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/rocprof_compute_soc/soc_base.py b/src/rocprof_compute_soc/soc_base.py
index cae37629b4..b1ff73254a 100644
--- a/src/rocprof_compute_soc/soc_base.py
+++ b/src/rocprof_compute_soc/soc_base.py
@@ -513,6 +513,7 @@ class OmniSoC_Base:
                 )
             )
             from rocprofv3_avail_module import avail
+
             avail.loadLibrary.libname = str(
                 Path(self.get_args().rocprofiler_sdk_library_path).parent.parent
                 / "libexec"
diff --git a/src/utils/parser.py b/src/utils/parser.py
index e99ec05416..164558e75c 100644
--- a/src/utils/parser.py
+++ b/src/utils/parser.py
@@ -1196,7 +1196,7 @@ def load_pc_sampling_data_per_kernel(
             next_index = i + 1
             if next_index < len(filtered_sorted_list):  # Ensure the next item exists
                 next_item = filtered_sorted_list[next_index]
-                kernel_info["potential_end_offset"] = item[
+                kernel_info["potential_end_offset"] = next_item[
                     "kernel_code_entry_byte_offset"
                 ]
             else: