Skip to content

Commit 8972feb

Browse files
author
Diptorup Deb
authored
Merge pull request #333 from IntelPython/fix/knn_failure
Fix failure in knn kernel version
2 parents 8b56970 + 233abbc commit 8972feb

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dpbench/benchmarks/knn/knn_numba_dpex_k.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,9 @@ def _knn_kernel( # noqa: C901: TODO: can we simplify logic?
6262
distance += diff * diff
6363
dist = sqrt(distance)
6464

65-
if dist < queue_neighbors[k - 1][0]:
66-
queue_neighbors[k - 1][0] = dist
67-
queue_neighbors[k - 1][1] = train_labels[j]
65+
if dist < queue_neighbors[k - 1, 0]:
66+
queue_neighbors[k - 1, 0] = dist
67+
queue_neighbors[k - 1, 1] = train_labels[j]
6868
new_distance = queue_neighbors[k - 1, 0]
6969
new_neighbor_label = queue_neighbors[k - 1, 1]
7070
index = k - 1

0 commit comments

Comments
 (0)