SWDEV-320640 - Fixes to return right error codes in hipStreamUpdateCaptureDependencies() for negative scenarios

Change-Id: I3c4c54f8955fa9bb17eb595544005fe9f0e8a5f0


[ROCm/clr commit: 5a057af564]
This commit is contained in:
Sourabh Betigeri
2022-05-23 17:11:28 -07:00
committed by Sourabh Betigeri
vanhempi de626141ee
commit ea9847c1fd
+4 -1
Näytä tiedosto
@@ -1321,7 +1321,10 @@ hipError_t hipStreamUpdateCaptureDependencies(hipStream_t stream, hipGraphNode_t
if (s->GetCaptureStatus() == hipStreamCaptureStatusNone) {
HIP_RETURN(hipErrorIllegalState);
}
if (numDependencies > 0 && dependencies == nullptr) {
if (numDependencies > 0 && dependencies == nullptr ||
(numDependencies > s->GetLastCapturedNodes().size()) ||
(flags != 0 && (flags != hipStreamAddCaptureDependencies ||
flags != hipStreamSetCaptureDependencies))) {
HIP_RETURN(hipErrorInvalidValue);
}
std::vector<hipGraphNode_t> depNodes;