SWDEV-254441 - Fixed review comments

Change-Id: Ica13f23638be04216d152821127b2b7316f96af0


[ROCm/clr commit: 35c8493237]
Tento commit je obsažen v:
Sarbojit Sarkar
2021-03-31 00:26:23 -07:00
rodič 4578277e13
revize 993f52cd8b
+18
Zobrazit soubor
@@ -204,6 +204,7 @@ hipError_t ihipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKin
return hipErrorInvalidValue;
}
amd::HostQueue* newQueue = nullptr;
amd::Command* command = nullptr;
amd::Command::EventWaitList waitList;
@@ -292,6 +293,9 @@ hipError_t ihipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKin
}
command = new amd::CopyMemoryCommand(*pQueue, CL_COMMAND_COPY_BUFFER, waitList,
*srcMemory->asBuffer(), *dstMemory->asBuffer(), sOffset, dOffset, sizeBytes);
if (pQueue != &queue) {
newQueue = pQueue;
}
}
}
if (command == nullptr) {
@@ -301,6 +305,20 @@ hipError_t ihipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKin
command->enqueue();
if (!isAsync) {
command->awaitCompletion();
} else {
if (newQueue != nullptr) {
amd::Command::EventWaitList waitList;
amd::Command* cmd = newQueue->getLastQueuedCommand(true);
if (cmd != nullptr) {
waitList.push_back(cmd);
amd::Command* depdentMarker = new amd::Marker(queue, true, waitList);
if (depdentMarker != nullptr) {
depdentMarker->enqueue();
depdentMarker->release();
}
cmd->release();
}
}
}
command->release();