SWDEV-301947 - Add new query for physical CU count

Add hipDeviceAttributePhysicalMultiprocessorCount to get all
available CUs for a device.

Change-Id: Ie6a2c6e958df9627f39c91d0651b89ef5bffd8fe


[ROCm/clr commit: 9a3b4a93d2]
Этот коммит содержится в:
Saleel Kudchadker
2022-01-31 16:17:06 -08:00
родитель fc67e37d9a
Коммит 5e46689191
+3
Просмотреть файл
@@ -319,6 +319,9 @@ hipError_t hipDeviceGetAttribute(int* pi, hipDeviceAttribute_t attr, int device)
case hipDeviceAttributeImageSupport:
*pi = static_cast<int>(g_devices[device]->devices()[0]->info().imageSupport_);
break;
case hipDeviceAttributePhysicalMultiProcessorCount:
*pi = g_devices[device]->devices()[0]->info().maxPhysicalComputeUnits_;
break;
default:
HIP_RETURN(hipErrorInvalidValue);
}