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]
Dieser Commit ist enthalten in:
1 Ursprung
fc67e37d9a
Commit
5e46689191
1 Datei geändert
+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);
|
||||
}
|
||||
|
||||
In neuem Issue referenzieren
Einen Benutzer sperren