Skip to content

Commit e644eaf

Browse files
authored
Merge 60844cd into 08928bf
2 parents 08928bf + 60844cd commit e644eaf

File tree

1 file changed

+0
-12
lines changed

1 file changed

+0
-12
lines changed

dpnp/dpnp_iface_searching.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -37,17 +37,12 @@
3737
3838
"""
3939

40-
# pylint: disable=no-name-in-module
41-
4240
import dpctl.tensor as dpt
4341
import dpctl.tensor._tensor_impl as dti
4442

4543
import dpnp
4644

4745
from .dpnp_array import dpnp_array
48-
from .dpnp_utils import (
49-
get_usm_allocations,
50-
)
5146
from .dpnp_utils.dpnp_utils_reduction import dpnp_wrap_reduction_call
5247

5348
__all__ = ["argmax", "argmin", "searchsorted", "where"]
@@ -395,13 +390,6 @@ def where(condition, x=None, y=None, /, *, order="K", out=None):
395390
usm_y = dpnp.get_usm_ndarray_or_scalar(y)
396391
usm_condition = dpnp.get_usm_ndarray(condition)
397392

398-
usm_type, queue = get_usm_allocations([condition, x, y])
399-
if dpnp.isscalar(usm_x):
400-
usm_x = dpt.asarray(usm_x, usm_type=usm_type, sycl_queue=queue)
401-
402-
if dpnp.isscalar(usm_y):
403-
usm_y = dpt.asarray(usm_y, usm_type=usm_type, sycl_queue=queue)
404-
405393
usm_out = None if out is None else dpnp.get_usm_ndarray(out)
406394
result = dpnp_array._create_from_usm_ndarray(
407395
dpt.where(usm_condition, usm_x, usm_y, order=order, out=usm_out)

0 commit comments

Comments
 (0)