Skip to content

Commit 59a2302

Browse files
authored
[SYCL] [L0] Correct the device id check for PVC. (#9503)
Account for the various device ids used for PVC GPUs.
1 parent c5f150a commit 59a2302

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

sycl/plugins/unified_runtime/ur/adapters/level_zero/ur_level_zero.hpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,14 @@ struct _ur_device_handle_t : _ur_object {
181181

182182
bool isSubDevice() { return RootDevice != nullptr; }
183183

184-
// Is this a Data Center GPU Max series (aka PVC).
185-
bool isPVC() { return (ZeDeviceProperties->deviceId & 0xff0) == 0xbd0; }
184+
// Is this a Data Center GPU Max series (aka PVC)?
185+
// TODO: change to use
186+
// https://spec.oneapi.io/level-zero/latest/core/api.html#ze-device-ip-version-ext-t
187+
// when that is stable.
188+
bool isPVC() {
189+
return (ZeDeviceProperties->deviceId & 0xff0) == 0xbd0 ||
190+
(ZeDeviceProperties->deviceId & 0xff0) == 0xb60;
191+
}
186192

187193
// Does this device represent a single compute slice?
188194
bool isCCS() const {

0 commit comments

Comments
 (0)