@@ -26,10 +26,14 @@ def _assert_shape(a, b):
26
26
assert a .shape == (), f"{ a .shape } != ()"
27
27
28
28
29
- def _get_dev_mask (device = None ):
29
+ def _get_dev_id (device = None ):
30
30
dev = dpctl .select_default_device () if device is None else device
31
31
dev_info = dpctl .utils .intel_device_info (dev )
32
- return dev_info .get ("device_id" , 0 ) & 0xFF00
32
+ return dev_info .get ("device_id" , 0 )
33
+
34
+
35
+ def _get_dev_mask (device = None ):
36
+ return _get_dev_id (device ) & 0xFF00
33
37
34
38
35
39
def assert_dtype_allclose (
@@ -448,13 +452,6 @@ def is_intel_numpy():
448
452
return all (dep ["name" ].startswith ("mkl" ) for dep in [blas , lapack ])
449
453
450
454
451
- def is_iris_xe (device = None ):
452
- """
453
- Return True if a test is running on Iris Xe GPU device, False otherwise.
454
- """
455
- return _get_dev_mask (device ) == 0x9A00
456
-
457
-
458
455
def is_lnl (device = None ):
459
456
"""
460
457
Return True if a test is running on Lunar Lake GPU device, False otherwise.
@@ -479,6 +476,14 @@ def is_ptl(device=None):
479
476
return _get_dev_mask (device ) in (0xB000 , 0xFD00 )
480
477
481
478
479
+ def is_tgllp_iris_xe (device = None ):
480
+ """
481
+ Return True if a test is running on Tiger Lake-LP with Iris Xe GPU device,
482
+ False otherwise.
483
+ """
484
+ return _get_dev_id (device ) in (0x9A49 , 0x9A40 )
485
+
486
+
482
487
def is_win_platform ():
483
488
"""
484
489
Return True if a test is running on Windows OS, False otherwise.
0 commit comments