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]
Этот коммит содержится в:
@@ -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);
|
||||
}
|
||||
|
||||
Ссылка в новой задаче
Block a user