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

SWDEV-316155 - Fix for P2P sample failure

Browse Source 
Change-Id: I95a70d44949c59319a9a09f84be771740c3f0521


[ROCm/clr commit: 845b923507]
This commit is contained in:
Sarbojit Sarkar
2021-12-22 11:52:24 +00:00
committed by Sarbojit Sarkar
parent 8929cc4544
commit 74269eb547
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_memory.cpp
+17
View File
@@ -357,6 +357,23 @@ hipError_t ihipMemcpyCommand(amd::Command*& command, void* dst, const void* src,
if ((srcMemory->getContext().devices()[0] != dstMemory->getContext().devices()[0]) &&
((srcMemory->getContext().devices().size() == 1) &&
(dstMemory->getContext().devices().size() == 1))) {
amd::HostQueue* currNullStream = hip::getNullStream();
if (currNullStream == &queue) {
if (queueDevice != srcMemory->getContext().devices()[0]) {
amd::HostQueue* pQueue = hip::getNullStream(srcMemory->getContext());
amd::Command* cmd = pQueue->getLastQueuedCommand(true);
if (cmd != nullptr) {
waitList.push_back(cmd);
}
}
if (queueDevice != dstMemory->getContext().devices()[0]) {
amd::HostQueue* pQueue = hip::getNullStream(dstMemory->getContext());
amd::Command* cmd = pQueue->getLastQueuedCommand(true);
if (cmd != nullptr) {
waitList.push_back(cmd);
}
}
}
command = new amd::CopyMemoryP2PCommand(queue, CL_COMMAND_COPY_BUFFER, waitList,
*srcMemory->asBuffer(), *dstMemory->asBuffer(), sOffset, dOffset, sizeBytes);
if (command == nullptr) {