fix knn failure on windows due to mismatch in types and precision

Author: adarshyoga

dpbench/benchmarks/default/knn/knn_initialize.py

@@ -24,7 +24,9 @@ def _gen_data_x(ip_size, data_dim, seed, dtype):
 
     def _gen_data_y(ip_size, classes_num, seed):
         default_rng.seed(seed)
-        data = default_rng.randint(classes_num, size=ip_size)
+        data = default_rng.randint(
+            classes_num, size=ip_size, dtype=types_dict["int"]
+        )
         return data
 
     def _gen_train_data(train_size, data_dim, classes_num, seed_train, dtype):

dpbench/benchmarks/default/knn/knn_sycl_native_ext/knn_sycl/_knn_kernel.hpp

@@ -7,16 +7,16 @@
 
 template <typename FpTy, typename IntTy> class theKernel;
 
-template <typename FpTy> struct neighbors
+template <typename FpTy, typename IntTy> struct neighbors
 {
     FpTy dist;
-    size_t label;
+    IntTy label;
 };
 
 template <typename FpTy, typename IntTy>
 sycl::event knn_impl(sycl::queue q,
                      FpTy *d_train,
-                     size_t *d_train_labels,
+                     IntTy *d_train_labels,
                      FpTy *d_test,
                      size_t k,
                      size_t classes_num,
@@ -33,7 +33,7 @@ sycl::event knn_impl(sycl::queue q,
 
                 // here k has to be 5 in order to match with numpy no. of
                 // neighbors
-                struct neighbors<FpTy> queue_neighbors[5];
+                struct neighbors<FpTy, IntTy> queue_neighbors[5];
 
                 // count distances
                 for (size_t j = 0; j < k; ++j) {
@@ -54,7 +54,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;
-                    FpTy new_neighbor_label = queue_neighbors[j].label;
+                    IntTy new_neighbor_label = queue_neighbors[j].label;
                     size_t index = j;
                     while (index > 0 &&
                            new_distance < queue_neighbors[index - 1].dist)
@@ -83,7 +83,7 @@ sycl::event knn_impl(sycl::queue q,
 
                         // push queue
                         FpTy new_distance = queue_neighbors[k - 1].dist;
-                        FpTy new_neighbor_label = queue_neighbors[k - 1].label;
+                        IntTy new_neighbor_label = queue_neighbors[k - 1].label;
                         size_t index = k - 1;
 
                         while (index > 0 &&

dpbench/benchmarks/default/knn/knn_sycl_native_ext/knn_sycl/_knn_sycl.cpp

@@ -43,7 +43,7 @@ void knn_sync(dpctl::tensor::usm_ndarray x_train,
     if (typenum == UAR_FLOAT) {
         sycl::event res_ev = knn_impl<float, unsigned int>(
             x_train.get_queue(), x_train.get_data<float>(),
-            y_train.get_data<size_t>(), x_test.get_data<float>(), k,
+            y_train.get_data<unsigned int>(), x_test.get_data<float>(), k,
             classes_num, train_size, test_size,
             predictions.get_data<unsigned int>(),
             votes_to_classes.get_data<float>(), data_dim);