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

SWDEV-258579: hipMemcpy2D better error handling

Browse Source 
Change-Id: I09378d6e31da99451e22cbe9276a5ff1ad4a1840
This commit is contained in:
Sarbojit Sarkar
2020-11-20 02:46:37 -05:00
committed by Sarbojit Sarkar
parent ef415ec610
commit a08ddb4b53
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/rocclr/hip_memory.cpp
+15 -6
View File
@@ -1098,13 +1098,16 @@ hipError_t ihipMemcpyDtoD(void* srcDevice,
return hipErrorOutOfMemory; return hipErrorOutOfMemory;
} }
hipError_t status = hipSuccess;
command->enqueue(); command->enqueue();
if (!isAsync) { if (!isAsync) {
command->awaitCompletion(); if (!command->awaitCompletion()) {
status = hipErrorUnknown;
}
} }
command->release(); command->release();
return hipSuccess; return status;
} }
hipError_t ihipMemcpyDtoH(void* srcDevice, hipError_t ihipMemcpyDtoH(void* srcDevice,
@@ -1157,13 +1160,16 @@ hipError_t ihipMemcpyDtoH(void* srcDevice,
return hipErrorOutOfMemory; return hipErrorOutOfMemory;
} }
hipError_t status = hipSuccess;
command->enqueue(); command->enqueue();
if (!isAsync) { if (!isAsync) {
command->awaitCompletion(); if (!command->awaitCompletion()) {
status = hipErrorUnknown;
}
} }
command->release(); command->release();
return hipSuccess; return status;
} }
hipError_t ihipMemcpyHtoD(const void* srcHost, hipError_t ihipMemcpyHtoD(const void* srcHost,
@@ -1216,13 +1222,16 @@ hipError_t ihipMemcpyHtoD(const void* srcHost,
return hipErrorOutOfMemory; return hipErrorOutOfMemory;
} }
hipError_t status = hipSuccess;
command->enqueue(); command->enqueue();
if (!isAsync) { if (!isAsync) {
command->awaitCompletion(); if (!command->awaitCompletion()) {
status = hipErrorUnknown;
}
} }
command->release(); command->release();
return hipSuccess; return status;
} }
hipError_t ihipMemcpyHtoH(const void* srcHost, hipError_t ihipMemcpyHtoH(const void* srcHost,