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

SWDEV-339742 - Check can access before set device to 1. (#2758)

Browse Source 
Change-Id: I5b06f98b0061ee162488e288c3bfac09ff4f94cf
This commit is contained in:
ROCm CI Service Account
2022-06-29 02:06:26 +05:30
committed by GitHub
parent 88d9a0e7ae
commit 60b60f78e6
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/catch/unit/memory/hipPtrGetAttribute.cc
+8 -4
View File
@@ -122,10 +122,14 @@ TEST_CASE("Unit_hipPtrGetAttribute_Simple") {
// HIP_POINTER_ATTRIBUTE_DEVICE_ORDINAL
if (numDevices > 1) {
HIP_CHECK(hipSetDevice(1));
HIP_CHECK(hipPointerGetAttribute(&datatype, HIP_POINTER_ATTRIBUTE_DEVICE_ORDINAL,
reinterpret_cast<hipDeviceptr_t>(A_d)));
REQUIRE(datatype == 0);
int canAccess = -1;
HIP_CHECK(hipDeviceCanAccessPeer(&canAccess, 1, 0));
if (canAccess == 1) {
HIP_CHECK(hipSetDevice(1));
HIP_CHECK(hipPointerGetAttribute(&datatype, HIP_POINTER_ATTRIBUTE_DEVICE_ORDINAL,
reinterpret_cast<hipDeviceptr_t>(A_d)));
REQUIRE(datatype == 0);
}
}
// HIP_POINTER_ATTRIBUTE_MAPPED