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:
@@ -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);
|
||||
}
|
||||
|
||||
Fai riferimento in un nuovo problema
Block a user