Skip to content

Commit f102a0b

Browse files
committed
Use is_gpu_device() where required
1 parent 19205fc commit f102a0b

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

dpnp/tests/test_random_state.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,12 @@
1515
from dpnp.dpnp_array import dpnp_array
1616
from dpnp.random import RandomState
1717

18-
from .helper import assert_dtype_allclose, get_array, is_cpu_device
18+
from .helper import (
19+
assert_dtype_allclose,
20+
get_array,
21+
is_cpu_device,
22+
is_gpu_device,
23+
)
1924

2025
# aspects of default device:
2126
_def_device = dpctl.SyclQueue().sycl_device
@@ -688,7 +693,7 @@ def test_scalar(self, func):
688693
],
689694
)
690695
def test_array_range(self, seed):
691-
if not is_cpu_device():
696+
if is_gpu_device():
692697
pytest.skip("seed as a scalar is only supported on GPU")
693698

694699
size = 15

0 commit comments

Comments
 (0)