SWDEV-279657 - Fix hipMallocManaged-N256M failure

Add concurrentManagedAccess detection in hipMallocManaged test.
Skip test when device doesn't support concurrentManagedAccess.

Change-Id: Ie54046feef3baba857a7068972ec1fc1a60c2dfd


[ROCm/clr commit: 20fd6b8a66]
This commit is contained in:
Tao Sang authored and Tao Sang committed 2021-04-20 09:07:31 -04:00
1 vanhempi 9834c99bfc
commit fe7d103d56
2 files changed
+12

No files matched your search

@@ -1561,6 +1561,9 @@ inline static hipError_t hipDeviceGetAttribute(int* pi, hipDeviceAttribute_t att
case hipDeviceAttributeCooperativeMultiDeviceLaunch:
cdattr = cudaDevAttrCooperativeMultiDeviceLaunch;
break;
case hipDeviceAttributeConcurrentManagedAccess:
cdattr = cudaDevAttrConcurrentManagedAccess;
break;
default:
return hipCUDAErrorTohipError(cudaErrorInvalidValue);
}
@@ -20,6 +20,15 @@ void add(int n, float *x, float *y)
int main(int argc, char *argv[])
{
HipTest::parseStandardArguments(argc, argv, true);
int concurrentManagedAccess = 0;
HIPCHECK(hipDeviceGetAttribute(&concurrentManagedAccess,
hipDeviceAttributeConcurrentManagedAccess,
p_gpuDevice));
if(!concurrentManagedAccess) {
printf("info: concurrent managed access not supported on device %d\n Skipped\n",
p_gpuDevice);
passed();
}
printf("info: set device to %d\n", p_gpuDevice);
HIPCHECK(hipSetDevice(p_gpuDevice));