temp knn

IntelPython · Apr 19, 2024 · 04b4fcb · 04b4fcb
1 parent 223e70b
commit 04b4fcb
Showing 1 changed file with 4 additions and 5 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
@@ -98,11 +98,10 @@ sycl::event knn_impl(sycl::queue q,
                     }
                 }
 
-                // // simple vote
-                // for (size_t j = 0; j < k; ++j) {
-                //     d_votes_to_classes[i * classes_num +
-                //                        queue_neighbors[j].label]++;
-                // }
+                // simple vote
+                for (size_t j = 0; j < k; ++j) {
+		  d_votes_to_classes[(i*classes_num) + (queue_neighbors[j].label)] += 1.0;
+                }
 
                 IntTy max_ind = 0;
                 FpTy max_value = 0.0;