We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 8b56970 + 233abbc commit 8972febCopy full SHA for 8972feb
dpbench/benchmarks/knn/knn_numba_dpex_k.py
@@ -62,9 +62,9 @@ def _knn_kernel( # noqa: C901: TODO: can we simplify logic?
62
distance += diff * diff
63
dist = sqrt(distance)
64
65
- if dist < queue_neighbors[k - 1][0]:
66
- queue_neighbors[k - 1][0] = dist
67
- queue_neighbors[k - 1][1] = train_labels[j]
+ if dist < queue_neighbors[k - 1, 0]:
+ queue_neighbors[k - 1, 0] = dist
+ queue_neighbors[k - 1, 1] = train_labels[j]
68
new_distance = queue_neighbors[k - 1, 0]
69
new_neighbor_label = queue_neighbors[k - 1, 1]
70
index = k - 1
0 commit comments