SWDEV-279657 - Fix hipMallocManaged-N256M failure

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

Change-Id: Ie54046feef3baba857a7068972ec1fc1a60c2dfd
Этот коммит содержится в:
Tao Sang
2021-04-19 21:58:36 -04:00
коммит произвёл Tao Sang
родитель 248785e3d1
Коммит 20fd6b8a66
2 изменённых файлов: 12 добавлений и 0 удалений
+3
Просмотреть файл
@@ -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);
}
+9
Просмотреть файл
@@ -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));