AI/rocm-systems
2
0
Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

SWDEV-374383 - Check from and to if no of dependencies are more than 0, Check from and to node are belongs to graph pass in.

Bladeren bron 
Change-Id: I0cdafb497147f9ac7f84d40203a21e9eefd595a3


[ROCm/clr commit: bb486b04c6]
This commit is contained in:
Jaydeep Patel
2023-01-05 12:55:26 +00:00
gecommit door Jaydeepkumar Patel
bovenliggende 87db95fad7
commit d12bfbbb4d
1 gewijzigde bestanden met toevoegingen van 3 en 2 verwijderingen
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files
projects/clr/hipamd/src/hip_graph.cpp
+3 -2
Bestand weergeven
@@ -1615,11 +1615,12 @@ hipError_t hipStreamUpdateCaptureDependencies(hipStream_t stream, hipGraphNode_t
hipError_t hipGraphRemoveDependencies(hipGraph_t graph, const hipGraphNode_t* from, hipError_t hipGraphRemoveDependencies(hipGraph_t graph, const hipGraphNode_t* from,
const hipGraphNode_t* to, size_t numDependencies) { const hipGraphNode_t* to, size_t numDependencies) {
HIP_INIT_API(hipGraphRemoveDependencies, graph, from, to, numDependencies); HIP_INIT_API(hipGraphRemoveDependencies, graph, from, to, numDependencies);
if (graph == nullptr || from == nullptr || to == nullptr) { if (graph == nullptr || (numDependencies > 0 && (from == nullptr || to == nullptr))) {
HIP_RETURN(hipErrorInvalidValue); HIP_RETURN(hipErrorInvalidValue);
} }
for (size_t i = 0; i < numDependencies; i++) { for (size_t i = 0; i < numDependencies; i++) {
if (from[i]->RemoveEdge(to[i]) == false) { if (to[i]->GetParentGraph() != graph || from[i]->GetParentGraph() != graph ||
from[i]->RemoveEdge(to[i]) == false) {
HIP_RETURN(hipErrorInvalidValue); HIP_RETURN(hipErrorInvalidValue);
} }
} }