AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adding agent-index (#189)

Browse Source 
* Adding agent-index

* review changes

* review comments addressed

* minor fix

* fix CI failure

* review comments

* Fix agent index test and address review comments

* Build Fixes

---------

Co-authored-by: Benjamin Welton <bewelton@amd.com>
This commit is contained in:
Nagaraj, Sriraksha
2025-03-14 02:51:32 -05:00
committed by GitHub
parent 2fe63d873e
commit c30bb7cbda
32 changed files with 604 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/rocprofv3/pc-sampling/host-trap/transpose-multiple-agents/validate.py
+5 -1
View File
@@ -49,7 +49,11 @@ def test_multi_agent_support(
# Determine the agent on which sample was generated
samples_df["Agent_Id"] = (
samples_df["Dispatch_Id"]
.map(input_kernel_trace_csv.set_index("Dispatch_Id")["Agent_Id"])
.map(
input_kernel_trace_csv.set_index("Dispatch_Id")["Agent_Id"]
.str.split(" ")
.str[1]
)
.astype(np.uint64)
)
sampled_agents = samples_df["Agent_Id"].unique()