@@ -27,8 +27,8 @@ def test_load_store_fn(supported_dtype):
27
27
"""A test for load/store atomic functions."""
28
28
29
29
@dpex_exp .kernel
30
- def _kernel (a , b ):
31
- i = dpex . get_global_id (0 )
30
+ def _kernel (item , a , b ):
31
+ i = item . get_id (0 )
32
32
a_ref = AtomicRef (a , index = i )
33
33
b_ref = AtomicRef (b , index = i )
34
34
val = b_ref .load ()
@@ -38,17 +38,6 @@ def _kernel(a, b):
38
38
a = dpnp .zeros (2 * N , dtype = supported_dtype )
39
39
b = dpnp .arange (N , dtype = supported_dtype )
40
40
41
- dev = a .sycl_device
42
- if (
43
- dev .backend == dpctl .backend_type .opencl
44
- and dev .device_type == dpctl .device_type .cpu
45
- and supported_dtype == dpnp .float64
46
- ):
47
- pytest .xfail (
48
- "Atomic load and store operations not working "
49
- " for fp64 on OpenCL CPU"
50
- )
51
-
52
41
dpex_exp .call_kernel (_kernel , dpex .Range (b .size ), a , b )
53
42
# Verify that `b[i]` loaded and stored into a[i] by kernel
54
43
# matches the `b[i]` loaded stored into a[i] using Python
@@ -66,25 +55,15 @@ def test_exchange_fn(supported_dtype):
66
55
"""A test for exchange atomic function."""
67
56
68
57
@dpex_exp .kernel
69
- def _kernel (a , b ):
70
- i = dpex . get_global_id (0 )
58
+ def _kernel (item , a , b ):
59
+ i = item . get_id (0 )
71
60
v = AtomicRef (a , index = i )
72
61
b [i ] = v .exchange (b [i ])
73
62
74
63
N = 10
75
64
a_orig = dpnp .zeros (2 * N , dtype = supported_dtype )
76
65
b_orig = dpnp .arange (N , dtype = supported_dtype )
77
66
78
- dev = a_orig .sycl_device
79
- if (
80
- dev .backend == dpctl .backend_type .opencl
81
- and dev .device_type == dpctl .device_type .cpu
82
- and supported_dtype == dpnp .float64
83
- ):
84
- pytest .xfail (
85
- "Atomic exchange operation not working " " for fp64 on OpenCL CPU"
86
- )
87
-
88
67
a_copy = dpnp .copy (a_orig )
89
68
b_copy = dpnp .copy (b_orig )
90
69
@@ -112,17 +91,6 @@ def _kernel(b):
112
91
113
92
b = dpnp .arange (4 , dtype = supported_dtype )
114
93
115
- dev = b .sycl_device
116
- if (
117
- dev .backend == dpctl .backend_type .opencl
118
- and dev .device_type == dpctl .device_type .cpu
119
- and supported_dtype == dpnp .float64
120
- ):
121
- pytest .xfail (
122
- "Atomic compare_exchange operation not working "
123
- " for fp64 on OpenCL CPU"
124
- )
125
-
126
94
dpex_exp .call_kernel (_kernel , dpex .Range (1 ), b )
127
95
128
96
# check for failure
@@ -142,8 +110,8 @@ def test_store_exchange_diff_types(store_exchange_fn):
142
110
"""
143
111
144
112
@dpex_exp .kernel
145
- def _kernel (a , b ):
146
- i = dpex . get_global_id (0 )
113
+ def _kernel (item , a , b ):
114
+ i = item . get_id (0 )
147
115
v = AtomicRef (b , index = 0 )
148
116
getattr (v , store_exchange_fn )(a [i ])
149
117
0 commit comments