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

SWDEV-457316 - Other graph can free mem alloc node and return invalid val only if there is double mem free node in all captured graphs.

Browse Source 
Change-Id: Icf12164bf0ecd171a4673ff4f384528e7671f944
This commit is contained in:
Jaydeep Patel
2024-05-20 17:37:23 +00:00
committed by Jaydeepkumar Patel
parent 4e77806ee5
commit 5c77e30b18
1 changed files with 8 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_graph.cpp
+8 -17
View File
@@ -2580,26 +2580,17 @@ hipError_t hipGraphAddMemFreeNode(hipGraphNode_t* pGraphNode, hipGraph_t graph,
HIP_RETURN(hipErrorInvalidValue);
}
hip::GraphNode* pNode;
bool AllocNodeFound = false;
hip::Graph* g = reinterpret_cast<hip::Graph*>(graph);
for (auto n : g->vertices_) {
if (n->GetType() == hipGraphNodeTypeMemAlloc) {
hipMemAllocNodeParams param = {};
reinterpret_cast<hip::GraphMemAllocNode*>(n)->GetParams(&param);
if (param.dptr == dev_ptr) {
AllocNodeFound = true;
}
} else if (n->GetType() == hipGraphNodeTypeMemFree) {
void* param;
reinterpret_cast<hip::GraphMemFreeNode*>(n)->GetParams(&param);
if (param == dev_ptr) {
HIP_RETURN(hipErrorInvalidValue);
for (auto it : hip::Graph::graphSet_) {
for (auto n : it->vertices_) {
if (n->GetType() == hipGraphNodeTypeMemFree) {
void* param;
reinterpret_cast<hip::GraphMemFreeNode*>(n)->GetParams(&param);
if (param == dev_ptr) {
HIP_RETURN(hipErrorInvalidValue);
}
}
}
}
if (!AllocNodeFound) {
HIP_RETURN(hipErrorInvalidValue);
}
auto status =
ihipGraphAddMemFreeNode(&pNode,
reinterpret_cast<hip::Graph*>(graph),