AI/rocm-systems
2
0
Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

SWDEV-325379 - Fix for remote copy crash

Bladeren bron 
Change-Id: I22152c0b3538cf7cfc80f82505bc255c01d98f7b
This commit is contained in:
Sarbojit Sarkar
2022-06-04 07:57:54 +00:00
gecommit door Sarbojit Sarkar
bovenliggende 73ec7bada6
commit 356e22f910
2 gewijzigde bestanden met toevoegingen van 9 en 3 verwijderingen
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files
rocclr/platform/command.cpp
+6 -3
Bestand weergeven
@@ -487,7 +487,8 @@ bool OneMemoryArgCommand::validatePeerMemory() {
// extra memory objects.
if (queue_device->settings().rocr_backend_) {
const std::vector<Device*>& srcDevices = memory_->getContext().devices();
if (srcDevices.size() == 1 && queue_device != srcDevices[0]) {
if (!memory_->isArena() &&
srcDevices.size() == 1 && queue_device != srcDevices[0]) {
// current device and source device are not same hence
// explicit allow access is needed for P2P access
device::Memory* mem = memory_->getDeviceMemory(*srcDevices[0]);
@@ -533,14 +534,16 @@ bool TwoMemoryArgsCommand::validatePeerMemory(){
const std::vector<Device*>& dstDevices = memory2_->getContext().devices();
// explicit allow access is needed for P2P access
device::Memory* mem1 = memory1_->getDeviceMemory(*srcDevices[0]);
if (!mem1->getAllowedPeerAccess() && srcDevices.size() == 1) {
if (!memory1_->isArena() &&
!mem1->getAllowedPeerAccess() && srcDevices.size() == 1) {
void* src = reinterpret_cast<void*>(mem1->originalDeviceAddress());
accessAllowed = srcDevices[0]->deviceAllowAccess(src);
mem1->setAllowedPeerAccess(true);
}
device::Memory* mem2 = memory2_->getDeviceMemory(*dstDevices[0]);
if (!mem2->getAllowedPeerAccess() && dstDevices.size() == 1) {
if (!memory2_->isArena() &&
!mem2->getAllowedPeerAccess() && dstDevices.size() == 1) {
void* dst = reinterpret_cast<void*>(mem2->originalDeviceAddress());
accessAllowed &= dstDevices[0]->deviceAllowAccess(dst);
mem2->setAllowedPeerAccess(true);