SWDEV-272543 - Fix hipDeviceGetP2PAttribute when both devices are same

Change-Id: I3db17a962807565ea134e3cbcefa66d373e2d363


[ROCm/hip commit: 0d7cd231e8]
This commit is contained in:
Satyanvesh Dittakavi
2021-02-15 01:54:00 -05:00
orang tua c91bc2ba9e
melakukan db0c3fdaaf
+1 -1
Melihat File
@@ -122,7 +122,7 @@ hipError_t hipDeviceGetP2PAttribute(int* value, hipDeviceP2PAttr attr,
HIP_RETURN(hipErrorInvalidValue);
}
if (srcDevice >= static_cast<int>(g_devices.size())
if (srcDevice == dstDevice || srcDevice >= static_cast<int>(g_devices.size())
|| dstDevice >= static_cast<int>(g_devices.size())) {
HIP_RETURN(hipErrorInvalidDevice);
}