File tree Expand file tree Collapse file tree 3 files changed +7
-7
lines changed
dpbench/benchmarks/default Expand file tree Collapse file tree 3 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -38,9 +38,9 @@ def count_weighted_pairs_3d_intel_no_slm_ker(
38
38
39
39
n_wi = 32
40
40
41
- dsq_mat = dpex . private . array (shape = (32 * 32 ), dtype = dtype )
42
- w0_vec = dpex . private . array (shape = (32 ), dtype = dtype )
43
- w1_vec = dpex . private . array (shape = (32 ), dtype = dtype )
41
+ dsq_mat = kapi . PrivateArray (shape = (32 * 32 ), dtype = dtype )
42
+ w0_vec = kapi . PrivateArray (shape = (32 ), dtype = dtype )
43
+ w1_vec = kapi . PrivateArray (shape = (32 ), dtype = dtype )
44
44
45
45
offset0 = gr0 * n_wi * lws0 + lid0
46
46
offset1 = gr1 * n_wi * lws1 + lid1
@@ -80,7 +80,7 @@ def count_weighted_pairs_3d_intel_no_slm_ker(
80
80
81
81
# update slm_hist. Use work-item private buffer of 16 tfloat elements
82
82
for k in range (0 , slm_hist_size , private_hist_size ):
83
- private_hist = dpex . private . array (shape = (32 ), dtype = dtype )
83
+ private_hist = kapi . PrivateArray (shape = (32 ), dtype = dtype )
84
84
for p in range (private_hist_size ):
85
85
private_hist [p ] = 0.0
86
86
Original file line number Diff line number Diff line change @@ -56,7 +56,7 @@ def groupByCluster(
56
56
for i in range (WorkPI ):
57
57
point_id = grid * WorkPI * local_size + i * local_size + lid
58
58
if point_id < numpoints :
59
- localP = dpex . private . array (dims , dtyp )
59
+ localP = kapi . PrivateArray (dims , dtyp )
60
60
for d in range (dims ):
61
61
localP [d ] = arrayP [point_id , d ]
62
62
@@ -179,7 +179,7 @@ def updateLabels(
179
179
for i in range (WorkPI ):
180
180
point_id = grid * WorkPI * local_size + i * local_size + lid
181
181
if point_id < numpoints :
182
- localP = dpex . private . array (dims , dtyp )
182
+ localP = kapi . PrivateArray (dims , dtyp )
183
183
for d in range (dims ):
184
184
localP [d ] = arrayP [point_id , d ]
185
185
Original file line number Diff line number Diff line change @@ -25,7 +25,7 @@ def _knn_kernel( # noqa: C901: TODO: can we simplify logic?
25
25
dtype = train .dtype
26
26
i = item .get_id (0 )
27
27
# here k has to be 5 in order to match with numpy
28
- queue_neighbors = dpex . private . array (shape = (5 , 2 ), dtype = dtype )
28
+ queue_neighbors = kapi . PrivateArray (shape = (5 , 2 ), dtype = dtype )
29
29
30
30
for j in range (k ):
31
31
x1 = train [j ]
You can’t perform that action at this time.
0 commit comments