[Misc] fix the agent_id field (#297)

* fix the agent_id field

* Fix shebang

---------

Co-authored-by: Jonathan R. Madsen <jonathanrmadsen@gmail.com>

[ROCm/rocprofiler-sdk commit: 35069a9d06]
This commit is contained in:
Nagaraj, Sriraksha
2025-04-27 19:56:45 -05:00
committed by GitHub
parent d10a9d855a
commit a83ee16d51
7 changed files with 127 additions and 34 deletions
@@ -35,11 +35,16 @@ kernel_list = sorted(
counters_list = ["SQ_WAVES", "GRBM_GUI_ACTIVE"]
def test_validate_counter_collection_pmc1(input_data: pd.DataFrame):
def test_validate_counter_collection_pmc1(
input_data: pd.DataFrame, retain_agent_prefix_flag
):
df = input_data
assert not df.empty
df_agent_id = df["Agent_Id"].str.split(" ").str[-1]
assert (df_agent_id.astype(int).values >= 0).all()
if retain_agent_prefix_flag == "true":
df_agent_id = df["Agent_Id"].str.split(" ").str[-1]
assert (df_agent_id.astype(int).values >= 0).all()
else:
assert (df["Agent_Id"].astype(int).values > 0).all()
assert (df["Queue_Id"].astype(int).values > 0).all()
assert len(df["Kernel_Name"]) > 0