testing knn

IntelPython · Apr 20, 2024 · bb7f537 · bb7f537
1 parent baa2741
commit bb7f537
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dpbench/benchmarks/default/knn/knn_sycl_native_ext/knn_sycl/_knn_kernel.hpp b/dpbench/benchmarks/default/knn/knn_sycl_native_ext/knn_sycl/_knn_kernel.hpp
@@ -55,7 +55,7 @@ sycl::event knn_impl(sycl::queue q,
                 for (size_t j = 0; j < k; ++j) {
                     // push queue
                     FpTy new_distance = queue_neighbors[j].dist;
-                    IntTy new_neighbor_label = queue_neighbors[j].label;
+                    size_t new_neighbor_label = queue_neighbors[j].label;
                     size_t index = j;
                     while (index > 0 &&
                            new_distance < queue_neighbors[index - 1].dist)
@@ -84,7 +84,7 @@ sycl::event knn_impl(sycl::queue q,
 
                         // push queue
                         FpTy new_distance = queue_neighbors[k - 1].dist;
-                        IntTy new_neighbor_label = queue_neighbors[k - 1].label;
+                        size_t new_neighbor_label = queue_neighbors[k - 1].label;
                         size_t index = k - 1;
 
                         while (index > 0 &&