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

SWDEV-445084 - Add DEBUG_CLR_GRAPH_PACKET_CAPTURE support for hipGraphInstantiateWithFlags/Params

Browse Source 
Change-Id: I5096b4c8d73d1faf972dfd23ab86a53d888946c4


[ROCm/clr commit: d6bc40e822]
This commit is contained in:
Anusha GodavarthySurya
2024-02-07 17:37:24 +00:00
committed by Anusha Godavarthy Surya
parent 3cad2e16f6
commit fe3c0a3fdc
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_graph.cpp
-4
View File
@@ -1262,10 +1262,6 @@ hipError_t hipGraphInstantiate(hipGraphExec_t* pGraphExec, hipGraph_t graph,
hipError_t status = ihipGraphInstantiate(&ge, reinterpret_cast<hip::Graph*>(graph));
if (status == hipSuccess) {
*pGraphExec = reinterpret_cast<hipGraphExec_t>(ge);
if (DEBUG_CLR_GRAPH_PACKET_CAPTURE) {
// For graph nodes capture AQL packets to dispatch them directly during graph launch.
status = ge->CaptureAQLPackets();
}
}
HIP_RETURN(status);
}
projects/clr/hipamd/src/hip_graph_internal.cpp
+8 -1
View File
@@ -344,11 +344,18 @@ hipError_t GraphExec::Init() {
for (auto& node : topoOrder_) {
status = node->GetNumParallelStreams(min_num_streams);
if(status != hipSuccess) {
if (status != hipSuccess) {
return status;
}
}
status = CreateStreams(parallelLists_.size() - 1 + min_num_streams);
if (status != hipSuccess) {
return status;
}
if (DEBUG_CLR_GRAPH_PACKET_CAPTURE) {
// For graph nodes capture AQL packets to dispatch them directly during graph launch.
status = CaptureAQLPackets();
}
return status;
}