Merge pull request #333 from IntelPython/fix/knn_failure

Fix failure in knn kernel version
IntelPython · Feb 7, 2024 · 8972feb · 8972feb
2 parents 8b56970 + 233abbc
commit 8972feb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dpbench/benchmarks/knn/knn_numba_dpex_k.py b/dpbench/benchmarks/knn/knn_numba_dpex_k.py
@@ -62,9 +62,9 @@ def _knn_kernel(  # noqa: C901: TODO: can we simplify logic?
             distance += diff * diff
         dist = sqrt(distance)
 
-        if dist < queue_neighbors[k - 1][0]:
-            queue_neighbors[k - 1][0] = dist
-            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]
             new_distance = queue_neighbors[k - 1, 0]
             new_neighbor_label = queue_neighbors[k - 1, 1]
             index = k - 1