SWDEV-301947 - Add new query for physical CU count

Add hipDeviceAttributePhysicalMultiprocessorCount to get all
available CUs for a device.

Change-Id: Ie6a2c6e958df9627f39c91d0651b89ef5bffd8fe
This commit is contained in:
Saleel Kudchadker
2022-01-31 16:17:06 -08:00
parent 1965dd5795
commit 9a3b4a93d2
+3
Vedi File
@@ -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);
}