Skip to content

Commit a472abe

Browse files
committed
Add a helper util to check LTS driver
1 parent bc5dd4f commit a472abe

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

dpnp/tests/helper.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -434,6 +434,15 @@ def is_intel_numpy():
434434
return all(dep["name"].startswith("mkl") for dep in [blas, lapack])
435435

436436

437+
def is_lts_driver(device=None):
438+
"""
439+
Return True if a test is running on a GPU device with LTS driver version,
440+
False otherwise.
441+
"""
442+
dev = dpctl.select_default_device() if device is None else device
443+
return dev.has_aspect_gpu and "1.3" in dev.driver_version
444+
445+
437446
def is_win_platform():
438447
"""
439448
Return True if a test is running on Windows OS, False otherwise.

0 commit comments

Comments
 (0)