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

SWDEV-386926 - Added check for correct node type in hipGraphMem[Alloc/Free]NodeGetParams

Browse Source 
Change-Id: Ic8446210a683d1329391288d0ed99d835c5aa140
This commit is contained in:
Ioannis Assiouras
2023-03-07 13:53:44 +00:00
parent 9e1e3035ec
commit 58e05898d1
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_graph.cpp
+4 -2
View File
@@ -2179,7 +2179,8 @@ hipError_t hipGraphAddMemAllocNode(hipGraphNode_t* pGraphNode, hipGraph_t graph,
// ================================================================================================
hipError_t hipGraphMemAllocNodeGetParams(hipGraphNode_t node, hipMemAllocNodeParams* pNodeParams) {
HIP_INIT_API(hipGraphMemAllocNodeGetParams, node, pNodeParams);
if (node == nullptr || pNodeParams == nullptr || !hipGraphNode::isNodeValid(node)) {
if (node == nullptr || pNodeParams == nullptr || !hipGraphNode::isNodeValid(node)
|| node->GetType() != hipGraphNodeTypeMemAlloc) {
HIP_RETURN(hipErrorInvalidValue);
}
reinterpret_cast<hipGraphMemAllocNode*>(node)->GetParams(pNodeParams);
@@ -2214,7 +2215,8 @@ hipError_t hipGraphAddMemFreeNode(hipGraphNode_t* pGraphNode, hipGraph_t graph,
// ================================================================================================
hipError_t hipGraphMemFreeNodeGetParams(hipGraphNode_t node, void* dev_ptr) {
HIP_INIT_API(hipGraphMemFreeNodeGetParams, node, dev_ptr);
if (node == nullptr || dev_ptr == nullptr || !hipGraphNode::isNodeValid(node)) {
if (node == nullptr || dev_ptr == nullptr || !hipGraphNode::isNodeValid(node)
|| node->GetType() != hipGraphNodeTypeMemFree) {
HIP_RETURN(hipErrorInvalidValue);
}
reinterpret_cast<hipGraphMemFreeNode*>(node)->GetParams(reinterpret_cast<void**>(dev_ptr));