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

SWDEV-329829 - Fixes to return right error codes for respective negative scenarios

Browse Source 
Change-Id: Icb615a336bb846df1cf2811316d88a0e88381da8
This commit is contained in:
Sourabh Betigeri
2022-04-26 09:32:18 -07:00
committed by Sourabh Betigeri
parent 84c46047e5
commit 5fa514fea0
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_graph.cpp
+9 -2
View File
@@ -776,10 +776,17 @@ hipError_t hipStreamBeginCapture(hipStream_t stream, hipStreamCaptureMode mode)
hipError_t hipStreamEndCapture(hipStream_t stream, hipGraph_t* pGraph) {
HIP_INIT_API(hipStreamEndCapture, stream, pGraph);
if (pGraph == nullptr || stream == nullptr || !hip::isValid(stream)) {
HIP_RETURN(hipErrorInvalidValue);
if (pGraph == nullptr || stream == nullptr) {
HIP_RETURN(hipErrorIllegalState);
}
if (!hip::isValid(stream)) {
HIP_RETURN(hipErrorContextIsDestroyed);
}
hip::Stream* s = reinterpret_cast<hip::Stream*>(stream);
// Capture status must be active before endCapture can be initiated
if (s->GetCaptureStatus() == hipStreamCaptureStatusNone) {
HIP_RETURN(hipErrorIllegalState);
}
// Capture must be ended on the same stream in which it was initiated
if (!s->IsOriginStream()) {
HIP_RETURN(hipErrorStreamCaptureUnmatched);