Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion conda-recipe/meta.yaml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
{% set max_compiler_and_mkl_version = environ.get("MAX_BUILD_CMPL_MKL_VERSION", "2026.0a0") %}
{% set required_compiler_and_mkl_version = "2025.0" %}
{% set required_dpctl_version = "0.21.0" %}
{% set required_dpctl_version = "0.22.0*" %}

{% set pyproject = load_file_data('pyproject.toml') %}
{% set py_build_deps = pyproject.get('build-system', {}).get('requires', []) %}
Expand Down
10 changes: 8 additions & 2 deletions dpnp/tests/helper.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import importlib.util
from enum import Enum
from sys import platform

import dpctl
Expand All @@ -11,6 +12,11 @@
from . import config


class LTS_VERSION(Enum):
V1_3 = "1.3"
V1_6 = "1.6"


def _assert_dtype(a_dt, b_dt, check_only_type_kind=False):
if check_only_type_kind:
assert a_dt.kind == b_dt.kind, f"{a_dt.kind} != {b_dt.kind}"
Expand Down Expand Up @@ -475,13 +481,13 @@ def is_lnl(device=None):
return _get_dev_mask(device) == 0x6400


def is_lts_driver(device=None):
def is_lts_driver(version=LTS_VERSION.V1_3, device=None):
"""
Return True if a test is running on a GPU device with LTS driver version,
False otherwise.
"""
dev = dpctl.select_default_device() if device is None else device
return dev.has_aspect_gpu and "1.3" in dev.driver_version
return dev.has_aspect_gpu and version.value in dev.driver_version


def is_ptl(device=None):
Expand Down
7 changes: 4 additions & 3 deletions dpnp/tests/test_mathematical.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
from dpnp.dpnp_utils import map_dtype_to_device

from .helper import (
LTS_VERSION,
assert_dtype_allclose,
generate_random_numpy_array,
get_abs_array,
Expand All @@ -33,7 +34,7 @@
has_support_aspect16,
has_support_aspect64,
is_intel_numpy,
is_ptl,
is_lts_driver,
numpy_version,
)
from .third_party.cupy import testing
Expand Down Expand Up @@ -218,7 +219,7 @@ def _get_exp_array(self, a, axis, dtype):
@pytest.mark.parametrize("axis", [None, 2, -1])
@pytest.mark.parametrize("include_initial", [True, False])
def test_basic(self, dtype, axis, include_initial):
if axis is None and is_ptl():
if axis is None and not is_lts_driver(version=LTS_VERSION.V1_6):
pytest.skip("due to SAT-8336")

a = dpnp.ones((3, 4, 5, 6, 7), dtype=dtype)
Expand All @@ -238,7 +239,7 @@ def test_basic(self, dtype, axis, include_initial):
@pytest.mark.parametrize("axis", [None, 2, -1])
@pytest.mark.parametrize("include_initial", [True, False])
def test_include_initial(self, dtype, axis, include_initial):
if axis is None and is_ptl():
if axis is None and not is_lts_driver(version=LTS_VERSION.V1_6):
pytest.skip("due to SAT-8336")

a = dpnp.ones((3, 4, 5, 6, 7), dtype=dtype)
Expand Down
2 changes: 1 addition & 1 deletion environments/dpctl_pkg.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
--index-url https://pypi.anaconda.org/dppy/label/dev/simple
dpctl>=0.21.0dev0
dpctl>=0.22.0dev0
2 changes: 1 addition & 1 deletion environments/dpctl_pkg.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,4 @@ name: Install dpctl package
channels:
- dppy/label/dev
dependencies:
- dpctl>=0.21.0dev0
- dpctl>=0.22.0dev0
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@ dependencies = [
# "dpcpp-cpp-rt>=0.59.0",
# "intel-cmplr-lib-rt>=0.59.0"
# WARNING: use the latest dpctl dev version, otherwise stable w/f will fail
"dpctl>=0.21.0dev0",
"dpctl>=0.22.0dev0",
"numpy>=1.26.0"
]
description = "Data Parallel Extension for NumPy"
Expand Down
Loading