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

SWDEV-315033,SWDEV-315028 - adding null checks for graph

Browse Source 
Change-Id: I9cbea5ea9ff59d7960ae11dfd73d5a744981495f


[ROCm/clr commit: 6a48e430d0]
This commit is contained in:
Payam
2022-03-10 21:13:20 -05:00
committed by Payam Ghafari
parent 6a849eb690
commit 566cd820ac
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_graph.cpp
+6 -1
View File
@@ -1175,9 +1175,14 @@ hipError_t hipGraphAddDependencies(hipGraph_t graph, const hipGraphNode_t* from,
}
for (size_t i = 0; i < numDependencies; i++) {
// When the same node is specified for both from and to
if (from[i] == to[i]) {
if (from[i] == nullptr || to[i] == nullptr || from[i] == to[i] ||
// making sure the nodes blong to the graph
to[i]->GetParentGraph() != graph || from[i]->GetParentGraph() != graph) {
HIP_RETURN(hipErrorInvalidValue);
}
}
for (size_t i = 0; i < numDependencies; i++) {
// When the same edge added from->to return invalid value
const std::vector<Node>& edges = from[i]->GetEdges();
for (auto edge : edges) {